ai-commit-cli 0.4.0

A CLI that writes your git commit messages for you with AI
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

mod commit;
mod complete;

use anyhow::Result;
use clap::builder::styling::AnsiColor;
use clap::builder::Styles;
use clap::{Parser, Subcommand};
use clap_verbosity_flag::{InfoLevel, Verbosity};

#[async_trait::async_trait]
pub trait Run {
    async fn run(&self) -> Result<()>;
}

#[derive(Debug, Parser)]
#[command(version, author, styles = STYLES)]
pub struct Cmd {
    #[command(subcommand)]
    sub_cmd: SubCmd,

    #[command(flatten)]
    verbosity: Verbosity<InfoLevel>,
}

const STYLES: Styles = Styles::styled()
    .header(AnsiColor::Blue.on_default().bold())
    .usage(AnsiColor::Cyan.on_default().bold())
    .literal(AnsiColor::Green.on_default().bold())
    .placeholder(AnsiColor::Magenta.on_default().bold());

#[derive(Debug, Subcommand)]
enum SubCmd {
    Commit(commit::Cmd),
    Complete(complete::Cmd),
}

#[async_trait::async_trait]
impl Run for Cmd {
    async fn run(&self) -> Result<()> {
        let level = self.verbosity.log_level().map(|level| match level {
            clap_verbosity_flag::Level::Error => tracing::Level::ERROR,
            clap_verbosity_flag::Level::Warn => tracing::Level::WARN,
            clap_verbosity_flag::Level::Info => tracing::Level::INFO,
            clap_verbosity_flag::Level::Debug => tracing::Level::DEBUG,
            clap_verbosity_flag::Level::Trace => tracing::Level::TRACE,
        });
        if let Some(level) = level {
            if level < tracing::Level::DEBUG {
                tracing_subscriber::fmt()
                    .pretty()
                    .with_file(false)
                    .with_line_number(false)
                    .with_max_level(level)
                    .with_target(false)
                    .without_time()
                    .init();
            } else {
                tracing_subscriber::fmt()
                    .pretty()
                    .with_max_level(level)
                    .init();
            }
        }
        match &self.sub_cmd {
            SubCmd::Commit(cmd) => cmd.run().await,
            SubCmd::Complete(cmd) => cmd.run().await,
        }
    }
}