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

//! Defines the `git` module, used in the lua engine.
//!
//! ---
//!
//! ## Lua Documentation
//!
//! The `git` module exposes functions for performing Git operations.
//!
//! ### Functions
//!
//! - `aip.git.restore(path: string): string`

use crate::hub::get_hub;
use crate::runtime::Runtime;
use crate::{Error, Result};
use mlua::{IntoLua, Lua, Table, Value};

pub fn init_module(lua: &Lua, runtime: &Runtime) -> Result<Table> {
	let table = lua.create_table()?;

	let rt = runtime.clone();
	let git_restore_fn = lua.create_function(move |lua, (path,): (String,)| git_restore(lua, &rt, path))?;

	table.set("restore", git_restore_fn)?;

	Ok(table)
}

// region: --- Lua Functions

/// ## Lua Documentation
///
/// Executes a `git restore` command in the workspace directory using the given file path.
///
/// ```lua
/// -- API Signature
/// aip.git.restore(path: string): string
/// ```
///
/// ### Arguments
///
/// - `path: string`: The file path to restore.
///
/// ### Returns
///
/// Returns the standard output as a string if the command is successful.
///
/// ### Example
///
/// ```lua
/// local result = aip.git.restore("src/main.rs")
/// print(result)
/// ```
///
/// ### Error
///
/// Throws an error if the command's stderr output is not empty.
fn git_restore(lua: &Lua, runtime: &Runtime, path: String) -> mlua::Result<Value> {
	let current_dir = runtime
		.dir_context()
		.try_wks_dir_with_err_ctx("aip.git.restore requires a aipack workspace setup")?;
	let output = std::process::Command::new("git")
		.current_dir(current_dir)
		.arg("restore")
		.arg(&path)
		.output()
		.expect("Failed to execute command");

	let stdout = String::from_utf8_lossy(&output.stdout);
	let stderr = String::from_utf8_lossy(&output.stderr);

	if !stderr.is_empty() {
		get_hub().publish_sync(format!("stderr: {}", stderr));
		return Err(Error::cc(format!("'git restore {path}' failed"), stderr).into());
	}

	stdout.into_lua(lua)
}

// endregion: --- Lua Functions