mirror of
https://github.com/mastodon/mastodon.git
synced 2024-11-24 13:35:07 +01:00
61 lines
2.2 KiB
Ruby
61 lines
2.2 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class BulkImportRowService
|
||
|
|
def call(row)
|
||
|
|
@account = row.bulk_import.account
|
||
|
|
@data = row.data
|
||
|
|
@type = row.bulk_import.type.to_sym
|
||
|
|
|
||
|
|
case @type
|
||
|
|
when :following, :blocking, :muting
|
||
|
|
target_acct = @data['acct']
|
||
|
|
target_domain = domain(target_acct)
|
||
|
|
@target_account = stoplight_wrap_request(target_domain) { ResolveAccountService.new.call(target_acct, { check_delivery_availability: true }) }
|
||
|
|
return false if @target_account.nil?
|
||
|
|
when :bookmarks
|
||
|
|
target_uri = @data['uri']
|
||
|
|
target_domain = Addressable::URI.parse(target_uri).normalized_host
|
||
|
|
@target_status = ActivityPub::TagManager.instance.uri_to_resource(target_uri, Status)
|
||
|
|
return false if @target_status.nil? && ActivityPub::TagManager.instance.local_uri?(target_uri)
|
||
|
|
|
||
|
|
@target_status ||= stoplight_wrap_request(target_domain) { ActivityPub::FetchRemoteStatusService.new.call(target_uri) }
|
||
|
|
return false if @target_status.nil?
|
||
|
|
end
|
||
|
|
|
||
|
|
case @type
|
||
|
|
when :following
|
||
|
|
FollowService.new.call(@account, @target_account, reblogs: @data['show_reblogs'], notify: @data['notify'], languages: @data['languages'])
|
||
|
|
when :blocking
|
||
|
|
BlockService.new.call(@account, @target_account)
|
||
|
|
when :muting
|
||
|
|
MuteService.new.call(@account, @target_account, notifications: @data['hide_notifications'])
|
||
|
|
when :bookmarks
|
||
|
|
return false unless StatusPolicy.new(@account, @target_status).show?
|
||
|
|
|
||
|
|
@account.bookmarks.find_or_create_by!(status: @target_status)
|
||
|
|
end
|
||
|
|
|
||
|
|
true
|
||
|
|
rescue ActiveRecord::RecordNotFound
|
||
|
|
false
|
||
|
|
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
|
||
|
|
yield
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|