Browse Source

Disable PuSH for blocked domains

master
Eugen Rochko 7 years ago
parent
commit
40a4053732
2 changed files with 7 additions and 3 deletions
  1. +3
    -2
      app/services/pubsubhubbub/subscribe_service.rb
  2. +4
    -1
      app/workers/pubsubhubbub/distribution_worker.rb

+ 3
- 2
app/services/pubsubhubbub/subscribe_service.rb View File

@@ -2,8 +2,9 @@

class Pubsubhubbub::SubscribeService < BaseService
def call(account, callback, secret, lease_seconds)
return ['Invalid topic URL', 422] if account.nil?
return ['Invalid callback URL', 422] unless !callback.blank? && callback =~ /\A#{URI.regexp(%w(http https))}\z/
return ['Invalid topic URL', 422] if account.nil?
return ['Invalid callback URL', 422] unless !callback.blank? && callback =~ /\A#{URI.regexp(%w(http https))}\z/
return ['Callback URL not allowed', 403] if DomainBlock.blocked?(Addressable::URI.parse(callback).host)

subscription = Subscription.where(account: account, callback_url: callback).first_or_create!(account: account, callback_url: callback)
Pubsubhubbub::ConfirmationWorker.perform_async(subscription.id, 'subscribe', secret, lease_seconds)


+ 4
- 1
app/workers/pubsubhubbub/distribution_worker.rb View File

@@ -13,8 +13,11 @@ class Pubsubhubbub::DistributionWorker
account = stream_entry.account
renderer = AccountsController.renderer.new(method: 'get', http_host: Rails.configuration.x.local_domain, https: Rails.configuration.x.use_https)
payload = renderer.render(:show, assigns: { account: account, entries: [stream_entry] }, formats: [:atom])
# domains = account.followers_domains

Subscription.where(account: account).active.select('id').find_each do |subscription|
Subscription.where(account: account).active.select('id, callback_url').find_each do |subscription|
host = Addressable::URI.parse(subscription.callback_url).host
next if DomainBlock.blocked?(host) # || !domains.include?(host)
Pubsubhubbub::DeliveryWorker.perform_async(subscription.id, payload)
end
rescue ActiveRecord::RecordNotFound


Loading…
Cancel
Save