harn_vm/
stdlib.rs

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

harn_vm/
stdlib.rs
