toptal/chewy

module Chewy
  class Index
    module Wrapper
      extend ActiveSupport::Concern

      included do
        attr_accessor :_data, :_object
        attr_reader :attributes
      end

      module ClassMethods
        def build(hit)
          attributes = (hit['_source'] || {})
            .reverse_merge(id: hit['_id'])
            .merge!(_score: hit['_score'])
            .merge!(_explanation: hit['_explanation'])

          wrapper = new(attributes)
          wrapper._data = hit
          wrapper
        end
      end

      def initialize(attributes = {})
        @attributes = attributes.stringify_keys
      end

      def ==(other)
        return true if super

        if other.is_a?(Chewy::Index)
          self.class == other.class && (respond_to?(:id) ? id == other.id : attributes == other.attributes)
        elsif other.respond_to?(:id)
          self.class.adapter.target.is_a?(Class) &&
            other.is_a?(self.class.adapter.target) &&
            id.to_s == other.id.to_s
        else
          false
        end
      end

      %w[_id _type _index].each do |name|
        define_method name do
          _data[name]
        end
      end

      def method_missing(method, *args, &block)
        m = method.to_s
        if (name = highlight_name(m))
          highlight(name)
        elsif (name = highlight_names(m))
          highlights(name)
        elsif @attributes.key?(m)
          @attributes[m]
        elsif attribute_defined?(m)
          nil
        else
          super
        end
      end

      def respond_to_missing?(method, include_private = false)
        m = method.to_s
        highlight_name(m) || highlight_names(m) || @attributes.key?(m) || attribute_defined?(m) || super
      end

    private

      def highlight_name(method)
        method.sub(/_highlight\z/, '') if method.end_with?('_highlight')
      end

      def highlight_names(method)
        method.sub(/_highlights\z/, '') if method.end_with?('_highlights')
      end

      def attribute_defined?(attribute)
        self.class.root && self.class.root.children.find { |a| a.name.to_s == attribute }.present?
      end

      def highlight(attribute)
        _data['highlight'][attribute].first if highlight?(attribute)
      end

      def highlights(attribute)
        _data['highlight'][attribute] if highlight?(attribute)
      end

      def highlight?(attribute)
        _data.key?('highlight') && _data['highlight'].key?(attribute)
      end
    end
  end
end