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

harn_vm/
stdlib.rs
