use anyhow::{Error, Result};
use rusqlite_migration::{Migrations, M};
use tokio_rusqlite::Connection;

pub mod categories;
pub mod incomes;
pub mod jobs;
pub mod payments;
pub mod users;
mod utils;

pub async fn init(path: &str) -> Result<Connection> {
    let connection = Connection::open(path).await.map_err(|err| {
        Error::msg(format!("Error opening connection: {err}"))
    })?;

    apply_migrations(&connection).await?;
    set_pragma(&connection, "foreign_keys", "ON").await?;
    set_pragma(&connection, "journal_mode", "wal").await?;
    Ok(connection)
}

async fn apply_migrations(conn: &Connection) -> Result<()> {
    let migrations = Migrations::new(vec![
        M::up(include_str!("migration/01-init.sql")),
        M::up(include_str!("migration/02-payment-category.sql")),
        M::up(include_str!("migration/03-sign-in-token.sql")),
        M::up(include_str!("migration/04-plural-naming.sql")),
        M::up(include_str!("migration/05-strict-tables.sql")),
    ]);

    Ok(conn
        .call(move |conn| {
            migrations.to_latest(conn).map_err(|migration_err| {
                tokio_rusqlite::Error::Other(Box::new(migration_err))
            })
        })
        .await?)
}

async fn set_pragma(conn: &Connection, key: impl Into<String>, value: impl Into<String>) -> Result<()> {
    let key = key.into();
    let value = value.into();
    Ok(conn
        .call(move |conn| {
            conn.pragma_update(None, &key, &value)
                .map_err(tokio_rusqlite::Error::Rusqlite)
        })
        .await?)
}