aipack 0.8.22

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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

//! Model-related support utilities for Lua script modules.
//!
//! This module provides shared functionality for `aip_run`, `aip_task`, and `aip_pin` modules
//! that need to interact with the model layer.

use crate::Result;
use crate::model::{PinBmc, PinForRunSave, PinForTaskSave, RuntimeCtx};
use crate::runtime::Runtime;
use crate::script::LuaValueExt;
use crate::types::uc;
use mlua::{FromLua, Lua, Value, Variadic};

// region:    --- Pin Support

/// Shared implementation for both `run.pin` and `task.pin`.
pub fn create_pin(lua: &Lua, runtime: &Runtime, for_task: bool, args: Variadic<Value>) -> Result<()> {
	let cmd = PinCommand::from_lua_variadic(lua, args)?;

	let ctx = RuntimeCtx::extract_from_global(lua)?;

	let mm = runtime.mm();
	let (run_id, task_id) = {
		let run_id = ctx.get_run_id(mm)?.ok_or("Cannot create pin – no RUN context available")?;
		let task_id = if for_task { ctx.get_task_id(mm)? } else { None };
		(run_id, task_id)
	};

	if for_task {
		let task_id = task_id.ok_or(
			"Cannot call 'aip.task.pin(...)' in a before all or after all code block.\nCall `aip.run.pin(..)`'",
		)?;
		let pin_c = PinForTaskSave {
			run_id,
			task_id,
			iden: cmd.iden,
			priority: cmd.priority,
			content: Some(cmd.content),
		};

		PinBmc::save_task_pin(mm, pin_c)?
	} else {
		let pin_c = PinForRunSave {
			run_id,
			iden: cmd.iden,
			priority: cmd.priority,
			content: Some(cmd.content),
		};

		PinBmc::save_run_pin(mm, pin_c)?
	};

	Ok(())
}

// -- PinCommand
// Captures the parsed arguments provided to the Lua `...pin(..)` helpers.
struct PinCommand {
	iden: String,
	priority: Option<f64>,
	content: String,
}

impl PinCommand {
	/// Parses the variadic Lua arguments for the two supported signatures:
	///
	/// 1. `pin(iden, priority, content)`
	/// 2. `pin(iden, content)`
	///
	/// `content` can be a string (or convertible primitive value) or a structured table
	/// `{label?: string, content: string}` (mapped to uc::Marker).
	///
	/// Returns an informative error if the arguments do not match either form.
	fn from_lua_variadic(lua: &Lua, args: Variadic<Value>) -> Result<Self> {
		match args.len() {
			2 => {
				let mut args = args.into_iter();
				let iden = args
					.next()
					.ok_or("aip...pin(iden, content) – expected <string> for parameter `iden`.")?;
				let iden = iden
					.x_as_lua_str()
					.ok_or("aip...pin(iden, content) – expected <string> for parameter `iden`.")?;

				let content = args.next().ok_or("aip...pin(iden, content) – expected content.")?;
				let content = Self::value_to_uc_string(lua, content)?;

				Ok(Self {
					iden: iden.to_string(),
					priority: None,
					content,
				})
			}
			3 => {
				let mut args = args.into_iter();
				let iden = args
					.next()
					.ok_or("aip...pin(iden, content) – expected <string> for parameter `iden`.")?;
				let iden = iden
					.x_as_lua_str()
					.ok_or("aip...pin(iden, content) – expected <string> for parameter `iden`.")?;

				let priority = args
					.next()
					.and_then(|v| v.x_as_f64())
					.ok_or("aip...pin(iden, priority, content) – expected <number> for parameter `priority`.")?;

				let content = args.next().ok_or("aip...pin(iden, content) – expected content.")?;
				let content = Self::value_to_uc_string(lua, content)?;

				Ok(Self {
					iden: iden.to_string(),
					priority: Some(priority),
					content,
				})
			}
			_ => Err(crate::Error::custom(
				"aip...pin(...) – expected 2 or 3 parameters: (iden, content) or (iden, priority, content).",
			)),
		}
	}

	/// Convert into a UC Component
	/// For now, only support uc::Marker
	fn value_to_uc_string(lua: &Lua, val: Value) -> Result<String> {
		let uc_comp = uc::Marker::from_lua(val, lua)?;
		let json_string = serde_json::to_string_pretty(&uc_comp)
			.map_err(|err| crate::Error::cc("Cannot seralize uc component", err))?;

		Ok(json_string)
	}
}

// endregion: --- Pin Support