# frozen_string_literal: true

module Esse
  class Document
    attr_reader :object, :options

    def initialize(object, **options)
      @object = object
      @options = options
    end

    # @return [String, Number] the document ID
    # @abstract Override this method to return the document ID
    def id
      raise NotImplementedError, 'Override this method to return the document ID'
    end

    # @return [String, nil] the document type
    # @abstract Override this method to return the document type
    def type
      nil
    end

    # @return [Boolean] whether the document has type
    def type?
      !type.nil?
    end

    # @return [String, nil] the document routing
    # @abstract Override this method to return the document routing
    def routing
      nil
    end

    # @return [Boolean] whether the document has routing
    def routing?
      !routing.nil?
    end

    # @TODO allow import, index, bulk to accept a suffix to tell which index to use
    # def index_suffix
    #   nil
    # end

    # @return [Hash] the document meta
    # @abstract Override this method to return the document meta
    def meta
      {}
    end

    # @return [Hash] the document source
    # @abstract Override this method to return the document source
    def source
      {}
    end

    # @return [Hash] the document data
    def to_h
      source.merge(
        _id: id,
      ).tap do |hash|
        hash[:_type] = type if type
        hash[:_routing] = routing if routing
        hash.merge!(meta)
      end
    end

    def to_bulk(data: true, operation: nil)
      doc_header.tap do |h|
        if data && operation == :update
          h[:data] = { doc: source_with_lazy_attributes }
        elsif data
          h[:data] = source_with_lazy_attributes
        end
        h.merge!(meta)
      end
    end

    def ignore_on_index?
      id.nil?
    end

    def ignore_on_delete?
      id.nil?
    end

    def ==(other)
      other.is_a?(self.class) && (
        id == other.id && type == other.type && routing == other.routing && meta == other.meta && source == other.source
      )
    end

    def doc_header
      { _id: id }.tap do |h|
        h[:_type] = type if type
        h[:_routing] = routing if routing?
      end
    end

    def inspect
      attributes = %i[id routing source].map do |attr|
        value = send(attr)
        "#{attr}: #{value.inspect}" if value
      end.compact.join(', ')
      "#<#{self.class.name || 'Esse::Document'} #{attributes}>"
    end

    protected

    def source_with_lazy_attributes
      return source unless @__lazy_source_data__

      @__lazy_source_data__.merge(source)
    end

    # api private
    def __add_lazy_data_to_source__(hash)
      return hash unless hash.is_a?(Hash)

      @__lazy_source_data__ ||= {}
      @__lazy_source_data__.merge!(hash)
    end
  end
end