Commit | Line | Data |
---|---|---|
f5223195 JM |
1 | --- step4_if_fn_do ------------------------------ |
2 | import types, reader, printer, env, core | |
3 | ||
4 | READ(str): return reader.read_str(str) | |
5 | ||
6 | eval_ast(ast,env): | |
7 | switch type(ast): | |
8 | symbol: return env.get(ast) | |
9 | list,vector: return ast.map((x) -> EVAL(x,env)) | |
10 | hash: return ast.map((k,v) -> list(k, EVAL(v,env))) | |
11 | _default_: return ast | |
12 | ||
13 | EVAL(ast,env): | |
14 | if not list?(ast): return eval_ast(ast, env) | |
15 | switch ast[0]: | |
16 | 'def!: return env.set(ast[1], EVAL(ast[2], env)) | |
899ff7fa | 17 | 'let*: let_env = ...; return EVAL(ast[2], let_env) |
f5223195 JM |
18 | 'do: return eval_ast(rest(ast), env)[-1] |
19 | 'if: return EVAL(EVAL(ast[1], env) ? ast[2] : ast[3], env) | |
20 | 'fn*: return (...a) -> EVAL(ast[2], new Env(env, ast[1], a)) | |
21 | _default_: f, args = eval_ast(ast, env) | |
22 | return apply(f, args) | |
23 | ||
24 | PRINT(exp): return printer.pr_str(exp) | |
25 | ||
26 | repl_env = new Env() | |
27 | rep(str): return PRINT(EVAL(READ(str),repl_env)) | |
28 | ||
29 | ;; core.EXT: defined using Racket | |
30 | core.ns.map((k,v) -> (repl_env.set(k, v))) | |
31 | ||
32 | ;; core.mal: defined using the language itself | |
33 | rep("(def! not (fn* (a) (if a false true)))") | |
34 | ||
35 | main loop: | |
36 | try: println(rep(readline("user> "))) | |
37 | catch e: println("Error: ", e) | |
38 | ||
39 | --- env module ---------------------------------- | |
40 | class Env (outer=null,binds=[],exprs=[]) | |
41 | data = hash_map() | |
42 | foreach b, i in binds: | |
43 | if binds[i] == '&: data[binds[i+1]] = exprs.drop(i); break | |
44 | else: data[binds[i]] = exprs[i] | |
45 | set(k,v): return data.set(k,v) | |
46 | find(k): return data.has(k) ? this : (if outer ? find(outer) : null) | |
47 | get(k): return data.find(k).get(k) OR raise "'" + k + "' not found" | |
fd888612 | 48 | |
f5223195 JM |
49 | --- core module --------------------------------- |
50 | ns = {'=: equal?, | |
f5223195 JM |
51 | |
52 | 'pr-str: (a) -> a.map(|s| pr_str(e,true)).join(" ")), | |
53 | 'str: (a) -> a.map(|s| pr_str(e,false)).join("")), | |
54 | 'prn: (a) -> println(a.map(|s| pr_str(e,true)).join(" ")), | |
55 | 'println: (a) -> println(a.map(|s| pr_str(e,false)).join(" ")), | |
56 | ||
57 | '<: lt, | |
58 | '<=: lte, | |
59 | '>: gt, | |
60 | '>=: gte, | |
61 | '+: add, | |
62 | '-: sub, | |
63 | '*: mult, | |
64 | '/: div, | |
65 | ||
66 | 'list: list, | |
67 | 'list?: list?, | |
f5223195 JM |
68 | |
69 | 'empty?: empty?, | |
70 | 'count: count} |