@ -0,0 +1 @@ | |||||
.crystal |
@ -0,0 +1,413 @@ | |||||
require "time" | |||||
require "./types" | |||||
require "./error" | |||||
require "./printer" | |||||
require "./reader" | |||||
require "./readline" | |||||
module Mal | |||||
macro calc_op(op) | |||||
-> (args : Array(Mal::Type)) { | |||||
x, y = args[0].unwrap, args[1].unwrap | |||||
eval_error "invalid arguments for binary operator {{op.id}}" unless x.is_a?(Int32) && y.is_a?(Int32) | |||||
Mal::Type.new(x {{op.id}} y) | |||||
} | |||||
end | |||||
def self.list(args) | |||||
args.to_mal | |||||
end | |||||
def self.list?(args) | |||||
args.first.unwrap.is_a? Mal::List | |||||
end | |||||
def self.empty?(args) | |||||
a = args.first.unwrap | |||||
a.is_a?(Array) ? a.empty? : false | |||||
end | |||||
def self.count(args) | |||||
a = args.first.unwrap | |||||
case a | |||||
when Array | |||||
a.size as Int32 | |||||
when Nil | |||||
0 | |||||
else | |||||
eval_error "invalid argument for function 'count'" | |||||
end | |||||
end | |||||
def self.pr_str_(args) | |||||
args.map{|a| pr_str(a)}.join(" ") | |||||
end | |||||
def self.str(args) | |||||
args.map{|a| pr_str(a, false)}.join | |||||
end | |||||
def self.prn(args) | |||||
puts self.pr_str_(args) | |||||
nil | |||||
end | |||||
def self.println(args) | |||||
puts args.map{|a| pr_str(a, false)}.join(" ") | |||||
nil | |||||
end | |||||
def self.read_string(args) | |||||
head = args.first.unwrap | |||||
eval_error "argument of read-str must be string" unless head.is_a? String | |||||
read_str head | |||||
end | |||||
def self.slurp(args) | |||||
head = args.first.unwrap | |||||
eval_error "argument of slurp must be string" unless head.is_a? String | |||||
begin | |||||
File.read head | |||||
rescue e : Errno | |||||
eval_error "no such file" | |||||
end | |||||
end | |||||
def self.cons(args) | |||||
head, tail = args[0] as Mal::Type, args[1].unwrap | |||||
eval_error "2nd arg of cons must be list" unless tail.is_a? Array | |||||
([head] + tail).to_mal | |||||
end | |||||
def self.concat(args) | |||||
args.each_with_object(Mal::List.new) do |arg, list| | |||||
a = arg.unwrap | |||||
eval_error "arguments of concat must be list" unless a.is_a?(Array) | |||||
a.each{|e| list << e} | |||||
end | |||||
end | |||||
def self.nth(args) | |||||
a0, a1 = args[0].unwrap, args[1].unwrap | |||||
eval_error "1st argument of nth must be list or vector" unless a0.is_a? Array | |||||
eval_error "2nd argument of nth must be integer" unless a1.is_a? Int32 | |||||
a0[a1] | |||||
end | |||||
def self.first(args) | |||||
a0 = args[0].unwrap | |||||
return nil if a0.nil? | |||||
eval_error "1st argument of first must be list or vector or nil" unless a0.is_a? Array | |||||
a0.empty? ? nil : a0.first | |||||
end | |||||
def self.rest(args) | |||||
a0 = args[0].unwrap | |||||
return Mal::List.new if a0.nil? | |||||
eval_error "1st argument of first must be list or vector or nil" unless a0.is_a? Array | |||||
return Mal::List.new if a0.empty? | |||||
a0[1..-1].to_mal | |||||
end | |||||
def self.apply(args) | |||||
eval_error "apply must take at least 2 arguments" unless args.size >= 2 | |||||
head = args.first.unwrap | |||||
last = args.last.unwrap | |||||
eval_error "last argument of apply must be list or vector" unless last.is_a? Array | |||||
case head | |||||
when Mal::Closure | |||||
head.fn.call(args[1..-2] + last) | |||||
when Mal::Func | |||||
head.call(args[1..-2] + last) | |||||
else | |||||
eval_error "1st argument of apply must be function or closure" | |||||
end | |||||
end | |||||
def self.map(args) | |||||
func = args.first.unwrap | |||||
list = args[1].unwrap | |||||
eval_error "2nd argument of map must be list or vector" unless list.is_a? Array | |||||
f = case func | |||||
when Mal::Closure then func.fn | |||||
when Mal::Func then func | |||||
else eval_error "1st argument of map must be function" | |||||
end | |||||
list.each_with_object(Mal::List.new) do |elem, mapped| | |||||
mapped << f.call([elem]) | |||||
end | |||||
end | |||||
def self.nil?(args) | |||||
args.first.unwrap.nil? | |||||
end | |||||
def self.true?(args) | |||||
a = args.first.unwrap | |||||
a.is_a?(Bool) && a | |||||
end | |||||
def self.false?(args) | |||||
a = args.first.unwrap | |||||
a.is_a?(Bool) && !a | |||||
end | |||||
def self.symbol?(args) | |||||
args.first.unwrap.is_a?(Mal::Symbol) | |||||
end | |||||
def self.symbol(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of symbol function must be string" unless head.is_a? String | |||||
Mal::Symbol.new head | |||||
end | |||||
def self.keyword(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of symbol function must be string" unless head.is_a? String | |||||
"\u029e" + head | |||||
end | |||||
def self.keyword?(args) | |||||
head = args.first.unwrap | |||||
head.is_a?(String) && !head.empty? && head[0] == '\u029e' | |||||
end | |||||
def self.vector(args) | |||||
args.to_mal(Mal::Vector) | |||||
end | |||||
def self.vector?(args) | |||||
args.first.unwrap.is_a? Mal::Vector | |||||
end | |||||
def self.hash_map(args) | |||||
eval_error "hash-map must take even number of arguments" unless args.size.even? | |||||
map = Mal::HashMap.new | |||||
args.each_slice(2) do |kv| | |||||
k = kv[0].unwrap | |||||
eval_error "key must be string" unless k.is_a? String | |||||
map[k] = kv[1] | |||||
end | |||||
map | |||||
end | |||||
def self.map?(args) | |||||
args.first.unwrap.is_a? Mal::HashMap | |||||
end | |||||
def self.assoc(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of assoc must be hashmap" unless head.is_a? Mal::HashMap | |||||
eval_error "assoc must take a list and even number of arguments" unless (args.size - 1).even? | |||||
map = Mal::HashMap.new | |||||
head.each{|k, v| map[k] = v} | |||||
args[1..-1].each_slice(2) do |kv| | |||||
k = kv[0].unwrap | |||||
eval_error "key must be string" unless k.is_a? String | |||||
map[k] = kv[1] | |||||
end | |||||
map | |||||
end | |||||
def self.dissoc(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of assoc must be hashmap" unless head.is_a? Mal::HashMap | |||||
map = Mal::HashMap.new | |||||
head.each{|k,v| map[k] = v} | |||||
args[1..-1].each do |arg| | |||||
key = arg.unwrap | |||||
eval_error "key must be string" unless key.is_a? String | |||||
map.delete key | |||||
end | |||||
map | |||||
end | |||||
def self.get(args) | |||||
a0, a1 = args[0].unwrap, args[1].unwrap | |||||
return nil unless a0.is_a? Mal::HashMap | |||||
eval_error "2nd argument of get must be string" unless a1.is_a? String | |||||
# a0[a1]? isn't available because type ofa0[a1] is infered NoReturn | |||||
a0.has_key?(a1) ? a0[a1] : nil | |||||
end | |||||
def self.contains?(args) | |||||
a0, a1 = args[0].unwrap, args[1].unwrap | |||||
eval_error "1st argument of get must be hashmap" unless a0.is_a? Mal::HashMap | |||||
eval_error "2nd argument of get must be string" unless a1.is_a? String | |||||
a0.has_key? a1 | |||||
end | |||||
def self.keys(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of assoc must be hashmap" unless head.is_a? Mal::HashMap | |||||
head.keys.each_with_object(Mal::List.new){|e,l| l << Mal::Type.new(e)} | |||||
end | |||||
def self.vals(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of assoc must be hashmap" unless head.is_a? Mal::HashMap | |||||
head.values.to_mal | |||||
end | |||||
def self.sequential?(args) | |||||
args.first.unwrap.is_a? Array | |||||
end | |||||
def self.readline(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of readline must be string" unless head.is_a? String | |||||
my_readline head | |||||
end | |||||
def self.meta(args) | |||||
m = args.first.meta | |||||
m.nil? ? nil : m | |||||
end | |||||
def self.with_meta(args) | |||||
t = args.first.dup | |||||
t.meta = args[1] | |||||
t | |||||
end | |||||
def self.atom(args) | |||||
Mal::Atom.new args.first | |||||
end | |||||
def self.atom?(args) | |||||
args.first.unwrap.is_a? Mal::Atom | |||||
end | |||||
def self.deref(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of deref must be atom" unless head.is_a? Mal::Atom | |||||
head.val | |||||
end | |||||
def self.reset!(args) | |||||
head = args.first.unwrap | |||||
eval_error "1st argument of reset! must be atom" unless head.is_a? Mal::Atom | |||||
head.val = args[1] | |||||
end | |||||
def self.swap!(args) | |||||
atom = args.first.unwrap | |||||
eval_error "1st argument of swap! must be atom" unless atom.is_a? Mal::Atom | |||||
a = [atom.val] + args[2..-1] | |||||
func = args[1].unwrap | |||||
case func | |||||
when Mal::Func | |||||
atom.val = func.call a | |||||
when Mal::Closure | |||||
atom.val = func.fn.call a | |||||
else | |||||
eval_error "2nd argumetn of swap! must be function" | |||||
end | |||||
end | |||||
def self.conj(args) | |||||
seq = args.first.unwrap | |||||
case seq | |||||
when Mal::List | |||||
(args[1..-1].reverse + seq).to_mal | |||||
when Mal::Vector | |||||
(seq + args[1..-1]).to_mal(Mal::Vector) | |||||
else | |||||
eval_error "1st argument of conj must be list or vector" | |||||
end | |||||
end | |||||
def self.time_ms(args) | |||||
(Time.now.to_i.to_i32) * 1000 | |||||
end | |||||
# Note: | |||||
# Simply using ->self.some_func doesn't work | |||||
macro func(name) | |||||
-> (args : Array(Mal::Type)) { Mal::Type.new self.{{name.id}}(args) } | |||||
end | |||||
macro rel_op(op) | |||||
-> (args : Array(Mal::Type)) { Mal::Type.new (args[0] {{op.id}} args[1]) } | |||||
end | |||||
NS = { | |||||
"+" => calc_op(:+) | |||||
"-" => calc_op(:-) | |||||
"*" => calc_op(:*) | |||||
"/" => calc_op(:/) | |||||
"list" => func(:list) | |||||
"list?" => func(:list?) | |||||
"empty?" => func(:empty?) | |||||
"count" => func(:count) | |||||
"=" => rel_op(:==) | |||||
"<" => rel_op(:<) | |||||
">" => rel_op(:>) | |||||
"<=" => rel_op(:<=) | |||||
">=" => rel_op(:>=) | |||||
"pr-str" => func(:pr_str_) | |||||
"str" => func(:str) | |||||
"prn" => func(:prn) | |||||
"println" => func(:println) | |||||
"read-string" => func(:read_string) | |||||
"slurp" => func(:slurp) | |||||
"cons" => func(:cons) | |||||
"concat" => func(:concat) | |||||
"nth" => func(:nth) | |||||
"first" => func(:first) | |||||
"rest" => func(:rest) | |||||
"throw" => -> (args : Array(Mal::Type)) { raise Mal::RuntimeException.new args[0] } | |||||
"apply" => func(:apply) | |||||
"map" => func(:map) | |||||
"nil?" => func(:nil?) | |||||
"true?" => func(:true?) | |||||
"false?" => func(:false?) | |||||
"symbol?" => func(:symbol?) | |||||
"symbol" => func(:symbol) | |||||
"keyword" => func(:keyword) | |||||
"keyword?" => func(:keyword?) | |||||
"vector" => func(:vector) | |||||
"vector?" => func(:vector?) | |||||
"hash-map" => func(:hash_map) | |||||
"map?" => func(:map?) | |||||
"assoc" => func(:assoc) | |||||
"dissoc" => func(:dissoc) | |||||
"get" => func(:get) | |||||
"contains?" => func(:contains?) | |||||
"keys" => func(:keys) | |||||
"vals" => func(:vals) | |||||
"sequential?" => func(:sequential?) | |||||
"readline" => func(:readline) | |||||
"meta" => func(:meta) | |||||
"with-meta" => func(:with_meta) | |||||
"atom" => func(:atom) | |||||
"atom?" => func(:atom?) | |||||
"deref" => func(:deref) | |||||
"deref" => func(:deref) | |||||
"reset!" => func(:reset!) | |||||
"swap!" => func(:swap!) | |||||
"conj" => func(:conj) | |||||
"time-ms" => func(:time_ms) | |||||
} of String => Mal::Func | |||||
end |
@ -0,0 +1,68 @@ | |||||
require "./types" | |||||
require "./error" | |||||
module Mal | |||||
class Env | |||||
property data | |||||
def initialize(@outer) | |||||
@data = {} of String => Mal::Type | |||||
end | |||||
def initialize(@outer, binds, exprs : Array(Mal::Type)) | |||||
@data = {} of String => Mal::Type | |||||
eval_error "binds must be list or vector" unless binds.is_a? Array | |||||
# Note: | |||||
# Array#zip() can't be used because overload resolution failed | |||||
(0...binds.size).each do |idx| | |||||
sym = binds[idx].unwrap | |||||
eval_error "bind name must be symbol" unless sym.is_a? Mal::Symbol | |||||
if sym.str == "&" | |||||
eval_error "missing variable parameter name" if binds.size == idx | |||||
next_param = binds[idx+1].unwrap | |||||
eval_error "bind name must be symbol" unless next_param.is_a? Mal::Symbol | |||||
var_args = Mal::List.new | |||||
exprs[idx..-1].each{|e| var_args << e} if idx < exprs.size | |||||
@data[next_param.str] = Mal::Type.new var_args | |||||
break | |||||
end | |||||
@data[sym.str] = exprs[idx] | |||||
end | |||||
end | |||||
def dump | |||||
puts "ENV BEGIN".colorize.red | |||||
@data.each do |k, v| | |||||
puts " #{k} -> #{print(v)}".colorize.red | |||||
end | |||||
puts "ENV END".colorize.red | |||||
end | |||||
def set(key, value) | |||||
@data[key] = value | |||||
end | |||||
def find(key) | |||||
return self if @data.has_key? key | |||||
o = @outer | |||||
if o | |||||
o.find key | |||||
else | |||||
nil | |||||
end | |||||
end | |||||
def get(key) | |||||
e = find key | |||||
eval_error "'#{key}' not found" unless e | |||||
e.data[key] | |||||
end | |||||
end | |||||
end |
@ -0,0 +1,22 @@ | |||||
module Mal | |||||
class ParseException < Exception | |||||
end | |||||
class EvalException < Exception | |||||
end | |||||
class RuntimeException < Exception | |||||
getter :thrown | |||||
def initialize(@thrown) | |||||
super() | |||||
end | |||||
end | |||||
end | |||||
def eval_error(msg) | |||||
raise Mal::EvalException.new msg | |||||
end | |||||
def parse_error(msg) | |||||
raise Mal::ParseException.new msg | |||||
end |
@ -0,0 +1,34 @@ | |||||
require "./types" | |||||
def pr_str(value, print_readably = true) | |||||
case value | |||||
when Nil then "nil" | |||||
when Bool then value.to_s | |||||
when Int32 then value.to_s | |||||
when Mal::List then "(#{value.map{|v| pr_str(v, print_readably) as String}.join(" ")})" | |||||
when Mal::Vector then "[#{value.map{|v| pr_str(v, print_readably) as String}.join(" ")}]" | |||||
when Mal::Symbol then value.str.to_s | |||||
when Mal::Func then "<function>" | |||||
when Mal::Closure then "<closure>" | |||||
when Mal::HashMap | |||||
# step1_read_print.cr requires specifying type | |||||
"{#{value.map{|k, v| "#{pr_str(k, print_readably)} #{pr_str(v, print_readably)}" as String}.join(" ")}}" | |||||
when String | |||||
case | |||||
when value.empty?() | |||||
print_readably ? value.inspect : value | |||||
when value[0] == '\u029e' | |||||
":#{value[1..-1]}" | |||||
else | |||||
print_readably ? value.inspect : value | |||||
end | |||||
when Mal::Atom | |||||
"(atom #{pr_str(value.val, print_readably)})" | |||||
else | |||||
raise "invalid MalType: #{value.to_s}" | |||||
end | |||||
end | |||||
def pr_str(t : Mal::Type, print_readably = true) | |||||
pr_str(t.unwrap, print_readably) + (t.macro? ? " (macro)" : "") | |||||
end |
@ -0,0 +1,137 @@ | |||||
require "./types" | |||||
require "./error" | |||||
class Reader | |||||
def initialize(@tokens) | |||||
@pos = 0 | |||||
end | |||||
def current_token | |||||
@tokens[@pos] rescue nil | |||||
end | |||||
def peek | |||||
t = current_token | |||||
if t && t[0] == ';' | |||||
@pos += 1 | |||||
peek | |||||
else | |||||
t | |||||
end | |||||
end | |||||
def next | |||||
peek | |||||
ensure | |||||
@pos += 1 | |||||
end | |||||
def read_sequence(init, open, close) | |||||
token = self.next | |||||
parse_error "expected '#{open}', got EOF" unless token | |||||
parse_error "expected '#{open}', got #{token}" unless token[0] == open | |||||
loop do | |||||
token = peek | |||||
parse_error "expected '#{close}', got EOF" unless token | |||||
break if token[0] == close | |||||
init << read_form | |||||
peek | |||||
end | |||||
self.next | |||||
init | |||||
end | |||||
def read_list | |||||
Mal::Type.new read_sequence(Mal::List.new, '(', ')') | |||||
end | |||||
def read_vector | |||||
Mal::Type.new read_sequence(Mal::Vector.new, '[', ']') | |||||
end | |||||
def read_hashmap | |||||
types = read_sequence([] of Mal::Type, '{', '}') | |||||
parse_error "odd number of elements for hash-map: #{types.size}" if types.size.odd? | |||||
map = Mal::HashMap.new | |||||
types.each_slice(2) do |kv| | |||||
k, v = kv[0].unwrap, kv[1] | |||||
case k | |||||
when String | |||||
map[k] = v | |||||
else | |||||
parse_error("key of hash-map must be string or keyword") | |||||
end | |||||
end | |||||
Mal::Type.new map | |||||
end | |||||
def read_atom | |||||
token = self.next | |||||
parse_error "expected Atom but got EOF" unless token | |||||
Mal::Type.new case | |||||
when token =~ /^-?\d+$/ then token.to_i | |||||
when token == "true" then true | |||||
when token == "false" then false | |||||
when token == "nil" then nil | |||||
when token[0] == '"' then token[1..-2].gsub(/\\"/, "\"") | |||||
when token[0] == ':' then "\u029e#{token[1..-1]}" | |||||
else Mal::Symbol.new token | |||||
end | |||||
end | |||||
def list_of(symname) | |||||
Mal::List.new << gen_type(Mal::Symbol, symname) << read_form | |||||
end | |||||
def read_form | |||||
token = peek | |||||
parse_error "unexpected EOF" unless token | |||||
parse_error "unexpected comment" if token[0] == ';' | |||||
Mal::Type.new case token | |||||
when "(" then read_list | |||||
when ")" then parse_error "unexpected ')'" | |||||
when "[" then read_vector | |||||
when "]" then parse_error "unexpected ']'" | |||||
when "{" then read_hashmap | |||||
when "}" then parse_error "unexpected '}'" | |||||
when "'" then self.next; list_of("quote") | |||||
when "`" then self.next; list_of("quasiquote") | |||||
when "~" then self.next; list_of("unquote") | |||||
when "~@" then self.next; list_of("splice-unquote") | |||||
when "@" then self.next; list_of("deref") | |||||
when "^" | |||||
self.next | |||||
meta = read_form | |||||
list_of("with-meta") << meta | |||||
else read_atom | |||||
end | |||||
end | |||||
end | |||||
def tokenize(str) | |||||
regex = /[\s,]*(~@|[\[\]{}()'`~^@]|"(?:\\.|[^\\"])*"|;.*|[^\s\[\]{}('"`,;)]*)/ | |||||
str.scan(regex).map{|m| m[1]}.reject(&.empty?) | |||||
end | |||||
def read_str(str) | |||||
r = Reader.new(tokenize(str)) | |||||
begin | |||||
r.read_form | |||||
ensure | |||||
unless r.peek.nil? | |||||
raise Mal::ParseException.new "expected EOF, got #{r.peek.to_s}" | |||||
end | |||||
end | |||||
end | |||||
@ -0,0 +1,21 @@ | |||||
# Note: | |||||
# Crystal already has "readline" library. | |||||
# I implemented a subset of it again for practice. | |||||
@[Link("readline")] | |||||
lib LibReadline | |||||
fun readline(prompt : UInt8*) : UInt8* | |||||
fun add_history(line : UInt8*) | |||||
end | |||||
def my_readline(prompt = "") | |||||
line = LibReadline.readline(prompt) | |||||
if line | |||||
LibReadline.add_history(line) | |||||
String.new(line) | |||||
else | |||||
nil | |||||
end | |||||
ensure | |||||
LibC.free(line as Void*) if line | |||||
end |
@ -0,0 +1,26 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def read(x) | |||||
x | |||||
end | |||||
def eval(x) | |||||
x | |||||
end | |||||
def print(x) | |||||
x | |||||
end | |||||
def rep(x) | |||||
read(eval(print(x))) | |||||
end | |||||
while line = my_readline("user> ") | |||||
puts rep(line) | |||||
end |
@ -0,0 +1,32 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
def eval(x) | |||||
x | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str))) | |||||
end | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,90 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def eval_error(msg) | |||||
raise Mal::EvalException.new msg | |||||
end | |||||
def num_func(func) | |||||
-> (args : Array(Mal::Type)) { | |||||
x, y = args[0].unwrap, args[1].unwrap | |||||
eval_error "invalid arguments" unless x.is_a?(Int32) && y.is_a?(Int32) | |||||
Mal::Type.new func.call(x, y) | |||||
} | |||||
end | |||||
$repl_env = { | |||||
"+" => num_func(->(x : Int32, y : Int32){ x + y }), | |||||
"-" => num_func(->(x : Int32, y : Int32){ x - y }), | |||||
"*" => num_func(->(x : Int32, y : Int32){ x * y }), | |||||
"/" => num_func(->(x : Int32, y : Int32){ x / y }), | |||||
} of String => Mal::Func | |||||
def eval_ast(a, env) | |||||
return a.map{|n| eval(n, env) as Mal::Type} if a.is_a? Mal::List | |||||
return a unless a | |||||
ast = a.unwrap | |||||
case ast | |||||
when Mal::Symbol | |||||
if env.has_key? ast.str | |||||
env[ast.str] | |||||
else | |||||
eval_error "'#{ast.str}' not found" | |||||
end | |||||
when Mal::List | |||||
ast.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
ast.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Mal::HashMap | |||||
ast.each{|k, v| ast[k] = eval(v, env)} | |||||
else | |||||
ast | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
def eval(t, env) | |||||
Mal::Type.new case ast = t.unwrap | |||||
when Mal::List | |||||
eval_error "empty list" if ast.empty? | |||||
f = eval_ast(ast.first, env) | |||||
ast.shift(1) | |||||
args = eval_ast(ast, env) | |||||
if f.is_a?(Mal::Func) | |||||
f.call(args) | |||||
else | |||||
eval_error "expected function symbol as the first symbol of list" | |||||
end | |||||
else | |||||
eval_ast(t, env) | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,115 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def eval_error(msg) | |||||
raise Mal::EvalException.new msg | |||||
end | |||||
def num_func(func) | |||||
-> (args : Array(Mal::Type)) { | |||||
x, y = args[0].unwrap, args[1].unwrap | |||||
eval_error "invalid arguments" unless x.is_a?(Int32) && y.is_a?(Int32) | |||||
Mal::Type.new func.call(x, y) | |||||
} | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
$repl_env.set("+", Mal::Type.new num_func(->(x : Int32, y : Int32){ x + y })) | |||||
$repl_env.set("-", Mal::Type.new num_func(->(x : Int32, y : Int32){ x - y })) | |||||
$repl_env.set("*", Mal::Type.new num_func(->(x : Int32, y : Int32){ x * y })) | |||||
$repl_env.set("/", Mal::Type.new num_func(->(x : Int32, y : Int32){ x / y })) | |||||
def eval_ast(a, env) | |||||
return a.map{|n| eval(n, env) } if a.is_a? Array | |||||
Mal::Type.new case ast = a.unwrap | |||||
when Mal::Symbol | |||||
if e = env.get(ast.str) | |||||
e | |||||
else | |||||
eval_error "'#{ast.str}' not found" | |||||
end | |||||
when Mal::List | |||||
ast.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
ast.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Mal::HashMap | |||||
new_map = Mal::HashMap.new | |||||
ast.each{|k, v| new_map[k] = eval(v, env)} | |||||
new_map | |||||
else | |||||
ast | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
def eval(t, env) | |||||
ast = t.unwrap | |||||
return eval_ast(t, env) unless ast.is_a?(Mal::List) | |||||
eval_error "empty list" if ast.empty? | |||||
sym = ast.first.unwrap | |||||
eval_error "first element of list must be a symbol" unless sym.is_a?(Mal::Symbol) | |||||
Mal::Type.new case sym.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless ast.size == 3 | |||||
a1 = ast[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol" unless a1.is_a?(Mal::Symbol) | |||||
env.set(a1.str, eval(ast[2], env) as Mal::Type) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless ast.size == 3 | |||||
bindings = ast[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a?(Array) | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
name, value = binding[0].unwrap, binding[1] | |||||
eval_error "name of binding must be specified as symbol" unless name.is_a?(Mal::Symbol) | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
eval(ast[2], new_env) | |||||
else | |||||
f = eval_ast(ast.first, env) | |||||
ast.shift(1) | |||||
args = eval_ast(ast, env) | |||||
if f.is_a?(Mal::Type) && (f2 = f.unwrap).is_a?(Mal::Func) | |||||
f2.call(args as Array(Mal::Type)) | |||||
else | |||||
eval_error "expected function symbol as the first symbol of list" | |||||
end | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,130 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
require "./core" | |||||
require "./error" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def func_of(env, binds, body) | |||||
-> (args : Array(Mal::Type)) { | |||||
new_env = Mal::Env.new(env, binds, args) | |||||
eval(body, new_env) | |||||
} as Mal::Func | |||||
end | |||||
def eval_ast(ast, env) | |||||
return ast.map{|n| eval(n, env) as Mal::Type} if ast.is_a? Mal::List | |||||
val = ast.unwrap | |||||
Mal::Type.new case val | |||||
when Mal::Symbol | |||||
if e = env.get(val.str) | |||||
e | |||||
else | |||||
eval_error "'#{val.str}' not found" | |||||
end | |||||
when Mal::List | |||||
val.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
val.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Mal::HashMap | |||||
val.each{|k, v| val[k] = eval(v, env)} | |||||
val | |||||
else | |||||
val | |||||
end | |||||
end | |||||
def eval_invocation(list, env) | |||||
f = eval(list.first, env).unwrap | |||||
eval_error "expected function symbol as the first symbol of list" unless f.is_a? Mal::Func | |||||
f.call eval_ast(list[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, env) | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
def eval(ast, env) | |||||
list = ast.unwrap | |||||
return eval_ast(ast, env) unless list.is_a? Mal::List | |||||
return gen_type Mal::List if list.empty? | |||||
head = list.first.unwrap | |||||
Mal::Type.new case head | |||||
when Mal::Symbol | |||||
case head.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env)) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
bindings = list[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a? Array | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
key, value = binding | |||||
name = key.unwrap | |||||
eval_error "name of binding must be specified as symbol" unless name.is_a? Mal::Symbol | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
eval(list[2], new_env) | |||||
when "do" | |||||
list.shift 1 | |||||
eval_ast(list, env).last | |||||
when "if" | |||||
cond = eval(list[1], env).unwrap | |||||
case cond | |||||
when Nil | |||||
list.size >= 4 ? eval(list[3], env) : nil | |||||
when false | |||||
list.size >= 4 ? eval(list[3], env) : nil | |||||
else | |||||
eval(list[2], env) | |||||
end | |||||
when "fn*" | |||||
# Note: | |||||
# If writing lambda expression here directly, compiler will fail to infer type of 'list'. (Error 'Nil for empty?') | |||||
func_of(env, list[1].unwrap, list[2]) | |||||
else | |||||
eval_invocation(list, env) | |||||
end | |||||
else | |||||
eval_invocation(list, env) | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
Mal::NS.each{|k,v| $repl_env.set(k, Mal::Type.new(v))} | |||||
rep "(def! not (fn* (a) (if a false true)))" | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,164 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
require "./core" | |||||
require "./error" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def func_of(env, binds, body) | |||||
-> (args : Array(Mal::Type)) { | |||||
new_env = Mal::Env.new(env, binds, args) | |||||
eval(body, new_env) | |||||
} as Mal::Func | |||||
end | |||||
def eval_ast(ast, env) | |||||
return ast.map{|n| eval(n, env) as Mal::Type} if ast.is_a? Mal::List | |||||
val = ast.unwrap | |||||
Mal::Type.new case val | |||||
when Mal::Symbol | |||||
if e = env.get(val.str) | |||||
e | |||||
else | |||||
eval_error "'#{val.str}' not found" | |||||
end | |||||
when Mal::List | |||||
val.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
val.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Array(Mal::Type) | |||||
val.map{|n| eval(n, env)} | |||||
when Mal::HashMap | |||||
val.each{|k, v| val[k] = eval(v, env)} | |||||
val | |||||
else | |||||
val | |||||
end | |||||
end | |||||
def eval_invocation(list, env) | |||||
f = eval(list.first, env).unwrap | |||||
case f | |||||
when Mal::Closure | |||||
f.fn.call eval_ast(list[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, env) | |||||
when Mal::Func | |||||
f.call eval_ast(list[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, env) | |||||
else | |||||
eval_error "expected function as the first argument" | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
macro invoke_list(l) | |||||
f = eval({{l}}.first, env).unwrap | |||||
args = eval_ast({{l}}[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, env) | |||||
case f | |||||
when Mal::Closure | |||||
ast = f.ast | |||||
env = Mal::Env.new(f.env, f.params, args) | |||||
next # TCO | |||||
when Mal::Func | |||||
return f.call args | |||||
else | |||||
eval_error "expected function as the first argument" | |||||
end | |||||
end | |||||
def eval(ast, env) | |||||
# 'next' in 'do...end' has a bug in crystal 0.7.1 | |||||
# https://github.com/manastech/crystal/issues/659 | |||||
while true | |||||
list = ast.unwrap | |||||
return eval_ast(ast, env) unless list.is_a? Mal::List | |||||
return gen_type Mal::List if list.empty? | |||||
head = list.first.unwrap | |||||
unless head.is_a? Mal::Symbol | |||||
invoke_list list | |||||
end | |||||
return Mal::Type.new case head.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env)) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
bindings = list[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a? Array | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
key, value = binding | |||||
name = key.unwrap | |||||
eval_error "name of binding must be specified as symbol" unless name.is_a? Mal::Symbol | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
ast, env = list[2], new_env | |||||
next # TCO | |||||
when "do" | |||||
if list.empty? | |||||
ast = Mal::Type.new nil | |||||
next | |||||
end | |||||
eval_ast(list[1..-2].each_with_object(Mal::List.new){|i,l| l << i}, env) | |||||
ast = list.last | |||||
next # TCO | |||||
when "if" | |||||
ast = unless eval(list[1], env).unwrap | |||||
list.size >= 4 ? list[3] : Mal::Type.new(nil) | |||||
else | |||||
list[2] | |||||
end | |||||
next # TCO | |||||
when "fn*" | |||||
# Note: | |||||
# If writing lambda expression here directly, compiler will fail to infer type of 'list'. (Error 'Nil for empty?') | |||||
params = list[1].unwrap | |||||
unless params.is_a?(Mal::List) || params.is_a?(Mal::Vector) | |||||
eval_error "'fn*' parameters must be list" | |||||
end | |||||
Mal::Closure.new(list[2], params, env, func_of(env, list[1].unwrap, list[2])) | |||||
else | |||||
invoke_list list | |||||
end | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
Mal::NS.each{|k,v| $repl_env.set(k, Mal::Type.new(v))} | |||||
rep "(def! not (fn* (a) (if a false true)))" | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,177 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
require "./core" | |||||
require "./error" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def func_of(env, binds, body) | |||||
-> (args : Array(Mal::Type)) { | |||||
new_env = Mal::Env.new(env, binds, args) | |||||
eval(body, new_env) | |||||
} as Mal::Func | |||||
end | |||||
def eval_ast(ast, env) | |||||
return ast.map{|n| eval(n, env) as Mal::Type} if ast.is_a? Mal::List | |||||
val = ast.unwrap | |||||
Mal::Type.new case val | |||||
when Mal::Symbol | |||||
if e = env.get(val.str) | |||||
e | |||||
else | |||||
eval_error "'#{val.str}' not found" | |||||
end | |||||
when Mal::List | |||||
val.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
val.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Array(Mal::Type) | |||||
val.map{|n| eval(n, env)} | |||||
when Mal::HashMap | |||||
val.each{|k, v| val[k] = eval(v, env)} | |||||
val | |||||
else | |||||
val | |||||
end | |||||
end | |||||
def eval_invocation(list, env) | |||||
f = eval(list.first, env).unwrap | |||||
case f | |||||
when Mal::Closure | |||||
f.fn.call eval_ast(list[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, env) | |||||
when Mal::Func | |||||
f.call eval_ast(list[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, env) | |||||
else | |||||
eval_error "expected function as the first argument" | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
macro invoke_list(l, env) | |||||
f = eval({{l}}.first, {{env}}).unwrap | |||||
args = eval_ast({{l}}[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, {{env}}) | |||||
case f | |||||
when Mal::Closure | |||||
ast = f.ast | |||||
{{env}} = Mal::Env.new(f.env, f.params, args) | |||||
next # TCO | |||||
when Mal::Func | |||||
return f.call args | |||||
else | |||||
eval_error "expected function as the first argument" | |||||
end | |||||
end | |||||
def eval(ast, env) | |||||
# 'next' in 'do...end' has a bug in crystal 0.7.1 | |||||
# https://github.com/manastech/crystal/issues/659 | |||||
while true | |||||
list = ast.unwrap | |||||
return eval_ast(ast, env) unless list.is_a? Mal::List | |||||
return gen_type Mal::List if list.empty? | |||||
head = list.first.unwrap | |||||
unless head.is_a? Mal::Symbol | |||||
invoke_list(list, env) | |||||
end | |||||
return Mal::Type.new case head.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env)) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
bindings = list[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a? Array | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
key, value = binding | |||||
name = key.unwrap | |||||
eval_error "name of binding must be specified as symbol" unless name.is_a? Mal::Symbol | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
ast, env = list[2], new_env | |||||
next # TCO | |||||
when "do" | |||||
if list.empty? | |||||
ast = Mal::Type.new nil | |||||
next | |||||
end | |||||
eval_ast(list[1..-2].each_with_object(Mal::List.new){|i,l| l << i}, env) | |||||
ast = list.last | |||||
next # TCO | |||||
when "if" | |||||
ast = unless eval(list[1], env).unwrap | |||||
list.size >= 4 ? list[3] : Mal::Type.new(nil) | |||||
else | |||||
list[2] | |||||
end | |||||
next # TCO | |||||
when "fn*" | |||||
params = list[1].unwrap | |||||
unless params.is_a?(Mal::List) || params.is_a?(Mal::Vector) | |||||
eval_error "'fn*' parameters must be list" | |||||
end | |||||
Mal::Closure.new(list[2], params, env, func_of(env, params, list[2])) | |||||
else | |||||
invoke_list(list, env) | |||||
end | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
Mal::NS.each{|k,v| $repl_env.set(k, Mal::Type.new(v))} | |||||
$repl_env.set("eval", Mal::Type.new -> (args: Array(Mal::Type)){ eval(args[0], $repl_env) }) | |||||
rep "(def! not (fn* (a) (if a false true)))" | |||||
rep "(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))" | |||||
$argv = Mal::List.new | |||||
$repl_env.set("*ARGV*", Mal::Type.new $argv) | |||||
unless ARGV.empty? | |||||
if ARGV.size > 1 | |||||
ARGV[1..-1].each do |a| | |||||
$argv << Mal::Type.new(a) | |||||
end | |||||
end | |||||
rep "(load-file \"#{ARGV[0]}\")" | |||||
exit | |||||
end | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,207 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
require "./core" | |||||
require "./error" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def func_of(env, binds, body) | |||||
-> (args : Array(Mal::Type)) { | |||||
new_env = Mal::Env.new(env, binds, args) | |||||
eval(body, new_env) | |||||
} as Mal::Func | |||||
end | |||||
def eval_ast(ast, env) | |||||
return ast.map{|n| eval(n, env) as Mal::Type} if ast.is_a? Mal::List | |||||
val = ast.unwrap | |||||
Mal::Type.new case val | |||||
when Mal::Symbol | |||||
if e = env.get(val.str) | |||||
e | |||||
else | |||||
eval_error "'#{val.str}' not found" | |||||
end | |||||
when Mal::List | |||||
val.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
val.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Array(Mal::Type) | |||||
val.map{|n| eval(n, env)} | |||||
when Mal::HashMap | |||||
val.each{|k, v| val[k] = eval(v, env)} | |||||
val | |||||
else | |||||
val | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
macro is_pair(list) | |||||
{{list}}.is_a?(Array) && !{{list}}.empty? | |||||
end | |||||
def quasiquote(ast) | |||||
list = ast.unwrap | |||||
unless is_pair(list) | |||||
return Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "quote") << ast | |||||
) | |||||
end | |||||
head = list.first.unwrap | |||||
case | |||||
# ("unquote" ...) | |||||
when head.is_a?(Mal::Symbol) && head.str == "unquote" | |||||
list[1] | |||||
# (("splice-unquote" ...) ...) | |||||
when is_pair(head) && (arg0 = head.first.unwrap).is_a?(Mal::Symbol) && arg0.str == "splice-unquote" | |||||
tail = Mal::Type.new list[1..-1].each_with_object(Mal::List.new){|e,l| l << e} | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "concat") << head[1] << quasiquote(tail) | |||||
) | |||||
else | |||||
tail = Mal::Type.new list[1..-1].each_with_object(Mal::List.new){|e,l| l << e} | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "cons") << quasiquote(list.first) << quasiquote(tail) | |||||
) | |||||
end | |||||
end | |||||
macro invoke_list(l, env) | |||||
f = eval({{l}}.first, {{env}}).unwrap | |||||
args = eval_ast({{l}}[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, {{env}}) | |||||
case f | |||||
when Mal::Closure | |||||
ast = f.ast | |||||
{{env}} = Mal::Env.new(f.env, f.params, args) | |||||
next # TCO | |||||
when Mal::Func | |||||
return f.call args | |||||
else | |||||
eval_error "expected function as the first argument" | |||||
end | |||||
end | |||||
def eval(ast, env) | |||||
# 'next' in 'do...end' has a bug in crystal 0.7.1 | |||||
# https://github.com/manastech/crystal/issues/659 | |||||
while true | |||||
list = ast.unwrap | |||||
return eval_ast(ast, env) unless list.is_a? Mal::List | |||||
return gen_type Mal::List if list.empty? | |||||
head = list.first.unwrap | |||||
unless head.is_a? Mal::Symbol | |||||
return invoke_list(list, env) | |||||
end | |||||
return Mal::Type.new case head.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env)) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
bindings = list[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a? Array | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
key, value = binding | |||||
name = key.unwrap | |||||
eval_error "name of binding must be specified as symbol" unless name.is_a? Mal::Symbol | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
ast, env = list[2], new_env | |||||
next # TCO | |||||
when "do" | |||||
if list.empty? | |||||
ast = Mal::Type.new nil | |||||
next | |||||
end | |||||
eval_ast(list[1..-2].each_with_object(Mal::List.new){|i,l| l << i}, env) | |||||
ast = list.last | |||||
next # TCO | |||||
when "if" | |||||
ast = unless eval(list[1], env).unwrap | |||||
list.size >= 4 ? list[3] : Mal::Type.new(nil) | |||||
else | |||||
list[2] | |||||
end | |||||
next # TCO | |||||
when "fn*" | |||||
params = list[1].unwrap | |||||
unless params.is_a?(Mal::List) || params.is_a?(Mal::Vector) | |||||
eval_error "'fn*' parameters must be list" | |||||
end | |||||
Mal::Closure.new(list[2], params, env, func_of(env, params, list[2])) | |||||
when "quote" | |||||
list[1] | |||||
when "quasiquote" | |||||
ast = quasiquote list[1] | |||||
next # TCO | |||||
else | |||||
invoke_list(list, env) | |||||
end | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
Mal::NS.each{|k,v| $repl_env.set(k, Mal::Type.new(v))} | |||||
$repl_env.set("eval", Mal::Type.new -> (args: Array(Mal::Type)){ eval(args[0], $repl_env) }) | |||||
rep "(def! not (fn* (a) (if a false true)))" | |||||
rep "(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))" | |||||
$argv = Mal::List.new | |||||
$repl_env.set("*ARGV*", Mal::Type.new $argv) | |||||
unless ARGV.empty? | |||||
if ARGV.size > 1 | |||||
ARGV[1..-1].each do |a| | |||||
$argv << Mal::Type.new(a) | |||||
end | |||||
end | |||||
begin | |||||
rep "(load-file \"#{ARGV[0]}\")" | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
exit | |||||
end | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,253 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
require "./core" | |||||
require "./error" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def func_of(env, binds, body) | |||||
-> (args : Array(Mal::Type)) { | |||||
new_env = Mal::Env.new(env, binds, args) | |||||
eval(body, new_env) | |||||
} as Mal::Func | |||||
end | |||||
def eval_ast(ast, env) | |||||
return ast.map{|n| eval(n, env) as Mal::Type} if ast.is_a? Mal::List | |||||
val = ast.unwrap | |||||
Mal::Type.new case val | |||||
when Mal::Symbol | |||||
if e = env.get(val.str) | |||||
e | |||||
else | |||||
eval_error "'#{val.str}' not found" | |||||
end | |||||
when Mal::List | |||||
val.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
val.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Array(Mal::Type) | |||||
val.map{|n| eval(n, env)} | |||||
when Mal::HashMap | |||||
val.each{|k, v| val[k] = eval(v, env)} | |||||
val | |||||
else | |||||
val | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
macro pair?(list) | |||||
{{list}}.is_a?(Array) && !{{list}}.empty? | |||||
end | |||||
def quasiquote(ast) | |||||
list = ast.unwrap | |||||
unless pair?(list) | |||||
return Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "quote") << ast | |||||
) | |||||
end | |||||
head = list.first.unwrap | |||||
case | |||||
# ("unquote" ...) | |||||
when head.is_a?(Mal::Symbol) && head.str == "unquote" | |||||
list[1] | |||||
# (("splice-unquote" ...) ...) | |||||
when pair?(head) && (arg0 = head.first.unwrap).is_a?(Mal::Symbol) && arg0.str == "splice-unquote" | |||||
tail = Mal::Type.new list[1..-1].each_with_object(Mal::List.new){|e,l| l << e} | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "concat") << head[1] << quasiquote(tail) | |||||
) | |||||
else | |||||
tail = Mal::Type.new list[1..-1].each_with_object(Mal::List.new){|e,l| l << e} | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "cons") << quasiquote(list.first) << quasiquote(tail) | |||||
) | |||||
end | |||||
end | |||||
def macro_call?(ast, env) | |||||
list = ast.unwrap | |||||
return false unless list.is_a? Mal::List | |||||
sym = list.first.unwrap | |||||
return false unless sym.is_a? Mal::Symbol | |||||
func = env.find(sym.str).try(&.data[sym.str]) | |||||
return false unless func && func.macro? | |||||
true | |||||
end | |||||
def macroexpand(ast, env) | |||||
while macro_call?(ast, env) | |||||
# Already checked in macro_call? | |||||
list = ast.unwrap as Mal::List | |||||
func_sym = list[0].unwrap as Mal::Symbol | |||||
func = env.get(func_sym.str).unwrap | |||||
case func | |||||
when Mal::Func | |||||
ast = func.call(list[1..-1]) | |||||
when Mal::Closure | |||||
ast = func.fn.call(list[1..-1]) | |||||
else | |||||
eval_error "macro '#{func_sym.str}' must be function: #{ast}" | |||||
end | |||||
end | |||||
ast | |||||
end | |||||
macro invoke_list(l, env) | |||||
f = eval({{l}}.first, {{env}}).unwrap | |||||
args = eval_ast({{l}}[1..-1].each_with_object(Mal::List.new){|i, l| l << i}, {{env}}) | |||||
case f | |||||
when Mal::Closure | |||||
ast = f.ast | |||||
{{env}} = Mal::Env.new(f.env, f.params, args) | |||||
next # TCO | |||||
when Mal::Func | |||||
return f.call args | |||||
else | |||||
eval_error "expected function as the first argument: #{f}" | |||||
end | |||||
end | |||||
def eval(ast, env) | |||||
# 'next' in 'do...end' has a bug in crystal 0.7.1 | |||||
# https://github.com/manastech/crystal/issues/659 | |||||
while true | |||||
return eval_ast(ast, env) unless ast.unwrap.is_a? Mal::List | |||||
ast = macroexpand(ast, env) | |||||
list = ast.unwrap | |||||
return ast unless list.is_a? Mal::List | |||||
return ast if list.empty? | |||||
head = list.first.unwrap | |||||
return invoke_list(list, env) unless head.is_a? Mal::Symbol | |||||
return Mal::Type.new case head.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol: #{a1}" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env)) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
bindings = list[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a? Array | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
key, value = binding | |||||
name = key.unwrap | |||||
eval_error "name of binding must be specified as symbol #{name}" unless name.is_a? Mal::Symbol | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
ast, env = list[2], new_env | |||||
next # TCO | |||||
when "do" | |||||
if list.empty? | |||||
ast = Mal::Type.new nil | |||||
next | |||||
end | |||||
eval_ast(list[1..-2].each_with_object(Mal::List.new){|i,l| l << i}, env) | |||||
ast = list.last | |||||
next # TCO | |||||
when "if" | |||||
ast = unless eval(list[1], env).unwrap | |||||
list.size >= 4 ? list[3] : Mal::Type.new(nil) | |||||
else | |||||
list[2] | |||||
end | |||||
next # TCO | |||||
when "fn*" | |||||
params = list[1].unwrap | |||||
unless params.is_a? Array | |||||
eval_error "'fn*' parameters must be list or vector: #{params}" | |||||
end | |||||
Mal::Closure.new(list[2], params, env, func_of(env, params, list[2])) | |||||
when "quote" | |||||
list[1] | |||||
when "quasiquote" | |||||
ast = quasiquote list[1] | |||||
next # TCO | |||||
when "defmacro!" | |||||
eval_error "wrong number of argument for 'defmacro!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'defmacro!' must be symbol: #{a1}" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env).tap{|n| n.is_macro = true}) | |||||
when "macroexpand" | |||||
macroexpand(list[1], env) | |||||
else | |||||
invoke_list(list, env) | |||||
end | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
Mal::NS.each{|k,v| $repl_env.set(k, Mal::Type.new(v))} | |||||
$repl_env.set("eval", Mal::Type.new -> (args: Array(Mal::Type)){ eval(args[0], $repl_env) }) | |||||
rep "(def! not (fn* (a) (if a false true)))" | |||||
rep "(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))" | |||||
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)))))))" | |||||
rep "(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))" | |||||
$argv = Mal::List.new | |||||
$repl_env.set("*ARGV*", Mal::Type.new $argv) | |||||
unless ARGV.empty? | |||||
if ARGV.size > 1 | |||||
ARGV[1..-1].each do |a| | |||||
$argv << Mal::Type.new(a) | |||||
end | |||||
end | |||||
begin | |||||
rep "(load-file \"#{ARGV[0]}\")" | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
exit | |||||
end | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,270 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
require "./core" | |||||
require "./error" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def func_of(env, binds, body) | |||||
-> (args : Array(Mal::Type)) { | |||||
new_env = Mal::Env.new(env, binds, args) | |||||
eval(body, new_env) | |||||
} as Mal::Func | |||||
end | |||||
def eval_ast(ast, env) | |||||
return ast.map{|n| eval(n, env) as Mal::Type} if ast.is_a? Mal::List | |||||
val = ast.unwrap | |||||
Mal::Type.new case val | |||||
when Mal::Symbol | |||||
if e = env.get(val.str) | |||||
e | |||||
else | |||||
eval_error "'#{val.str}' not found" | |||||
end | |||||
when Mal::List | |||||
val.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
val.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Array(Mal::Type) | |||||
val.map{|n| eval(n, env)} | |||||
when Mal::HashMap | |||||
val.each{|k, v| val[k] = eval(v, env)} | |||||
val | |||||
else | |||||
val | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
macro pair?(list) | |||||
{{list}}.is_a?(Array) && !{{list}}.empty? | |||||
end | |||||
def quasiquote(ast) | |||||
list = ast.unwrap | |||||
unless pair?(list) | |||||
return Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "quote") << ast | |||||
) | |||||
end | |||||
head = list.first.unwrap | |||||
case | |||||
# ("unquote" ...) | |||||
when head.is_a?(Mal::Symbol) && head.str == "unquote" | |||||
list[1] | |||||
# (("splice-unquote" ...) ...) | |||||
when pair?(head) && (arg0 = head.first.unwrap).is_a?(Mal::Symbol) && arg0.str == "splice-unquote" | |||||
tail = Mal::Type.new list[1..-1].to_mal | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "concat") << head[1] << quasiquote(tail) | |||||
) | |||||
else | |||||
tail = Mal::Type.new list[1..-1].to_mal | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "cons") << quasiquote(list.first) << quasiquote(tail) | |||||
) | |||||
end | |||||
end | |||||
def macro_call?(ast, env) | |||||
list = ast.unwrap | |||||
return false unless list.is_a? Mal::List | |||||
sym = list.first.unwrap | |||||
return false unless sym.is_a? Mal::Symbol | |||||
func = env.find(sym.str).try(&.data[sym.str]) | |||||
return false unless func && func.macro? | |||||
true | |||||
end | |||||
def macroexpand(ast, env) | |||||
while macro_call?(ast, env) | |||||
# Already checked in macro_call? | |||||
list = ast.unwrap as Mal::List | |||||
func_sym = list[0].unwrap as Mal::Symbol | |||||
func = env.get(func_sym.str).unwrap | |||||
case func | |||||
when Mal::Func | |||||
ast = func.call(list[1..-1]) | |||||
when Mal::Closure | |||||
ast = func.fn.call(list[1..-1]) | |||||
else | |||||
eval_error "macro '#{func_sym.str}' must be function: #{ast}" | |||||
end | |||||
end | |||||
ast | |||||
end | |||||
macro invoke_list(l, env) | |||||
f = eval({{l}}.first, {{env}}).unwrap | |||||
args = eval_ast({{l}}[1..-1].to_mal, {{env}}) | |||||
case f | |||||
when Mal::Closure | |||||
ast = f.ast | |||||
{{env}} = Mal::Env.new(f.env, f.params, args) | |||||
next # TCO | |||||
when Mal::Func | |||||
return f.call args | |||||
else | |||||
eval_error "expected function as the first argument: #{f}" | |||||
end | |||||
end | |||||
def eval(ast, env) | |||||
# 'next' in 'do...end' has a bug in crystal 0.7.1 | |||||
# https://github.com/manastech/crystal/issues/659 | |||||
while true | |||||
return eval_ast(ast, env) unless ast.unwrap.is_a? Mal::List | |||||
ast = macroexpand(ast, env) | |||||
list = ast.unwrap | |||||
return ast unless list.is_a? Mal::List | |||||
return ast if list.empty? | |||||
head = list.first.unwrap | |||||
return invoke_list(list, env) unless head.is_a? Mal::Symbol | |||||
return Mal::Type.new case head.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol: #{a1}" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env)) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
bindings = list[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a? Array | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
key, value = binding | |||||
name = key.unwrap | |||||
eval_error "name of binding must be specified as symbol #{name}" unless name.is_a? Mal::Symbol | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
ast, env = list[2], new_env | |||||
next # TCO | |||||
when "do" | |||||
if list.empty? | |||||
ast = Mal::Type.new nil | |||||
next | |||||
end | |||||
eval_ast(list[1..-2].to_mal, env) | |||||
ast = list.last | |||||
next # TCO | |||||
when "if" | |||||
ast = unless eval(list[1], env).unwrap | |||||
list.size >= 4 ? list[3] : Mal::Type.new(nil) | |||||
else | |||||
list[2] | |||||
end | |||||
next # TCO | |||||
when "fn*" | |||||
params = list[1].unwrap | |||||
unless params.is_a? Array | |||||
eval_error "'fn*' parameters must be list or vector: #{params}" | |||||
end | |||||
Mal::Closure.new(list[2], params, env, func_of(env, params, list[2])) | |||||
when "quote" | |||||
list[1] | |||||
when "quasiquote" | |||||
ast = quasiquote list[1] | |||||
next # TCO | |||||
when "defmacro!" | |||||
eval_error "wrong number of argument for 'defmacro!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'defmacro!' must be symbol: #{a1}" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env).tap{|n| n.is_macro = true}) | |||||
when "macroexpand" | |||||
macroexpand(list[1], env) | |||||
when "try*" | |||||
catch_list = list[2].unwrap | |||||
return eval(list[1], env) unless catch_list.is_a? Mal::List | |||||
catch_head = catch_list.first.unwrap | |||||
return eval(list[1], env) unless catch_head.is_a? Mal::Symbol | |||||
return eval(list[1], env) unless catch_head.str == "catch*" | |||||
begin | |||||
eval(list[1], env) | |||||
rescue e : Mal::RuntimeException | |||||
new_env = Mal::Env.new(env, [catch_list[1]], [e.thrown]) | |||||
eval(catch_list[2], new_env) | |||||
rescue e | |||||
new_env = Mal::Env.new(env, [catch_list[1]], [Mal::Type.new e.message]) | |||||
eval(catch_list[2], new_env) | |||||
end | |||||
else | |||||
invoke_list(list, env) | |||||
end | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
Mal::NS.each{|k,v| $repl_env.set(k, Mal::Type.new(v))} | |||||
$repl_env.set("eval", Mal::Type.new -> (args: Array(Mal::Type)){ eval(args[0], $repl_env) }) | |||||
rep "(def! not (fn* (a) (if a false true)))" | |||||
rep "(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))" | |||||
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)))))))" | |||||
rep "(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))" | |||||
$argv = Mal::List.new | |||||
$repl_env.set("*ARGV*", Mal::Type.new $argv) | |||||
unless ARGV.empty? | |||||
if ARGV.size > 1 | |||||
ARGV[1..-1].each do |a| | |||||
$argv << Mal::Type.new(a) | |||||
end | |||||
end | |||||
begin | |||||
rep "(load-file \"#{ARGV[0]}\")" | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
exit | |||||
end | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,280 @@ | |||||
#! /usr/bin/env crystal run | |||||
require "colorize" | |||||
require "./readline" | |||||
require "./reader" | |||||
require "./printer" | |||||
require "./types" | |||||
require "./env" | |||||
require "./core" | |||||
require "./error" | |||||
# Note: | |||||
# Employed downcase names because Crystal prohibits uppercase names for methods | |||||
def func_of(env, binds, body) | |||||
-> (args : Array(Mal::Type)) { | |||||
new_env = Mal::Env.new(env, binds, args) | |||||
eval(body, new_env) | |||||
} as Mal::Func | |||||
end | |||||
def eval_ast(ast, env) | |||||
return ast.map{|n| eval(n, env) as Mal::Type} if ast.is_a? Array | |||||
val = ast.unwrap | |||||
Mal::Type.new case val | |||||
when Mal::Symbol | |||||
if e = env.get(val.str) | |||||
e | |||||
else | |||||
eval_error "'#{val.str}' not found" | |||||
end | |||||
when Mal::List | |||||
val.each_with_object(Mal::List.new){|n, l| l << eval(n, env)} | |||||
when Mal::Vector | |||||
val.each_with_object(Mal::Vector.new){|n, l| l << eval(n, env)} | |||||
when Mal::HashMap | |||||
new_map = Mal::HashMap.new | |||||
val.each{|k, v| new_map[k] = eval(v, env)} | |||||
new_map | |||||
else | |||||
val | |||||
end | |||||
end | |||||
def read(str) | |||||
read_str str | |||||
end | |||||
macro pair?(list) | |||||
{{list}}.is_a?(Array) && !{{list}}.empty? | |||||
end | |||||
def quasiquote(ast) | |||||
list = ast.unwrap | |||||
unless pair?(list) | |||||
return Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "quote") << ast | |||||
) | |||||
end | |||||
head = list.first.unwrap | |||||
case | |||||
# ("unquote" ...) | |||||
when head.is_a?(Mal::Symbol) && head.str == "unquote" | |||||
list[1] | |||||
# (("splice-unquote" ...) ...) | |||||
when pair?(head) && (arg0 = head.first.unwrap).is_a?(Mal::Symbol) && arg0.str == "splice-unquote" | |||||
tail = Mal::Type.new list[1..-1].to_mal | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "concat") << head[1] << quasiquote(tail) | |||||
) | |||||
else | |||||
tail = Mal::Type.new list[1..-1].to_mal | |||||
Mal::Type.new( | |||||
Mal::List.new << gen_type(Mal::Symbol, "cons") << quasiquote(list.first) << quasiquote(tail) | |||||
) | |||||
end | |||||
end | |||||
def macro_call?(ast, env) | |||||
list = ast.unwrap | |||||
return false unless list.is_a? Mal::List | |||||
return false if list.empty? | |||||
sym = list.first.unwrap | |||||
return false unless sym.is_a? Mal::Symbol | |||||
func = env.find(sym.str).try(&.data[sym.str]) | |||||
return false unless func && func.macro? | |||||
true | |||||
end | |||||
def macroexpand(ast, env) | |||||
while macro_call?(ast, env) | |||||
# Already checked in macro_call? | |||||
list = ast.unwrap as Mal::List | |||||
func_sym = list[0].unwrap as Mal::Symbol | |||||
func = env.get(func_sym.str).unwrap | |||||
case func | |||||
when Mal::Func | |||||
ast = func.call(list[1..-1]) | |||||
when Mal::Closure | |||||
ast = func.fn.call(list[1..-1]) | |||||
else | |||||
eval_error "macro '#{func_sym.str}' must be function: #{ast}" | |||||
end | |||||
end | |||||
ast | |||||
end | |||||
macro invoke_list(l, env) | |||||
f = eval({{l}}.first, {{env}}).unwrap | |||||
args = eval_ast({{l}}[1..-1], {{env}}) as Array | |||||
case f | |||||
when Mal::Closure | |||||
ast = f.ast | |||||
{{env}} = Mal::Env.new(f.env, f.params, args) | |||||
next # TCO | |||||
when Mal::Func | |||||
return f.call args | |||||
else | |||||
eval_error "expected function as the first argument: #{f}" | |||||
end | |||||
end | |||||
def debug(ast) | |||||
puts print(ast).colorize.red | |||||
end | |||||
def eval(ast, env) | |||||
# 'next' in 'do...end' has a bug in crystal 0.7.1 | |||||
# https://github.com/manastech/crystal/issues/659 | |||||
while true | |||||
return eval_ast(ast, env) unless ast.unwrap.is_a? Mal::List | |||||
ast = macroexpand(ast, env) | |||||
list = ast.unwrap | |||||
return ast unless list.is_a? Mal::List | |||||
return ast if list.empty? | |||||
head = list.first.unwrap | |||||
return invoke_list(list, env) unless head.is_a? Mal::Symbol | |||||
return Mal::Type.new case head.str | |||||
when "def!" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'def!' must be symbol: #{a1}" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env)) | |||||
when "let*" | |||||
eval_error "wrong number of argument for 'def!'" unless list.size == 3 | |||||
bindings = list[1].unwrap | |||||
eval_error "1st argument of 'let*' must be list or vector" unless bindings.is_a? Array | |||||
eval_error "size of binding list must be even" unless bindings.size.even? | |||||
new_env = Mal::Env.new env | |||||
bindings.each_slice(2) do |binding| | |||||
key, value = binding | |||||
name = key.unwrap | |||||
eval_error "name of binding must be specified as symbol #{name}" unless name.is_a? Mal::Symbol | |||||
new_env.set(name.str, eval(value, new_env)) | |||||
end | |||||
ast, env = list[2], new_env | |||||
next # TCO | |||||
when "do" | |||||
if list.empty? | |||||
ast = Mal::Type.new nil | |||||
next | |||||
end | |||||
eval_ast(list[1..-2].to_mal, env) | |||||
ast = list.last | |||||
next # TCO | |||||
when "if" | |||||
ast = unless eval(list[1], env).unwrap | |||||
list.size >= 4 ? list[3] : Mal::Type.new(nil) | |||||
else | |||||
list[2] | |||||
end | |||||
next # TCO | |||||
when "fn*" | |||||
params = list[1].unwrap | |||||
unless params.is_a? Array | |||||
eval_error "'fn*' parameters must be list or vector: #{params}" | |||||
end | |||||
Mal::Closure.new(list[2], params, env, func_of(env, params, list[2])) | |||||
when "quote" | |||||
list[1] | |||||
when "quasiquote" | |||||
ast = quasiquote list[1] | |||||
next # TCO | |||||
when "defmacro!" | |||||
eval_error "wrong number of argument for 'defmacro!'" unless list.size == 3 | |||||
a1 = list[1].unwrap | |||||
eval_error "1st argument of 'defmacro!' must be symbol: #{a1}" unless a1.is_a? Mal::Symbol | |||||
env.set(a1.str, eval(list[2], env).tap{|n| n.is_macro = true}) | |||||
when "macroexpand" | |||||
macroexpand(list[1], env) | |||||
when "try*" | |||||
catch_list = list[2].unwrap | |||||
return eval(list[1], env) unless catch_list.is_a? Mal::List | |||||
catch_head = catch_list.first.unwrap | |||||
return eval(list[1], env) unless catch_head.is_a? Mal::Symbol | |||||
return eval(list[1], env) unless catch_head.str == "catch*" | |||||
begin | |||||
eval(list[1], env) | |||||
rescue e : Mal::RuntimeException | |||||
new_env = Mal::Env.new(env, [catch_list[1]], [e.thrown]) | |||||
eval(catch_list[2], new_env) | |||||
rescue e | |||||
new_env = Mal::Env.new(env, [catch_list[1]], [Mal::Type.new e.message]) | |||||
eval(catch_list[2], new_env) | |||||
end | |||||
else | |||||
invoke_list(list, env) | |||||
end | |||||
end | |||||
end | |||||
def print(result) | |||||
pr_str(result, true) | |||||
end | |||||
def rep(str) | |||||
print(eval(read(str), $repl_env)) | |||||
end | |||||
$repl_env = Mal::Env.new nil | |||||
Mal::NS.each{|k,v| $repl_env.set(k, Mal::Type.new(v))} | |||||
$repl_env.set("eval", Mal::Type.new -> (args: Array(Mal::Type)){ eval(args[0], $repl_env) }) | |||||
rep "(def! not (fn* (a) (if a false true)))" | |||||
rep "(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))" | |||||
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)))))))" | |||||
rep "(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))" | |||||
rep("(def! *host-language* \"crystal\")") | |||||
$argv = Mal::List.new | |||||
$repl_env.set("*ARGV*", Mal::Type.new $argv) | |||||
unless ARGV.empty? | |||||
if ARGV.size > 1 | |||||
ARGV[1..-1].each do |a| | |||||
$argv << Mal::Type.new(a) | |||||
end | |||||
end | |||||
begin | |||||
rep "(load-file \"#{ARGV[0]}\")" | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
exit | |||||
end | |||||
rep("(println (str \"Mal [\" *host-language* \"]\"))") | |||||
while line = my_readline("user> ") | |||||
begin | |||||
puts rep(line) | |||||
rescue e | |||||
STDERR.puts e | |||||
end | |||||
end |
@ -0,0 +1,114 @@ | |||||
require "./printer" | |||||
module Mal | |||||
class Symbol | |||||
property :str | |||||
def initialize(@str) | |||||
end | |||||
def ==(other : Symbol) | |||||
@str == other.str | |||||
end | |||||
end | |||||
class List < Array(Type) | |||||
end | |||||
class Vector < Array(Type) | |||||
end | |||||
class HashMap < Hash(String, Type) | |||||
end | |||||
class Atom | |||||
property :val | |||||
def initialize(@val) | |||||
end | |||||
def ==(rhs : Atom) | |||||
@val == rhs.val | |||||
end | |||||
end | |||||
class Closure | |||||
property :ast, :params, :env, :fn | |||||
def initialize(@ast, @params, @env, @fn) | |||||
end | |||||
end | |||||
class Type | |||||
alias Func = (Array(Type) -> Type) | |||||
alias ValueType = Nil | Bool | Int32 | String | Symbol | List | Vector | HashMap | Func | Closure | Atom | |||||
is_macro :: Bool | |||||
meta :: Type | |||||
property :is_macro, :meta | |||||
def initialize(@val : ValueType) | |||||
@is_macro = false | |||||
@meta = nil | |||||
end | |||||
def initialize(other : Type) | |||||
@val = other.unwrap | |||||
@is_macro = other.is_macro | |||||
@meta = other.meta | |||||
end | |||||
def unwrap | |||||
@val | |||||
end | |||||
def macro? | |||||
@is_macro | |||||
end | |||||
def to_s | |||||
pr_str(self) | |||||
end | |||||
def dup | |||||
Type.new(@val).tap do |t| | |||||
t.is_macro = @is_macro | |||||
t.meta = @meta | |||||
end | |||||
end | |||||
def ==(other : Type) | |||||
@val == other.unwrap | |||||
end | |||||
macro rel_op(*ops) | |||||
{% for op in ops %} | |||||
def {{op.id}}(other : Mal::Type) | |||||
l, r = @val, other.unwrap | |||||
{% for t in [Int32, String] %} | |||||
if l.is_a?({{t}}) && r.is_a?({{t}}) | |||||
return (l) {{op.id}} (r) | |||||
end | |||||
{% end %} | |||||
if l.is_a?(Symbol) && r.is_a?(Symbol) | |||||
return l.str {{op.id}} r.str | |||||
end | |||||
false | |||||
end | |||||
{% end %} | |||||
end | |||||
rel_op :<, :>, :<=, :>= | |||||
end | |||||
alias Func = Type::Func | |||||
end | |||||
macro gen_type(t, *args) | |||||
Mal::Type.new {{t.id}}.new({{*args}}) | |||||
end | |||||
class Array | |||||
def to_mal(t = Mal::List) | |||||
each_with_object(t.new){|e, l| l << e} | |||||
end | |||||
end | |||||