#   Copyright (c) 2010, Diaspora Inc.  This file is
#   licensed under the Affero General Public License version 3.  See
#   the COPYRIGHT file.



module Diaspora
  module WebSocket
    def self.queue_to_user(uid, data)
      channel = Magent::GenericChannel.new('websocket')
      channel.enqueue({:uid => uid, :data => data})
    end

    def self.initialize_channels
      @channels = {}
    end

    def self.push_to_user(uid, data)
      Rails.logger.debug "Websocketing to #{uid}"
      @channels[uid.to_s][0].push(data) if @channels[uid.to_s]
    end

    def self.subscribe(uid, ws)
      Rails.logger.debug "Subscribing socket to #{uid}"
      self.ensure_channel(uid)
      @channels[uid][0].subscribe{ |msg| ws.send msg }
      @channels[uid][1] += 1
    end

    def self.ensure_channel(uid)
      @channels[uid] ||= [EM::Channel.new, 0 ]
    end

    def self.unsubscribe(uid,sid)
      Rails.logger.debug "Unsubscribing socket #{sid} from #{uid}"
      @channels[uid][0].unsubscribe(sid) if @channels[uid]
      @channels[uid][1] -= 1
      if @channels[uid][1] <= 0
        @channels.delete(uid)
      end
    end
  end

  module Socketable
    def socket_to_uid(id, opts={})
      SocketsController.new.outgoing(id, self, opts)
    end

    def unsocket_from_uid(id, opts={})
      SocketsController.new.outgoing(id, Retraction.for(self), opts)
    end
  end
end