require 'apress/yandex_market/readers/base'
require 'facets/string/squish'

module Apress
  module YandexMarket
    module Readers
      # Ридер категорий. Читает из API Яндекс.Маркета переданные категории и их подкатегории
      #
      # Examples
      #
      #   reader = Apress::YandexMarket::Readers::Category.new(region_id: 225,
      #                                                        token: 'qwerty',
      #                                                        categories: 'Авто; Дом и дача')
      #   # => #<Apress::YandexMarket::Readers::Category ...>
      #
      #   reader.each_row |category|
      #     puts category
      #   end
      #   # => {
      #          :id=>90402,
      #          :name=>"Авто",
      #          :fullName=>"Товары для авто- и мототехники",
      #          :link=>
      #            "https://market.yandex.ru/catalog/90402/list?hid=90402&onstock=1&pp=1001&clid=2326601&distr_type=7",
      #          :childCount=>12,
      #          :advertisingModel=>"HYBRID",
      #          :viewType=>"LIST"
      #        }
      #        {
      #          :id=>90403,
      #          ...
      #        }
      class Category < Base
        FIELDS = %w(PARENT).join(',').freeze

        class << self
          def allowed_options
            super + %i(categories)
          end
        end

        # Public: инициализация ридера
        #
        # options - Hash, параметры ридера
        #   :region_id  - идентификатор региона (необязательный)
        #   :token      - токен для доступа к API Яндекс.Маркета
        #   :categories - список категорий для загрузки моделей Яндекс.Маркета (разделитель - ';')
        #
        # Returns an instance of Readers::Category
        def initialize(options)
          super
          @categories = options.fetch(:categories).split(';').map(&:squish)
        end

        # Public: читаем категории из API и для каждой выполняем блок
        #
        # Returns nothing
        def each_row
          root_categories.each do |root_category|
            yield root_category

            get_children_categories(root_category[:id]) do |category|
              yield category
            end
          end
        end

        private

        def get_children_categories(parent_id, &block)
          page = 1

          loop do
            sleep(SLEEP_TIME)
            categories =
              get_categories("categories/#{parent_id}/children", page)

            page += 1

            categories.each do |category|
              yield category

              next if category[:childCount].zero?

              get_children_categories(category[:id], &block)
            end

            break if categories.empty? || categories.count < PAGE_SIZE
          end
        end

        def root_categories(page = 1)
          categories = get_categories('categories', page)
          categories += root_categories(page + 1) unless categories.size < PAGE_SIZE

          categories.select! { |category| @categories.include?(category.fetch(:name)) }
        end

        def get_categories(resource, page)
          with_rescue_temporary_errors do
            client.
              get(
                resource,
                geo_id: @region_id,
                sort: 'BY_NAME'.freeze,
                count: PAGE_SIZE,
                page: page
              ).
              fetch(:categories)
          end
        rescue Api::PageError
          []
        end
      end
    end
  end
end