mastodon/app/workers/pubsubhubbub/confirmation_worker.rb

37 lines
1.3 KiB
Ruby
Raw Normal View History

2016-11-28 04:36:47 -08:00
# frozen_string_literal: true
class Pubsubhubbub::ConfirmationWorker
include Sidekiq::Worker
include RoutingHelper
sidekiq_options queue: 'push', retry: false
2016-11-28 04:36:47 -08:00
def perform(subscription_id, mode, secret = nil, lease_seconds = nil)
subscription = Subscription.find(subscription_id)
challenge = SecureRandom.hex
subscription.secret = secret
subscription.lease_seconds = lease_seconds
subscription.confirmed = true
2016-11-28 04:36:47 -08:00
response = HTTP.headers(user_agent: 'Mastodon/PubSubHubbub')
.timeout(:per_operation, write: 20, connect: 20, read: 50)
.get(subscription.callback_url, params: {
'hub.topic' => account_url(subscription.account, format: :atom),
'hub.mode' => mode,
'hub.challenge' => challenge,
'hub.lease_seconds' => subscription.lease_seconds,
})
2016-11-28 10:15:28 -08:00
body = response.body.to_s
logger.debug "Confirming PuSH subscription for #{subscription.callback_url} with challenge #{challenge}: #{body}"
2016-11-28 10:15:28 -08:00
if mode == 'subscribe' && body == challenge
2016-11-28 04:36:47 -08:00
subscription.save!
2016-11-28 10:15:28 -08:00
elsif (mode == 'unsubscribe' && body == challenge) || !subscription.confirmed?
2016-11-28 04:36:47 -08:00
subscription.destroy!
end
end
end