* Spread out crawling randomly to avoid DDoSing the link Fix #4486 * Remove trailing whitespace
		
			
				
	
	
		
			133 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			133 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| class ActivityPub::Activity
 | |
|   include JsonLdHelper
 | |
| 
 | |
|   def initialize(json, account, **options)
 | |
|     @json    = json
 | |
|     @account = account
 | |
|     @object  = @json['object']
 | |
|     @options = options
 | |
|   end
 | |
| 
 | |
|   def perform
 | |
|     raise NotImplementedError
 | |
|   end
 | |
| 
 | |
|   class << self
 | |
|     def factory(json, account, **options)
 | |
|       @json = json
 | |
|       klass&.new(json, account, options)
 | |
|     end
 | |
| 
 | |
|     private
 | |
| 
 | |
|     def klass
 | |
|       case @json['type']
 | |
|       when 'Create'
 | |
|         ActivityPub::Activity::Create
 | |
|       when 'Announce'
 | |
|         ActivityPub::Activity::Announce
 | |
|       when 'Delete'
 | |
|         ActivityPub::Activity::Delete
 | |
|       when 'Follow'
 | |
|         ActivityPub::Activity::Follow
 | |
|       when 'Like'
 | |
|         ActivityPub::Activity::Like
 | |
|       when 'Block'
 | |
|         ActivityPub::Activity::Block
 | |
|       when 'Update'
 | |
|         ActivityPub::Activity::Update
 | |
|       when 'Undo'
 | |
|         ActivityPub::Activity::Undo
 | |
|       when 'Accept'
 | |
|         ActivityPub::Activity::Accept
 | |
|       when 'Reject'
 | |
|         ActivityPub::Activity::Reject
 | |
|       when 'Flag'
 | |
|         ActivityPub::Activity::Flag
 | |
|       when 'Add'
 | |
|         ActivityPub::Activity::Add
 | |
|       when 'Remove'
 | |
|         ActivityPub::Activity::Remove
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   protected
 | |
| 
 | |
|   def status_from_uri(uri)
 | |
|     ActivityPub::TagManager.instance.uri_to_resource(uri, Status)
 | |
|   end
 | |
| 
 | |
|   def account_from_uri(uri)
 | |
|     ActivityPub::TagManager.instance.uri_to_resource(uri, Account)
 | |
|   end
 | |
| 
 | |
|   def object_uri
 | |
|     @object_uri ||= value_or_id(@object)
 | |
|   end
 | |
| 
 | |
|   def redis
 | |
|     Redis.current
 | |
|   end
 | |
| 
 | |
|   def distribute(status)
 | |
|     crawl_links(status)
 | |
| 
 | |
|     notify_about_reblog(status) if reblog_of_local_account?(status)
 | |
|     notify_about_mentions(status)
 | |
| 
 | |
|     # Only continue if the status is supposed to have arrived in real-time.
 | |
|     # Note that if @options[:override_timestamps] isn't set, the status
 | |
|     # may have a lower snowflake id than other existing statuses, potentially
 | |
|     # "hiding" it from paginated API calls
 | |
|     return unless @options[:override_timestamps] || status.within_realtime_window?
 | |
| 
 | |
|     distribute_to_followers(status)
 | |
|   end
 | |
| 
 | |
|   def reblog_of_local_account?(status)
 | |
|     status.reblog? && status.reblog.account.local?
 | |
|   end
 | |
| 
 | |
|   def notify_about_reblog(status)
 | |
|     NotifyService.new.call(status.reblog.account, status)
 | |
|   end
 | |
| 
 | |
|   def notify_about_mentions(status)
 | |
|     status.mentions.includes(:account).each do |mention|
 | |
|       next unless mention.account.local? && audience_includes?(mention.account)
 | |
|       NotifyService.new.call(mention.account, mention)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def crawl_links(status)
 | |
|     return if status.spoiler_text?
 | |
| 
 | |
|     # Spread out crawling randomly to avoid DDoSing the link
 | |
|     LinkCrawlWorker.perform_in(rand(1..59).seconds, status.id)
 | |
|   end
 | |
| 
 | |
|   def distribute_to_followers(status)
 | |
|     ::DistributionWorker.perform_async(status.id)
 | |
|   end
 | |
| 
 | |
|   def delete_arrived_first?(uri)
 | |
|     redis.exists("delete_upon_arrival:#{@account.id}:#{uri}")
 | |
|   end
 | |
| 
 | |
|   def delete_later!(uri)
 | |
|     redis.setex("delete_upon_arrival:#{@account.id}:#{uri}", 6.hours.seconds, uri)
 | |
|   end
 | |
| 
 | |
|   def fetch_remote_original_status
 | |
|     if object_uri.start_with?('http')
 | |
|       return if ActivityPub::TagManager.instance.local_uri?(object_uri)
 | |
|       ActivityPub::FetchRemoteStatusService.new.call(object_uri, id: true, on_behalf_of: @account.followers.local.first)
 | |
|     elsif @object['url'].present?
 | |
|       ::FetchRemoteStatusService.new.call(@object['url'])
 | |
|     end
 | |
|   end
 | |
| end
 |