2016-11-28 07:36:47 -05:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
class Pubsubhubbub::DistributionWorker
|
|
|
|
include Sidekiq::Worker
|
|
|
|
|
2016-11-28 20:07:14 -05:00
|
|
|
sidekiq_options queue: 'push'
|
|
|
|
|
2017-06-14 12:01:35 -04:00
|
|
|
def perform(stream_entry_ids)
|
|
|
|
stream_entries = StreamEntry.where(id: stream_entry_ids).includes(:status).reject { |e| e.status&.direct_visibility? }
|
2017-01-02 18:44:23 -05:00
|
|
|
|
2017-06-14 12:01:35 -04:00
|
|
|
return if stream_entries.empty?
|
2017-01-02 18:44:23 -05:00
|
|
|
|
2017-06-14 12:01:35 -04:00
|
|
|
@account = stream_entries.first.account
|
|
|
|
@subscriptions = active_subscriptions.to_a
|
|
|
|
|
|
|
|
distribute_public!(stream_entries.reject(&:hidden?))
|
|
|
|
distribute_hidden!(stream_entries.reject { |s| !s.hidden? })
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def distribute_public!(stream_entries)
|
|
|
|
return if stream_entries.empty?
|
|
|
|
|
2017-07-18 10:39:47 -04:00
|
|
|
@payload = Ostatus::AtomSerializer.render(Ostatus::AtomSerializer.new.feed(@account, stream_entries))
|
2017-06-14 12:01:35 -04:00
|
|
|
|
|
|
|
Pubsubhubbub::DeliveryWorker.push_bulk(@subscriptions) do |subscription|
|
|
|
|
[subscription.id, @payload]
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def distribute_hidden!(stream_entries)
|
|
|
|
return if stream_entries.empty?
|
|
|
|
|
2017-07-18 10:39:47 -04:00
|
|
|
@payload = Ostatus::AtomSerializer.render(Ostatus::AtomSerializer.new.feed(@account, stream_entries))
|
2017-06-22 12:34:27 -04:00
|
|
|
@domains = @account.followers.domains
|
2016-11-28 07:36:47 -05:00
|
|
|
|
2017-07-14 17:01:20 -04:00
|
|
|
Pubsubhubbub::DeliveryWorker.push_bulk(@subscriptions.reject { |s| !allowed_to_receive?(s.callback_url, s.domain) }) do |subscription|
|
2017-06-14 12:01:35 -04:00
|
|
|
[subscription.id, @payload]
|
2016-11-28 07:36:47 -05:00
|
|
|
end
|
|
|
|
end
|
2017-05-09 20:55:43 -04:00
|
|
|
|
2017-06-14 12:01:35 -04:00
|
|
|
def active_subscriptions
|
2017-07-14 17:01:20 -04:00
|
|
|
Subscription.where(account: @account).active.select('id, callback_url, domain')
|
2017-06-14 12:01:35 -04:00
|
|
|
end
|
2017-05-09 20:55:43 -04:00
|
|
|
|
2017-07-14 17:01:20 -04:00
|
|
|
def allowed_to_receive?(callback_url, domain)
|
|
|
|
(!domain.nil? && @domains.include?(domain)) || @domains.include?(Addressable::URI.parse(callback_url).host)
|
2017-05-09 20:55:43 -04:00
|
|
|
end
|
2016-11-28 07:36:47 -05:00
|
|
|
end
|