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

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

    let applied_migrations = migration_manager.applied_migration_names(conn)?;
    let migrations = config.migrations()?;

    let rollback_migrations_number = if opts.all_migrations {
        applied_migrations.len()
    } else {
        cmp::min(opts.migrations_number, applied_migrations.len())
    };

    for migration_name in &applied_migrations[..rollback_migrations_number] {
        if let Some(migration) = migrations.iter().find(|m| m.name() == migration_name) {
            println!("downgrade {}...", migration.name());
            with_transaction(conn, &mut |conn| {
                migration_manager.downgrade(conn, &migration)
            })?;
        }
    }

    Ok(())
}