diff options
Diffstat (limited to 'app')
-rw-r--r-- | app/services/pubsubhubbub/subscribe_service.rb | 8 | ||||
-rw-r--r-- | app/services/pubsubhubbub/unsubscribe_service.rb | 10 | ||||
-rw-r--r-- | app/workers/pubsubhubbub/distribution_worker.rb | 18 |
3 files changed, 21 insertions, 15 deletions
diff --git a/app/services/pubsubhubbub/subscribe_service.rb b/app/services/pubsubhubbub/subscribe_service.rb index 67d7f6598..eeb7ab258 100644 --- a/app/services/pubsubhubbub/subscribe_service.rb +++ b/app/services/pubsubhubbub/subscribe_service.rb @@ -6,9 +6,9 @@ class Pubsubhubbub::SubscribeService < BaseService attr_reader :account, :callback, :secret, :lease_seconds def call(account, callback, secret, lease_seconds) - @account = account - @callback = callback - @secret = secret + @account = account + @callback = Addressable::URI.parse(callback).normalize.to_s + @secret = secret @lease_seconds = lease_seconds process_subscribe @@ -52,7 +52,7 @@ class Pubsubhubbub::SubscribeService < BaseService end def blocked_domain? - DomainBlock.blocked? Addressable::URI.parse(callback).normalize.host + DomainBlock.blocked? Addressable::URI.parse(callback).host end def locate_subscription diff --git a/app/services/pubsubhubbub/unsubscribe_service.rb b/app/services/pubsubhubbub/unsubscribe_service.rb index 99fec8b14..646150f7b 100644 --- a/app/services/pubsubhubbub/unsubscribe_service.rb +++ b/app/services/pubsubhubbub/unsubscribe_service.rb @@ -1,11 +1,11 @@ # frozen_string_literal: true class Pubsubhubbub::UnsubscribeService < BaseService - attr_reader :account, :callback_url + attr_reader :account, :callback - def call(account, callback_url) - @account = account - @callback_url = callback_url + def call(account, callback) + @account = account + @callback = Addressable::URI.parse(callback).normalize.to_s process_unsubscribe end @@ -26,6 +26,6 @@ class Pubsubhubbub::UnsubscribeService < BaseService end def subscription - @_subscription ||= Subscription.find_by(account: account, callback_url: callback_url) + @_subscription ||= Subscription.find_by(account: account, callback_url: callback) end end diff --git a/app/workers/pubsubhubbub/distribution_worker.rb b/app/workers/pubsubhubbub/distribution_worker.rb index c0e03990a..b8f5c35e1 100644 --- a/app/workers/pubsubhubbub/distribution_worker.rb +++ b/app/workers/pubsubhubbub/distribution_worker.rb @@ -10,15 +10,21 @@ class Pubsubhubbub::DistributionWorker return if stream_entry.status&.direct_visibility? - account = stream_entry.account - payload = AtomSerializer.render(AtomSerializer.new.feed(account, [stream_entry])) - domains = account.followers_domains + @account = stream_entry.account + @payload = AtomSerializer.render(AtomSerializer.new.feed(@account, [stream_entry])) + @domains = @account.followers_domains - Subscription.where(account: account).active.select('id, callback_url').find_each do |subscription| - next unless domains.include?(Addressable::URI.parse(subscription.callback_url).host) - Pubsubhubbub::DeliveryWorker.perform_async(subscription.id, payload) + Subscription.where(account: @account).active.select('id, callback_url').find_each do |subscription| + next if stream_entry.hidden? && !allowed_to_receive?(subscription.callback_url) + Pubsubhubbub::DeliveryWorker.perform_async(subscription.id, @payload) end rescue ActiveRecord::RecordNotFound true end + + private + + def allowed_to_receive?(callback_url) + @domains.include?(Addressable::URI.parse(callback_url).host) + end end |