Skip to main content

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 io;
11mod json;
12mod logging;
13mod math;
14pub mod process;
15mod regex;
16mod sets;
17mod shapes;
18mod strings;
19mod testing;
20mod tools;
21mod tracing;
22mod types;
23
24use crate::http::register_http_builtins;
25use crate::llm::register_llm_builtins;
26use crate::mcp::register_mcp_builtins;
27use crate::mcp_server::register_mcp_server_builtins;
28use crate::vm::Vm;
29
30// Re-export helpers used by other modules in harn-vm
31pub(crate) use json::json_to_vm_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: types, math, strings, json, datetime, regex, crypto,
37/// sets, shapes, testing. These are pure/deterministic and require no I/O.
38pub fn register_core_stdlib(vm: &mut Vm) {
39    types::register_type_builtins(vm);
40    math::register_math_builtins(vm);
41    strings::register_string_builtins(vm);
42    json::register_json_builtins(vm);
43    datetime::register_datetime_builtins(vm);
44    regex::register_regex_builtins(vm);
45    crypto::register_crypto_builtins(vm);
46    sets::register_set_builtins(vm);
47    shapes::register_shape_builtins(vm);
48    testing::register_testing_builtins(vm);
49}
50
51/// Register I/O builtins: filesystem, process, logging, tracing, I/O.
52/// Requires OS access (file reads, process spawning, environment vars).
53pub fn register_io_stdlib(vm: &mut Vm) {
54    io::register_io_builtins(vm);
55    fs::register_fs_builtins(vm);
56    process::register_process_builtins(vm);
57    process::register_path_builtins(vm);
58    tracing::register_tracing_builtins(vm);
59}
60
61/// Register agent builtins: concurrency, tools, agents, HTTP, LLM, MCP.
62/// 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/// Reset thread-local stdlib state (logging, tracing, source dir). Call between test runs.
81pub fn reset_stdlib_state() {
82    logging::reset_logging_state();
83    process::reset_process_state();
84}