add return types to builtins

1 files changed, 6 insertions, 6 deletions

diff --git a/neb/std/strings.py b/neb/std/strings.py
index 68f02d5..7e3177e 100644
--- a/neb/std/strings.py
+++ b/neb/std/strings.py
@@ -10,12 +10,12 @@ def interpretConcat(symbol, args, env, ns):
     return String(out)
 
 string_arg = Arg("arg", TypeEnum.STRING)
-STRINGS.register("concat", Builtin("concat", interpretConcat, [string_arg, string_arg], string_arg))
+STRINGS.register("concat", Builtin("concat", interpretConcat, [string_arg, string_arg], string_arg, Type(":string")))
 
 def interpretStrip(symbol, args, env, ns):
     return String(args[0].value.strip())
 
-STRINGS.register("strip", Builtin("strip", interpretStrip, [Arg("filename", TypeEnum.STRING)]))
+STRINGS.register("strip", Builtin("strip", interpretStrip, [Arg("filename", TypeEnum.STRING)], return_type=Type(":string")))
 
 def interpretSplit(symbol, args, env, ns):
     target = args[0]
@@ -25,7 +25,7 @@ def interpretSplit(symbol, args, env, ns):
     ret = target.value.split(splitter.value)
     return List([String(r) for r in ret])
 
-STRINGS.register("split", Builtin("split", interpretSplit, [Arg("target", TypeEnum.STRING)], Arg("splitter", TypeEnum.STRING, optional=True)))
+STRINGS.register("split", Builtin("split", interpretSplit, [Arg("target", TypeEnum.STRING)], Arg("splitter", TypeEnum.STRING, optional=True), Type(":list")))
 
 def interpretJoin(symbol, args, env, ns):
     lst = args[0]
@@ -34,16 +34,16 @@ def interpretJoin(symbol, args, env, ns):
 
 join_list_arg = Arg("list", TypeEnum.LIST)
 join_string_arg = Arg("joiner", TypeEnum.STRING)
-STRINGS.register("join", Builtin("join", interpretJoin, [join_list_arg, join_string_arg]))
+STRINGS.register("join", Builtin("join", interpretJoin, [join_list_arg, join_string_arg], return_type=Type(":string")))
 
 def interpretFirstChar(symbol, args, env, ns):
     if len(args[0].value) == 0:
         raise InterpretPanic(symbol, ":string is empty", ev)
     return String(args[0].value[0])
 
-STRINGS.register("first-char", Builtin("first-char", interpretFirstChar, [Arg("string", TypeEnum.STRING)]))
+STRINGS.register("first-char", Builtin("first-char", interpretFirstChar, [Arg("string", TypeEnum.STRING)], return_type=Type(":string")))
 
 def interpretRestChar(symbol, args, env, ns):
     return String(args[0].value[1:])
 
-STRINGS.register("rest-char", Builtin("rest-char", interpretRestChar, [Arg("string", TypeEnum.STRING)]))
+STRINGS.register("rest-char", Builtin("rest-char", interpretRestChar, [Arg("string", TypeEnum.STRING)], return_type=Type(":string")))