~cytrogen/masto-fe

ref: 53f5b27bd10e5f471b59cd0597d67ea12587c95a masto-fe/app/services/follow_migration_service.rb -rw-r--r-- 2.3 KiB
53f5b27b — Claire Merge commit '640421f661ee4d7e76a2aab607e7b15687940b6f' into glitch-soc/merge-upstream 2 years ago
                                                                                
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
# frozen_string_literal: true

class FollowMigrationService < FollowService
  # Follow an account with the same settings as another account, and unfollow the old account once the request is sent
  # @param [Account] source_account From which to follow
  # @param [Account] target_account Account to follow
  # @param [Account] old_target_account Account to unfollow once the follow request has been sent to the new one
  # @option [Boolean] bypass_locked Whether to immediately follow the new account even if it is locked
  def call(source_account, target_account, old_target_account, bypass_locked: false)
    @old_target_account = old_target_account

    @original_follow = source_account.active_relationships.find_by(target_account: old_target_account)
    reblogs          = @original_follow&.show_reblogs?
    notify           = @original_follow&.notify?
    languages        = @original_follow&.languages

    super(source_account, target_account, reblogs: reblogs, notify: notify, languages: languages, bypass_locked: bypass_locked, bypass_limit: true)
  end

  private

  def request_follow!
    follow_request = @source_account.request_follow!(@target_account, **follow_options.merge(rate_limit: @options[:with_rate_limit], bypass_limit: @options[:bypass_limit]))
    migrate_list_accounts!

    if @target_account.local?
      LocalNotificationWorker.perform_async(@target_account.id, follow_request.id, follow_request.class.name, 'follow_request')
      UnfollowService.new.call(@source_account, @old_target_account, skip_unmerge: true)
    elsif @target_account.activitypub?
      ActivityPub::MigratedFollowDeliveryWorker.perform_async(build_json(follow_request), @source_account.id, @target_account.inbox_url, @old_target_account.id)
    end

    follow_request
  end

  def change_follow_options!
    migrate_list_accounts!
    super
  end

  def change_follow_request_options!
    migrate_list_accounts!
    super
  end

  def direct_follow!
    follow = super

    migrate_list_accounts!
    UnfollowService.new.call(@source_account, @old_target_account, skip_unmerge: true)

    follow
  end

  def migrate_list_accounts!
    ListAccount.where(follow_id: @original_follow.id).includes(:list).find_each do |list_account|
      list_account.list.accounts << @target_account
    rescue ActiveRecord::RecordInvalid
      nil
    end
  end
end