harn_vm/
stdlib.rs

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

harn_vm/
stdlib.rs
