3a4d994c40
* Add test to handle suspended and missing users in BootstrapTimelineService * Fix BootstrapTimelineService crashing when bootstrapped accounts are invalid
52 lines
1.5 KiB
Ruby
52 lines
1.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class BootstrapTimelineService < BaseService
|
|
def call(source_account)
|
|
@source_account = source_account
|
|
|
|
autofollow_inviter!
|
|
autofollow_bootstrap_timeline_accounts!
|
|
end
|
|
|
|
private
|
|
|
|
def autofollow_inviter!
|
|
return unless @source_account&.user&.invite&.autofollow?
|
|
FollowService.new.call(@source_account, @source_account.user.invite.user.account)
|
|
end
|
|
|
|
def autofollow_bootstrap_timeline_accounts!
|
|
bootstrap_timeline_accounts.each do |target_account|
|
|
begin
|
|
FollowService.new.call(@source_account, target_account)
|
|
rescue ActiveRecord::RecordNotFound, Mastodon::NotPermittedError
|
|
nil
|
|
end
|
|
end
|
|
end
|
|
|
|
def bootstrap_timeline_accounts
|
|
return @bootstrap_timeline_accounts if defined?(@bootstrap_timeline_accounts)
|
|
|
|
@bootstrap_timeline_accounts = bootstrap_timeline_accounts_usernames.empty? ? admin_accounts : local_unlocked_accounts(bootstrap_timeline_accounts_usernames)
|
|
end
|
|
|
|
def bootstrap_timeline_accounts_usernames
|
|
@bootstrap_timeline_accounts_usernames ||= (Setting.bootstrap_timeline_accounts || '').split(',').map { |str| str.strip.gsub(/\A@/, '') }.reject(&:blank?)
|
|
end
|
|
|
|
def admin_accounts
|
|
User.admins
|
|
.includes(:account)
|
|
.where(accounts: { locked: false })
|
|
.map(&:account)
|
|
end
|
|
|
|
def local_unlocked_accounts(usernames)
|
|
Account.local
|
|
.without_suspended
|
|
.where(username: usernames)
|
|
.where(locked: false)
|
|
.where(moved_to_account_id: nil)
|
|
end
|
|
end
|