about summary refs log tree commit diff
path: root/app/services/activitypub/fetch_account_statuses_service.rb
blob: 27fa8bd6c20d5dd3b4e0f1cbd1626555682c2d04 (plain) (blame)
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# frozen_string_literal: true

class ActivityPub::FetchAccountStatusesService < BaseService
  include JsonLdHelper
  include Redisable

  MAX_PAGES = 50

  def call(account, url = nil)
    @account = account
    return if account.local? || account.suspended? || redis.get('busy_key')

    redis.set(busy_key, 1, ex: 3.days)

    @items = Rails.cache.fetch(sync_key) || []

    return if redis.get(cooldown_key) && @items.empty?
    redis.set(cooldown_key, 1, ex: 1.day)

    @json = fetch_collection(url || account.outbox_url)
    page = 1

    if @items.empty?
      until page == MAX_PAGES || @json.blank?
        items = collection_items(@json).select { |item| item['type'] == 'Create' }
        @items.concat(items)
        break if @json['next'].blank?
        page += 1
        @json = fetch_collection(@json['next'])
      end
    end

    Rails.cache.write(sync_key, @items, expires_in: 1.day)

    process_items(@items)

    Rails.cache.delete(sync_key)
    redis.expire(cooldown_key, 1.week)

    @items
  ensure
    redis.del(busy_key)
  end

  private

  def sync_key
    "account_sync:#{@account.id}"
  end

  def busy_key
    "account_sync:#{@account.id}:busy"
  end

  def cooldown_key
    "account_sync:#{@account.id}:cooldown"
  end

  def fetch_collection(collection_or_uri)
    return collection_or_uri if collection_or_uri.is_a?(Hash)

    collection = _fetch_collection(collection_or_uri)
    return unless collection.is_a?(Hash)

    if collection['first'].present?
      collection = _fetch_collection(collection['first'])
      return unless collection.is_a?(Hash)
    end

    collection
  end

  def _fetch_collection(collection_or_uri)
    return collection_or_uri if collection_or_uri.is_a?(Hash)
    return if invalid_origin?(collection_or_uri)
    fetch_resource_without_id_validation(collection_or_uri, nil, true)
  end

  def collection_items(collection)
    case collection['type']
    when 'Collection', 'CollectionPage'
      collection['items']
    when 'OrderedCollection', 'OrderedCollectionPage'
      collection['orderedItems']
    end
  end

  def process_items(items)
    items.reverse_each.map { |item| process_item(item) }.compact
  end

  def process_item(item)
    return unless item.is_a?(Hash) && item['type'].present?
    ActivityPub::Activity.factory(item, @account, override_timestamps: true, requested: true)&.perform
  rescue => e
    Rails.logger.error("Failed to process #{item['type']} #{item['id']} due to #{e}: #{e.message}")
    Rails.logger.error("Stack trace: #{backtrace.map {|l| "  #{l}\n"}.join}")
  end
end