machinio/solrb

require 'solr/update/commands/add'
require 'solr/update/commands/delete'
require 'solr/update/commands/commit'
require 'solr/update/commands/optimize'

module Solr
  module Update
    class Request
      PATH = '/update/json'.freeze

      attr_reader :commands

      def initialize(commands = [])
        @commands = commands
      end

      def run(runner_options: nil)
        http_request = build_http_request
        options = default_runner_options.merge(runner_options || {})
        Solr::Request::Runner.call(request: http_request, **options)
      end

      private

      def default_runner_options
        if Solr.cloud_enabled?
          { node_selection_strategy: Solr::Request::Cloud::LeaderNodeSelectionStrategy }
        elsif Solr.master_slave_enabled?
          { node_selection_strategy: Solr::Request::MasterSlave::MasterNodeSelectionStrategy }
        else
          {}
        end
      end

      def build_http_request
        Solr::Request::HttpRequest.new(path: PATH, body: body, url_params: {}, method: :post)
      end

      # Solr supports two body formats:
      # 1) Group "add" operations into an array (currently we use this way):
      # {
      #   "add": [
      #     { "id": "1", "name": "sun" },
      #     { "id": "2", "name": "moon" }
      #   ],
      #   "commit": {}
      # }
      #
      # 2) Use duplicate "add" keys, but ruby doesn't allow to do it
      #    (we can build this string manually, but that may create issues)
      # {
      #   "add": { "doc": { "id": "1", "name": "sun"  } },
      #   "add": { "doc": { "id": "2", "name": "moon" } },
      #   "commit": {}
      # }
      def body
        commands.group_by(&:class).reduce({}) do |acc, (command_class, commands_group)|
          acc[command_class::COMMAND_KEY] = command_class.unnest(commands_group)
          acc
        end
      end
    end
  end
end