mumuarray/
lib.rs

1// src/lib.rs
2#![allow(dead_code)]
3#![allow(unused_imports)]
4
5use mumu::{
6    parser::interpreter::Interpreter,
7    parser::types::{Value, FunctionValue},
8};
9use std::sync::{Arc, Mutex};
10
11mod simple;
12mod append;
13mod apply;
14mod assoc;
15mod collect;
16mod each;
17mod join;
18mod keys;
19mod key_eq;
20mod length;
21mod map;
22mod multi;
23mod nth;
24mod prop;
25mod range;
26mod reduce;
27mod search;
28mod sort;
29mod tail;
30mod util;
31mod transform; // directory of transforms
32
33/// Register every `array:*` bridge into the provided interpreter.
34/// This is suitable for *embedded/static* usage (e.g., WASM builds),
35/// so callers don’t need `extend("array")`.
36pub fn register_all(interp: &mut Interpreter) {
37    macro_rules! reg {
38        ($name:expr, $f:expr) => {{
39            let func = Arc::new(Mutex::new($f));
40            interp.register_dynamic_function($name, func);
41            interp.set_variable(
42                $name,
43                Value::Function(Box::new(FunctionValue::Named($name.into()))),
44            );
45        }};
46    }
47
48    // Keep this list in sync with the Cargo_lock FFI path below.
49    reg!("array:filter", simple::array_filter);
50    reg!("array:sort", sort::array_sort_bridge);
51    reg!("array:flatten", simple::array_flatten);
52    reg!("array:join", join::array_join_bridge);
53    reg!("array:zip", simple::array_zip);
54    reg!("array:zip_with", multi::array_zip_with_bridge);
55    reg!("array:prop", prop::array_prop_bridge);
56
57    reg!("array:append", append::array_append_bridge);
58    reg!("array:apply", apply::array_apply_bridge);
59    reg!("array:assoc", assoc::array_assoc_bridge);
60    reg!("array:chunk", transform::chunk::array_chunk_bridge);
61    reg!("array:collect", collect::array_collect_bridge);
62    reg!("array:compact", util::array_compact_bridge);
63    reg!("array:drop", transform::drop::array_drop_bridge);
64    reg!("array:drop_while", transform::drop_while::array_drop_while_bridge);
65    reg!("array:each", each::array_each_bridge);
66    reg!("array:every", search::array_every_bridge);
67    reg!("array:find", search::array_find_bridge);
68    reg!("array:findIndex", search::array_find_index_bridge);
69    reg!("array:flatten_deep", util::array_flatten_deep_bridge);
70    reg!("array:group_by", transform::group_by::array_group_by_bridge);
71    reg!("array:head", util::array_head_bridge);
72    reg!("array:includes", search::array_includes_bridge);
73    reg!("array:init", transform::init::array_init_bridge);
74    reg!("array:last", util::array_last_bridge);
75    reg!("array:length", length::array_length_bridge);
76    reg!("array:map", map::array_map_bridge);
77    reg!("array:multi_concat", multi::array_concat_bridge);
78    reg!("array:multi_difference", multi::array_difference_bridge);
79    reg!("array:multi_intersection", multi::array_intersection_bridge);
80    reg!("array:multi_union", multi::array_union_bridge);
81    reg!("array:multi_zip", multi::array_zip_bridge);
82    reg!("array:multi_zip_with", multi::array_zip_with_bridge);
83    reg!("array:nth", nth::array_nth_bridge);
84    reg!("array:partition", transform::partition::array_partition_bridge);
85    reg!("array:pluck", transform::pluck::array_pluck_bridge);
86    reg!("array:prop_partial", prop::array_prop_bridge);
87    reg!("array:range", range::array_range_bridge);
88    reg!("array:reduce", reduce::array_reduce_bridge);
89    reg!("array:repeat", transform::fill::array_repeat_bridge);
90    reg!("array:reverse", util::array_reverse_bridge);
91    reg!("array:slice", transform::slice::array_slice_bridge);
92    reg!("array:some", search::array_some_bridge);
93    reg!("array:sort_by", sort::array_sort_by_bridge);
94    reg!("array:sort_with", sort::array_sort_with_bridge);
95    reg!("array:tail", tail::array_tail_bridge);
96    reg!("array:take", transform::take::array_take_bridge);
97    reg!("array:take_while", transform::take_while::array_take_while_bridge);
98    reg!("array:uniq", util::array_uniq_bridge);
99    reg!("array:shuffle", util::array_shuffle_bridge);
100    reg!("array:keys", keys::array_keys_bridge);
101
102    // Ramda-like propEq
103    reg!("array:key_eq", key_eq::array_key_eq_bridge);
104}
105
106/// Dynamic-loader entry point used by native “host” builds.
107/// For embedded/static (e.g., WASM), call `register_all(...)` directly.
108#[no_mangle]
109pub unsafe extern "C" fn Cargo_lock(
110    interp_ptr: *mut std::ffi::c_void,
111    _extra_str: *const std::ffi::c_void,
112) -> i32 {
113    let interp = &mut *(interp_ptr as *mut Interpreter);
114    register_all(interp);
115    0
116}