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
|
from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic
from ..structs import *
import shlex
import subprocess
import sys
from datetime import datetime
SYS = Environment()
def interpretArgv(symbol, args, env, ns):
out = []
for arg in sys.argv[1:]:
out.append(String(arg))
return List(out)
SYS.register("argv", Builtin("argv", interpretArgv, [], return_type=Type(":list")))
def interpretShell(symbol, args, env, ns):
ret = subprocess.run(shlex.split(args[0].value), capture_output=True)
return List([String(r) for r in ret.stdout.decode("utf-8").split("\n")])
SYS.register("$", Builtin("$", interpretShell, [Arg("command", TypeEnum.STRING)], return_type=Type(":list")))
def interpretExit(symbol, args, env, ns):
status = 0 if len(args) == 0 else args[0].value
sys.exit(status)
return List([])
exit_arg = Arg("status", TypeEnum.INT, optional=True)
SYS.register("exit", Builtin("exit", interpretExit, [exit_arg], return_type=Type(":list")))
def interpretPrint(symbol, args, env, ns):
print(args[0].value)
return List([]) # print returns nothing
SYS.register("print", Builtin("print", interpretPrint, [Arg("arg", TypeEnum.STRING)], return_type=Type(":list")))
def interpretBench(symbol, args, env, ns):
before = datetime.now()
ret = evaluate(args[0], env, ns)
after = datetime.now()
print(f"bench [{symbol.line}]: {args[0]} => {after - before}")
return ret
SYS.register("bench", Builtin("bench", interpretBench, [Arg("command", TypeEnum.ANY, lazy=True)], return_type=Type(":any")))
|
