tetratto_core/database/
services.rs

1use crate::model::{
2    auth::User,
3    littleweb::{Service, ServiceFsEntry},
4    permissions::{FinePermission, SecondaryPermission},
5    Error, Result,
6};
7use crate::{auto_method, DataManager};
8use oiseau::{cache::Cache, execute, get, params, query_rows, PostgresRow};
9
10impl DataManager {
11    /// Get a [`Service`] from an SQL row.
12    pub(crate) fn get_service_from_row(x: &PostgresRow) -> Service {
13        Service {
14            id: get!(x->0(i64)) as usize,
15            created: get!(x->1(i64)) as usize,
16            owner: get!(x->2(i64)) as usize,
17            name: get!(x->3(String)),
18            files: serde_json::from_str(&get!(x->4(String))).unwrap(),
19            revision: get!(x->5(i64)) as usize,
20        }
21    }
22
23    auto_method!(get_service_by_id(usize as i64)@get_service_from_row -> "SELECT * FROM services WHERE id = $1" --name="service" --returns=Service --cache-key-tmpl="atto.service:{}");
24
25    /// Get all services by user.
26    ///
27    /// # Arguments
28    /// * `id` - the ID of the user to fetch services for
29    pub async fn get_services_by_user(&self, id: usize) -> Result<Vec<Service>> {
30        let conn = match self.0.connect().await {
31            Ok(c) => c,
32            Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
33        };
34
35        let res = query_rows!(
36            &conn,
37            "SELECT * FROM services WHERE owner = $1 ORDER BY created DESC",
38            &[&(id as i64)],
39            |x| { Self::get_service_from_row(x) }
40        );
41
42        if res.is_err() {
43            return Err(Error::GeneralNotFound("service".to_string()));
44        }
45
46        Ok(res.unwrap())
47    }
48
49    const MAXIMUM_FREE_SERVICES: usize = 10;
50
51    /// Create a new service in the database.
52    ///
53    /// # Arguments
54    /// * `data` - a mock [`Service`] object to insert
55    pub async fn create_service(&self, data: Service) -> Result<Service> {
56        // check values
57        if data.name.trim().len() < 2 {
58            return Err(Error::DataTooShort("name".to_string()));
59        } else if data.name.len() > 128 {
60            return Err(Error::DataTooLong("name".to_string()));
61        }
62
63        // check number of services
64        let owner = self.get_user_by_id(data.owner).await?;
65
66        if !owner.permissions.check(FinePermission::SUPPORTER) {
67            let services = self.get_services_by_user(data.owner).await?;
68
69            if services.len() >= Self::MAXIMUM_FREE_SERVICES {
70                return Err(Error::MiscError(
71                    "You already have the maximum number of services you can have".to_string(),
72                ));
73            }
74        }
75
76        // ...
77        let conn = match self.0.connect().await {
78            Ok(c) => c,
79            Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
80        };
81
82        let res = execute!(
83            &conn,
84            "INSERT INTO services VALUES ($1, $2, $3, $4, $5, $6)",
85            params![
86                &(data.id as i64),
87                &(data.created as i64),
88                &(data.owner as i64),
89                &data.name,
90                &serde_json::to_string(&data.files).unwrap(),
91                &(data.created as i64)
92            ]
93        );
94
95        if let Err(e) = res {
96            return Err(Error::DatabaseError(e.to_string()));
97        }
98
99        Ok(data)
100    }
101
102    pub async fn delete_service(&self, id: usize, user: &User) -> Result<()> {
103        let service = self.get_service_by_id(id).await?;
104
105        // check user permission
106        if user.id != service.owner
107            && !user
108                .secondary_permissions
109                .check(SecondaryPermission::MANAGE_SERVICES)
110        {
111            return Err(Error::NotAllowed);
112        }
113
114        // ...
115        let conn = match self.0.connect().await {
116            Ok(c) => c,
117            Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
118        };
119
120        let res = execute!(&conn, "DELETE FROM services WHERE id = $1", &[&(id as i64)]);
121
122        if let Err(e) = res {
123            return Err(Error::DatabaseError(e.to_string()));
124        }
125
126        // ...
127        self.0.1.remove(format!("atto.service:{}", id)).await;
128        Ok(())
129    }
130
131    auto_method!(update_service_name(&str)@get_service_by_id:FinePermission::MANAGE_USERS; -> "UPDATE services SET name = $1 WHERE id = $2" --cache-key-tmpl="atto.service:{}");
132    auto_method!(update_service_files(Vec<ServiceFsEntry>)@get_service_by_id:FinePermission::MANAGE_USERS; -> "UPDATE services SET files = $1 WHERE id = $2" --serde --cache-key-tmpl="atto.service:{}");
133    auto_method!(update_service_revision(i64) -> "UPDATE services SET revision = $1 WHERE id = $2" --cache-key-tmpl="atto.service:{}");
134}
tetratto_core/database/services.rs

tetratto_core/database/
services.rs
