janezicmatej/ahab
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: return an exit code from main

Browse Source
This commit is contained in:
Matej Janezic
2026-09-08 09:54:58 +00:00
parent aec7afb274
commit 11ecd0ac89
2 changed files with 54 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
src/main.rs
View File

@@ -1,9 +1,11 @@
use std::process::ExitCode;
use ahab::{cli, command_builder, scripts};
use anyhow::Result;
use clap::Parser;
fn main() -> Result<()> {
fn main() -> ExitCode {
let args = cli::Ahab::parse();
command_builder::set_options(command_builder::Options {
@@ -11,26 +13,49 @@ fn main() -> Result<()> {
dry_run: args.dry_run,
});
match args.command {
cli::Commands::Django { command } => match command {
cli::Django::Bash => scripts::django::bash(),
cli::Django::Run { rest } => scripts::django::run(&rest),
cli::Django::MakeCommand { app, name } => scripts::django::make_command(&app, &name),
cli::Django::Makemigrations => scripts::django::makemigrations(),
cli::Django::Manage { rest } => scripts::django::manage(&rest),
cli::Django::Migrate { rest } => scripts::django::migrate(&rest),
cli::Django::Shell => scripts::django::shell(),
cli::Django::Test => scripts::django::test(),
},
cli::Commands::Postgres { command } => match command {
cli::Postgres::Import { path } => scripts::postgres::import(&path),
cli::Postgres::Dump { path, format, gzip } => {
scripts::postgres::dump(&path, format, gzip)
}
},
match run(args.command) {
Ok(code) => code,
Err(e) => {
eprintln!("Error: {e:#}");
ExitCode::FAILURE
}
}
}
fn run(command: cli::Commands) -> Result<ExitCode> {
let done = ExitCode::SUCCESS;
match command {
cli::Commands::Django { command } => {
match command {
cli::Django::Bash => scripts::django::bash(),
cli::Django::Run { rest } => scripts::django::run(&rest),
cli::Django::MakeCommand { app, name } => {
scripts::django::make_command(&app, &name)
}
cli::Django::Makemigrations => scripts::django::makemigrations(),
cli::Django::Manage { rest } => scripts::django::manage(&rest),
cli::Django::Migrate { rest } => scripts::django::migrate(&rest),
cli::Django::Shell => scripts::django::shell(),
cli::Django::Test => scripts::django::test(),
}?;
Ok(done)
}
cli::Commands::Postgres { command } => {
match command {
cli::Postgres::Import { path } => scripts::postgres::import(&path),
cli::Postgres::Dump { path, format, gzip } => {
scripts::postgres::dump(&path, format, gzip)
}
}?;
Ok(done)
}
cli::Commands::Link { command } => match command {
cli::Link::Add { paths, force } => scripts::link::add(&paths, force),
cli::Link::Restore { paths, all } => scripts::link::restore(&paths, all),
cli::Link::Add { paths, force } => scripts::link::add(&paths, force).map(|()| done),
cli::Link::Restore { paths, all } => scripts::link::restore(&paths, all).map(|()| done),
// the one command with something to say through its exit code
cli::Link::Check {
paths,
porcelain,
@@ -38,6 +63,10 @@ fn main() -> Result<()> {
exit_code,
} => scripts::link::check(&paths, porcelain, null, exit_code),
},
cli::Commands::Completions { shell } => scripts::completions::completions(shell),
cli::Commands::Completions { shell } => {
scripts::completions::completions(shell)?;
Ok(done)
}
}
}