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

harn_vm/
stdlib.rs
