From 3e70f2ce47ee954640df9ee3b432964c1111b5ed Mon Sep 17 00:00:00 2001
From: mryouse
Date: Tue, 10 May 2022 02:46:42 +0000
Subject: refactor: std to own file, implement 'exit', multiple sigs

---
 std.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 std.py

(limited to 'std.py')

diff --git a/std.py b/std.py
new file mode 100644
index 0000000..dfa1cef
--- /dev/null
+++ b/std.py
@@ -0,0 +1,32 @@
+from tokens import *
+import sys
+from collections import namedtuple
+
+FuncImpl = namedtuple("FuncImpl", ("func", "impl"))
+
+STD = {}
+
+def std_exit(status=0):
+    sys.exit(status)
+    return NebLiteral(NebType.BOOL, True)
+
+def std_print(arg):
+    print(arg.value)
+    #return []  # TODO this should return empty list
+    return NebLiteral(NebType.BOOL, True)
+
+def build_std():
+    print_string = FuncImpl(NebFunction("print", [NebType.STRING], NebType.BOOL), std_print)
+    STD["print"] = {
+        print_string.func.in_sig(): print_string
+    }
+
+    exit_ = FuncImpl(NebFunction("exit", [], NebType.BOOL), std_exit)
+    exit_int = FuncImpl(NebFunction("exit", [NebType.INT], NebType.BOOL), std_exit)
+
+    STD["exit"] = {
+        exit_.func.in_sig(): exit_,
+        exit_int.func.in_sig(): exit_int
+    }
+    
+build_std()
-- 
cgit v1.2.3