aipack 0.7.7-WIP

Command Agent runner to accelerate production coding with genai.
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

#![allow(unused)] // because of some mac os only (will clean later)
use crate::Result;
use crossterm::cursor::MoveTo;
use crossterm::terminal::ClearType;
use crossterm::{execute, terminal};
use keyring::Entry;
use std::io::{self, Write};

const KEY_SERVICE: &str = "aipack_secrets";

// NOT USED NOW
fn _clear_api_key(key_name: &str) -> Result<()> {
	_clear_key(KEY_SERVICE, key_name)?;
	Ok(())
}

pub fn get_or_prompt_api_key(key_name: &str) -> Result<String> {
	let api_key = get_and_save_pwd(KEY_SERVICE, key_name)?;

	Ok(api_key)
}

// region:    --- Support

// Get the value from the local keychain, or prompt the user to save and return.
fn get_and_save_pwd(service: &str, name: &str) -> Result<String> {
	let entry = Entry::new(service, name)?;
	let pwd = match entry.get_password() {
		Ok(pwd) => pwd,
		Err(keyring::Error::NoEntry) => prompt_and_save(entry, name)?,
		Err(other) => return Err(format!("Failed to execute keyring: {other}").into()),
	};

	Ok(pwd)
}

/// Clear the secret key if present
/// - Returns true if it was present
/// - If not present, will do nothing (and no error)
///
/// NOT USED FOR NOW
fn _clear_key(service: &str, name: &str) -> Result<bool> {
	if let Ok(entry) = Entry::new(service, name) {
		entry.delete_credential()?;
		Ok(true)
	} else {
		Ok(false)
	}
}

// Ask the user for a value and store it in the entry.
fn prompt_and_save(entry: Entry, disp_name: &str) -> Result<String> {
	// -- exit of raw mode for now to get the input.
	terminal::disable_raw_mode().expect("Failed to disable crossterm raw mode");

	// -- Prompt the user

	print!(
		r#"
API KEY for '{}' not found in environment vairable or keychain 
Please enter value (will store key in Mac keychain, under aipack_secrets/{}): 
"#,
		disp_name, disp_name
	);
	io::stdout().flush()?;

	// NOTE: Sometime on VSCode integrated terminal, we do not get the input.
	// TODO: Might want to change that with the cross term.
	let mut input = String::new();
	io::stdin().read_line(&mut input)?;

	// -- Validate the answer
	let val = input.trim().to_string();
	if val.is_empty() {
		return Err("Value cannot be empty.".into());
	}

	// -- Save and get again
	entry.set_password(&val)?;
	// Making sure we get the value from store
	let val = entry.get_password()?;

	// -- put back in raw mode
	clear_terminal()?;
	// sleep 200ms
	std::thread::sleep(std::time::Duration::from_millis(200));
	terminal::enable_raw_mode().expect("Failed to enable crossterm raw mode");

	// TODO: We need to have the TUI taking care of the prompt

	Ok(val)
}

fn clear_terminal() -> Result<()> {
	let mut stdout = io::stdout();
	execute!(stdout, terminal::Clear(ClearType::All))?;
	execute!(stdout, MoveTo(0, 0))?; // Move the cursor to the top-left
	Ok(())
}
// endregion: --- Support