git-ai 0.1.30

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

// Hook: prepare-commit-msg

use std::time::Duration;

use indicatif::{ProgressBar, ProgressStyle};
use anyhow::{Context, Result};
use git2::{Oid, Repository};
use ai::{commit, config};
use ai::commit::Session;
use clap::Parser;
use ai::hook::*;

#[tokio::main]
async fn main() -> Result<()> {
  env_logger::init();

  let args = Args::parse();
  let max_tokens = config::APP.max_diff_tokens;
  let pb = ProgressBar::new_spinner();
  let repo = Repository::open_from_env().context("Failed to open repository")?;

  // If defined, then the user already provided a commit message
  if args.commit_type.is_some() {
    return Ok(());
  }

  // Loading bar to indicate that the program is running
  let style = ProgressStyle::default_spinner()
    .tick_strings(&["-", "\\", "|", "/"])
    .template("{spinner:.blue} {msg}")
    .context("Failed to create progress bar style")?;

  pb.set_style(style);
  pb.set_message("Generating commit message...");
  pb.enable_steady_tick(Duration::from_millis(150));

  let tree = match args.sha1.as_deref() {
    // git commit --amend or git commit -c
    Some("HEAD") | None => repo.head().ok().and_then(|head| head.peel_to_tree().ok()),
    // git rebase
    Some(sha1) => {
      repo
        .find_object(Oid::from_str(sha1)?, None)
        .ok()
        .and_then(|obj| obj.peel_to_tree().ok())
    }
  };

  let patch = repo
    .to_patch(tree, max_tokens)
    .context("Failed to get patch")?;

  if patch.is_empty() {
    Err(HookError::EmptyDiffOutput)?;
  }

  let pb_clone = pb.clone();
  ctrlc::set_handler(move || {
    pb_clone.finish_and_clear();
    console::Term::stdout()
      .show_cursor()
      .expect("Failed to show cursor");
    std::process::exit(1);
  })?;

  // Create a new session from the client
  let session = Session::load_from_repo(&repo).await.unwrap();

  // If the user has a session, then we can use it to generate the commit message
  let response = commit::generate(patch.to_string(), session.into(), pb.clone().into()).await?;

  // Write the response to the commit message file
  args
    .commit_msg_file
    .write(response.response.trim().to_string())
    .unwrap();

  // Save the session to the repository
  response.session.save_to_repo(&repo).await.unwrap();

  pb.finish_and_clear();

  Ok(())
}