module Gummi
  module Document
    module Search
      class Result

        attr_reader :took, :total, :hits, :facets

        def initialize(result)
          @took = result["took"]
          @total = result["hits"]["total"]
          @hits = result["hits"]["hits"]
          @facets = result["facets"]
        end

        def records
          hits.map do |hit|
            model = "DB::#{hit["_type"].humanize}".constantize
            doc_hash = {id: hit["_id"]}.merge(hit["_source"])
            model.new(doc_hash)
          end
        end
      end
    end
  end
end