require 'tamashii/manager/channel_pool'

module Tamashii
  module Manager
    class Channel < Set
      SERVER_ID = 0

      class << self
        def pool
          @pool ||= ChannelPool.new
        end

        def get(id)
          pool[id]
        end

        def servers
          @servers ||= Channel.new(SERVER_ID)
        end

        def select_channel(client)
          case client.type
          when :checkin
            servers
          else
            return pool.get_idle || pool.create! if pool[client.tag].nil?
            pool[client.tag]
          end
        end

        def subscribe(client)
          channel = select_channel(client)
          channel.add(client)
          client.tag = channel.id

          pool.ready(channel)

          Manager.logger.info("Client #{client.id} subscribe to Channel ##{channel.id}")

          channel
        end

        def unsubscribe(client)
          channel = select_channel(client)
          channel.delete(client)

          Manager.logger.info("Client #{client.id} unsubscribe to Channel ##{channel.id}")

          if channel.empty? && channel.id != SERVER_ID
            pool.idle(channel.id)
            Manager.logger.debug("Channel Pool add - ##{channel.id}, available channels: #{pool.idle.size}")
          end
        end
      end

      attr_reader :id

      def initialize(id, *args)
        super(*args)
        @id = id
      end

      def send_to(channel_id, packet)
        return unless channel = Channel.pool[channel_id]
        channel.broadcast(packet)
      end

      def broadcast(packet, exclude_server = false)
        Manager.logger.info("Broadcast \"#{packet}\" to Channel ##{@id}")
        each do |client|
          client.send(packet)
        end
        Channel.servers.broadcast(packet) unless id == SERVER_ID || exclude_server
      end

      def broadcast_all(packet)
        Channel.pool.each do |id, channel|
          channel.broadcast(packet, true) unless channel.nil?
        end
        Channel.servers.broadcast(packet) unless id == SERVER_ID
      end
    end
  end
end