aboutsummaryrefslogtreecommitdiff
path: root/neb/std/repl.py
blob: 3168770970bf12ac427c26b787382a68d0e16844 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from .. import TypeEnum, Environment, Arg, Builtin, UserFunction, Function, evaluate, InterpretPanic, Callable, NebSyntax, lex, parse
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 List([])

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

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)

REPL.register("symbols", Builtin("symbols", interpretSymbols, [Arg("all?", TypeEnum.BOOL, optional=True)], return_type=Type(":list")))

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, Function)]
    else:
        keys = [Symbol(k, -1, True) for k,v in sorted(env.get_all().items()) if isinstance(v, Function) and not k.startswith(".")]
    return List(keys)

REPL.register("funcs", Builtin("funcs", interpretFuncs, [Arg("all?", TypeEnum.BOOL, optional=True)], return_type=Type(":list")))

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

REPL.register("user-symbols", Builtin("user-symbols", interpretUserSymbols, [Arg("all?", TypeEnum.BOOL, optional=True)], return_type=Type(":list")))

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

REPL.register("user-funcs", Builtin("user-funcs", interpretUserFuncs, [Arg("all?", TypeEnum.BOOL, optional=True)], return_type=Type(":list")))

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)

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

def interpretParseNeb(symbol, args, env, ns):
    return interpretQuote(None, [parse(lex(args[0].value))[0]], env, ns)

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