use anyhow::Result;
use generate::generate;
use crate::opt::{Command, DatabaseCommand, MigrateCommand};
mod database;
mod generate;
mod migrate;
mod opt;
mod prepare;
pub use crate::opt::Opt;
pub async fn run(opt: Opt) -> Result<()> {
match opt.command {
Command::Migrate(migrate) => match migrate.command {
MigrateCommand::Add {
description,
reversible,
} => migrate::add(&migrate.source, &description, reversible).await?,
MigrateCommand::Run {
dry_run,
ignore_missing,
database_url,
} => migrate::run(&migrate.source, &database_url, dry_run, ignore_missing).await?,
MigrateCommand::Revert {
dry_run,
ignore_missing,
database_url,
} => migrate::revert(&migrate.source, &database_url, dry_run, ignore_missing).await?,
MigrateCommand::Info { database_url } => {
migrate::info(&migrate.source, &database_url).await?
}
MigrateCommand::BuildScript { force } => migrate::build_script(&migrate.source, force)?,
MigrateCommand::Generate {
database_url,
table,
} => generate(&database_url, table, &migrate.source).await,
},
Command::Database(database) => match database.command {
DatabaseCommand::Create { database_url } => database::create(&database_url).await?,
DatabaseCommand::Drop { yes, database_url } => {
database::drop(&database_url, !yes).await?
}
DatabaseCommand::Reset {
yes,
source,
database_url,
} => database::reset(&source, &database_url, !yes).await?,
DatabaseCommand::Setup {
source,
database_url,
} => database::setup(&source, &database_url).await?,
},
Command::Prepare {
check: false,
merged,
args,
database_url,
} => prepare::run(&database_url, merged, args)?,
Command::Prepare {
check: true,
merged,
args,
database_url,
} => prepare::check(&database_url, merged, args)?,
};
Ok(())
}