Merge branch 'Haxe'

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

diff --git a/crystal/env.cr b/crystal/env.cr
index 8ad0039..9f38d68 100644 (file)
--- a/crystal/env.cr
+++ b/crystal/env.cr
@@ -18,10 +18,29 @@ module Mal
       # Note:
       # Array#zip() can't be used because overload resolution failed
       (0...binds.size).each do |idx|
-        sym, expr = binds[idx].unwrap, exprs[idx]
-        eval_error "bind list must be symbol" unless sym.is_a? Mal::Symbol
-        @data[sym.str] = 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)
@@ -40,8 +59,8 @@ module Mal
     end
 
     def get(key)
-      e = find(key)
-      eval_error "#{key} not found" unless e
+      e = find key
+      eval_error "'#{key}' not found" unless e
       e.data[key]
     end
   end