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 monitors;
22mod path;
23pub mod process;
24mod project;
25mod project_catalog;
26mod project_enrich;
27mod regex;
28mod review;
29pub(crate) mod sandbox;
30pub mod secret_scan;
31mod sets;
32mod shapes;
33mod skills;
34mod strings;
35pub mod template;
36mod testing;
37pub(crate) mod tools;
38pub mod tracing;
39mod transcript_compact;
40mod triggers_stdlib;
41mod types;
42mod vision;
43pub(crate) mod waitpoint;
44mod waitpoints;
45pub mod workflow_messages;
46
47use crate::http::register_http_builtins;
48use crate::llm::register_llm_builtins;
49use crate::mcp::register_mcp_builtins;
50use crate::mcp_server::register_mcp_server_builtins;
51use crate::vm::Vm;
52
53pub(crate) use crate::schema::{json_to_vm_value, schema_result_value};
54pub(crate) fn set_thread_source_dir(dir: &std::path::Path) {
55    process::set_thread_source_dir(dir);
56}
57
58/// Register core builtins: pure/deterministic, no I/O.
59pub fn register_core_stdlib(vm: &mut Vm) {
60    crate::runtime_context::register_runtime_context_builtins(vm);
61    types::register_type_builtins(vm);
62    math::register_math_builtins(vm);
63    strings::register_string_builtins(vm);
64    json::register_json_builtins(vm);
65    datetime::register_datetime_builtins(vm);
66    regex::register_regex_builtins(vm);
67    bytes::register_bytes_builtins(vm);
68    crypto::register_crypto_builtins(vm);
69    path::register_path_helper_builtins(vm);
70    sets::register_set_builtins(vm);
71    iter::register_iter_builtins(vm);
72    shapes::register_shape_builtins(vm);
73    testing::register_testing_builtins(vm);
74}
75
76/// Register I/O builtins (requires OS access).
77pub fn register_io_stdlib(vm: &mut Vm) {
78    io::register_io_builtins(vm);
79    host::register_host_builtins(vm);
80    fs::register_fs_builtins(vm);
81    vision::register_vision_builtins(vm);
82    agent_state::register_agent_state_builtins(vm);
83    process::register_process_builtins(vm);
84    process::register_path_builtins(vm);
85    project::register_project_builtins(vm);
86    tracing::register_tracing_builtins(vm);
87}
88
89/// Register agent builtins (requires network access and async runtime).
90pub fn register_agent_stdlib(vm: &mut Vm) {
91    concurrency::register_concurrency_builtins(vm);
92    connectors::register_connector_builtins(vm);
93    review::register_review_builtins(vm);
94    secret_scan::register_secret_scan_builtins(vm);
95    tools::register_tool_builtins(vm);
96    skills::register_skill_builtins(vm);
97    agents_daemon::register_daemon_builtins(vm);
98    triggers_stdlib::register_trigger_builtins(vm);
99    waitpoints::register_waitpoint_builtins(vm);
100    monitors::register_monitor_builtins(vm);
101    hitl::register_hitl_builtins(vm);
102    hitl_read::register_hitl_read_builtins(vm);
103    waitpoint::register_waitpoint_builtins(vm);
104    agents::register_agent_builtins(vm);
105    agent_sessions::register_agent_session_builtins(vm);
106    workflow_messages::register_workflow_message_builtins(vm);
107    transcript_compact::register_transcript_compaction_builtins(vm);
108    register_http_builtins(vm);
109    register_llm_builtins(vm);
110    register_mcp_builtins(vm);
111    register_mcp_server_builtins(vm);
112}
113
114/// Register all standard builtins on a VM (core + io + agent).
115pub fn register_vm_stdlib(vm: &mut Vm) {
116    register_core_stdlib(vm);
117    register_io_stdlib(vm);
118    register_agent_stdlib(vm);
119}
120
121/// Return the canonical list of all stdlib builtin names. Used by
122/// harn-lint and harn-lsp to avoid hardcoded duplicate lists.
123pub fn stdlib_builtin_names() -> Vec<String> {
124    let mut vm = Vm::new();
125    register_vm_stdlib(&mut vm);
126    // Name-only introspection — the path is never accessed, but passing
127    // a real per-platform temp dir keeps the registration logic honest
128    // when the callee someday decides it needs a valid parent.
129    let tmp = std::env::temp_dir();
130    crate::store::register_store_builtins(&mut vm, &tmp);
131    crate::checkpoint::register_checkpoint_builtins(&mut vm, &tmp, "default");
132    crate::metadata::register_metadata_builtins(&mut vm, &tmp);
133    crate::metadata::register_scan_builtins(&mut vm);
134    let mut names = vm.builtin_names();
135    // Special opcodes/keywords, not registered builtins, but linter
136    // should recognize them as valid function calls.
137    for extra in [
138        "spawn",
139        "await",
140        "cancel",
141        "cancel_graceful",
142        "is_cancelled",
143    ] {
144        names.push(extra.to_string());
145    }
146    names
147}
148
149/// Reset thread-local stdlib state. Call between test runs.
150pub fn reset_stdlib_state() {
151    logging::reset_logging_state();
152    process::reset_process_state();
153    sandbox::reset_sandbox_state();
154    fs::reset_fs_state();
155    json::reset_json_state();
156    host::reset_host_state();
157    hitl::reset_hitl_state();
158    monitors::reset_monitor_state();
159    waitpoints::reset_waitpoint_state();
160    waitpoint::reset_waitpoint_state();
161    agents::records::reset_eval_metrics();
162    tools::clear_current_tool_registry();
163    vision::reset_vision_state();
164    crate::skills::clear_current_skill_registry();
165    template::reset_prompt_registry();
166}
harn_vm/stdlib.rs

harn_vm/
stdlib.rs
