feat: runtime dict/list type

erg-lang · Sep 19, 2024 · 65deb68 · 65deb68
1 parent 222c2c3
commit 65deb68
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/crates/erg_compiler/context/initialize/classes.rs b/crates/erg_compiler/context/initialize/classes.rs
@@ -1685,6 +1685,19 @@ impl Context {
             Immutable,
             Visibility::BUILTIN_PUBLIC,
         );
+        if PYTHON_MODE {
+            let t_getitem = func1(
+                t_singleton(T.clone()),
+                t_singleton(unknown_len_list_t(T.clone())),
+            )
+            .quantify();
+            generic_list.register_builtin_erg_impl(
+                FUNDAMENTAL_GETITEM,
+                t_getitem,
+                Immutable,
+                Visibility::BUILTIN_PUBLIC,
+            );
+        }
         let mut list_hash = Self::builtin_methods(Some(mono(HASH)), 1);
         list_hash.register_builtin_erg_impl(
             OP_HASH,
@@ -2211,6 +2224,19 @@ impl Context {
             Immutable,
             Visibility::BUILTIN_PUBLIC,
         );
+        if PYTHON_MODE {
+            let t_getitem = func1(
+                tuple_t(vec![t_singleton(T.clone()), t_singleton(U.clone())]),
+                t_singleton(Type::from(dict! { T.clone() => U.clone() })),
+            )
+            .quantify();
+            generic_dict.register_builtin_erg_impl(
+                FUNDAMENTAL_GETITEM,
+                t_getitem,
+                Immutable,
+                Visibility::BUILTIN_PUBLIC,
+            );
+        }
         let dict_t = poly(DICT, vec![D.clone()]);
         let mut dict_ =
             // TODO: D <: GenericDict

diff --git a/crates/erg_compiler/ty/constructors.rs b/crates/erg_compiler/ty/constructors.rs
@@ -141,6 +141,15 @@ pub fn v_enum(s: Set<ValueObj>) -> Type {
     try_v_enum(s).unwrap_or_else(|set| panic!("not homogeneous: {}", set))
 }
 
+pub fn t_enum(s: Set<Type>) -> Type {
+    try_v_enum(s.into_iter().map(ValueObj::builtin_type).collect())
+        .unwrap_or_else(|set| panic!("not homogeneous: {}", set))
+}
+
+pub fn t_singleton(t: Type) -> Type {
+    t_enum(set! {t})
+}
+
 pub fn tp_enum(ty: Type, s: Set<TyParam>) -> Type {
     let name = FRESH_GEN.fresh_varname();
     let preds = s