require 'concurrent'

module SplitIoClient
  module Cache
    module Repositories
      class SplitsRepository < Repository
        SPLITS_SLICE = 10

        def initialize(adapter)
          @adapter = adapter

          @adapter.set_string(namespace_key('split.till'), '-1')
          @adapter.initialize_map(namespace_key('segments.registered'))
        end

        def add_split(split)
          @adapter.set_string(namespace_key("split.#{split[:name]}"), split.to_json)
        end

        def remove_split(name)
          @adapter.delete(namespace_key("split.#{name}"))
        end

        def get_splits(names, slice = SPLITS_SLICE)
          splits = {}

          names.each_slice(slice) do |splits_slice|
            splits.merge!(
              @adapter
                .multiple_strings(splits_slice.map { |name| namespace_key("split.#{name}") })
                .map { |name, data| [name.gsub(namespace_key('split.'), ''), data] }.to_h
            )
          end

          splits.map do |name, data|
            parsed_data = data ? JSON.parse(data, symbolize_names: true) : nil
            [name.to_sym, parsed_data]
          end.to_h
        end

        def get_split(name)
          split = @adapter.string(namespace_key("split.#{name}"))

          JSON.parse(split, symbolize_names: true) if split
        end

        def splits
          splits_hash = {}

          split_names.each do |name|
            splits_hash[name] = get_split(name)
          end

          splits_hash
        end

        # Return an array of Split Names excluding control keys like split.till
        def split_names
          @adapter.find_strings_by_prefix(namespace_key('split'))
            .reject { |split| split == namespace_key('split.till') }
            .map { |split| split.gsub(namespace_key('split.'), '') }
        end

        def set_change_number(since)
          @adapter.set_string(namespace_key('split.till'), since)
        end

        def get_change_number
          @adapter.string(namespace_key('split.till'))
        end

        def set_segment_names(names)
          return if names.nil? || names.empty?

          names.each do |name|
            @adapter.add_to_set(namespace_key('segments.registered'), name)
          end
        end

        def exists?(name)
          @adapter.exists?(namespace_key("split.#{name}"))
        end
      end
    end
  end
end