1
1
mirror of https://github.com/kanaka/mal.git synced 2024-11-09 18:06:35 +03:00
mal/js/step6_file.js
2016-03-27 22:16:57 -04:00

133 lines
3.5 KiB
JavaScript

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 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);
}
if (ast.length === 0) {
return ast;
}
// apply list
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 "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) \")\")))))");
if (typeof process !== 'undefined' && process.argv.length > 2) {
repl_env.set('*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); }
}
}
}