harn_vm/
stdlib.rs

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

harn_vm/
stdlib.rs
