from .. import TypeEnum, Environment, Arg, Builtin, UserFunction, evaluate, InterpretPanic, Callable, NebSyntax, lex, parse, MultiFunction
from .core import interpretQuote
from ..structs import *

REPL = Environment()

def interpretHowTo(symbol, args, env, ns):
    target = args[0]
    if isinstance(target, Symbol):
        target = evaluate(target, env, ns)
    if not isinstance(target, Callable):
        raise InterpretPanic(symbol, "expects a func or symbol", args[0])
    des = target.describe()
    if isinstance(des, str):
        print(des)
    else:
        print("\n".join(des))
    return Nil()

howto_func = Builtin("howto", interpretHowTo, [Arg("symbol", TypeEnum.ANY)], return_type=Type(":nil"))
howto_multi = MultiFunction("howto")
howto_multi.register(howto_func)
REPL.register("howto", howto_multi)

def interpretSymbols(symbol, args, env, ns):
    if len(args) == 1 and args[0].value:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items())]
    else:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if not k.startswith(".")]
    return List(keys)

symbols_nonhidden_func = Builtin("symbols", interpretSymbols, [], return_type=Type(":[]"))
symbols_all_func = Builtin("symbols", interpretSymbols, [Arg("all?", TypeEnum.BOOL)], return_type=Type(":[]"))
symbols_multi = MultiFunction("symbols")
symbols_multi.register(symbols_nonhidden_func)
symbols_multi.register(symbols_all_func)
REPL.register("symbols", symbols_multi)

def interpretFuncs(symbol, args, env, ns):
    if len(args) == 1 and args[0].value:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if isinstance(v, MultiFunction)]
    else:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if isinstance(v, MultiFunction) and not k.startswith(".")]
    return List(keys)

funcs_nonhidden_func = Builtin("funcs", interpretFuncs, [], return_type=Type(":[]"))
funcs_all_func = Builtin("funcs", interpretFuncs, [Arg("all?", TypeEnum.BOOL)], return_type=Type(":[]"))
funcs_multi = MultiFunction("funcs")
funcs_multi.register(funcs_nonhidden_func)
funcs_multi.register(funcs_all_func)
REPL.register("funcs", funcs_multi)

def interpretUserSymbols(symbol, args, env, ns):
    if len(args) == 1 and args[0].value:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if (isinstance(v, MultiFunction) and v.user_impl()) or isinstance(v, Literal)]
    else:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if ((isinstance(v, MultiFunction) and v.user_impl()) or isinstance(v, Literal)) and not k.startswith(".")]
    return List(keys)

usersymbols_nonhidden_func = Builtin("user-symbols", interpretUserSymbols, [], return_type=Type(":[]"))
usersymbols_all_func = Builtin("user-symbols", interpretUserSymbols, [Arg("all?", TypeEnum.BOOL)], return_type=Type(":[]"))
usersymbols_multi = MultiFunction("user-symbols")
usersymbols_multi.register(usersymbols_nonhidden_func)
usersymbols_multi.register(usersymbols_all_func)
REPL.register("user-symbols", usersymbols_multi)

def interpretUserFuncs(symbol, args, env, ns):
    if len(args) == 1 and args[0].value:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if isinstance(v, MultiFunction) and v.user_impl()]
    else:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if isinstance(v, MultiFunction) and v.user_impl() and not k.startswith(".")]
    return List(keys)

userfuncs_nonhidden_func = Builtin("user-funcs", interpretUserFuncs, [], return_type=Type(":[]"))
userfuncs_all_func = Builtin("user-funcs", interpretUserFuncs, [Arg("all?", TypeEnum.BOOL)], return_type=Type(":[]"))
userfuncs_multi = MultiFunction("user-funcs")
userfuncs_multi.register(userfuncs_nonhidden_func)
userfuncs_multi.register(userfuncs_all_func)
REPL.register("user-funcs", userfuncs_multi)

def interpretSyntax(symbol, args, env, ns):
    keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if isinstance(v, NebSyntax)]
    return List(keys)

syntax_func = Builtin("syntax", interpretSyntax, [], return_type=Type(":[]"))
syntax_multi = MultiFunction("syntax")
syntax_multi.register(syntax_func)
REPL.register("syntax", syntax_multi)

def interpretParseNeb(symbol, args, env, ns):
    #out = []
    #for item in parse(lex(args[0].value)):
    #    out.append(Expr([Symbol("quote", symbol.line), item]))
    #return List(out)

    #return interpretQuote(None, [parse(lex(args[0].value))[0]], env, ns)
    return interpretQuote(None, [List(parse(lex(args[0].value)))], env, ns)
    #return List(parse(lex(args[0].value)))

parseneb_func = Builtin("parse-neb", interpretParseNeb, [Arg("string", TypeEnum.STRING)], return_type=Type(":any"))
parseneb_multi = MultiFunction("parse-neb")
parseneb_multi.register(parseneb_func)
REPL.register("parse-neb", parseneb_multi)