* Eliminate extra accounts select query from FollowService * Optimistically update follow state in web UI and hide loading bar Fix #6205 * Asynchronize NotifyService in FollowService And fix failing test * Skip Webfinger resolve routine when called from FollowService if possible If an account is ActivityPub, then webfinger re-resolving is not necessary when called from FollowService. Improve options of ResolveAccountService
		
			
				
	
	
		
			92 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| class FollowService < BaseService
 | |
|   include StreamEntryRenderer
 | |
| 
 | |
|   # Follow a remote user, notify remote user about the follow
 | |
|   # @param [Account] source_account From which to follow
 | |
|   # @param [String, Account] uri User URI to follow in the form of username@domain (or account record)
 | |
|   # @param [true, false, nil] reblogs Whether or not to show reblogs, defaults to true
 | |
|   def call(source_account, target_account, reblogs: nil)
 | |
|     reblogs = true if reblogs.nil?
 | |
|     target_account = ResolveAccountService.new.call(target_account, skip_webfinger: true)
 | |
| 
 | |
|     raise ActiveRecord::RecordNotFound if target_account.nil? || target_account.id == source_account.id || target_account.suspended?
 | |
|     raise Mastodon::NotPermittedError  if target_account.blocking?(source_account) || source_account.blocking?(target_account)
 | |
| 
 | |
|     if source_account.following?(target_account)
 | |
|       # We're already following this account, but we'll call follow! again to
 | |
|       # make sure the reblogs status is set correctly.
 | |
|       source_account.follow!(target_account, reblogs: reblogs)
 | |
|       return
 | |
|     elsif source_account.requested?(target_account)
 | |
|       # This isn't managed by a method in AccountInteractions, so we modify it
 | |
|       # ourselves if necessary.
 | |
|       req = source_account.follow_requests.find_by(target_account: target_account)
 | |
|       req.update!(show_reblogs: reblogs)
 | |
|       return
 | |
|     end
 | |
| 
 | |
|     ActivityTracker.increment('activity:interactions')
 | |
| 
 | |
|     if target_account.locked? || target_account.activitypub?
 | |
|       request_follow(source_account, target_account, reblogs: reblogs)
 | |
|     else
 | |
|       direct_follow(source_account, target_account, reblogs: reblogs)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   private
 | |
| 
 | |
|   def request_follow(source_account, target_account, reblogs: true)
 | |
|     follow_request = FollowRequest.create!(account: source_account, target_account: target_account, show_reblogs: reblogs)
 | |
| 
 | |
|     if target_account.local?
 | |
|       LocalNotificationWorker.perform_async(target_account.id, follow_request.id, follow_request.class.name)
 | |
|     elsif target_account.ostatus?
 | |
|       NotificationWorker.perform_async(build_follow_request_xml(follow_request), source_account.id, target_account.id)
 | |
|       AfterRemoteFollowRequestWorker.perform_async(follow_request.id)
 | |
|     elsif target_account.activitypub?
 | |
|       ActivityPub::DeliveryWorker.perform_async(build_json(follow_request), source_account.id, target_account.inbox_url)
 | |
|     end
 | |
| 
 | |
|     follow_request
 | |
|   end
 | |
| 
 | |
|   def direct_follow(source_account, target_account, reblogs: true)
 | |
|     follow = source_account.follow!(target_account, reblogs: reblogs)
 | |
| 
 | |
|     if target_account.local?
 | |
|       LocalNotificationWorker.perform_async(target_account.id, follow.id, follow.class.name)
 | |
|     else
 | |
|       Pubsubhubbub::SubscribeWorker.perform_async(target_account.id) unless target_account.subscribed?
 | |
|       NotificationWorker.perform_async(build_follow_xml(follow), source_account.id, target_account.id)
 | |
|       AfterRemoteFollowWorker.perform_async(follow.id)
 | |
|     end
 | |
| 
 | |
|     MergeWorker.perform_async(target_account.id, source_account.id)
 | |
| 
 | |
|     follow
 | |
|   end
 | |
| 
 | |
|   def redis
 | |
|     Redis.current
 | |
|   end
 | |
| 
 | |
|   def build_follow_request_xml(follow_request)
 | |
|     OStatus::AtomSerializer.render(OStatus::AtomSerializer.new.follow_request_salmon(follow_request))
 | |
|   end
 | |
| 
 | |
|   def build_follow_xml(follow)
 | |
|     OStatus::AtomSerializer.render(OStatus::AtomSerializer.new.follow_salmon(follow))
 | |
|   end
 | |
| 
 | |
|   def build_json(follow_request)
 | |
|     Oj.dump(ActivityPub::LinkedDataSignature.new(ActiveModelSerializers::SerializableResource.new(
 | |
|       follow_request,
 | |
|       serializer: ActivityPub::FollowSerializer,
 | |
|       adapter: ActivityPub::Adapter
 | |
|     ).as_json).sign!(follow_request.account))
 | |
|   end
 | |
| end
 |