1 --- stepA_interop -------------------------------
2 import types, reader, printer, env, core
4 READ(str): return reader.read_str(str)
6 pair?(ast): return ... // true if non-empty sequence
7 quasiquote(ast): return ... // quasiquote
9 macro?(ast, env): return ... // true if macro function
10 macroexpand(ast, env): return ... // recursive macro expansion
14 symbol: return env.get(ast)
15 list,vector: return ast.map((x) -> EVAL(x,env))
16 hash: return ast.map((k,v) -> list(k, EVAL(v,env)))
21 if not list?(ast): return eval_ast(ast, env)
23 'def!: return env.set(ast[1], EVAL(ast[2], env))
24 'let*: env = ...; ast = ast[2] // TCO
26 'quasiquote: ast = quasiquote(ast[1]) // TCO
27 'defmacro!: return ... // like def!, but set macro property
28 'macroexpand: return macroexpand(ast[1], env)
29 'try*: return ... // try/catch native and malval exceptions
30 'do: ast = eval_ast(ast[1..-1], env)[-1] // TCO
31 'if: EVAL(ast[1], env) ? ast = ast[2] : ast = ast[3] // TCO
32 'fn*: return new MalFunc(...)
33 _default_: f, args = eval_ast(ast, env)
34 if malfunc?(ast[0]): ast = ast[0].fn; env = ... // TCO
35 else: return apply(f, args)
37 PRINT(exp): return printer.pr_str(exp)
40 rep(str): return PRINT(EVAL(READ(str),repl_env))
42 ;; core.EXT: defined using Racket
43 core.ns.map((k,v) -> (repl_env.set(k, v)))
44 repl_env.set('eval, (ast) -> EVAL(ast, repl-env))
45 repl_env.set('*ARGV*, cmdline_args[1..])
47 ;; core.mal: defined using the language itself
48 rep("(def! *host-language* \"racket\")")
49 rep("(def! not (fn* (a) (if a false true)))")
50 rep("(def! load-file (fn* (f) ...))")
51 rep("(defmacro! cond (fn* (& xs) ...))")
52 rep("(defmacro! or (fn* (& xs) ...))")
54 if cmdline_args: rep("(load-file \"" + args[0] + "\")"); exit 0
56 rep("(println (str \"Mal [\" *host-language* \"]\"))")
58 try: println(rep(readline("user> ")))
59 catch e: println("Error: ", e)
61 --- env module ----------------------------------
62 class Env (outer=null,binds=[],exprs=[])
64 foreach b, i in binds:
65 if binds[i] == '&: data[binds[i+1]] = exprs.drop(i); break
66 else: data[binds[i]] = exprs[i]
67 set(k,v): return data.set(k,v)
68 find(k): return data.has(k) ? this : (if outer ? find(outer) : null)
69 get(k): return data.find(k).get(k) OR raise "'" + k + "' not found"
71 --- core module ---------------------------------
83 'pr-str: (a) -> a.map(|s| pr_str(e,true)).join(" ")),
84 'str: (a) -> a.map(|s| pr_str(e,false)).join("")),
85 'prn: (a) -> println(a.map(|s| pr_str(e,true)).join(" ")),
86 'println: (a) -> println(a.map(|s| pr_str(e,false)).join(" ")),
87 'read-string: read_str,
109 'contains?: contains?,
113 'sequential? sequential?,
114 'cons: (a) -> concat([a[0]], a[1]),
115 'concat: (a) -> reduce(concat, [], a),
116 'nth: (a) -> a[0][a[1]] OR raise "nth: index out of range",
117 'first: (a) -> a[0][0] OR nil,
118 'rest: (a) -> a[0][1..] OR list(),
125 'meta: (a) -> a[0].meta,
126 'with-meta: (a) -> a[0].with_meta(a[1]),
127 'atom: (a) -> new Atom(a[0]),
128 'atom?: (a) -> type(a[0]) == "atom",
129 'deref: (a) -> a[0].val,
130 'reset!: (a) -> a[0].val = a[1],