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

harn_vm/
stdlib.rs
