from .. import TypeEnum, Environment, Arg, Builtin, evaluate, InterpretPanic
from ..structs import *
from pathlib import Path
from glob import glob
from io import UnsupportedOperation
import sys

FS = Environment()

FS.register("_stdin_", Handle(sys.stdin))
FS.register("_stdout_", Handle(sys.stdout))
FS.register("_stderr_", Handle(sys.stderr))

def interpretExists(symbol, args, env, ns):
    return Bool(Path(args[0].value).resolve().exists())

FS.register("exists?", Builtin("exists?", interpretExists, [Arg("filename", TypeEnum.STRING)], return_type=Type(":bool")))

def interpretGlob(symbol, args, env, ns):
    items = glob(args[0].value)
    return List([String(item) for item in items])

FS.register("glob", Builtin("glob", interpretGlob, [Arg("regex", TypeEnum.STRING)], return_type=Type(":string")))

def interpretUnlink(symbol, args, env, ns):
    target_path = Path(args[0].value).resolve()
    if not target_path.exists():
        raise InterpretPanic(symbol, "target file does not exist", target_path)
    target_path.unlink()
    return List([])

FS.register("unlink", Builtin("unlink", interpretUnlink, [Arg("filename", TypeEnum.STRING)], return_type=Type(":list")))

def interpretWithWrite(symbol, args, env, ns):
    target_file = args[0]
    new_env = Environment(env)
    target_path = Path(target_file.value).resolve()
    ret = Literal([])
    with open(str(target_path), "w") as fil:
        new_env.register("_file_", List([fil])) # TODO wrong!
        for arg in args[1:]:
            ret = evaluate(arg, new_env, ns)
    return ret

FS.register("with-write", Builtin("with-write", interpretWithWrite, [Arg("filename", TypeEnum.STRING)], Arg("exprs", TypeEnum.ANY, lazy=True)))

def interpretWrite(symbol, args, env, ns):
    string = args[0]
    handle = args[1]
    try:
        handle.file.write(string.value)
    except UnsupportedOperation:
        raise InterpretPanic(symbol, f"{handle} is not writable!")
    except ValueError:
        raise InterpretPanic(symbol, f"{handle} is closed")
    return List([])

FS.register("write", Builtin("write", interpretWrite, [Arg("string", TypeEnum.STRING), Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.LIST))

def interpretOpenRead(symbol, args, env, ns):
    name = args[0].value
    fil = Path(name)
    if not fil.exists():
        raise InterpretPanic(symbol, "file does not exist", fil)
    try:
        f = open(str(fil), "r")
    except:
        raise InterpretPanic(symbol, "cannot open {fil} for reading")
    return Handle(f)

FS.register("open-read", Builtin("open-read", interpretOpenRead, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE))

def interpretOpenWrite(symbol, args, env, ns):
    name = args[0].value
    fil = Path(name)
    try:
        f = open(str(fil), "w")
    except:
        raise InterpretPanic(symbol, "cannot open {fil} for writing")
    return Handle(f)

FS.register("open-write", Builtin("open-write", interpretOpenWrite, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE))

def interpretOpenAppend(symbol, args, env, ns):
    name = args[0].value
    fil = Path(name)
    try:
        f = open(str(fil), "a")
    except:
        raise InterpretPanic(symbol, "cannot open {fil} for appending")
    return Handle(f)

FS.register("open-append", Builtin("open-append", interpretOpenAppend, [Arg("filename", TypeEnum.STRING)], return_type=TypeEnum.HANDLE))

def interpretClose(symbol, args, env, ns):
    try:
        args[0].file.close()
        # TODO ideally we'd be able to remove it from the env
        # but by this point, we don't know its symbol
        # though it may not be in the env, e.g.
        # (close (print (read (open-read "fil.txt"))))
    except:
        raise InterpretPanic(symbol, "cannot close {args[0]}")
    return List([])

FS.register("close", Builtin("close", interpretClose, [Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.LIST))

def interpretRead(symbol, args, env, ns):
    handle = args[0]
    try:
        inp = args[0].file.read()
    except UnsupportedOperation:
        raise InterpretPanic(symbol, f"{handle} is not writable!")
    except ValueError:
        raise InterpretPanic(symbol, f"{handle} is closed")
    return String(inp)

FS.register("read", Builtin("read", interpretRead, [Arg("handle", TypeEnum.HANDLE)], return_type=TypeEnum.STRING))

def interpretReadLines(symbol, args, env, ns):
    target_file_name = args[0].value
    target_file = Path(target_file_name).resolve()
    if not target_file.exists():
        raise InterpretPanic(symbol, "no such file", target_file)
    with open(target_file, "r") as fil:
        data = fil.readlines()
    out = List([String(d) for d in data]) # all lines are strings
    return out

FS.register("read-lines", Builtin("read-lines", interpretReadLines, [Arg("filename", TypeEnum.STRING)], return_type=Type(":list")))