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 String(ret.stdout.decode("utf-8"))

SYS.register("$", Builtin("$", interpretShell, [Arg("command", TypeEnum.STRING)], return_type=Type(":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], 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 interpretEnv(symbol, args, env, ns):
    items = os.environ[args[0].value].split(":")
    return List([String(item) for item in items])

SYS.register("env", Builtin("env", interpretEnv, [Arg("regex", TypeEnum.STRING)], return_type=Type(":string")))