Merge pull request #337 from AndreaCrotti/upgrade-libraries

[jackhill/mal.git] / ruby / env.rb

class Env
    attr_accessor :data
    def initialize(outer=nil, binds=[], exprs=[])
        @data = {}
        @outer = outer
        binds.each_index do |i|
            if binds[i] == :"&"
                data[binds[i+1]] = exprs.drop(i)
                break
            else
                data[binds[i]] = exprs[i]
            end
        end
        return self
    end

    def find(key)
        if @data.key? key
            return self
        elsif @outer
            return @outer.find(key)
        else
            return nil
        end
    end

    def set(key, value)
        @data[key] = value
        return value
    end

    def get(key)
        env = find(key)
        raise "'" + key.to_s + "' not found" if not env
        env.data[key]
    end
end