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

class Postzord::Dispatcher::Private
  # @note Takes :additional_subscribers param to add to subscribers to dispatch to
  def initialize(user, object, opts={})
    unless object.respond_to? :to_diaspora_xml
      raise 'this object does not respond_to? to_diaspora xml.  try including Diaspora::Webhooks into your object'
    end
    @sender = user
    @sender_person = @sender.person
    @object = object
    @xml = @object.to_diaspora_xml
    @subscribers = @object.subscribers(@sender)
    @subscribers = @subscribers | [*opts[:additional_subscribers]] if opts[:additional_subscribers]
  end

  def salmon
    @salmon_factory ||= Salmon::EncryptedSlap.create_by_user_and_activity(@sender, @xml)
  end

  def post(opts = {})
    unless @subscribers == nil
      remote_people, local_people = @subscribers.partition{ |person| person.owner_id.nil? }

      if @object.respond_to?(:relayable?) && @sender.owns?(@object.parent)
        user_ids = [*local_people].map{|x| x.owner_id }
        local_users = User.where(:id => user_ids)
        self.notify_users(local_users)
        local_users << @sender if @object.author.local?
        self.socket_to_users(local_users)
      else
        self.deliver_to_local(local_people)
      end

      self.deliver_to_remote(remote_people) unless @sender.username == 'diasporahq' #NOTE: 09/08/11 this is temporary (~3days max) till we fix fanout in federation
    end
    self.deliver_to_services(opts[:url], opts[:services] || [])
    @object.after_dispatch(@sender)
  end

  protected

  def deliver_to_remote(people)
    Resque.enqueue(Job::HttpMulti, @sender.id, Base64.encode64(@object.to_diaspora_xml), people.map{|p| p.id}) unless people.empty?
  end

  def deliver_to_local(people)
    return if people.blank? || @object.is_a?(Profile)
    if @object.is_a?(Post)
      batch_deliver_to_local(people)
    else
      people.each do |person|
        Rails.logger.info("event=push route=local sender=#{@sender_person.diaspora_handle} recipient=#{person.diaspora_handle} payload_type=#{@object.class}")
        Resque.enqueue(Job::Receive, person.owner_id, @xml, @sender_person.id)
      end
    end
  end

  def batch_deliver_to_local(people)
    ids = people.map{ |p| p.owner_id }
    Resque.enqueue(Job::ReceiveLocalBatch, @object.id, ids)
    Rails.logger.info("event=push route=local sender=#{@sender_person.diaspora_handle} recipients=#{ids.join(',')} payload_type=#{@object.class}")
  end

  def deliver_to_hub
    Rails.logger.debug("event=post_to_service type=pubsub sender_handle=#{@sender.diaspora_handle}")
    Resque.enqueue(Job::PublishToHub, @sender.public_url)
  end

  def deliver_to_services(url, services)
    if @object.respond_to?(:public) && @object.public
      deliver_to_hub
    end
    if @object.instance_of?(StatusMessage)
      services.each do |service|
        Resque.enqueue(Job::PostToService, service.id, @object.id, url)
      end
    end
  end

  def socket_and_notify_users(users)
    notify_users(users)
    socket_to_users(users)
  end


  def notify_users(users)
    Resque.enqueue(Job::NotifyLocalUsers, users.map{|u| u.id}, @object.class.to_s, @object.id, @object.author.id)
  end
  def socket_to_users(users)
    return unless @object.respond_to?(:socket_to_user)
    users.each do |user|
      @object.socket_to_user(user)
    end
  end
end