132 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			132 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| class ActivityPub::Activity::Undo < ActivityPub::Activity
 | |
|   def perform
 | |
|     case @object['type']
 | |
|     when 'Announce'
 | |
|       undo_announce
 | |
|     when 'Accept'
 | |
|       undo_accept
 | |
|     when 'Follow'
 | |
|       undo_follow
 | |
|     when 'Like'
 | |
|       undo_like
 | |
|     when 'Block'
 | |
|       undo_block
 | |
|     when nil
 | |
|       handle_reference
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   private
 | |
| 
 | |
|   def handle_reference
 | |
|     # Some implementations do not inline the object, and as we don't have a
 | |
|     # global index, we have to guess what object it is.
 | |
|     return if object_uri.nil?
 | |
| 
 | |
|     try_undo_announce || try_undo_accept || try_undo_follow || try_undo_like || try_undo_block || delete_later!(object_uri)
 | |
|   end
 | |
| 
 | |
|   def try_undo_announce
 | |
|     status = Status.where.not(reblog_of_id: nil).find_by(uri: object_uri, account: @account)
 | |
|     if status.present?
 | |
|       RemoveStatusService.new.call(status)
 | |
|       true
 | |
|     else
 | |
|       false
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def try_undo_accept
 | |
|     # We can't currently handle `Undo Accept` as we don't record `Accept`'s uri
 | |
|     false
 | |
|   end
 | |
| 
 | |
|   def try_undo_follow
 | |
|     follow = @account.follow_requests.find_by(uri: object_uri) || @account.active_relationships.find_by(uri: object_uri)
 | |
| 
 | |
|     if follow.present?
 | |
|       follow.destroy
 | |
|       true
 | |
|     else
 | |
|       false
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def try_undo_like
 | |
|     # There is an index on accounts, but an account may have *many* favs, so this may be too costly
 | |
|     false
 | |
|   end
 | |
| 
 | |
|   def try_undo_block
 | |
|     block = @account.block_relationships.find_by(uri: object_uri)
 | |
|     if block.present?
 | |
|       UnblockService.new.call(@account, block.target_account)
 | |
|       true
 | |
|     else
 | |
|       false
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def undo_announce
 | |
|     return if object_uri.nil?
 | |
| 
 | |
|     status   = Status.find_by(uri: object_uri, account: @account)
 | |
|     status ||= Status.find_by(uri: @object['atomUri'], account: @account) if @object.is_a?(Hash) && @object['atomUri'].present?
 | |
| 
 | |
|     if status.nil?
 | |
|       delete_later!(object_uri)
 | |
|     else
 | |
|       RemoveStatusService.new.call(status)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def undo_accept
 | |
|     ::Follow.find_by(target_account: @account, uri: target_uri)&.revoke_request!
 | |
|   end
 | |
| 
 | |
|   def undo_follow
 | |
|     target_account = account_from_uri(target_uri)
 | |
| 
 | |
|     return if target_account.nil? || !target_account.local?
 | |
| 
 | |
|     if @account.following?(target_account)
 | |
|       @account.unfollow!(target_account)
 | |
|     elsif @account.requested?(target_account)
 | |
|       FollowRequest.find_by(account: @account, target_account: target_account)&.destroy
 | |
|     else
 | |
|       delete_later!(object_uri)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def undo_like
 | |
|     status = status_from_uri(target_uri)
 | |
| 
 | |
|     return if status.nil? || !status.account.local?
 | |
| 
 | |
|     if @account.favourited?(status)
 | |
|       favourite = status.favourites.where(account: @account).first
 | |
|       favourite&.destroy
 | |
|     else
 | |
|       delete_later!(object_uri)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def undo_block
 | |
|     target_account = account_from_uri(target_uri)
 | |
| 
 | |
|     return if target_account.nil? || !target_account.local?
 | |
| 
 | |
|     if @account.blocking?(target_account)
 | |
|       UnblockService.new.call(@account, target_account)
 | |
|     else
 | |
|       delete_later!(object_uri)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def target_uri
 | |
|     @target_uri ||= value_or_id(@object['object'])
 | |
|   end
 | |
| end
 |