app/services/subscribe_service.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

# frozen_string_literal: true

class SubscribeService < BaseService
  def call(account)
    return if account.hub_url.blank?

    @account        = account
    @account.secret = SecureRandom.hex
    @response       = build_request.perform

    if response_failed_permanently?
      # We're not allowed to subscribe. Fail and move on.
      @account.secret = ''
      @account.save!
    elsif response_successful?
      # The subscription will be confirmed asynchronously.
      @account.save!
    else
      # The response was either a 429 rate limit, or a 5xx error.
      # We need to retry at a later time. Fail loudly!
      raise Mastodon::UnexpectedResponseError, @response
    end
    @response.connection&.close
  end

  private

  def build_request
    request = Request.new(:post, @account.hub_url, form: subscription_params)
    request.on_behalf_of(some_local_account) if some_local_account
    request
  end

  def subscription_params
    {
      'hub.topic': @account.remote_url,
      'hub.mode': 'subscribe',
      'hub.callback': api_subscription_url(@account.id),
      'hub.verify': 'async',
      'hub.secret': @account.secret,
      'hub.lease_seconds': 7.days.seconds,
    }
  end

  def some_local_account
    @some_local_account ||= Account.local.where(suspended: false).first
  end

  # Any response in the 3xx or 4xx range, except for 429 (rate limit)
  def response_failed_permanently?
    (@response.status.redirect? || @response.status.client_error?) && !@response.status.too_many_requests?
  end

  # Any response in the 2xx range
  def response_successful?
    @response.status.success?
  end
end