devai 0.5.12

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

// region:    --- Modules

mod file_common;
mod file_md;

use crate::run::RuntimeContext;
use crate::script::lua_script::utils_file::file_common::{
	file_append, file_ensure_exists, file_first, file_list, file_list_load, file_load, file_save, EnsureExistsOptions,
};
use crate::script::lua_script::utils_file::file_md::{file_load_md_sections, file_load_md_split_first};
use crate::Result;
use mlua::{Lua, Table, Value};

// endregion: --- Modules

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

	// -- load
	let ctx = runtime_context.clone();
	let file_load_fn =
		lua.create_function(move |lua, (path, options): (String, Option<Value>)| file_load(lua, &ctx, path, options))?;

	// -- save
	let ctx = runtime_context.clone();
	let file_save_fn =
		lua.create_function(move |lua, (path, content): (String, String)| file_save(lua, &ctx, path, content))?;

	// -- append
	let ctx = runtime_context.clone();
	let file_append_fn =
		lua.create_function(move |lua, (path, content): (String, String)| file_append(lua, &ctx, path, content))?;

	// -- ensure_exists
	// (md_content, lang_name): (String, Option<String>)
	let ctx = runtime_context.clone();
	let file_ensure_exists_fn = lua.create_function(
		move |lua, (path, content, options): (String, Option<String>, Option<EnsureExistsOptions>)| {
			file_ensure_exists(lua, &ctx, path, content, options)
		},
	)?;

	// -- list
	let ctx = runtime_context.clone();
	let file_list_fn =
		lua.create_function(move |lua, (globs, options): (Value, Option<Value>)| file_list(lua, &ctx, globs, options))?;

	// -- list_load
	let ctx = runtime_context.clone();
	let file_list_load_fn = lua.create_function(move |lua, (globs, options): (Value, Option<Value>)| {
		file_list_load(lua, &ctx, globs, options)
	})?;

	// -- first
	let ctx = runtime_context.clone();
	let file_first_fn = lua
		.create_function(move |lua, (globs, options): (Value, Option<Value>)| file_first(lua, &ctx, globs, options))?;

	// -- load_md_sections
	let ctx = runtime_context.clone();
	let file_load_md_sections_fn = lua.create_function(move |lua, (path, headings): (String, Option<Value>)| {
		file_load_md_sections(lua, &ctx, path, headings)
	})?;

	// -- load_md_split_first
	let ctx = runtime_context.clone();
	let file_load_md_split_first_fn =
		lua.create_function(move |lua, (path,): (String,)| file_load_md_split_first(lua, &ctx, path))?;

	// -- All all function to the module
	table.set("load", file_load_fn)?;
	table.set("save", file_save_fn)?;
	table.set("append", file_append_fn)?;
	table.set("ensure_exists", file_ensure_exists_fn)?;
	table.set("list", file_list_fn)?;
	table.set("list_load", file_list_load_fn)?;
	table.set("first", file_first_fn)?;
	table.set("load_md_sections", file_load_md_sections_fn)?;
	table.set("load_md_split_first", file_load_md_split_first_fn)?;

	Ok(table)
}