jcommit 0.1.13

An AI-powered tool to generate git commit messages
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use anyhow::Result;
use clap::Parser;
use std::io::Write;
use futures_util::StreamExt;
mod git;
mod ai;
mod config;

#[derive(Parser)]
#[command(author, version, about)]
struct Cli {
    /// Additional hints provided by the user
    #[arg(short, long)]
    message: Option<String>,

    /// Specify Git repository path, defaults to current directory
    #[arg(short, long, default_value = ".")]
    path: String,

    /// Whether to include commit message body
    #[arg(short, long)]
    body: bool,

    /// Whether to execute git commit command directly
    #[arg(short = 'c', long)]
    commit: bool,

    /// Whether to print detailed AI conversation logs
    #[arg(short = 'v', long = "verbose")]
    verbose: bool,

    /// Generate a summary of changes between current HEAD and specified base commit/branch
    #[arg(short = 's', long)]
    summary: Option<String>,

    /// Automatically stage all changes before generating commit message
    #[arg(short = 'a', long)]
    add: bool,
}

#[tokio::main]
async fn main() -> Result<()> {
    let cli = Cli::parse();
    
    // Read Git repository diff information
    let git_diff = git::GitDiff::new(&cli.path)?;
    
    let diff_content = if let Some(base) = cli.summary {
        git_diff.get_summary_diff(&base)?
    } else {
        let diff = git_diff.get_staged_diff(cli.add)?;
        if diff.trim().is_empty() {
            println!("Error: No staged changes found. Please use 'git add' command to stage your changes.");
            return Ok(());
        }
        diff
    };
    // println!("Git diff content: {}\n", diff_content);
    
    // Read configuration
    let config = config::Config::new()?;
    
    // Call AI service to generate commit message
    let ai_service = ai::AiService::new(
        config.api_endpoint,
        config.model,
        config.api_key,
        config.is_azure.unwrap_or(false),
        config.api_version,
        config.prompt
    );
    let mut commit_message = String::new();
    let mut stream = ai_service.generate_commit_message(diff_content, cli.message, cli.body, cli.verbose).await?;
    println!("Commit message:\n");
    while let Some(content) = stream.next().await {
        let content = content?;
        if !content.is_empty() {
            print!("{}", content);
            std::io::stdout().flush()?;
            commit_message.push_str(&content);
        }
    }
    println!("\n");

    if cli.commit {
        // Automatically stage changes if -a flag is provided
        if cli.add {
            git_diff.add_all()?;
        }

        git_diff.commit(&commit_message)?;
        println!("Successfully committed changes.");
    } else {
        print!("Do you want to commit these changes? [y/N] ");
        std::io::stdout().flush()?;
        
        let mut input = String::new();
        std::io::stdin().read_line(&mut input)?;
        
        if input.trim().to_lowercase() == "y" {
            if cli.add {
                git_diff.add_all()?;
            }
            git_diff.commit(&commit_message)?;
            println!("Successfully committed changes.");
        }
    }
    
    Ok(())
}