1
1
mirror of https://github.com/kanaka/mal.git synced 2024-09-20 01:57:09 +03:00
mal/es6/step4_if_fn_do.js
Joel Martin a05c086f05 ES6: more use of ES6, simplifications, newer babel.
- Use Vector class derived from Array
- Use Array/Vector.from for initializing/cloning of Array/Vector
- Remove most semi-colon line endings
- More use of arrow functions
- Use Object.assign to copy properties in _malfunc and function
  cloning.
- Remove or inline a bunch of types.js functions that don't really
  need to be separate functions: _obj_type, _sequential_Q, _symbol,
  _symbol_Q, _vector, _vector_Q, _hash_map, _hash_map_Q
- Simplify dependency list in Makefile
- Remove some separate core.js functions by moving them into the
  core_ns declaration: _nth, keys, vals, with_meta.

With node 7, babel is mostly just used for translating imports into
CommonJS requires for node.
2017-02-10 23:02:30 -06:00

82 lines
2.4 KiB
JavaScript

import { readline } from './node_readline'
import { _list_Q } from './types'
import { BlankException, read_str } from './reader'
import { pr_str } from './printer'
import { new_env, env_set, env_get } from './env'
import { core_ns } from './core'
// read
const READ = str => read_str(str)
// eval
const eval_ast = (ast, env) => {
if (typeof ast === 'symbol') {
return env_get(env, ast)
} else if (ast instanceof Array) {
return ast.map(x => EVAL(x, env))
} else if (ast instanceof Map) {
let new_hm = new Map()
ast.forEach((v, k) => new_hm.set(EVAL(k, env), EVAL(v, env)))
return new_hm
} else {
return ast
}
}
const EVAL = (ast, env) => {
//console.log('EVAL:', pr_str(ast, true))
if (!_list_Q(ast)) { return eval_ast(ast, env) }
if (ast.length === 0) { return ast }
const [a0, a1, a2, a3] = ast
switch (typeof a0 === 'symbol' ? Symbol.keyFor(a0) : Symbol(':default')) {
case 'def!':
return env_set(env, a1, EVAL(a2, env))
case 'let*':
let let_env = new_env(env)
for (let i=0; i < a1.length; i+=2) {
env_set(let_env, a1[i], EVAL(a1[i+1], let_env))
}
return EVAL(a2, let_env)
case 'do':
return eval_ast(ast.slice(1), env)[ast.length-2]
case 'if':
let cond = EVAL(a1, env)
if (cond === null || cond === false) {
return typeof a3 !== 'undefined' ? EVAL(a3, env) : null
} else {
return EVAL(a2, env)
}
case 'fn*':
return (...args) => EVAL(a2, new_env(env, a1, args))
default:
let [f, ...args] = eval_ast(ast, env)
return f(...args)
}
}
// print
const PRINT = exp => pr_str(exp, true)
// repl
let repl_env = new_env()
const REP = str => PRINT(EVAL(READ(str), repl_env))
// core.EXT: defined using ES6
for (let [k, v] of core_ns) { env_set(repl_env, Symbol.for(k), v) }
// core.mal: defined using language itself
REP('(def! not (fn* (a) (if a false true)))')
while (true) {
let line = readline('user> ')
if (line == null) break
try {
if (line) { console.log(REP(line)) }
} catch (exc) {
if (exc instanceof BlankException) { continue }
if (exc.stack) { console.log(exc.stack) }
else { console.log(`Error: ${exc}`) }
}
}