aboutsummaryrefslogtreecommitdiff
path: root/neb/std/repl.py
blob: fbb0366d0bd53acd1ad53c3c1f6be65af10a2b33 (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
from .. import TypeEnum, Environment, Arg, Builtin, UserFunction, Function, evaluate, InterpretPanic, Callable, Macro
from ..structs import *

REPL = Environment()

def interpretHowTo(symbol, args, env, ns):
    if not isinstance(args[0], Callable):
        raise InterpretPanic(symbol, "expects a :func", args[0])
    print(args[0].describe())
    return List([])

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

def interpretSymbols(symbol, args, env, ns):
    keys = [Symbol(k, -1) for k,v in sorted(env.environment.items())]
    return List(keys)

REPL.register("symbols", Builtin("symbols", interpretSymbols, [], return_type=Type(":list")))

def interpretFuncs(symbol, args, env, ns):
    keys = [Symbol(k, -1) for k,v in sorted(env.environment.items()) if isinstance(v, Function)]
    return List(keys)

REPL.register("funcs", Builtin("funcs", interpretFuncs, [], return_type=Type(":list")))

def interpretUserSymbols(symbol, args, env, ns):
    keys = [Symbol(k, -1) for k,v in env.environment.items() if isinstance(v, UserFunction) or isinstance(v, Literal)]
    return List(keys)

REPL.register("user-symbols", Builtin("user-symbols", interpretUserSymbols, [], return_type=Type(":list")))

def interpretMacros(symbol, args, env, ns):
    keys = [Symbol(k, -1) for k,v in sorted(env.environment.items()) if isinstance(v, Macro)]
    return List(keys)

REPL.register("macros", Builtin("macros", interpretMacros, [], return_type=Type(":list")))