1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic, MultiFunction
from ..structs import *
import shlex
import subprocess
import sys
from datetime import datetime
import os
SYS = Environment()
def interpretArgv(symbol, args, env, ns):
out = []
for arg in sys.argv[1:]:
out.append(String(arg))
return List(out)
argv_func = Builtin("argv", interpretArgv, [], return_type=Type(":[]"))
argv_multi = MultiFunction("argv")
argv_multi.register(argv_func)
SYS.register("argv", argv_multi)
def interpretShell(symbol, args, env, ns):
ret = subprocess.run(shlex.split(args[0].value), capture_output=True)
return String(ret.stdout.decode("utf-8"))
shell_func = Builtin("$", interpretShell, [Arg("command", TypeEnum.STRING)], return_type=Type(":string"))
shell_multi = MultiFunction("$")
shell_multi.register(shell_func)
SYS.register("$", shell_multi)
def interpretExit(symbol, args, env, ns):
status = 0 if len(args) == 0 else args[0].value
sys.exit(status)
return Nil()
exit_func = Builtin("exit", interpretExit, [], return_type=Type(":nil"))
exit_status_func = Builtin("exit", interpretExit, [Arg("status", TypeEnum.INT)], return_type=Type(":nil"))
exit_multi = MultiFunction("exit")
exit_multi.register(exit_func)
exit_multi.register(exit_status_func)
SYS.register("exit", exit_multi)
def interpretPrint(symbol, args, env, ns):
print(args[0].value)
return Nil()
print_func = Builtin("print", interpretPrint, [Arg("arg", TypeEnum.STRING)], return_type=Type(":nil"))
print_multi = MultiFunction("print")
print_multi.register(print_func)
SYS.register("print", print_multi)
def interpretEnv(symbol, args, env, ns):
items = os.environ[args[0].value].split(":")
return List([String(item) for item in items])
env_func = Builtin("env", interpretEnv, [Arg("regex", TypeEnum.STRING)], return_type=Type(":string"))
env_multi = MultiFunction("env")
env_multi.register(env_func)
SYS.register("env", env_multi)
|
