Kordophone/kordophone-db/src/database.rs

use anyhow::Result;
use diesel::prelude::*;

use crate::repository::Repository;
use crate::settings::Settings;

use diesel_migrations::{embed_migrations, EmbeddedMigrations, MigrationHarness};
pub const MIGRATIONS: EmbeddedMigrations = embed_migrations!();

pub struct Database {
    connection: SqliteConnection,
}

impl Database {
    pub fn new(path: &str) -> Result<Self> {
        let mut connection = SqliteConnection::establish(path)?;
        connection.run_pending_migrations(MIGRATIONS)
            .map_err(|e| anyhow::anyhow!("Error running migrations: {}", e))?;

        Ok(Self { connection })
    }

    pub fn new_in_memory() -> Result<Self> {
        Self::new(":memory:")
    }

    pub fn with_repository<F, R>(&mut self, f: F) -> R
    where
        F: FnOnce(&mut Repository) -> R,
    {
        let mut repository = Repository::new(&mut self.connection);
        f(&mut repository)
    }

    pub fn with_settings<F, R>(&mut self, f: F) -> R
    where
        F: FnOnce(&mut Settings) -> R,
    {
        let mut settings = Settings::new(&mut self.connection);
        f(&mut settings)
    }
}