1
1
mirror of https://github.com/kanaka/mal.git synced 2024-08-17 17:50:24 +03:00

step A scaffolding

This commit is contained in:
Fabian 2021-04-05 21:22:32 +02:00 committed by Joel Martin
parent 4cb61f5426
commit 6b14be1285
3 changed files with 170 additions and 1 deletions

View File

@ -1,4 +1,4 @@
STEP_BINS = step0_repl step1_read_print step2_eval step3_env step4_if_fn_do step6_file step7_quote step8_macros step9_try
STEP_BINS = step0_repl step1_read_print step2_eval step3_env step4_if_fn_do step6_file step7_quote step8_macros step9_try stepA_mal
sml_MODE_DEFAULT = polyml
sml_MODE_CONFIG = .smlmode

12
impls/sml/stepA_mal.mlb Normal file
View File

@ -0,0 +1,12 @@
local
$(SML_LIB)/basis/basis.mlb
util.sml
types.sml
printer.sml
reader.sml
env.sml
core.sml
stepA_mal.sml
in
main.sml
end

157
impls/sml/stepA_mal.sml Normal file
View File

@ -0,0 +1,157 @@
fun read s =
readStr s
fun eval e ast = eval' e (expandMacro e [ast])
and eval' e (LIST (a::args)) = (case specialEval a of SOME special => special e args | _ => evalApply e (eval e a) args)
| eval' e (SYMBOL s) = evalSymbol e s
| eval' e (VECTOR v) = VECTOR (map (eval e) v)
| eval' e (MAP m) = MAP (List.map (fn (k, v) => (eval e k, eval e v)) m)
| eval' e ast = ast
and specialEval (SYMBOL "def!") = SOME evalDef
| specialEval (SYMBOL "let*") = SOME evalLet
| specialEval (SYMBOL "do") = SOME evalDo
| specialEval (SYMBOL "if") = SOME evalIf
| specialEval (SYMBOL "fn*") = SOME evalFn
| specialEval (SYMBOL "quote") = SOME evalQuote
| specialEval (SYMBOL "quasiquote") = SOME evalQuasiquote
| specialEval (SYMBOL "quasiquoteexpand") = SOME (fn _ => expandQuasiquote)
| specialEval (SYMBOL "defmacro!") = SOME evalDefmacro
| specialEval (SYMBOL "macroexpand") = SOME expandMacro
| specialEval (SYMBOL "try*") = SOME evalTry
| specialEval _ = NONE
and evalDef e [SYMBOL s, ast] = let val v = eval e ast in (def s v e; v) end
| evalDef _ _ = raise NotApplicable "def! needs a symbol and a form to evaluate"
and evalLet e [LIST bs, ast] = eval (bindLet bs (inside e)) ast
| evalLet e [VECTOR bs, ast] = eval (bindLet bs (inside e)) ast
| evalLet _ _ = raise NotApplicable "let* needs a list of bindings and a form to evaluate"
and evalDo e (x::xs) = foldl (fn (x, _) => eval e x) (eval e x) xs
| evalDo _ _ = raise NotApplicable "do needs at least one argument"
and evalIf e [c,a,b] = if truthy (eval e c) then eval e a else eval e b
| evalIf e [c,a] = evalIf e [c,a,NIL]
| evalIf _ _ = raise NotApplicable "if needs two or three arguments"
and evalFn e [(LIST binds),body] = makeFn e binds body
| evalFn e [(VECTOR binds),body] = makeFn e binds body
| evalFn _ _ = raise NotApplicable "fn* needs a list of bindings and a body"
and makeFn e binds body = FN (fn (exprs) => eval (bind (interleave binds exprs) (inside e)) body)
and evalQuote e [x] = x
| evalQuote _ _ = raise NotApplicable "quote needs one argument"
and evalQuasiquote e args = eval e (expandQuasiquote args)
and expandQuasiquote [LIST [SYMBOL "unquote", x]] = x
| expandQuasiquote [LIST l] = LIST (foldr quasiFolder [] l)
| expandQuasiquote [VECTOR v] = LIST [SYMBOL "vec", LIST (foldr quasiFolder [] v)]
| expandQuasiquote [m as MAP _] = LIST [SYMBOL "quote", m]
| expandQuasiquote [s as SYMBOL _] = LIST [SYMBOL "quote", s]
| expandQuasiquote [x] = x
| expandQuasiquote _ = raise NotApplicable "quasiquote needs one argument"
and quasiFolder (LIST [SYMBOL "splice-unquote", x], acc) = [SYMBOL "concat", x, LIST acc]
| quasiFolder (x, acc) = [SYMBOL "cons", expandQuasiquote [x], LIST acc]
and evalDefmacro e [SYMBOL s, LIST [SYMBOL "fn*", LIST binds, body]] = let val m = makeMacro e binds body in (def s m e; m) end
| evalDefmacro e [SYMBOL s, LIST [SYMBOL "fn*", VECTOR binds, body]] = let val m = makeMacro e binds body in (def s m e; m) end
| evalDefmacro _ _ = raise NotApplicable "defmacro! needs a name, a list of bindings, and a body"
and makeMacro e binds body = MACRO (fn (exprs) => eval (bind (interleave binds exprs) (inside e)) body)
and expandMacro e [(ast as LIST (SYMBOL s::args))] = (case lookup e s of SOME (MACRO m) => m args | _ => ast)
| expandMacro _ [ast] = ast
| expandMacro _ _ = raise NotApplicable "macroexpand needs one argument"
and evalTry e [a, LIST [SYMBOL "catch*", b, c]] = (eval e a handle ex => evalCatch (inside e) b ex c)
| evalTry e [a] = eval e a
| evalTry _ _ = raise NotApplicable "try* needs a form to evaluate"
and evalCatch e b ex body = eval (bind [b, exnVal ex] e) body
and exnVal (MalException x) = x
| exnVal (NotDefined msg) = STRING msg
| exnVal (NotApplicable msg) = STRING msg
| exnVal (OutOfBounds msg) = STRING msg
| exnVal exn = STRING (exnMessage exn)
and evalApply e (FN f) args = f (map (eval e) args)
| evalApply _ x args = raise NotApplicable (prStr x ^ " is not applicable on " ^ prStr (LIST args))
and evalSymbol e s = valOrElse (lookup e s)
(fn _ => raise NotDefined ("'" ^ s ^ "' not found"))
and bindLet args e = bind' (eval e) args e
and bind args e = bind' identity args e
and bind' evl (SYMBOL "&"::v::(SYMBOL s)::vs) e = (def s (LIST (map evl (v::vs))) e; e)
| bind' _ [SYMBOL "&", SYMBOL s] e = (def s (LIST []) e; e)
| bind' evl (SYMBOL s::v::rest) e = (def s (evl v) e; bind' evl rest e)
| bind' _ [] e = e
| bind' _ _ _ = raise NotApplicable "bindings must be a list of symbol/form pairs"
fun print f =
prReadableStr f
fun rep e s =
s |> read |> eval e |> print
handle Nothing => ""
| SyntaxError msg => "SYNTAX ERROR: " ^ msg
| NotApplicable msg => "CANNOT APPLY: " ^ msg
| NotDefined msg => "NOT DEFINED: " ^ msg
| MalException e => "ERROR: " ^ (prStr e)
| e => "ERROR: " ^ (exnMessage e)
val initEnv = ENV (NS (ref [])) |> bind coreNs
fun repl e =
let open TextIO
in (
print("user> ");
case inputLine(stdIn) of
SOME(line) =>
let val s = rep e line
val _ = print(s ^ "\n")
in
repl e
end
| NONE => ()
) end
val prelude = " \
\\
\(def! not (fn* (a) (if a false true))) \
\\
\(def! \
\ load-file \
\ (fn* (f) \
\ (eval (read-string (str \"(do \" (slurp f) \"\nnil)\")))))\
\\
\(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)))))))"
fun main () = (
bind [
SYMBOL "eval",
FN (fn ([x]) => eval initEnv x
| _ => raise NotApplicable "'eval' requires one argument")
] initEnv;
rep initEnv ("(do " ^ prelude ^ " nil)");
case CommandLine.arguments () of
prog::args => (
def "*ARGV*" (LIST (map STRING args)) initEnv;
rep initEnv ("(load-file \"" ^ prog ^ "\")");
()
)
| args => (
def "*ARGV*" (LIST (map STRING args)) initEnv;
repl initEnv
)
)