aboutsummaryrefslogtreecommitdiff
path: root/neb/std/sys.py
blob: 252a0b09f2ed68ed25de6fd41883a2fb5820b6bb (plain)
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
from .. import TypeEnum, Environment, Arg, Builtin, evaluate
from ..structs import *
import shlex
import subprocess
import sys

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(interpretArgv, []))

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)]))

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(interpretExit, [exit_arg]))

def interpretPrint(symbol, args, env, ns):
    print(args[0].value)
    return List([])  # print returns nothing

SYS.register("print", Builtin(interpretPrint, [Arg("arg", TypeEnum.STRING)]))