module Waistband
  class Result

    def initialize(result_hash)
      @result_hash = result_hash
    end

    def to_hash
      @result_hash['_source']
    end

    def [](key)
      @result_hash['_source'][key.to_s]
    end

    def method_missing(method_name, *args, &block)
      return @result_hash[method_name.to_s] if @result_hash.has_key?(method_name.to_s)
      return @result_hash['_source'][method_name.to_s] if @result_hash['_source'] && @result_hash['_source'].has_key?(method_name.to_s)
      nil
    end

    def respond_to_missing?(method_name, include_private = false)
      return true if @result_hash.has_key?(method_name.to_s)
      return true if @result_hash['_source'] && @result_hash['_source'].has_key?(method_name.to_s)
      super
    end

  end
end