from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic
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("argv", 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("exit", interpretExit, [exit_arg]))

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