63 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			63 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # 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
 |