giton 0.1.0

Giton is AI-powered utility for Git 
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

use core::db;
use std::process::Command;

#[cfg(not(debug_assertions))]
use human_panic::setup_panic;

#[cfg(debug_assertions)]
extern crate better_panic;

use utils::app_config::AppConfig;
use utils::error::{GitonError, Result};

/// The main entry point of the application.
fn main() -> Result<()> {
    // Human Panic. Only enabled when *not* debugging.
    #[cfg(not(debug_assertions))]
    {
        setup_panic!();
    }

    // Better Panic. Only enabled *when* debugging.
    #[cfg(debug_assertions)]
    {
        better_panic::Settings::debug()
            .most_recent_first(false)
            .lineno_suffix(true)
            .verbosity(better_panic::Verbosity::Full)
            .install();
    }

    let _guard = utils::logger::setup_logging()?;

    // Initialize Configuration
    let config_contents = include_str!("resources/default_config.toml");
    AppConfig::init(Some(config_contents))?;

    // Throw error if no OpenAI key is set
    if AppConfig::get::<String>("openai_key")?.is_empty() {
        println!("No OpenAI API key set. Please set one with `export GITON_OPENAI_KEY=<key>`");
        return Ok(());
    }

    // Match Commands
    let cli_match = cli::cli_match();

    match cli_match {
        Ok(_) => {}
        Err(e) => {
            match e {
                GitonError::Clap(_clap_error) => {
                    // extract args
                    let args: Vec<String> = std::env::args().skip(1).collect();
                    let args_string: String = args.join(" ");

                    Command::new("git")
                        .args(args)
                        .spawn()
                        .expect("git command failed to start");

                    // if success, insert command into file
                    db::insert_command(args_string)?;
                }
                _ => {
                    println!("{}", e);
                }
            }
        }
    }

    Ok(())
}