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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154

//! Defines the `file_md` module for the Lua engine.
//!
//! ---
//!
//! ## Lua Documentation
//!
//! The `file_md` module exposes functions to load and convert Markdown content.
//!
//! ### Functions
//!
//! - `aip.file.load_md_sections(path: string, headings?: string | list<string>): list<MdSection>`  
//! - `aip.file.load_md_split_first(path: string): MdSplitFirst`  
//!
//! ## Error
//!
//! Returns an error if file I/O, parsing, or conversion fails.
//
// region:    --- Modules

use crate::dir_context::PathResolver;
use crate::runtime::Runtime;
use crate::script::support::into_vec_of_strings;
use crate::support::md::MdSectionIter;
use mlua::{IntoLua, Lua, Value};

// endregion: --- Modules

/// ## Lua Documentation
///
/// Loads markdown sections from a file, optionally filtering by specific heading names.
///
/// ```lua
/// -- API Signature
/// aip.file.load_md_sections(
///   path: string,
///   headings?: string | list<string>
/// ): list<MdSection>
/// ```
pub(super) fn file_load_md_sections(
	lua: &Lua,
	runtime: &Runtime,
	path: String,
	headings: Option<Value>,
) -> mlua::Result<Value> {
	let headings = headings
		.map(|headings| into_vec_of_strings(headings, "aip.file.load_md_sections headings argument"))
		.transpose()?;

	let headings: Option<Vec<&str>> = headings
		.as_deref()
		.map(|vec| vec.iter().map(|s| s.as_str()).collect::<Vec<&str>>());

	let path = runtime
		.dir_context()
		.resolve_path(runtime.session(), path.into(), PathResolver::WksDir, None)?;
	let sec_iter = MdSectionIter::from_path(path, headings.as_deref())?;
	let sections = sec_iter.collect::<Vec<_>>();
	sections.into_lua(lua)
}

/// ## Lua Documentation
///
/// Splits a markdown file into three parts based on the *first* heading encountered.
///
/// ```lua
/// -- API Signature
/// aip.file.load_md_split_first(path: string): MdSplitFirst
/// ```
pub(super) fn file_load_md_split_first(lua: &Lua, runtime: &Runtime, path: String) -> mlua::Result<Value> {
	let path = runtime
		.dir_context()
		.resolve_path(runtime.session(), path.into(), PathResolver::WksDir, None)?;
	let mut sec_iter = MdSectionIter::from_path(path, None)?;
	let split_first = sec_iter.split_first();
	split_first.into_lua(lua)
}

// region:    --- Tests

#[cfg(test)]
mod tests {
	type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>;

	use crate::_test_support::{assert_contains, assert_not_contains, run_reflective_agent};
	use value_ext::JsonValueExt;

	#[tokio::test]
	async fn test_lua_file_load_md_sections_heading_1_top_ok() -> Result<()> {
		// -- Setup & Fixtures
		let fx_path = "other/md-sections.md";

		// -- Exec
		let mut res = run_reflective_agent(
			&format!(r##"return aip.file.load_md_sections("{fx_path}", {{"# Heading 1   "}})"##),
			None,
		)
		.await?;

		// -- Check
		let first_item = res
			.as_array_mut()
			.ok_or("Res should be array")?
			.iter_mut()
			.next()
			.ok_or("Should have at least one item")?;

		let content = first_item.x_get_str("content")?;
		let heading_content = first_item.x_get_str("/heading_content")?;
		let heading_level = first_item.x_get_i64("/heading_level")?;
		let heading_name = first_item.x_get_str("/heading_name")?;
		assert_eq!(heading_level, 1, "heading level");
		assert_contains(heading_content, "# Heading 1");
		assert_contains(heading_name, "Heading 1");
		assert_contains(content, "heading-1-content");
		assert_contains(content, "sub heading 1-a");
		assert_contains(content, "heading-1-a-blockquote");
		assert_not_contains(content, "{content}-2");
		assert_not_contains(content, "heading-2-blockquote");

		Ok(())
	}

	#[tokio::test]
	async fn test_lua_file_load_md_split_first_ok() -> Result<()> {
		// -- Setup & Fixtures
		let fx_path = "other/md-sections.md";

		// -- Exec
		let res = run_reflective_agent(&format!(r##"return aip.file.load_md_split_first("{fx_path}")"##), None).await?;

		// -- Check
		let before = res.x_get_str("/before")?;
		assert_eq!(before, "", "before should be empty");
		assert_eq!(
			res.x_get_str("/first/heading_content")?,
			"",
			"heading_content should be empty"
		);
		assert_eq!(res.x_get_i64("/first/heading_level")?, 0, "heading level should be 0");
		let content = res.x_get_str("/first/content")?;
		assert_contains(content, "Some early text");
		assert_contains(content, "- and more early text");
		assert_not_contains(content, "# Heading 1");
		assert_not_contains(content, "Some heading-1-content");
		let after = res.x_get_str("/after")?;
		assert_contains(after, "# Heading 1");
		assert_contains(after, "Some heading-1-content");
		assert_contains(after, "# Heading three");

		Ok(())
	}
}

// endregion: --- Tests