2020-07-21 18:01:48 +02:00
|
|
|
import functools
|
2014-03-24 16:32:24 -05:00
|
|
|
import sys, traceback
|
|
|
|
|
import mal_readline
|
2014-04-06 16:17:38 -05:00
|
|
|
import mal_types as types
|
|
|
|
|
import reader, printer
|
|
|
|
|
from env import Env
|
|
|
|
|
import core
|
2014-03-24 16:32:24 -05:00
|
|
|
|
|
|
|
|
# read
|
2024-08-09 19:12:09 +08:00
|
|
|
READ = reader.read_str
|
2014-03-24 16:32:24 -05:00
|
|
|
|
|
|
|
|
# eval
|
2020-07-21 18:01:48 +02:00
|
|
|
def qq_loop(acc, elt):
|
2024-08-09 19:12:09 +08:00
|
|
|
if types._list_Q(elt) \
|
|
|
|
|
and len(elt) == 2 \
|
|
|
|
|
and types._symbol_Q(elt[0]) \
|
|
|
|
|
and elt[0] == 'splice-unquote':
|
|
|
|
|
return types._list(types._symbol('concat'), elt[1], acc)
|
2020-07-21 18:01:48 +02:00
|
|
|
else:
|
2024-08-09 19:12:09 +08:00
|
|
|
return types._list(types._symbol('cons'), quasiquote(elt), acc)
|
2020-07-21 18:01:48 +02:00
|
|
|
|
|
|
|
|
def qq_foldr(seq):
|
|
|
|
|
return functools.reduce(qq_loop, reversed(seq), types._list())
|
2014-03-24 16:32:24 -05:00
|
|
|
|
|
|
|
|
def quasiquote(ast):
|
2020-07-21 18:01:48 +02:00
|
|
|
if types._list_Q(ast):
|
2024-08-09 19:12:09 +08:00
|
|
|
if len(ast) == 2 \
|
|
|
|
|
and types._symbol_Q(ast[0]) \
|
|
|
|
|
and ast[0] == 'unquote':
|
2020-07-21 18:01:48 +02:00
|
|
|
return ast[1]
|
|
|
|
|
else:
|
|
|
|
|
return qq_foldr(ast)
|
|
|
|
|
elif types._hash_map_Q(ast) or types._symbol_Q(ast):
|
2024-08-09 19:12:09 +08:00
|
|
|
return types._list(types._symbol('quote'), ast)
|
|
|
|
|
elif types._vector_Q(ast):
|
|
|
|
|
return types._list(types._symbol('vec'), qq_foldr(ast))
|
2014-03-24 16:32:24 -05:00
|
|
|
else:
|
2020-07-21 18:01:48 +02:00
|
|
|
return ast
|
2014-03-24 16:32:24 -05:00
|
|
|
|
2022-01-10 00:15:40 +01:00
|
|
|
def EVAL(ast, env):
|
|
|
|
|
while True:
|
2014-03-24 16:32:24 -05:00
|
|
|
|
2024-08-09 19:12:09 +08:00
|
|
|
dbgeval = env.get(types._symbol('DEBUG-EVAL'), return_nil=True)
|
2022-01-10 00:15:40 +01:00
|
|
|
if dbgeval is not None and dbgeval is not False:
|
|
|
|
|
print('EVAL: ' + printer._pr_str(ast))
|
2014-03-24 16:32:24 -05:00
|
|
|
|
2014-04-06 16:17:38 -05:00
|
|
|
if types._symbol_Q(ast):
|
2014-03-24 16:32:24 -05:00
|
|
|
return env.get(ast)
|
2014-04-06 16:17:38 -05:00
|
|
|
elif types._vector_Q(ast):
|
2024-08-09 19:12:09 +08:00
|
|
|
return types.Vector(EVAL(a, env) for a in ast)
|
2014-04-06 16:17:38 -05:00
|
|
|
elif types._hash_map_Q(ast):
|
2021-08-21 19:08:17 +02:00
|
|
|
return types.Hash_Map((k, EVAL(v, env)) for k, v in ast.items())
|
2022-01-10 00:15:40 +01:00
|
|
|
elif not types._list_Q(ast):
|
2014-03-24 16:32:24 -05:00
|
|
|
return ast # primitive value, return unchanged
|
2022-01-10 00:15:40 +01:00
|
|
|
else:
|
2014-04-06 16:17:38 -05:00
|
|
|
|
2014-03-24 16:32:24 -05:00
|
|
|
# apply list
|
|
|
|
|
if len(ast) == 0: return ast
|
2014-04-06 16:17:38 -05:00
|
|
|
a0 = ast[0]
|
2014-03-24 16:32:24 -05:00
|
|
|
|
2024-08-09 19:12:09 +08:00
|
|
|
if types._symbol_Q(a0):
|
2014-03-24 16:32:24 -05:00
|
|
|
if "def!" == a0:
|
|
|
|
|
a1, a2 = ast[1], ast[2]
|
|
|
|
|
res = EVAL(a2, env)
|
|
|
|
|
return env.set(a1, res)
|
|
|
|
|
elif "let*" == a0:
|
|
|
|
|
a1, a2 = ast[1], ast[2]
|
|
|
|
|
let_env = Env(env)
|
2024-08-09 19:12:09 +08:00
|
|
|
for k, v in types.asPairs(a1):
|
|
|
|
|
let_env.set(k, EVAL(v, let_env))
|
2014-04-23 21:59:50 -05:00
|
|
|
ast = a2
|
|
|
|
|
env = let_env
|
2024-08-09 19:12:09 +08:00
|
|
|
continue # TCO
|
2014-03-24 16:32:24 -05:00
|
|
|
elif "quote" == a0:
|
|
|
|
|
return ast[1]
|
|
|
|
|
elif "quasiquote" == a0:
|
2024-08-09 19:12:09 +08:00
|
|
|
ast = quasiquote(ast[1])
|
|
|
|
|
continue # TCO
|
2014-03-24 16:32:24 -05:00
|
|
|
elif 'defmacro!' == a0:
|
2024-08-09 19:12:09 +08:00
|
|
|
func = EVAL(ast[2], env)
|
|
|
|
|
func = types._clone(func)
|
2014-03-24 16:32:24 -05:00
|
|
|
func._ismacro_ = True
|
|
|
|
|
return env.set(ast[1], func)
|
|
|
|
|
elif "try*" == a0:
|
Test uncaught throw, catchless try* . Fix 46 impls.
Fixes made to: ada, c, chuck, clojure, coffee, common-lisp, cpp,
crystal, d, dart, elm, erlang, es6, factor, fsharp, gnu-smalltalk,
groovy, guile, haxe, hy, js, livescript, matlab, miniMAL, nasm, nim,
objc, objpascal, ocaml, perl, perl6, php, plsql, ps, python, r,
rpython, ruby, scheme, swift3, tcl, ts, vb, vimscript, wasm, yorick.
Catchless try* test is an optional test. Not all implementations
support catchless try* but a number were fixed so they at least don't
crash on catchless try*.
2018-12-03 13:20:44 -06:00
|
|
|
if len(ast) < 3:
|
2024-08-09 19:12:09 +08:00
|
|
|
ast = ast[1]
|
|
|
|
|
continue # TCO
|
|
|
|
|
else:
|
|
|
|
|
a1, a2 = ast[1], ast[2]
|
Test uncaught throw, catchless try* . Fix 46 impls.
Fixes made to: ada, c, chuck, clojure, coffee, common-lisp, cpp,
crystal, d, dart, elm, erlang, es6, factor, fsharp, gnu-smalltalk,
groovy, guile, haxe, hy, js, livescript, matlab, miniMAL, nasm, nim,
objc, objpascal, ocaml, perl, perl6, php, plsql, ps, python, r,
rpython, ruby, scheme, swift3, tcl, ts, vb, vimscript, wasm, yorick.
Catchless try* test is an optional test. Not all implementations
support catchless try* but a number were fixed so they at least don't
crash on catchless try*.
2018-12-03 13:20:44 -06:00
|
|
|
err = None
|
2014-03-24 16:32:24 -05:00
|
|
|
try:
|
Test uncaught throw, catchless try* . Fix 46 impls.
Fixes made to: ada, c, chuck, clojure, coffee, common-lisp, cpp,
crystal, d, dart, elm, erlang, es6, factor, fsharp, gnu-smalltalk,
groovy, guile, haxe, hy, js, livescript, matlab, miniMAL, nasm, nim,
objc, objpascal, ocaml, perl, perl6, php, plsql, ps, python, r,
rpython, ruby, scheme, swift3, tcl, ts, vb, vimscript, wasm, yorick.
Catchless try* test is an optional test. Not all implementations
support catchless try* but a number were fixed so they at least don't
crash on catchless try*.
2018-12-03 13:20:44 -06:00
|
|
|
return EVAL(a1, env)
|
|
|
|
|
except types.MalException as exc:
|
|
|
|
|
err = exc.object
|
2014-03-24 16:32:24 -05:00
|
|
|
except Exception as exc:
|
Test uncaught throw, catchless try* . Fix 46 impls.
Fixes made to: ada, c, chuck, clojure, coffee, common-lisp, cpp,
crystal, d, dart, elm, erlang, es6, factor, fsharp, gnu-smalltalk,
groovy, guile, haxe, hy, js, livescript, matlab, miniMAL, nasm, nim,
objc, objpascal, ocaml, perl, perl6, php, plsql, ps, python, r,
rpython, ruby, scheme, swift3, tcl, ts, vb, vimscript, wasm, yorick.
Catchless try* test is an optional test. Not all implementations
support catchless try* but a number were fixed so they at least don't
crash on catchless try*.
2018-12-03 13:20:44 -06:00
|
|
|
err = exc.args[0]
|
|
|
|
|
catch_env = Env(env, [a2[1]], [err])
|
2024-08-09 19:12:09 +08:00
|
|
|
ast = a2[2]
|
|
|
|
|
env = catch_env
|
|
|
|
|
continue # TCO
|
2014-03-24 16:32:24 -05:00
|
|
|
elif "do" == a0:
|
2022-01-10 00:15:40 +01:00
|
|
|
for i in range(1, len(ast)-1):
|
|
|
|
|
EVAL(ast[i], env)
|
2014-03-24 16:32:24 -05:00
|
|
|
ast = ast[-1]
|
2024-08-09 19:12:09 +08:00
|
|
|
continue # TCO
|
2014-03-24 16:32:24 -05:00
|
|
|
elif "if" == a0:
|
|
|
|
|
a1, a2 = ast[1], ast[2]
|
|
|
|
|
cond = EVAL(a1, env)
|
|
|
|
|
if cond is None or cond is False:
|
2024-08-09 19:12:09 +08:00
|
|
|
if len(ast) > 3:
|
|
|
|
|
ast = ast[3]
|
|
|
|
|
continue # TCO
|
|
|
|
|
else:
|
|
|
|
|
return None
|
2014-03-24 16:32:24 -05:00
|
|
|
else:
|
|
|
|
|
ast = a2
|
2024-08-09 19:12:09 +08:00
|
|
|
continue # TCO
|
2014-03-24 16:32:24 -05:00
|
|
|
elif "fn*" == a0:
|
|
|
|
|
a1, a2 = ast[1], ast[2]
|
2024-08-09 19:12:09 +08:00
|
|
|
def fn(*args):
|
|
|
|
|
return EVAL(a2, Env(env, a1, args))
|
|
|
|
|
fn.__ast__ = a2
|
|
|
|
|
fn.__gen_env__ = lambda args: Env(env, a1, args)
|
|
|
|
|
return fn
|
|
|
|
|
|
|
|
|
|
f = EVAL(a0, env)
|
|
|
|
|
if types._function_Q(f):
|
2022-01-10 00:15:40 +01:00
|
|
|
args = ast[1:]
|
|
|
|
|
if hasattr(f, '_ismacro_'):
|
|
|
|
|
ast = f(*args)
|
|
|
|
|
continue # TCO
|
2014-04-09 23:58:27 -05:00
|
|
|
if hasattr(f, '__ast__'):
|
|
|
|
|
ast = f.__ast__
|
2024-08-09 19:12:09 +08:00
|
|
|
env = f.__gen_env__(EVAL(a, env) for a in args)
|
|
|
|
|
continue # TCO
|
2014-03-24 16:32:24 -05:00
|
|
|
else:
|
2022-01-10 00:15:40 +01:00
|
|
|
return f(*(EVAL(a, env) for a in args))
|
2024-08-09 19:12:09 +08:00
|
|
|
else:
|
|
|
|
|
raise Exception('Can only apply functions')
|
2014-03-24 16:32:24 -05:00
|
|
|
|
|
|
|
|
# print
|
2024-08-09 19:12:09 +08:00
|
|
|
PRINT = printer._pr_str
|
2014-03-24 16:32:24 -05:00
|
|
|
|
|
|
|
|
# repl
|
|
|
|
|
repl_env = Env()
|
|
|
|
|
def REP(str):
|
|
|
|
|
return PRINT(EVAL(READ(str), repl_env))
|
|
|
|
|
|
2014-04-16 23:57:50 -05:00
|
|
|
# core.py: defined using python
|
2014-12-18 20:33:49 -06:00
|
|
|
for k, v in core.ns.items(): repl_env.set(types._symbol(k), v)
|
|
|
|
|
repl_env.set(types._symbol('eval'), lambda ast: EVAL(ast, repl_env))
|
2024-08-09 19:12:09 +08:00
|
|
|
repl_env.set(types._symbol('*ARGV*'), types.List(sys.argv[2:]))
|
2014-03-24 16:32:24 -05:00
|
|
|
|
2014-04-16 23:57:50 -05:00
|
|
|
# core.mal: defined using the language itself
|
2014-03-24 16:32:24 -05:00
|
|
|
REP("(def! not (fn* (a) (if a false true)))")
|
2024-08-09 19:12:09 +08:00
|
|
|
REP('(def! load-file (fn* (f) (eval (read-string (str "(do " (slurp f) "\nnil)")))))')
|
|
|
|
|
REP("""(defmacro! cond (fn* (& xs)
|
|
|
|
|
(if (> (count xs) 0)
|
|
|
|
|
(list 'if (first xs)
|
|
|
|
|
(if (> (count xs) 1) (nth xs 1) (throw "odd number of forms to cond"))
|
|
|
|
|
(cons 'cond (rest (rest xs)))))))""")
|
2014-03-24 16:32:24 -05:00
|
|
|
|
|
|
|
|
if len(sys.argv) >= 2:
|
|
|
|
|
REP('(load-file "' + sys.argv[1] + '")')
|
2014-04-19 13:04:09 -05:00
|
|
|
sys.exit(0)
|
|
|
|
|
|
|
|
|
|
# repl loop
|
|
|
|
|
while True:
|
|
|
|
|
try:
|
|
|
|
|
line = mal_readline.readline("user> ")
|
|
|
|
|
print(REP(line))
|
2024-08-09 19:12:09 +08:00
|
|
|
except EOFError:
|
|
|
|
|
print()
|
|
|
|
|
break
|
2014-04-19 13:04:09 -05:00
|
|
|
except reader.Blank: continue
|
Test uncaught throw, catchless try* . Fix 46 impls.
Fixes made to: ada, c, chuck, clojure, coffee, common-lisp, cpp,
crystal, d, dart, elm, erlang, es6, factor, fsharp, gnu-smalltalk,
groovy, guile, haxe, hy, js, livescript, matlab, miniMAL, nasm, nim,
objc, objpascal, ocaml, perl, perl6, php, plsql, ps, python, r,
rpython, ruby, scheme, swift3, tcl, ts, vb, vimscript, wasm, yorick.
Catchless try* test is an optional test. Not all implementations
support catchless try* but a number were fixed so they at least don't
crash on catchless try*.
2018-12-03 13:20:44 -06:00
|
|
|
except types.MalException as e:
|
|
|
|
|
print("Error:", printer._pr_str(e.object))
|
2024-08-09 19:12:09 +08:00
|
|
|
except Exception:
|
|
|
|
|
# See tests/step5_tco.mal in this directory.
|
|
|
|
|
print("".join(traceback.format_exception(*sys.exc_info())[0:100]))
|
