if (typeof module !== 'undefined') { var types = require('./types'); var readline = require('./node_readline'); var reader = require('./reader'); var printer = require('./printer'); var Env = require('./env').Env; var core = require('./core'); } // read function READ(str) { return reader.read_str(str); } // eval function is_pair(x) { return types._sequential_Q(x) && x.length > 0; } function quasiquote(ast) { if (!is_pair(ast)) { return [types._symbol("quote"), ast]; } else if (types._symbol_Q(ast[0]) && ast[0].value === 'unquote') { return ast[1]; } else if (is_pair(ast[0]) && ast[0][0].value === 'splice-unquote') { return [types._symbol("concat"), ast[0][1], quasiquote(ast.slice(1))]; } else { return [types._symbol("cons"), quasiquote(ast[0]), quasiquote(ast.slice(1))]; } } function is_macro_call(ast, env) { return types._list_Q(ast) && types._symbol_Q(ast[0]) && env.find(ast[0]) && env.get(ast[0])._ismacro_; } function macroexpand(ast, env) { while (is_macro_call(ast, env)) { var mac = env.get(ast[0]); ast = mac.apply(mac, ast.slice(1)); } return ast; } function eval_ast(ast, env) { if (types._symbol_Q(ast)) { return env.get(ast); } else if (types._list_Q(ast)) { return ast.map(function(a) { return EVAL(a, env); }); } else if (types._vector_Q(ast)) { var v = ast.map(function(a) { return EVAL(a, env); }); v.__isvector__ = true; return v; } else if (types._hash_map_Q(ast)) { var new_hm = {}; for (k in ast) { new_hm[EVAL(k, env)] = EVAL(ast[k], env); } return new_hm; } else { return ast; } } function _EVAL(ast, env) { while (true) { //printer.println("EVAL:", printer._pr_str(ast, true)); if (!types._list_Q(ast)) { return eval_ast(ast, env); } // apply list ast = macroexpand(ast, env); if (!types._list_Q(ast)) { return eval_ast(ast, env); } if (ast.length === 0) { return ast; } var a0 = ast[0], a1 = ast[1], a2 = ast[2], a3 = ast[3]; switch (a0.value) { case "def!": var res = EVAL(a2, env); return env.set(a1, res); case "let*": var let_env = new Env(env); for (var i=0; i < a1.length; i+=2) { let_env.set(a1[i], EVAL(a1[i+1], let_env)); } ast = a2; env = let_env; break; case "quote": return a1; case "quasiquote": ast = quasiquote(a1); break; case 'defmacro!': var func = EVAL(a2, env); func._ismacro_ = true; return env.set(a1, func); case 'macroexpand': return macroexpand(a1, env); case "do": eval_ast(ast.slice(1, -1), env); ast = ast[ast.length-1]; break; case "if": var cond = EVAL(a1, env); if (cond === null || cond === false) { ast = (typeof a3 !== "undefined") ? a3 : null; } else { ast = a2; } break; case "fn*": return types._function(EVAL, Env, a2, env, a1); default: var el = eval_ast(ast, env), f = el[0]; if (f.__ast__) { ast = f.__ast__; env = f.__gen_env__(el.slice(1)); } else { return f.apply(f, el.slice(1)); } } } } function EVAL(ast, env) { var result = _EVAL(ast, env); return (typeof result !== "undefined") ? result : null; } // print function PRINT(exp) { return printer._pr_str(exp, true); } // repl var repl_env = new Env(); var rep = function(str) { return PRINT(EVAL(READ(str), repl_env)); }; // core.js: defined using javascript for (var n in core.ns) { repl_env.set(types._symbol(n), core.ns[n]); } repl_env.set(types._symbol('eval'), function(ast) { return EVAL(ast, repl_env); }); repl_env.set(types._symbol('*ARGV*'), []); // core.mal: defined using the language itself rep("(def! not (fn* (a) (if a false true)))"); rep("(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))"); 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)))))))"); rep("(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))"); if (typeof process !== 'undefined' && process.argv.length > 2) { repl_env.set(types._symbol('*ARGV*'), process.argv.slice(3)); rep('(load-file "' + process.argv[2] + '")'); process.exit(0); } // repl loop if (typeof require !== 'undefined' && require.main === module) { // Synchronous node.js commandline mode while (true) { var line = readline.readline("user> "); if (line === null) { break; } try { if (line) { printer.println(rep(line)); } } catch (exc) { if (exc instanceof reader.BlankException) { continue; } if (exc.stack) { printer.println(exc.stack); } else { printer.println(exc); } } } }