mirror of
https://github.com/kanaka/mal.git
synced 2024-11-13 01:43:50 +03:00
14ab099cea
tests: check that `inc` is present in stepA. nasm: split lines in mal_startup_string for readability. objpascal: remove obsolete .orig file swift: remove an unneeded line in template swift4: remove duplicate definition of `or` macro
318 lines
10 KiB
Smalltalk
318 lines
10 KiB
Smalltalk
String extend [
|
|
String >> loadRelative [
|
|
| scriptPath scriptDirectory |
|
|
scriptPath := thisContext currentFileName.
|
|
scriptDirectory := FilePath stripFileNameFor: scriptPath.
|
|
FileStream fileIn: (FilePath append: self to: scriptDirectory)
|
|
]
|
|
]
|
|
|
|
'readline.st' loadRelative.
|
|
'util.st' loadRelative.
|
|
'types.st' loadRelative.
|
|
'reader.st' loadRelative.
|
|
'printer.st' loadRelative.
|
|
'env.st' loadRelative.
|
|
'func.st' loadRelative.
|
|
'core.st' loadRelative.
|
|
|
|
Object subclass: MAL [
|
|
MAL class >> READ: input [
|
|
^Reader readStr: input
|
|
]
|
|
|
|
MAL class >> evalAst: sexp env: env [
|
|
sexp type = #symbol ifTrue: [
|
|
^env get: sexp value
|
|
].
|
|
|
|
sexp type = #list ifTrue: [
|
|
^self evalList: sexp env: env class: MALList
|
|
].
|
|
sexp type = #vector ifTrue: [
|
|
^self evalList: sexp env: env class: MALVector
|
|
].
|
|
sexp type = #map ifTrue: [
|
|
^self evalList: sexp env: env class: MALMap
|
|
].
|
|
|
|
^sexp
|
|
]
|
|
|
|
MAL class >> evalList: sexp env: env class: aClass [
|
|
| items |
|
|
items := sexp value collect:
|
|
[ :item | self EVAL: item env: env ].
|
|
^aClass new: items
|
|
]
|
|
|
|
MAL class >> quasiquote: ast [
|
|
| result a a0 a0_ a0_0 rest |
|
|
ast isPair ifFalse: [
|
|
result := {MALSymbol new: #quote. ast}.
|
|
^MALList new: (OrderedCollection from: result)
|
|
].
|
|
|
|
a := ast value.
|
|
a0 := a first.
|
|
|
|
(a0 type = #symbol and: [ a0 value = #unquote ]) ifTrue: [ ^a second ].
|
|
|
|
a0 isPair ifTrue: [
|
|
a0_ := a0 value.
|
|
a0_0 := a0_ first.
|
|
|
|
(a0_0 type = #symbol and:
|
|
[ a0_0 value = #'splice-unquote' ]) ifTrue: [
|
|
rest := MALList new: a allButFirst.
|
|
result := {MALSymbol new: #concat.
|
|
a0_ second.
|
|
self quasiquote: rest}.
|
|
^MALList new: (OrderedCollection from: result)
|
|
]
|
|
].
|
|
|
|
rest := MALList new: a allButFirst.
|
|
result := {MALSymbol new: #cons. self quasiquote: a0.
|
|
self quasiquote: rest}.
|
|
^MALList new: (OrderedCollection from: result)
|
|
]
|
|
|
|
MAL class >> isMacroCall: ast env: env [
|
|
| a0 a0_ f |
|
|
ast type = #list ifTrue: [
|
|
a0 := ast value first.
|
|
a0_ := a0 value.
|
|
a0 type = #symbol ifTrue: [
|
|
f := env find: a0_.
|
|
(f notNil and: [ f type = #func ]) ifTrue: [
|
|
^f isMacro
|
|
]
|
|
]
|
|
].
|
|
^false
|
|
]
|
|
|
|
MAL class >> macroexpand: aSexp env: env [
|
|
| sexp |
|
|
|
|
"NOTE: redefinition of method arguments is not allowed"
|
|
sexp := aSexp.
|
|
|
|
[ self isMacroCall: sexp env: env ] whileTrue: [
|
|
| ast a0_ macro rest |
|
|
ast := sexp value.
|
|
a0_ := ast first value.
|
|
macro := env find: a0_.
|
|
rest := ast allButFirst.
|
|
sexp := macro fn value: rest.
|
|
].
|
|
|
|
^sexp
|
|
]
|
|
|
|
MAL class >> EVAL: aSexp env: anEnv [
|
|
| sexp env ast a0 a0_ a1 a1_ a2 a2_ a3 forms function args |
|
|
|
|
"NOTE: redefinition of method arguments is not allowed"
|
|
sexp := aSexp.
|
|
env := anEnv.
|
|
|
|
[
|
|
[ :continue |
|
|
sexp type ~= #list ifTrue: [
|
|
^self evalAst: sexp env: env
|
|
].
|
|
sexp value isEmpty ifTrue: [
|
|
^sexp
|
|
].
|
|
|
|
sexp := self macroexpand: sexp env: env.
|
|
sexp type ~= #list ifTrue: [
|
|
^self evalAst: sexp env: env
|
|
].
|
|
|
|
ast := sexp value.
|
|
a0 := ast first.
|
|
|
|
a0_ := ast first value.
|
|
a0_ = #'def!' ifTrue: [
|
|
| result |
|
|
a1_ := ast second value.
|
|
a2 := ast third.
|
|
result := self EVAL: a2 env: env.
|
|
env set: a1_ value: result.
|
|
^result
|
|
].
|
|
|
|
a0_ = #'defmacro!' ifTrue: [
|
|
| result |
|
|
a1_ := ast second value.
|
|
a2 := ast third.
|
|
result := self EVAL: a2 env: env.
|
|
result isMacro: true.
|
|
env set: a1_ value: result.
|
|
^result
|
|
].
|
|
|
|
a0_ = #'macroexpand' ifTrue: [
|
|
a1 := ast second.
|
|
^self macroexpand: a1 env: env
|
|
].
|
|
|
|
a0_ = #'let*' ifTrue: [
|
|
| env_ |
|
|
env_ := Env new: env.
|
|
a1_ := ast second value.
|
|
a2 := ast third.
|
|
1 to: a1_ size by: 2 do:
|
|
[ :i | env_ set: (a1_ at: i) value
|
|
value: (self EVAL: (a1_ at: i + 1)
|
|
env: env_) ].
|
|
env := env_.
|
|
sexp := a2.
|
|
continue value "TCO"
|
|
].
|
|
|
|
a0_ = #do ifTrue: [
|
|
| forms last |
|
|
ast size < 2 ifTrue: [
|
|
forms := {}.
|
|
last := MALObject Nil.
|
|
] ifFalse: [
|
|
forms := ast copyFrom: 2 to: ast size - 1.
|
|
last := ast last.
|
|
].
|
|
|
|
forms do: [ :form | self EVAL: form env: env ].
|
|
sexp := last.
|
|
continue value "TCO"
|
|
].
|
|
|
|
a0_ = #if ifTrue: [
|
|
| condition |
|
|
a1 := ast second.
|
|
a2 := ast third.
|
|
a3 := ast at: 4 ifAbsent: [ MALObject Nil ].
|
|
condition := self EVAL: a1 env: env.
|
|
|
|
(condition type = #false or:
|
|
[ condition type = #nil ]) ifTrue: [
|
|
sexp := a3
|
|
] ifFalse: [
|
|
sexp := a2
|
|
].
|
|
continue value "TCO"
|
|
].
|
|
|
|
a0_ = #quote ifTrue: [
|
|
a1 := ast second.
|
|
^a1
|
|
].
|
|
|
|
a0_ = #quasiquote ifTrue: [
|
|
| result |
|
|
a1 := ast second.
|
|
sexp := self quasiquote: a1.
|
|
continue value "TCO"
|
|
].
|
|
|
|
a0_ = #'try*' ifTrue: [
|
|
| A B C |
|
|
A := ast second.
|
|
ast at: 3 ifAbsent: [
|
|
^self EVAL: A env: env.
|
|
].
|
|
a2_ := ast third value.
|
|
B := a2_ second value.
|
|
C := a2_ third.
|
|
^[ self EVAL: A env: env ] on: MALError do:
|
|
[ :err |
|
|
| data env_ result |
|
|
data := err data.
|
|
data isString ifTrue: [
|
|
data := MALString new: data
|
|
].
|
|
env_ := Env new: env binds: {B} exprs: {data}.
|
|
err return: (self EVAL: C env: env_)
|
|
]
|
|
].
|
|
|
|
a0_ = #'fn*' ifTrue: [
|
|
| binds env_ fn |
|
|
a1_ := ast second value.
|
|
binds := a1_ collect: [ :item | item value ].
|
|
a2 := ast third.
|
|
fn := [ :args |
|
|
self EVAL: a2 env:
|
|
(Env new: env binds: binds exprs: args) ].
|
|
^Func new: a2 params: binds env: env fn: fn
|
|
].
|
|
|
|
forms := (self evalAst: sexp env: env) value.
|
|
function := forms first.
|
|
args := forms allButFirst asArray.
|
|
|
|
function type = #fn ifTrue: [ ^function fn value: args ].
|
|
function type = #func ifTrue: [
|
|
| env_ |
|
|
sexp := function ast.
|
|
env_ := Env new: function env binds: function params
|
|
exprs: args.
|
|
env := env_.
|
|
continue value "TCO"
|
|
]
|
|
] valueWithExit
|
|
] repeat.
|
|
]
|
|
|
|
MAL class >> PRINT: sexp [
|
|
^Printer prStr: sexp printReadably: true
|
|
]
|
|
|
|
MAL class >> rep: input env: env [
|
|
^self PRINT: (self EVAL: (self READ: input) env: env)
|
|
]
|
|
]
|
|
|
|
| input historyFile replEnv argv |
|
|
|
|
historyFile := '.mal_history'.
|
|
ReadLine readHistory: historyFile.
|
|
replEnv := Env new: nil.
|
|
|
|
argv := Smalltalk arguments.
|
|
argv notEmpty ifTrue: [ argv := argv allButFirst ].
|
|
argv := OrderedCollection from: (argv collect: [ :arg | MALString new: arg ]).
|
|
|
|
Core Ns keysAndValuesDo: [ :op :block | replEnv set: op value: block ].
|
|
replEnv set: #eval value: (Fn new: [ :args | MAL EVAL: args first env: replEnv ]).
|
|
replEnv set: #'*ARGV*' value: (MALList new: argv).
|
|
replEnv set: #'*host-language*' value: (MALString new: 'smalltalk').
|
|
|
|
MAL rep: '(def! not (fn* (a) (if a false true)))' env: replEnv.
|
|
MAL rep: '(def! load-file (fn* (f) (eval (read-string (str "(do " (slurp f) ")")))))' env: replEnv.
|
|
|
|
MAL 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)))))))' env: replEnv.
|
|
MAL rep: '(def! inc (fn* [x] (+ x 1)))' env: replEnv.
|
|
MAL rep: '(def! gensym (let* [counter (atom 0)] (fn* [] (symbol (str "G__" (swap! counter inc))))))' env: replEnv.
|
|
MAL rep: '(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) (let* (condvar (gensym)) `(let* (~condvar ~(first xs)) (if ~condvar ~condvar (or ~@(rest xs)))))))))' env: replEnv.
|
|
|
|
Smalltalk arguments notEmpty ifTrue: [
|
|
MAL rep: '(load-file "', Smalltalk arguments first, '")' env: replEnv
|
|
] ifFalse: [
|
|
MAL rep: '(println (str "Mal [" *host-language* "]"))' env: replEnv.
|
|
[ input := ReadLine readLine: 'user> '. input isNil ] whileFalse: [
|
|
input isEmpty ifFalse: [
|
|
ReadLine addHistory: input.
|
|
ReadLine writeHistory: historyFile.
|
|
[ (MAL rep: input env: replEnv) displayNl ]
|
|
on: MALEmptyInput do: [ #return ]
|
|
on: MALError do:
|
|
[ :err | ('error: ', err messageText) displayNl. #return ].
|
|
]
|
|
].
|
|
|
|
'' displayNl.
|
|
]
|