attempt to add a generic list type

1 files changed, 28 insertions, 4 deletions

diff --git a/neb/std/types.py b/neb/std/types.py
index f8d5405..3517384 100644
--- a/neb/std/types.py
+++ b/neb/std/types.py
@@ -1,4 +1,4 @@
-from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic, NebType
+from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic, NebType, TypeWrap
 from ..structs import *
 
 TYPES = Environment()
@@ -25,7 +25,29 @@ def interpretIsInt(symbol, args, env, ns):
 TYPES.register("int?", Builtin("int?", interpretIsInt, [Arg("arg", TypeEnum.ANY)], return_type=Type(":bool")))
 
 def interpretIsList(symbol, args, env, ns):
-    return Bool(isinstance(args[0], List))
+
+    # if it's not a list, we're done
+    if not isinstance(args[0], List):
+        return Bool(False)
+
+    # if the symbol isn't a type, we're done
+    elif not isinstance(symbol, Type):
+        return Bool(True)
+
+    inner = symbol.inner
+    if inner is None or inner.name == ":any":
+        return Bool(True)
+
+    inner_func = env.get(inner.name).is_func
+
+    for arg in args[0].args:
+        if inner_func(inner, [arg], env, ns).value is False:
+            return Bool(False)
+
+        #if not isinstance(arg, String):
+        #    return Bool(False)
+    return Bool(True)
+
 
 TYPES.register("list?", Builtin("list?", interpretIsList, [Arg("arg", TypeEnum.ANY)], return_type=Type(":bool")))
 
@@ -73,7 +95,8 @@ TYPES.register("literal?", Builtin("literal?", interpretIsLiteral, [Arg("arg", T
 any_type = NebType(":any", None, interpretIsAny)
 literal_type = NebType(":literal", any_type, interpretIsLiteral)
 string_type = NebType(":string", literal_type, interpretIsString)
-list_type = NebType(":list", any_type, interpretIsList)
+#list_type = NebType(":list", any_type, interpretIsList)
+bracket_type = NebType(":[]", any_type, interpretIsList)
 bool_type = NebType(":bool", literal_type, interpretIsBool)
 number_type = NebType(":number", literal_type, interpretIsNumber)
 int_type = NebType(":int", number_type, interpretIsInt)
@@ -82,7 +105,8 @@ float_type = NebType(":float", number_type, interpretIsFloat)
 TYPES.register(":any", any_type)
 TYPES.register(":literal", literal_type)
 TYPES.register(":string", string_type)
-TYPES.register(":list", list_type)
+#TYPES.register(":list", list_type)
+TYPES.register(":[]", bracket_type)
 TYPES.register(":bool", bool_type)
 TYPES.register(":number", number_type)
 TYPES.register(":int", int_type)