harn_vm/
stdlib.rs

1//! Standard library builtins for the Harn VM.
2
3mod agent_sessions;
4pub mod agent_state;
5mod agents;
6mod agents_daemon;
7mod concurrency;
8mod crypto;
9mod datetime;
10mod fs;
11pub mod host;
12mod io;
13mod iter;
14pub(crate) mod json;
15mod logging;
16mod math;
17mod path;
18pub mod process;
19mod project;
20mod project_catalog;
21mod project_enrich;
22mod regex;
23mod sets;
24mod shapes;
25mod skills;
26mod strings;
27pub mod template;
28mod testing;
29pub(crate) mod tools;
30pub mod tracing;
31mod transcript_compact;
32mod types;
33
34use crate::http::register_http_builtins;
35use crate::llm::register_llm_builtins;
36use crate::mcp::register_mcp_builtins;
37use crate::mcp_server::register_mcp_server_builtins;
38use crate::vm::Vm;
39
40pub(crate) use crate::schema::{json_to_vm_value, schema_result_value};
41pub(crate) fn set_thread_source_dir(dir: &std::path::Path) {
42    process::set_thread_source_dir(dir);
43}
44
45/// Register core builtins: pure/deterministic, no I/O.
46pub fn register_core_stdlib(vm: &mut Vm) {
47    types::register_type_builtins(vm);
48    math::register_math_builtins(vm);
49    strings::register_string_builtins(vm);
50    json::register_json_builtins(vm);
51    datetime::register_datetime_builtins(vm);
52    regex::register_regex_builtins(vm);
53    crypto::register_crypto_builtins(vm);
54    path::register_path_helper_builtins(vm);
55    sets::register_set_builtins(vm);
56    iter::register_iter_builtins(vm);
57    shapes::register_shape_builtins(vm);
58    testing::register_testing_builtins(vm);
59}
60
61/// Register I/O builtins (requires OS access).
62pub fn register_io_stdlib(vm: &mut Vm) {
63    io::register_io_builtins(vm);
64    host::register_host_builtins(vm);
65    fs::register_fs_builtins(vm);
66    agent_state::register_agent_state_builtins(vm);
67    process::register_process_builtins(vm);
68    process::register_path_builtins(vm);
69    project::register_project_builtins(vm);
70    tracing::register_tracing_builtins(vm);
71}
72
73/// Register agent builtins (requires network access and async runtime).
74pub fn register_agent_stdlib(vm: &mut Vm) {
75    concurrency::register_concurrency_builtins(vm);
76    tools::register_tool_builtins(vm);
77    skills::register_skill_builtins(vm);
78    agents_daemon::register_daemon_builtins(vm);
79    agents::register_agent_builtins(vm);
80    agent_sessions::register_agent_session_builtins(vm);
81    transcript_compact::register_transcript_compaction_builtins(vm);
82    register_http_builtins(vm);
83    register_llm_builtins(vm);
84    register_mcp_builtins(vm);
85    register_mcp_server_builtins(vm);
86}
87
88/// Register all standard builtins on a VM (core + io + agent).
89pub fn register_vm_stdlib(vm: &mut Vm) {
90    register_core_stdlib(vm);
91    register_io_stdlib(vm);
92    register_agent_stdlib(vm);
93}
94
95/// Return the canonical list of all stdlib builtin names. Used by
96/// harn-lint and harn-lsp to avoid hardcoded duplicate lists.
97pub fn stdlib_builtin_names() -> Vec<String> {
98    let mut vm = Vm::new();
99    register_vm_stdlib(&mut vm);
100    let tmp = std::path::PathBuf::from("/tmp");
101    crate::store::register_store_builtins(&mut vm, &tmp);
102    crate::checkpoint::register_checkpoint_builtins(&mut vm, &tmp, "default");
103    crate::metadata::register_metadata_builtins(&mut vm, &tmp);
104    crate::metadata::register_scan_builtins(&mut vm);
105    let mut names = vm.builtin_names();
106    // Special opcodes/keywords, not registered builtins, but linter
107    // should recognize them as valid function calls.
108    for extra in [
109        "spawn",
110        "await",
111        "cancel",
112        "cancel_graceful",
113        "is_cancelled",
114    ] {
115        names.push(extra.to_string());
116    }
117    names
118}
119
120/// Reset thread-local stdlib state. Call between test runs.
121pub fn reset_stdlib_state() {
122    logging::reset_logging_state();
123    process::reset_process_state();
124    fs::reset_fs_state();
125    json::reset_json_state();
126    host::reset_host_state();
127    agents::records::reset_eval_metrics();
128    tools::clear_current_tool_registry();
129    template::reset_prompt_registry();
130}
harn_vm/stdlib.rs

harn_vm/
stdlib.rs
