mastodon/app/workers/pubsubhubbub/confirmation_worker.rb

91 lines
2 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
attr_reader :subscription, :mode, :secret, :lease_seconds
2016-11-28 04:36:47 -08:00
def perform(subscription_id, mode, secret = nil, lease_seconds = nil)
@subscription = Subscription.find(subscription_id)
@mode = mode
@secret = secret
@lease_seconds = lease_seconds
process_confirmation
end
2016-11-28 04:36:47 -08:00
private
2016-11-28 04:36:47 -08:00
def process_confirmation
prepare_subscription
2016-11-28 04:36:47 -08:00
confirm_callback
logger.debug "Confirming PuSH subscription for #{subscription.callback_url} with challenge #{challenge}: #{callback_response_body}"
2016-11-28 10:15:28 -08:00
update_subscription
end
2016-11-28 10:15:28 -08:00
def update_subscription
if successful_subscribe?
2016-11-28 04:36:47 -08:00
subscription.save!
elsif successful_unsubscribe?
2016-11-28 04:36:47 -08:00
subscription.destroy!
end
end
def successful_subscribe?
subscribing? && response_matches_challenge?
end
def successful_unsubscribe?
(unsubscribing? && response_matches_challenge?) || !subscription.confirmed?
end
def response_matches_challenge?
callback_response_body == challenge
end
def subscribing?
mode == 'subscribe'
end
def unsubscribing?
mode == 'unsubscribe'
end
def confirm_callback
@_confirm_callback ||= callback_get_with_params
end
def callback_get_with_params
HTTP.headers(user_agent: 'Mastodon/PubSubHubbub')
.timeout(:per_operation, write: 20, connect: 20, read: 50)
.get(subscription.callback_url, params: callback_params)
end
def callback_response_body
confirm_callback.body.to_s
end
def callback_params
{
'hub.topic' => account_url(subscription.account, format: :atom),
'hub.mode' => mode,
'hub.challenge' => challenge,
'hub.lease_seconds' => subscription.lease_seconds,
}
end
def prepare_subscription
subscription.secret = secret
subscription.lease_seconds = lease_seconds
subscription.confirmed = true
end
def challenge
@_challenge ||= SecureRandom.hex
end
2016-11-28 04:36:47 -08:00
end