Merge branch 'Haxe'

[jackhill/mal.git] / crystal / env.cr

diff --git a/crystal/env.cr b/crystal/env.cr
index 59666df..9f38d68 100644 (file)
--- a/crystal/env.cr
+++ b/crystal/env.cr
@@ -1,4 +1,5 @@
 require "./types"
+require "./error"
 
 module Mal
 
@@ -12,15 +13,34 @@ module Mal
     def initialize(@outer, binds, exprs : Array(Mal::Type))
       @data = {} of String => Mal::Type
 
-      raise EvalException.new "binds must be list or vector" unless binds.is_a?(Array)
+      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, expr = binds[idx].val, exprs[idx]
-        raise EvalException.new "bind list must be symbol" unless sym.is_a?(Mal::Symbol)
-        @data[sym.val] = expr
+        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)
@@ -39,8 +59,8 @@ module Mal
     end
 
     def get(key)
-      e = find(key)
-      raise EvalException.new "#{key} not found" unless e
+      e = find key
+      eval_error "'#{key}' not found" unless e
       e.data[key]
     end
   end