WIP add multifunc to fs

1 files changed, 41 insertions, 11 deletions

diff --git a/neb/std/fs.py b/neb/std/fs.py
index 144df0f..b49d63c 100644
--- a/neb/std/fs.py
+++ b/neb/std/fs.py
@@ -1,4 +1,4 @@
-from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic
+from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic, MultiFunction
 from ..structs import *
 from pathlib import Path
 from glob import glob
@@ -14,13 +14,19 @@ FS.register("_stderr_", Handle(sys.stderr))
 def interpretExists(symbol, args, env, ns):
     return Bool(Path(args[0].value).resolve().exists())
 
-FS.register("exists?", Builtin("exists?", interpretExists, [Arg("filename", TypeEnum.STRING)], return_type=Type(":bool")))
+exists_func = Builtin("exists?", interpretExists, [Arg("filename", TypeEnum.STRING)], return_type=Type(":bool"))
+exists_multi = MultiFunction("exists?")
+exists_multi.register(exists_func)
+FS.register("exists?", exists_multi)
 
 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)], return_type=Type(":string")))
+glob_func = Builtin("glob", interpretGlob, [Arg("regex", TypeEnum.STRING)], return_type=Type(":string"))
+glob_multi = MultiFunction("glob")
+glob_multi.register(glob_func)
+FS.register("glob", glob_multi)
 
 def interpretUnlink(symbol, args, env, ns):
     target_path = Path(args[0].value).resolve()
@@ -29,7 +35,10 @@ def interpretUnlink(symbol, args, env, ns):
     target_path.unlink()
     return List([])
 
-FS.register("unlink", Builtin("unlink", interpretUnlink, [Arg("filename", TypeEnum.STRING)], return_type=Type(":list")))
+unlink_func = Builtin("unlink", interpretUnlink, [Arg("filename", TypeEnum.STRING)], return_type=Type(":list"))
+unlink_multi = MultiFunction("unlink")
+unlink_multi.register(unlink_func)
+FS.register("unlink", unlink_multi)
 
 def interpretWrite(symbol, args, env, ns):
     string = args[0]
@@ -42,7 +51,10 @@ def interpretWrite(symbol, args, env, ns):
         raise InterpretPanic(symbol, f"{handle} is closed")
     return List([])
 
-FS.register("write", Builtin("write", interpretWrite, [Arg("string", TypeEnum.STRING), Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.LIST))
+write_func = Builtin("write", interpretWrite, [Arg("string", TypeEnum.STRING), Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.LIST)
+write_multi = MultiFunction("write")
+write_multi.register(write_func)
+FS.register("write", write_multi)
 
 def interpretOpenRead(symbol, args, env, ns):
     name = args[0].value
@@ -55,7 +67,10 @@ def interpretOpenRead(symbol, args, env, ns):
         raise InterpretPanic(symbol, "cannot open {fil} for reading")
     return Handle(f)
 
-FS.register("open-read", Builtin("open-read", interpretOpenRead, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE))
+openread_func = Builtin("open-read", interpretOpenRead, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE)
+openread_multi = MultiFunction("open-read")
+openread_multi.register(openread_func)
+FS.register("open-read", openread_multi)
 
 def interpretOpenWrite(symbol, args, env, ns):
     name = args[0].value
@@ -66,7 +81,10 @@ def interpretOpenWrite(symbol, args, env, ns):
         raise InterpretPanic(symbol, "cannot open {fil} for writing")
     return Handle(f)
 
-FS.register("open-write", Builtin("open-write", interpretOpenWrite, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE))
+openwrite_func = Builtin("open-write", interpretOpenWrite, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE)
+openwrite_multi = MultiFunction("open-write")
+openwrite_multi.register(openwrite_func)
+FS.register("open-write", openwrite_multi)
 
 def interpretOpenAppend(symbol, args, env, ns):
     name = args[0].value
@@ -77,7 +95,10 @@ def interpretOpenAppend(symbol, args, env, ns):
         raise InterpretPanic(symbol, "cannot open {fil} for appending")
     return Handle(f)
 
-FS.register("open-append", Builtin("open-append", interpretOpenAppend, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE))
+openappend_func = Builtin("open-append", interpretOpenAppend, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE)
+openappend_multi = MultiFunction("open-append")
+openappend_multi.register(openappend_func)
+FS.register("open-append", openappend_multi)
 
 def interpretClose(symbol, args, env, ns):
     try:
@@ -90,7 +111,10 @@ def interpretClose(symbol, args, env, ns):
         raise InterpretPanic(symbol, "cannot close {args[0]}")
     return List([])
 
-FS.register("close", Builtin("close", interpretClose, [Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.LIST))
+close_func = Builtin("close", interpretClose, [Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.LIST)
+close_multi = MultiFunction("close")
+close_multi.register(close_func)
+FS.register("close", close_multi)
 
 def interpretRead(symbol, args, env, ns):
     handle = args[0]
@@ -102,7 +126,10 @@ def interpretRead(symbol, args, env, ns):
         raise InterpretPanic(symbol, f"{handle} is closed")
     return String(inp)
 
-FS.register("read", Builtin("read", interpretRead, [Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.STRING))
+read_func = Builtin("read", interpretRead, [Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.STRING)
+read_multi = MultiFunction("read")
+read_multi.register(read_func)
+FS.register("read", read_multi)
 
 def interpretReadLines(symbol, args, env, ns):
     target_file_name = args[0].value
@@ -114,5 +141,8 @@ 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)], return_type=Type(":list")))
+readlines_func = Builtin("read-lines", interpretReadLines, [Arg("filename", TypeEnum.STRING)], return_type=Type(":list"))
+readlines_multi = MultiFunction("read-lines")
+readlines_multi.register(readlines_func)
+FS.register("read-lines", readlines_multi)