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

harn_vm/
stdlib.rs
