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 json;
13mod logging;
14mod math;
15pub mod process;
16mod regex;
17mod sets;
18mod shapes;
19mod strings;
20mod testing;
21mod tools;
22mod 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
31// Re-export helpers used by other modules in harn-vm
32pub(crate) use json::json_to_vm_value;
33pub(crate) fn set_thread_source_dir(dir: &std::path::Path) {
34    process::set_thread_source_dir(dir);
35}
36
37/// Register core builtins: types, math, strings, json, datetime, regex, crypto,
38/// sets, shapes, testing. These are pure/deterministic and require 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    sets::register_set_builtins(vm);
48    shapes::register_shape_builtins(vm);
49    testing::register_testing_builtins(vm);
50}
51
52/// Register I/O builtins: filesystem, process, logging, tracing, I/O.
53/// Requires OS access (file reads, process spawning, environment vars).
54pub fn register_io_stdlib(vm: &mut Vm) {
55    io::register_io_builtins(vm);
56    host::register_host_builtins(vm);
57    fs::register_fs_builtins(vm);
58    process::register_process_builtins(vm);
59    process::register_path_builtins(vm);
60    tracing::register_tracing_builtins(vm);
61}
62
63/// Register agent builtins: concurrency, tools, agents, HTTP, LLM, MCP.
64/// 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    register_http_builtins(vm);
70    register_llm_builtins(vm);
71    register_mcp_builtins(vm);
72    register_mcp_server_builtins(vm);
73}
74
75/// Register all standard builtins on a VM (core + io + agent).
76pub fn register_vm_stdlib(vm: &mut Vm) {
77    register_core_stdlib(vm);
78    register_io_stdlib(vm);
79    register_agent_stdlib(vm);
80}
81
82/// Return the canonical list of all stdlib builtin names.
83/// This creates a temporary VM, registers all builtins, and collects the names.
84/// Used by 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    // Register path-dependent builtins with a dummy path so we capture their names.
89    let tmp = std::path::PathBuf::from("/tmp");
90    crate::store::register_store_builtins(&mut vm, &tmp);
91    crate::checkpoint::register_checkpoint_builtins(&mut vm, &tmp, "default");
92    crate::metadata::register_metadata_builtins(&mut vm, &tmp);
93    crate::metadata::register_scan_builtins(&mut vm, &tmp);
94    let mut names = vm.builtin_names();
95    // These are handled as special opcodes/keywords, not registered builtins,
96    // but the linter should recognize them as valid function calls.
97    for extra in [
98        "spawn",
99        "await",
100        "cancel",
101        "cancel_graceful",
102        "is_cancelled",
103    ] {
104        names.push(extra.to_string());
105    }
106    names
107}
108
109/// Reset thread-local stdlib state (logging, tracing, source dir). Call between test runs.
110pub fn reset_stdlib_state() {
111    logging::reset_logging_state();
112    process::reset_process_state();
113}
harn_vm/stdlib.rs

harn_vm/
stdlib.rs
