jetro_core/eval/
builtins.rs

1//! Static builtin registry — zero vtable overhead via raw function pointers.
2//!
3//! All builtins are `fn(Val, &[Arg], &Env) -> Result<Val, EvalError>`.
4//! The registry is initialized once on first use via `OnceLock`.
5
6use std::collections::HashMap;
7use std::sync::OnceLock;
8use std::sync::Arc;
9
10use crate::ast::Arg;
11
12use super::{Env, EvalError, eval_pos, apply_item};
13use super::value::Val;
14use super::util::{val_to_string, val_str, field_exists_nested};
15use super::{func_strings, func_arrays, func_objects, func_paths, func_aggregates, func_csv};
16
17macro_rules! err {
18    ($($t:tt)*) => { Err(EvalError(format!($($t)*))) };
19}
20
21// ── Type alias ────────────────────────────────────────────────────────────────
22
23pub type BuiltinFn = fn(Val, &[Arg], &Env) -> Result<Val, EvalError>;
24
25// ── Registry ──────────────────────────────────────────────────────────────────
26
27pub struct BuiltinRegistry {
28    table: HashMap<&'static str, BuiltinFn>,
29}
30
31impl BuiltinRegistry {
32    #[inline]
33    pub fn get(&self, name: &str) -> Option<BuiltinFn> {
34        self.table.get(name).copied()
35    }
36
37    /// Iterate all registered builtin names (includes snake_case and camelCase aliases).
38    pub fn names(&self) -> impl Iterator<Item = &'static str> + '_ {
39        self.table.keys().copied()
40    }
41}
42
43/// Convenience — all builtin method names as a sorted `Vec<&'static str>`.
44pub fn all_names() -> Vec<&'static str> {
45    let mut v: Vec<&'static str> = global().names().collect();
46    v.sort_unstable();
47    v
48}
49
50static BUILTINS: OnceLock<BuiltinRegistry> = OnceLock::new();
51
52pub fn global() -> &'static BuiltinRegistry {
53    BUILTINS.get_or_init(build)
54}
55
56// ── Build ─────────────────────────────────────────────────────────────────────
57
58fn build() -> BuiltinRegistry {
59    let mut t: HashMap<&'static str, BuiltinFn> = HashMap::with_capacity(128);
60
61    // Basics
62    t.insert("len",        b_len);
63    t.insert("type",       b_type);
64    t.insert("to_string",  b_to_string);
65    t.insert("toString",   b_to_string);
66    t.insert("to_json",    b_to_json);
67    t.insert("toJson",     b_to_json);
68    t.insert("from_json",  b_from_json);
69    t.insert("fromJson",   b_from_json);
70
71    // Object
72    t.insert("keys",        b_keys);
73    t.insert("values",      b_values);
74    t.insert("entries",     b_entries);
75    t.insert("to_pairs",    b_to_pairs);
76    t.insert("toPairs",     b_to_pairs);
77    t.insert("from_pairs",  b_from_pairs);
78    t.insert("fromPairs",   b_from_pairs);
79    t.insert("invert",      b_invert);
80    t.insert("pick",        func_objects::pick);
81    t.insert("omit",        func_objects::omit);
82    t.insert("merge",       func_objects::merge);
83    t.insert("deep_merge",  func_objects::deep_merge_method);
84    t.insert("deepMerge",   func_objects::deep_merge_method);
85    t.insert("defaults",    func_objects::defaults);
86    t.insert("rename",      func_objects::rename);
87    t.insert("transform_keys",   func_objects::transform_keys);
88    t.insert("transformKeys",    func_objects::transform_keys);
89    t.insert("transform_values", func_objects::transform_values);
90    t.insert("transformValues",  func_objects::transform_values);
91    t.insert("filter_keys",   func_objects::filter_keys);
92    t.insert("filterKeys",    func_objects::filter_keys);
93    t.insert("filter_values", func_objects::filter_values);
94    t.insert("filterValues",  func_objects::filter_values);
95    t.insert("pivot",       func_objects::pivot);
96
97    // Arrays — full signature
98    t.insert("filter",    func_arrays::filter);
99    t.insert("map",       func_arrays::map);
100    t.insert("flatMap",   func_arrays::flat_map);
101    t.insert("flat_map",  func_arrays::flat_map);
102    t.insert("sort",      func_arrays::sort);
103    t.insert("flatten",   func_arrays::flatten);
104    t.insert("join",      func_arrays::join);
105    t.insert("equi_join", func_arrays::equi_join);
106    t.insert("equiJoin",  func_arrays::equi_join);
107    t.insert("first",     func_arrays::first);
108    t.insert("last",      func_arrays::last);
109    t.insert("nth",       func_arrays::nth);
110    t.insert("append",    func_arrays::append);
111    t.insert("prepend",   func_arrays::prepend);
112    t.insert("remove",    func_arrays::remove);
113    t.insert("diff",      func_arrays::diff);
114    t.insert("intersect", func_arrays::intersect);
115    t.insert("union",     func_arrays::union);
116    t.insert("enumerate", func_arrays::enumerate);
117    t.insert("window",    func_arrays::window);
118    t.insert("chunk",     func_arrays::chunk);
119    t.insert("batch",     func_arrays::chunk);
120    t.insert("takewhile", func_arrays::takewhile);
121    t.insert("take_while",func_arrays::takewhile);
122    t.insert("dropwhile", func_arrays::dropwhile);
123    t.insert("drop_while",func_arrays::dropwhile);
124    t.insert("accumulate",func_arrays::accumulate);
125    t.insert("partition", func_arrays::partition);
126    t.insert("zip",       func_arrays::zip_method);
127    t.insert("zip_longest",  func_arrays::zip_longest_method);
128    t.insert("zipLongest",   func_arrays::zip_longest_method);
129
130    // Arrays — zero-extra-arg wrappers
131    t.insert("reverse",  b_reverse);
132    t.insert("unique",   b_unique);
133    t.insert("distinct", b_unique);
134    t.insert("compact",  b_compact);
135    t.insert("pairwise", b_pairwise);
136
137    // Aggregates — full signature
138    t.insert("sum",      func_aggregates::sum);
139    t.insert("avg",      func_aggregates::avg);
140    t.insert("count",    func_aggregates::count);
141    t.insert("groupBy",  func_aggregates::group_by);
142    t.insert("group_by", func_aggregates::group_by);
143    t.insert("countBy",  func_aggregates::count_by);
144    t.insert("count_by", func_aggregates::count_by);
145    t.insert("indexBy",  func_aggregates::index_by);
146    t.insert("index_by", func_aggregates::index_by);
147
148    // Aggregates — bool-flag wrappers
149    t.insert("min", b_min);
150    t.insert("max", b_max);
151    t.insert("any", b_any);
152    t.insert("all", b_all);
153
154    // Paths
155    t.insert("get_path",       func_paths::get_path);
156    t.insert("getPath",        func_paths::get_path);
157    t.insert("set_path",       func_paths::set_path);
158    t.insert("setPath",        func_paths::set_path);
159    t.insert("del_path",       func_paths::del_path);
160    t.insert("delPath",        func_paths::del_path);
161    t.insert("del_paths",      func_paths::del_paths);
162    t.insert("delPaths",       func_paths::del_paths);
163    t.insert("has_path",       func_paths::has_path);
164    t.insert("hasPath",        func_paths::has_path);
165    t.insert("flatten_keys",   func_paths::flatten_keys);
166    t.insert("flattenKeys",    func_paths::flatten_keys);
167    t.insert("unflatten_keys", func_paths::unflatten_keys);
168    t.insert("unflattenKeys",  func_paths::unflatten_keys);
169
170    // CSV
171    t.insert("to_csv", b_to_csv);
172    t.insert("toCsv",  b_to_csv);
173    t.insert("to_tsv", b_to_tsv);
174    t.insert("toTsv",  b_to_tsv);
175
176    // Null safety / existence
177    t.insert("or",       b_or);
178    t.insert("has",      b_has);
179    t.insert("missing",  b_missing);
180    t.insert("includes", b_includes);
181    t.insert("contains", b_includes);
182
183    // Update / set
184    t.insert("set",    b_set);
185    t.insert("update", b_update);
186
187    // String methods
188    t.insert("upper",          func_strings::upper);
189    t.insert("lower",          func_strings::lower);
190    t.insert("capitalize",     func_strings::capitalize);
191    t.insert("title_case",     func_strings::title_case);
192    t.insert("titleCase",      func_strings::title_case);
193    t.insert("trim",           func_strings::trim);
194    t.insert("trim_left",      func_strings::trim_left);
195    t.insert("trimLeft",       func_strings::trim_left);
196    t.insert("trim_right",     func_strings::trim_right);
197    t.insert("trimRight",      func_strings::trim_right);
198    t.insert("lines",          func_strings::lines);
199    t.insert("words",          func_strings::words);
200    t.insert("chars",          func_strings::chars);
201    t.insert("to_number",      func_strings::to_number);
202    t.insert("toNumber",       func_strings::to_number);
203    t.insert("to_bool",        func_strings::to_bool);
204    t.insert("toBool",         func_strings::to_bool);
205    t.insert("to_base64",      func_strings::to_base64);
206    t.insert("toBase64",       func_strings::to_base64);
207    t.insert("from_base64",    func_strings::from_base64);
208    t.insert("fromBase64",     func_strings::from_base64);
209    t.insert("url_encode",     func_strings::url_encode);
210    t.insert("urlEncode",      func_strings::url_encode);
211    t.insert("url_decode",     func_strings::url_decode);
212    t.insert("urlDecode",      func_strings::url_decode);
213    t.insert("html_escape",    func_strings::html_escape);
214    t.insert("htmlEscape",     func_strings::html_escape);
215    t.insert("html_unescape",  func_strings::html_unescape);
216    t.insert("htmlUnescape",   func_strings::html_unescape);
217    t.insert("repeat",         func_strings::repeat);
218    t.insert("pad_left",       func_strings::pad_left);
219    t.insert("padLeft",        func_strings::pad_left);
220    t.insert("pad_right",      func_strings::pad_right);
221    t.insert("padRight",       func_strings::pad_right);
222    t.insert("starts_with",    func_strings::starts_with);
223    t.insert("startsWith",     func_strings::starts_with);
224    t.insert("ends_with",      func_strings::ends_with);
225    t.insert("endsWith",       func_strings::ends_with);
226    t.insert("index_of",       func_strings::index_of);
227    t.insert("indexOf",        func_strings::index_of);
228    t.insert("last_index_of",  func_strings::last_index_of);
229    t.insert("lastIndexOf",    func_strings::last_index_of);
230    t.insert("replace",        func_strings::replace);
231    t.insert("replace_all",    func_strings::replace_all);
232    t.insert("replaceAll",     func_strings::replace_all);
233    t.insert("strip_prefix",   func_strings::strip_prefix);
234    t.insert("stripPrefix",    func_strings::strip_prefix);
235    t.insert("strip_suffix",   func_strings::strip_suffix);
236    t.insert("stripSuffix",    func_strings::strip_suffix);
237    t.insert("slice",          func_strings::str_slice);
238    t.insert("split",          func_strings::split);
239    t.insert("indent",         func_strings::indent);
240    t.insert("dedent",         func_strings::dedent);
241    t.insert("matches",        func_strings::str_matches);
242    t.insert("scan",           func_strings::scan);
243
244    BuiltinRegistry { table: t }
245}
246
247// ── Wrapper functions ─────────────────────────────────────────────────────────
248
249fn b_len(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
250    Ok(Val::Int(match &recv {
251        Val::Arr(a) => a.len() as i64,
252        Val::Obj(m) => m.len() as i64,
253        Val::Str(s) => s.chars().count() as i64,
254        _ => return err!("len: unsupported type"),
255    }))
256}
257
258fn b_type(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
259    Ok(Val::Str(Arc::from(recv.type_name())))
260}
261
262fn b_to_string(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
263    Ok(val_str(&val_to_string(&recv)))
264}
265
266fn b_to_json(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
267    let sv: serde_json::Value = recv.into();
268    Ok(val_str(&serde_json::to_string(&sv).unwrap_or_default()))
269}
270
271fn b_from_json(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
272    let s = val_to_string(&recv);
273    let sv: serde_json::Value = serde_json::from_str(&s)
274        .map_err(|e| EvalError(format!("from_json: {}", e)))?;
275    Ok(Val::from(&sv))
276}
277
278fn b_keys(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
279    func_objects::keys(recv)
280}
281
282fn b_values(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
283    func_objects::values(recv)
284}
285
286fn b_entries(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
287    func_objects::entries(recv)
288}
289
290fn b_to_pairs(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
291    func_objects::to_pairs(recv)
292}
293
294fn b_from_pairs(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
295    func_objects::from_pairs(recv)
296}
297
298fn b_invert(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
299    func_objects::invert(recv)
300}
301
302fn b_reverse(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
303    func_arrays::reverse(recv)
304}
305
306fn b_unique(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
307    func_arrays::unique(recv)
308}
309
310fn b_compact(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
311    func_arrays::compact(recv)
312}
313
314fn b_pairwise(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
315    func_arrays::pairwise(recv)
316}
317
318fn b_min(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
319    func_aggregates::minmax(recv, args, env, false)
320}
321
322fn b_max(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
323    func_aggregates::minmax(recv, args, env, true)
324}
325
326fn b_any(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
327    func_aggregates::any_all(recv, args, env, false)
328}
329
330fn b_all(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
331    func_aggregates::any_all(recv, args, env, true)
332}
333
334fn b_to_csv(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
335    Ok(val_str(&func_csv::to_csv(&recv)))
336}
337
338fn b_to_tsv(recv: Val, _: &[Arg], _: &Env) -> Result<Val, EvalError> {
339    Ok(val_str(&func_csv::to_tsv(&recv)))
340}
341
342fn b_or(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
343    let default = args.first().map(|a| eval_pos(a, env)).transpose()?.unwrap_or(Val::Null);
344    if recv.is_null() { Ok(default) } else { Ok(recv) }
345}
346
347fn b_has(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
348    let result = args.iter().all(|a| {
349        eval_pos(a, env).ok()
350            .and_then(|v| v.as_str().map(|s| s.to_string()))
351            .map(|key| field_exists_nested(&recv, &key))
352            .unwrap_or(false)
353    });
354    Ok(Val::Bool(result))
355}
356
357fn b_missing(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
358    let key = args.first().map(|a| eval_pos(a, env)).transpose()?
359        .and_then(|v| v.as_str().map(|s| s.to_string()))
360        .unwrap_or_default();
361    Ok(Val::Bool(!field_exists_nested(&recv, &key)))
362}
363
364fn b_includes(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
365    let item = args.first().map(|a| eval_pos(a, env)).transpose()?.unwrap_or(Val::Null);
366    use super::util::val_to_key;
367    let key = val_to_key(&item);
368    Ok(Val::Bool(match &recv {
369        Val::Arr(a) => a.iter().any(|v| val_to_key(v) == key),
370        Val::Str(s) => s.contains(item.as_str().unwrap_or_default()),
371        _ => false,
372    }))
373}
374
375fn b_set(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
376    let _ = recv;
377    Ok(args.first().map(|a| eval_pos(a, env)).transpose()?.unwrap_or(Val::Null))
378}
379
380fn b_update(recv: Val, args: &[Arg], env: &Env) -> Result<Val, EvalError> {
381    let lam = args.first().ok_or_else(|| EvalError("update: requires lambda".into()))?;
382    apply_item(recv, lam, env)
383}
jetro_core/eval/builtins.rs

jetro_core/eval/
builtins.rs
