harn_vm/
stdlib.rs

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

harn_vm/
stdlib.rs
