autter_core/database/
mod.rs

1mod audit_log;
2mod ip_bans;
3mod notifications;
4mod organization_memberships;
5mod organizations;
6mod permissions_lists;
7mod properties;
8mod sql;
9mod user_warnings;
10mod users;
11
12use std::collections::HashMap;
13
14use crate::config::Config;
15use buckets_core::{Config as BucketsConfig, DataManager as BucketsManager};
16use oiseau::{execute, postgres::DataManager as OiseauManager, postgres::Result as PgResult};
17use tetratto_core::model::{Error, Result};
18
19#[derive(Clone)]
20pub struct DataManager(pub OiseauManager<Config>, pub BucketsManager);
21
22impl DataManager {
23    /// Create a new [`DataManager`].
24    pub async fn new(config: Config) -> PgResult<Self> {
25        let buckets_manager = BucketsManager::new(BucketsConfig {
26            directory: format!("{}/{}", config.dirs.media, "uploads"),
27            bucket_defaults: HashMap::new(),
28            database: config.database.clone(),
29        })
30        .await
31        .expect("failed to create buckets manager");
32
33        Ok(Self(OiseauManager::new(config).await?, buckets_manager))
34    }
35
36    /// Initialize tables.
37    pub async fn init(&self) -> Result<()> {
38        let conn = match self.0.connect().await {
39            Ok(c) => c,
40            Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
41        };
42
43        execute!(&conn, sql::CREATE_TABLE_USERS).unwrap();
44        execute!(&conn, sql::CREATE_TABLE_NOTIFICATIONS).unwrap();
45        execute!(&conn, sql::CREATE_TABLE_AUDIT_LOG).unwrap();
46        execute!(&conn, sql::CREATE_TABLE_IP_BANS).unwrap();
47        execute!(&conn, sql::CREATE_TABLE_USER_WARNINGS).unwrap();
48        execute!(&conn, sql::CREATE_TABLE_PROPERTIES).unwrap();
49        execute!(&conn, sql::CREATE_TABLE_ORGANIZATIONS).unwrap();
50        execute!(&conn, sql::CREATE_TABLE_ORGANIZATION_MEMBERSHIPS).unwrap();
51        execute!(&conn, sql::CREATE_TABLE_PERMISSIONS_LISTS).unwrap();
52
53        for x in sql::VERSION_MIGRATIONS.split(";") {
54            execute!(&conn, x).unwrap();
55        }
56
57        Ok(())
58    }
59}
autter_core/database/mod.rs

autter_core/database/
mod.rs
