harn_vm/
stdlib.rs

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

harn_vm/
stdlib.rs
