From 182a05b85113631a611169a6724162ae9247b55e Mon Sep 17 00:00:00 2001
From: mryouse
Date: Sun, 26 Jun 2022 01:31:14 +0000
Subject: add return types to builtins

---
 neb/std/fs.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'neb/std/fs.py')

diff --git a/neb/std/fs.py b/neb/std/fs.py
index 5983e20..495c7ea 100644
--- a/neb/std/fs.py
+++ b/neb/std/fs.py
@@ -8,13 +8,13 @@ FS = Environment()
 def interpretExists(symbol, args, env, ns):
     return Bool(Path(args[0].value).resolve().exists())
 
-FS.register("exists?", Builtin("exists?", interpretExists, [Arg("filename", TypeEnum.STRING)]))
+FS.register("exists?", Builtin("exists?", interpretExists, [Arg("filename", TypeEnum.STRING)], return_type=Type(":bool")))
 
 def interpretGlob(symbol, args, env, ns):
     items = glob(args[0].value)
     return List([String(item) for item in items])
 
-FS.register("glob", Builtin("glob", interpretGlob, [Arg("regex", TypeEnum.STRING)]))
+FS.register("glob", Builtin("glob", interpretGlob, [Arg("regex", TypeEnum.STRING)], return_type=Type(":string")))
 
 def interpretUnlink(symbol, args, env, ns):
     target_path = Path(args[0].value).resolve()
@@ -23,7 +23,7 @@ def interpretUnlink(symbol, args, env, ns):
     target_path.unlink()
     return List([])
 
-FS.register("unlink", Builtin("unlink", interpretUnlink, [Arg("filename", TypeEnum.STRING)]))
+FS.register("unlink", Builtin("unlink", interpretUnlink, [Arg("filename", TypeEnum.STRING)], return_type=Type(":list")))
 
 def interpretWithWrite(symbol, args, env, ns):
     target_file = args[0]
@@ -57,5 +57,5 @@ def interpretReadLines(symbol, args, env, ns):
     out = List([String(d) for d in data]) # all lines are strings
     return out
 
-FS.register("read-lines", Builtin("read-lines", interpretReadLines, [Arg("filename", TypeEnum.STRING)]))
+FS.register("read-lines", Builtin("read-lines", interpretReadLines, [Arg("filename", TypeEnum.STRING)], return_type=Type(":list")))
 
-- 
cgit v1.2.3