finish engram-lang: protocols, decorators, imports, Result, closures, stdlib, integration tests

2026-04-27 20:22:23 -05:00
parent 316c0a85ce
commit c427a0adc0
32 changed files with 2878 additions and 32 deletions
@@ -0,0 +1,56 @@
+//! Map<K, V> operations: get, set, remove, contains_key, keys, values, entries, merge.
+
+use el_types::{Type, TypeEnv};
+use super::fn_type;
+
+pub fn register(env: &mut TypeEnv) {
+    let map_unk = Type::Map { key: Box::new(Type::Unknown), value: Box::new(Type::Unknown) };
+    let arr_unk = Type::Array(Box::new(Type::Unknown));
+
+    env.register_fn("map_get",          fn_type(vec![map_unk.clone(), Type::Unknown], Type::Optional(Box::new(Type::Unknown))));
+    env.register_fn("map_set",          fn_type(vec![map_unk.clone(), Type::Unknown, Type::Unknown], map_unk.clone()));
+    env.register_fn("map_remove",       fn_type(vec![map_unk.clone(), Type::Unknown], map_unk.clone()));
+    env.register_fn("map_contains_key", fn_type(vec![map_unk.clone(), Type::Unknown], Type::Bool));
+    env.register_fn("map_keys",         fn_type(vec![map_unk.clone()], arr_unk.clone()));
+    env.register_fn("map_values",       fn_type(vec![map_unk.clone()], arr_unk.clone()));
+    env.register_fn("map_entries",      fn_type(vec![map_unk.clone()], arr_unk.clone()));
+    env.register_fn("map_merge",        fn_type(vec![map_unk.clone(), map_unk.clone()], map_unk.clone()));
+    env.register_fn("map_size",         fn_type(vec![map_unk.clone()], Type::Int));
+    env.register_fn("map_is_empty",     fn_type(vec![map_unk.clone()], Type::Bool));
+    env.register_fn("map_new",          fn_type(vec![], map_unk));
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    fn env() -> TypeEnv {
+        let mut e = TypeEnv::with_builtins();
+        register(&mut e);
+        e
+    }
+
+    #[test]
+    fn test_map_get_registered() {
+        assert!(env().lookup_fn("map_get").is_some());
+    }
+
+    #[test]
+    fn test_map_get_returns_optional() {
+        let e = env();
+        let ty = e.lookup_fn("map_get").unwrap();
+        assert!(matches!(ty, Type::Fn { return_type, .. } if matches!(return_type.as_ref(), Type::Optional(_))));
+    }
+
+    #[test]
+    fn test_map_contains_key_returns_bool() {
+        let e = env();
+        let ty = e.lookup_fn("map_contains_key").unwrap();
+        assert!(matches!(ty, Type::Fn { return_type, .. } if matches!(return_type.as_ref(), Type::Bool)));
+    }
+
+    #[test]
+    fn test_map_merge_registered() {
+        assert!(env().lookup_fn("map_merge").is_some());
+    }
+}