git-ai 0.1.28

Git AI: Automates commit messages using ChatGPT. Stage your files, and Git AI generates the messages.
Documentation 
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

// Hook: prepare-commit-msg

use std::path::Path;
use std::time::Duration;

use colored::Colorize;
use git2::{DiffOptions, Repository, RepositoryOpenFlags};
use anyhow::{Context, Result};
use ai::config::APP;
use ai::commit;

const MAX_NUMBER_OF_COMMITS: usize = 5;

trait RepositoryExt {
  fn get_last_n_commits(&self, max_commits: usize) -> Result<Vec<git2::Commit>, git2::Error>;
}

impl RepositoryExt for Repository {
  fn get_last_n_commits(&self, max_commits: usize) -> Result<Vec<git2::Commit>, git2::Error> {
    let mut revwalk = self.revwalk()?;
    revwalk.push_head()?;
    Ok(
      revwalk
        .take(max_commits)
        .map(move |id| {
          self
            .find_commit(id.unwrap())
            .expect("Failed to find commit")
        })
        .collect()
    )
  }
}

// TODO: Duplicate code from src/commit.rs
trait CommitExt {
  fn show(&self, repo: &Repository, max_tokens: usize) -> Result<String, git2::Error>;
}

impl CommitExt for git2::Commit<'_> {
  fn show(&self, repo: &Repository, max_tokens: usize) -> Result<String, git2::Error> {
    let mut commit_info = "".to_string();
    let mut opts = DiffOptions::new();
    let tree = self.tree()?;
    let parent_tree = self
      .parent(0)
      .ok()
      .as_ref()
      .map(|c| c.tree().ok())
      .flatten();
    let diff = repo.diff_tree_to_tree(parent_tree.as_ref(), Some(&tree), Some(&mut opts))?;

    _ = diff
      .print(git2::DiffFormat::Patch, |_delta, _hunk, line| {
        commit_info.push_str(std::str::from_utf8(line.content()).unwrap());
        commit_info.len() < max_tokens
      })
      .ok();

    Ok(commit_info)
  }
}

use console::Style;
use indicatif::{ProgressBar, ProgressStyle};

pub async fn run(_args: &clap::ArgMatches) -> Result<()> {
  let max_tokens = APP.max_diff_tokens;

  let current_dir = std::env::current_dir().context("Failed to get current directory")?;
  let repo = Repository::open_ext(&current_dir, RepositoryOpenFlags::empty(), Vec::<&Path>::new())?;
  let commits = repo
    .get_last_n_commits(MAX_NUMBER_OF_COMMITS)
    .context("Failed to get last commits")?;

  // Create and configure the progress bar
  let spinner_style = ProgressStyle::default_spinner()
    .tick_strings(&["-", "\\", "|", "/"])
    .template("{spinner:.blue} {msg}")
    .context("Failed to create progress bar style")?;

  let pb = ProgressBar::new_spinner();
  pb.set_style(spinner_style);
  pb.enable_steady_tick(Duration::from_millis(100));

  let header_style = Style::new().bold();
  println!("{}", header_style.apply_to("🛠️  AI-Generated Commit Message Examples"));

  for (index, commit) in commits.iter().enumerate() {
    pb.set_message(format!("Loading commit #{} ...\n", index + 1));
    let response = commit::generate(commit.show(&repo, max_tokens)?, None, None).await?;

    let commit_message = response.response.trim();
    pb.println(format!("Commit #{}:", index + 1));
    pb.println(format!("\tOriginal: {}", commit.message().unwrap_or_default().trim().italic()));
    pb.println(format!("\tGenerated: {}", commit_message.italic()));
  }

  Ok(())
}