use std::time;

use tracing::Instrument;

use sqlx::sqlite::{SqliteConnectOptions, SqlitePoolOptions};
use sqlx::ConnectOptions;
pub use sqlx::{Pool, Sqlite};

use std::str::FromStr as _;

use crate::StartError;

#[tracing::instrument]
pub async fn init_database_pool(url: &str) -> Result<Pool<Sqlite>, StartError> {
    Ok(SqlitePoolOptions::new()
        .max_connections(5)
        .connect_with(
            SqliteConnectOptions::from_str(url)?
                .log_statements(log::LevelFilter::Debug)
                .log_slow_statements(log::LevelFilter::Warn, time::Duration::from_millis(100))
                .tracing_span(tracing::info_span!("sqlx_pool"))
                .pragma("foreign_keys", "1"),
        )
        .await?)
}

#[tracing::instrument]
pub async fn migrate(url: &str) -> Result<(), StartError> {
    let pool = SqlitePoolOptions::new()
        .max_connections(5)
        .connect_with(
            SqliteConnectOptions::from_str(url)?
                .pragma("foreign_keys", "0")
                .log_statements(log::LevelFilter::Warn)
                .tracing_span(tracing::info_span!("sqlx_migration")),
        )
        .await?;

    async { sqlx::migrate!().run(&pool).await }
        .instrument(tracing::info_span!("packager::query", "migration"))
        .await?;

    Ok(())
}