use crate::config::Config;
use crate::database::prelude::*;
use crate::database::transaction::with_transaction;
use crate::database::{DatabaseConnectionManager, MigrationManager};
use crate::opts::ApplyCommandOpt;
use crate::StdResult;

pub(crate) fn apply_sql(config: Config, opts: ApplyCommandOpt) -> StdResult<()> {
    let mut connection_manager = DatabaseConnectionManager::connect(&config.database)?;
    let conn = connection_manager.connection();

    let migration_manager = MigrationManager::new();

    let file_path = {
        let mut file_path = config.directory_path().join(opts.file_name);
        if file_path.extension().is_none() {
            file_path.set_extension("sql");
        }
        file_path
    };

    let content = std::fs::read_to_string(file_path)?;
    with_transaction(conn, &mut |conn| {
        migration_manager.apply_sql(conn, &content)
    })?;

    Ok(())
}