mirror of
https://github.com/whippyshou/mastodon
synced 2024-12-11 13:18:56 +09:00
f1f96ebf02
* Fix being able to import more than allowed number of follows Without this commit, if someone tries importing a second list of accounts to follow before the first one has been processed, this will queue imports for the two whole lists, even if they exceed the account's allowed number of outgoing follows. This commit changes it so the individual queued imports aren't exempt from the follow limit check (they remain exempt from the rate-limiting check though). * Catch validation errors to not re-queue failed follows Co-authored-by: Claire <claire.github-309c@sitedethib.com>
56 lines
1.8 KiB
Ruby
56 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class Import::RelationshipWorker
|
|
include Sidekiq::Worker
|
|
|
|
sidekiq_options queue: 'pull', retry: 8, dead: false
|
|
|
|
def perform(account_id, target_account_uri, relationship, options = {})
|
|
from_account = Account.find(account_id)
|
|
target_domain = domain(target_account_uri)
|
|
target_account = stoplight_wrap_request(target_domain) { ResolveAccountService.new.call(target_account_uri, { check_delivery_availability: true }) }
|
|
options.symbolize_keys!
|
|
|
|
return if target_account.nil?
|
|
|
|
case relationship
|
|
when 'follow'
|
|
begin
|
|
FollowService.new.call(from_account, target_account, options)
|
|
rescue ActiveRecord::RecordInvalid
|
|
raise if FollowLimitValidator.limit_for_account(from_account) < from_account.following_count
|
|
end
|
|
when 'unfollow'
|
|
UnfollowService.new.call(from_account, target_account)
|
|
when 'block'
|
|
BlockService.new.call(from_account, target_account)
|
|
when 'unblock'
|
|
UnblockService.new.call(from_account, target_account)
|
|
when 'mute'
|
|
MuteService.new.call(from_account, target_account, options)
|
|
when 'unmute'
|
|
UnmuteService.new.call(from_account, target_account)
|
|
end
|
|
rescue ActiveRecord::RecordNotFound
|
|
true
|
|
end
|
|
|
|
def domain(uri)
|
|
domain = uri.is_a?(Account) ? uri.domain : uri.split('@')[1]
|
|
TagManager.instance.local_domain?(domain) ? nil : TagManager.instance.normalize_domain(domain)
|
|
end
|
|
|
|
def stoplight_wrap_request(domain, &block)
|
|
if domain.present?
|
|
Stoplight("source:#{domain}", &block)
|
|
.with_fallback { nil }
|
|
.with_threshold(1)
|
|
.with_cool_off_time(5.minutes.seconds)
|
|
.with_error_handler { |error, handle| error.is_a?(HTTP::Error) || error.is_a?(OpenSSL::SSL::SSLError) ? handle.call(error) : raise(error) }
|
|
.run
|
|
else
|
|
block.call
|
|
end
|
|
end
|
|
end
|