about summary refs log tree commit diff
path: root/app/services/activitypub/fetch_account_statuses_service.rb
blob: 6a0d90f3cef78a0709e1412447d4a7bc65621eef (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 = 100

  def call(account, url = nil)
    return if account.local? || account.suspended?

    @account = account
    @items = Rails.cache.fetch(sync_key) || []
    return if redis.get(cooldown_key).present? && @items.empty?

    RedisLock.acquire(lock_options) do |lock|
      return unless lock.acquired?

      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)
    end

    @items
  end

  private

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

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

  def lock_options
    { redis: Redis.current, key: "account_sync:#{@account.id}:lock" }
  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, @account.followers.local.first, false)
  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, 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: #{e.backtrace.map {|l| "  #{l}\n"}.join}")
  end
end