forked from fedi/mastodon
81584779cb
* Fix #2473 - Use sidekiq scheduler to refresh PuSH subscriptions instead of cron Fix an issue where / in domain would raise exception in TagManager#normalize_domain PuSH subscriptions refresh done in a round-robin way to avoid hammering a single server's hub in sequence. Correct handling of failures/retries through Sidekiq (see also #2613). Optimize Account#with_followers scope. Also, since subscriptions are now delegated to Sidekiq jobs, an uncaught exception will not stop the entire refreshing operation halfway through Fix #2702 - Correct user agent header on outgoing http requests * Add test for SubscribeService * Extract #expiring_accounts into method * Make mastodon:push:refresh no-op * Queues are now defined in sidekiq.yml * Queues are now in sidekiq.yml
49 lines
1.6 KiB
Ruby
49 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class Pubsubhubbub::DeliveryWorker
|
|
include Sidekiq::Worker
|
|
include RoutingHelper
|
|
|
|
sidekiq_options queue: 'push', retry: 3, dead: false
|
|
|
|
sidekiq_retry_in do |count|
|
|
5 * (count + 1)
|
|
end
|
|
|
|
def perform(subscription_id, payload)
|
|
subscription = Subscription.find(subscription_id)
|
|
headers = {}
|
|
host = Addressable::URI.parse(subscription.callback_url).normalize.host
|
|
|
|
return if DomainBlock.blocked?(host)
|
|
|
|
headers['User-Agent'] = 'Mastodon/PubSubHubbub'
|
|
headers['Link'] = LinkHeader.new([[api_push_url, [%w(rel hub)]], [account_url(subscription.account, format: :atom), [%w(rel self)]]]).to_s
|
|
headers['X-Hub-Signature'] = signature(subscription.secret, payload) if subscription.secret?
|
|
|
|
response = HTTP.timeout(:per_operation, write: 50, connect: 20, read: 50)
|
|
.headers(headers)
|
|
.post(subscription.callback_url, body: payload)
|
|
|
|
return subscription.destroy! if response_failed_permanently?(response) # HTTP 4xx means error is not temporary, except for 429 (throttling)
|
|
raise "Delivery failed for #{subscription.callback_url}: HTTP #{response.code}" unless response_successful?(response)
|
|
|
|
subscription.touch(:last_successful_delivery_at)
|
|
end
|
|
|
|
private
|
|
|
|
def signature(secret, payload)
|
|
hmac = OpenSSL::HMAC.hexdigest(OpenSSL::Digest.new('sha1'), secret, payload)
|
|
"sha1=#{hmac}"
|
|
end
|
|
|
|
def response_failed_permanently?(response)
|
|
response.code > 299 && response.code < 500 && response.code != 429
|
|
end
|
|
|
|
def response_successful?(response)
|
|
response.code > 199 && response.code < 300
|
|
end
|
|
end
|