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(":list"))
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)