use crate::{
cmd::{CompletionsArgs, DataArgs, EditArgs, ExecArgs, ListArgs, PickArgs, StatArgs, TestArgs},
err::Error,
};
use clap::{CommandFactory, Parser, Subcommand};
use env_logger::Env;
use log::LevelFilter;
pub fn reset_signal_pipe_handler() {
#[cfg(target_family = "unix")]
{
use nix::sys::signal;
unsafe {
let _ = signal::signal(signal::Signal::SIGPIPE, signal::SigHandler::SigDfl)
.map_err(|e| println!("{:?}", e));
}
}
}
#[derive(Parser)]
#[command(name = "leetcode", version, about = "May the Code be with You 👻")]
#[command(arg_required_else_help = true)]
pub struct Cli {
#[arg(short, long)]
pub debug: bool,
#[command(subcommand)]
pub command: Option<Commands>,
}
#[derive(Subcommand)]
pub enum Commands {
#[command(visible_alias = "d", display_order = 1)]
Data(DataArgs),
#[command(visible_alias = "e", display_order = 2)]
Edit(EditArgs),
#[command(visible_alias = "x", display_order = 3)]
Exec(ExecArgs),
#[command(visible_alias = "l", display_order = 4)]
List(ListArgs),
#[command(visible_alias = "p", display_order = 5)]
Pick(PickArgs),
#[command(visible_alias = "s", display_order = 6)]
Stat(StatArgs),
#[command(visible_alias = "t", display_order = 7)]
Test(TestArgs),
#[command(visible_alias = "c", display_order = 8)]
Completions(CompletionsArgs),
}
pub async fn main() -> Result<(), Error> {
reset_signal_pipe_handler();
let cli = Cli::parse();
if cli.debug {
env_logger::Builder::from_env(Env::default().default_filter_or("debug")).init();
} else {
env_logger::Builder::new()
.filter_level(LevelFilter::Info)
.format_timestamp(None)
.init();
}
match cli.command {
Some(Commands::Data(args)) => args.run().await,
Some(Commands::Edit(args)) => args.run().await,
Some(Commands::Exec(args)) => args.run().await,
Some(Commands::List(args)) => args.run().await,
Some(Commands::Pick(args)) => args.run().await,
Some(Commands::Stat(args)) => args.run().await,
Some(Commands::Test(args)) => args.run().await,
Some(Commands::Completions(args)) => {
let mut cmd = Cli::command();
args.run(&mut cmd)
}
None => Err(Error::MatchError),
}
}