mirror of
https://github.com/kanaka/mal.git
synced 2024-09-20 10:07:45 +03:00
142 lines
3.1 KiB
VimL
142 lines
3.1 KiB
VimL
|
source readline.vim
|
||
|
source types.vim
|
||
|
source reader.vim
|
||
|
source printer.vim
|
||
|
source env.vim
|
||
|
source core.vim
|
||
|
|
||
|
function READ(str)
|
||
|
return ReadStr(a:str)
|
||
|
endfunction
|
||
|
|
||
|
function EvalAst(ast, env)
|
||
|
if SymbolQ(a:ast)
|
||
|
let varname = ObjValue(a:ast)
|
||
|
return a:env.get(varname)
|
||
|
elseif ListQ(a:ast)
|
||
|
let ret = []
|
||
|
for e in ObjValue(a:ast)
|
||
|
call add(ret, EVAL(e, a:env))
|
||
|
endfor
|
||
|
return ListNew(ret)
|
||
|
elseif VectorQ(a:ast)
|
||
|
let ret = []
|
||
|
for e in ObjValue(a:ast)
|
||
|
call add(ret, EVAL(e, a:env))
|
||
|
endfor
|
||
|
return VectorNew(ret)
|
||
|
elseif HashQ(a:ast)
|
||
|
let ret = {}
|
||
|
for [k,v] in items(ObjValue(a:ast))
|
||
|
let keyobj = HashParseKey(k)
|
||
|
let newkey = EVAL(keyobj, a:env)
|
||
|
let newval = EVAL(v, a:env)
|
||
|
let keystring = HashMakeKey(newkey)
|
||
|
let ret[keystring] = newval
|
||
|
endfor
|
||
|
return HashNew(ret)
|
||
|
else
|
||
|
return a:ast
|
||
|
end
|
||
|
endfunction
|
||
|
|
||
|
function EVAL(ast, env)
|
||
|
let ast = a:ast
|
||
|
let env = a:env
|
||
|
|
||
|
while 1
|
||
|
if !ListQ(ast)
|
||
|
return EvalAst(ast, env)
|
||
|
end
|
||
|
|
||
|
let first = ListFirst(ast)
|
||
|
let first_symbol = SymbolQ(first) ? ObjValue(first) : ""
|
||
|
if first_symbol == "def!"
|
||
|
let a1 = ObjValue(ast)[1]
|
||
|
let a2 = ObjValue(ast)[2]
|
||
|
let ret = env.set(ObjValue(a1), EVAL(a2, env))
|
||
|
return ret
|
||
|
elseif first_symbol == "let*"
|
||
|
let a1 = ObjValue(ast)[1]
|
||
|
let a2 = ObjValue(ast)[2]
|
||
|
let env = NewEnv(env)
|
||
|
let let_binds = ObjValue(a1)
|
||
|
let i = 0
|
||
|
while i < len(let_binds)
|
||
|
call env.set(ObjValue(let_binds[i]), EVAL(let_binds[i+1], env))
|
||
|
let i = i + 2
|
||
|
endwhile
|
||
|
let ast = a2
|
||
|
" TCO
|
||
|
elseif first_symbol == "if"
|
||
|
let condvalue = EVAL(ObjValue(ast)[1], env)
|
||
|
if FalseQ(condvalue) || NilQ(condvalue)
|
||
|
if len(ObjValue(ast)) < 4
|
||
|
return g:MalNil
|
||
|
else
|
||
|
let ast = ObjValue(ast)[3]
|
||
|
endif
|
||
|
else
|
||
|
let ast = ObjValue(ast)[2]
|
||
|
endif
|
||
|
" TCO
|
||
|
elseif first_symbol == "do"
|
||
|
let astlist = ObjValue(ast)
|
||
|
call EvalAst(ListNew(astlist[1:-2]), env)
|
||
|
let ast = astlist[-1]
|
||
|
" TCO
|
||
|
elseif first_symbol == "fn*"
|
||
|
let fn = NewFn(ListNth(ast, 2), env, ListNth(ast, 1))
|
||
|
return fn
|
||
|
else
|
||
|
" apply list
|
||
|
let el = EvalAst(ast, env)
|
||
|
let funcobj = ListFirst(el)
|
||
|
let args = ListRest(el)
|
||
|
if NativeFunctionQ(funcobj)
|
||
|
return NativeFuncInvoke(funcobj, args)
|
||
|
elseif FunctionQ(funcobj)
|
||
|
let fn = ObjValue(funcobj)
|
||
|
let ast = fn.ast
|
||
|
let env = NewEnvWithBinds(fn.env, fn.params, args)
|
||
|
" TCO
|
||
|
else
|
||
|
throw "Not a function"
|
||
|
endif
|
||
|
endif
|
||
|
endwhile
|
||
|
endfunction
|
||
|
|
||
|
function PRINT(exp)
|
||
|
return PrStr(a:exp, 1)
|
||
|
endfunction
|
||
|
|
||
|
function REP(str, env)
|
||
|
return PRINT(EVAL(READ(a:str), a:env))
|
||
|
endfunction
|
||
|
|
||
|
set maxfuncdepth=10000
|
||
|
let repl_env = NewEnv("")
|
||
|
|
||
|
for [k, v] in items(CoreNs)
|
||
|
call repl_env.set(k, v)
|
||
|
endfor
|
||
|
|
||
|
call REP("(def! not (fn* (a) (if a false true)))", repl_env)
|
||
|
|
||
|
while 1
|
||
|
let [eof, line] = Readline("user> ")
|
||
|
if eof
|
||
|
break
|
||
|
endif
|
||
|
if line == ""
|
||
|
continue
|
||
|
endif
|
||
|
try
|
||
|
call PrintLn(REP(line, repl_env))
|
||
|
catch
|
||
|
call PrintLn("Error: " . v:exception)
|
||
|
endtry
|
||
|
endwhile
|
||
|
qall!
|