# frozen_string_literal: true

require 'concurrent/atomics'

module SplitIoClient
  module SSE
    class NotificationManagerKeeper
      def initialize(config)
        @config = config
        @publisher_available = Concurrent::AtomicBoolean.new(true)
        @on = { occupancy: ->(_) {}, push_shutdown: ->(_) {} }

        yield self if block_given?
      end

      def handle_incoming_occupancy_event(event)
        if event.data['type'] == 'CONTROL'
          process_event_control(event.data['controlType'])
        elsif event.channel == SplitIoClient::Constants::CONTROL_PRI
          process_event_occupancy(event.data['metrics']['publishers'])
        end
      rescue StandardError => e
        @config.logger.error(e)
      end

      def on_occupancy(&action)
        @on[:occupancy] = action
      end

      def on_push_shutdown(&action)
        @on[:push_shutdown] = action
      end

      private

      def process_event_control(type)
        case type
        when 'STREAMING_PAUSED'
          dispatch_occupancy_event(false)
        when 'STREAMING_RESUMED'
          dispatch_occupancy_event(true) if @publisher_available.value
        when 'STREAMING_DISABLED'
          dispatch_push_shutdown
        else
          @config.logger.error("Incorrect event type: #{incoming_notification}")
        end
      end

      def process_event_occupancy(publishers)
        @config.logger.debug(
          "Occupancy process event with #{publishers} publishers and polling on: #{@publisher_available.value}"
        )
        if publishers <= 0 && @publisher_available.value
          @publisher_available.make_false
          dispatch_occupancy_event(false)
        elsif publishers >= 1 && !@publisher_available.value
          @publisher_available.make_true
          dispatch_occupancy_event(true)
        end
      end

      def dispatch_occupancy_event(push_enable)
        @config.logger.debug("Dispatching occupancy event with publisher avaliable: #{push_enable}")
        @on[:occupancy].call(push_enable)
      end

      def dispatch_push_shutdown
        @config.logger.debug('Dispatching push shutdown')
        @on[:push_shutdown].call
      end
    end
  end
end