525 lines
16 KiB
Ruby
525 lines
16 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'set'
|
|
require_relative '../../config/boot'
|
|
require_relative '../../config/environment'
|
|
require_relative 'cli_helper'
|
|
|
|
module GabSocial
|
|
class AccountsCLI < Thor
|
|
def self.exit_on_failure?
|
|
true
|
|
end
|
|
|
|
option :all, type: :boolean
|
|
desc 'rotate [USERNAME]', 'Generate and broadcast new keys'
|
|
long_desc <<-LONG_DESC
|
|
Generate and broadcast new RSA keys as part of security
|
|
maintenance.
|
|
|
|
With the --all option, all local accounts will be subject
|
|
to the rotation. Otherwise, and by default, only a single
|
|
account specified by the USERNAME argument will be
|
|
processed.
|
|
LONG_DESC
|
|
def rotate(username = nil)
|
|
if options[:all]
|
|
processed = 0
|
|
delay = 0
|
|
|
|
Account.local.without_suspended.find_in_batches do |accounts|
|
|
accounts.each do |account|
|
|
rotate_keys_for_account(account, delay)
|
|
processed += 1
|
|
say('.', :green, false)
|
|
end
|
|
|
|
delay += 5.minutes
|
|
end
|
|
|
|
say
|
|
say("OK, rotated keys for #{processed} accounts", :green)
|
|
elsif username.present?
|
|
rotate_keys_for_account(Account.find_local(username))
|
|
say('OK', :green)
|
|
else
|
|
say('No account(s) given', :red)
|
|
exit(1)
|
|
end
|
|
end
|
|
|
|
option :email, required: true
|
|
option :confirmed, type: :boolean
|
|
option :role, default: 'user'
|
|
option :reattach, type: :boolean
|
|
option :force, type: :boolean
|
|
desc 'create USERNAME', 'Create a new user'
|
|
long_desc <<-LONG_DESC
|
|
Create a new user account with a given USERNAME and an
|
|
e-mail address provided with --email.
|
|
|
|
With the --confirmed option, the confirmation e-mail will
|
|
be skipped and the account will be active straight away.
|
|
|
|
With the --role option one of "user", "admin" or "moderator"
|
|
can be supplied. Defaults to "user"
|
|
|
|
With the --reattach option, the new user will be reattached
|
|
to a given existing username of an old account. If the old
|
|
account is still in use by someone else, you can supply
|
|
the --force option to delete the old record and reattach the
|
|
username to the new account anyway.
|
|
LONG_DESC
|
|
def create(username)
|
|
account = Account.new(username: username)
|
|
password = SecureRandom.hex
|
|
user = User.new(email: options[:email], password: password, agreement: true, approved: true, admin: options[:role] == 'admin', moderator: options[:role] == 'moderator', confirmed_at: options[:confirmed] ? Time.now.utc : nil)
|
|
|
|
if options[:reattach]
|
|
account = Account.find_local(username) || Account.new(username: username)
|
|
|
|
if account.user.present? && !options[:force]
|
|
say('The chosen username is currently in use', :red)
|
|
say('Use --force to reattach it anyway and delete the other user')
|
|
return
|
|
elsif account.user.present?
|
|
account.user.destroy!
|
|
end
|
|
end
|
|
|
|
account.suspended_at = nil
|
|
user.account = account
|
|
|
|
if user.save
|
|
if options[:confirmed]
|
|
user.confirmed_at = nil
|
|
user.confirm!
|
|
end
|
|
|
|
say('OK', :green)
|
|
say("New password: #{password}")
|
|
else
|
|
user.errors.to_h.each do |key, error|
|
|
say('Failure/Error: ', :red)
|
|
say(key)
|
|
say(' ' + error, :red)
|
|
end
|
|
|
|
exit(1)
|
|
end
|
|
end
|
|
|
|
option :role
|
|
option :email
|
|
option :confirm, type: :boolean
|
|
option :enable, type: :boolean
|
|
option :disable, type: :boolean
|
|
option :disable_2fa, type: :boolean
|
|
option :approve, type: :boolean
|
|
desc 'modify USERNAME', 'Modify a user'
|
|
long_desc <<-LONG_DESC
|
|
Modify a user account.
|
|
|
|
With the --role option, update the user's role to one of "user",
|
|
"moderator" or "admin".
|
|
|
|
With the --email option, update the user's e-mail address. With
|
|
the --confirm option, mark the user's e-mail as confirmed.
|
|
|
|
With the --disable option, lock the user out of their account. The
|
|
--enable option is the opposite.
|
|
|
|
With the --approve option, the account will be approved, if it was
|
|
previously not due to not having open registrations.
|
|
|
|
With the --disable-2fa option, the two-factor authentication
|
|
requirement for the user can be removed.
|
|
LONG_DESC
|
|
def modify(username)
|
|
user = Account.find_local(username)&.user
|
|
|
|
if user.nil?
|
|
say('No user with such username', :red)
|
|
exit(1)
|
|
end
|
|
|
|
if options[:role]
|
|
user.admin = options[:role] == 'admin'
|
|
user.moderator = options[:role] == 'moderator'
|
|
end
|
|
|
|
user.email = options[:email] if options[:email]
|
|
user.disabled = false if options[:enable]
|
|
user.disabled = true if options[:disable]
|
|
user.approved = true if options[:approve]
|
|
user.otp_required_for_login = false if options[:disable_2fa]
|
|
user.confirm if options[:confirm]
|
|
|
|
if user.save
|
|
say('OK', :green)
|
|
else
|
|
user.errors.to_h.each do |key, error|
|
|
say('Failure/Error: ', :red)
|
|
say(key)
|
|
say(' ' + error, :red)
|
|
end
|
|
|
|
exit(1)
|
|
end
|
|
end
|
|
|
|
desc 'delete USERNAME', 'Delete a user'
|
|
long_desc <<-LONG_DESC
|
|
Suspend the account matching USERNAME and remove the associated user (can't be undone).
|
|
LONG_DESC
|
|
def delete(username)
|
|
account = Account.find_local(username)
|
|
|
|
if account.nil?
|
|
say('No user with such username', :red)
|
|
exit(1)
|
|
end
|
|
|
|
say("Suspending account, removing #{account.statuses_count} statuses and user...")
|
|
SuspendAccountService.new.call(account, including_user: true)
|
|
say('OK', :green)
|
|
end
|
|
|
|
desc 'backup USERNAME', 'Request a backup for a user'
|
|
long_desc <<-LONG_DESC
|
|
Request a new backup for an account with a given USERNAME.
|
|
|
|
The backup will be created in Sidekiq asynchronously, and
|
|
the user will receive an e-mail with a link to it once
|
|
it's done.
|
|
LONG_DESC
|
|
def backup(username)
|
|
account = Account.find_local(username)
|
|
|
|
if account.nil?
|
|
say('No user with such username', :red)
|
|
exit(1)
|
|
end
|
|
|
|
backup = account.user.backups.create!
|
|
BackupWorker.perform_async(backup.id)
|
|
say('OK', :green)
|
|
end
|
|
|
|
option :dry_run, type: :boolean
|
|
desc 'cull', 'Remove remote accounts that no longer exist'
|
|
long_desc <<-LONG_DESC
|
|
Query every single remote account in the database to determine
|
|
if it still exists on the origin server, and if it doesn't,
|
|
remove it from the database.
|
|
|
|
Accounts that have had confirmed activity within the last week
|
|
are excluded from the checks.
|
|
|
|
Domains that are unreachable are not checked.
|
|
|
|
With the --dry-run option, no deletes will actually be carried
|
|
out.
|
|
LONG_DESC
|
|
def cull
|
|
skip_threshold = 7.days.ago
|
|
culled = 0
|
|
dry_run_culled = []
|
|
skip_domains = Set.new
|
|
dry_run = options[:dry_run] ? ' (DRY RUN)' : ''
|
|
|
|
Account.remote.where(protocol: :activitypub).partitioned.find_each do |account|
|
|
next if account.updated_at >= skip_threshold || (account.last_webfingered_at.present? && account.last_webfingered_at >= skip_threshold)
|
|
|
|
code = 0
|
|
unless skip_domains.include?(account.domain)
|
|
begin
|
|
code = Request.new(:head, account.uri).perform(&:code)
|
|
rescue HTTP::ConnectionError
|
|
skip_domains << account.domain
|
|
rescue StandardError
|
|
next
|
|
end
|
|
end
|
|
|
|
if [404, 410].include?(code)
|
|
if options[:dry_run]
|
|
dry_run_culled << account.acct
|
|
else
|
|
SuspendAccountService.new.call(account, destroy: true)
|
|
end
|
|
culled += 1
|
|
say('+', :green, false)
|
|
else
|
|
account.touch # Touch account even during dry run to avoid getting the account into the window again
|
|
say('.', nil, false)
|
|
end
|
|
end
|
|
|
|
say
|
|
say("Removed #{culled} accounts. #{skip_domains.size} servers skipped#{dry_run}", skip_domains.empty? ? :green : :yellow)
|
|
|
|
unless skip_domains.empty?
|
|
say('The following servers were not available during the check:', :yellow)
|
|
skip_domains.each { |domain| say(' ' + domain) }
|
|
end
|
|
|
|
unless dry_run_culled.empty?
|
|
say('The following accounts would have been deleted:', :green)
|
|
dry_run_culled.each { |account| say(' ' + account) }
|
|
end
|
|
end
|
|
|
|
option :all, type: :boolean
|
|
option :domain
|
|
desc 'refresh [USERNAME]', 'Fetch remote user data and files'
|
|
long_desc <<-LONG_DESC
|
|
Fetch remote user data and files for one or multiple accounts.
|
|
|
|
With the --all option, all remote accounts will be processed.
|
|
Through the --domain option, this can be narrowed down to a
|
|
specific domain only. Otherwise, a single remote account must
|
|
be specified with USERNAME.
|
|
|
|
All processing is done in the background through Sidekiq.
|
|
LONG_DESC
|
|
def refresh(username = nil)
|
|
if options[:domain] || options[:all]
|
|
queued = 0
|
|
scope = Account.remote
|
|
scope = scope.where(domain: options[:domain]) if options[:domain]
|
|
|
|
scope.select(:id).reorder(nil).find_in_batches do |accounts|
|
|
Maintenance::RedownloadAccountMediaWorker.push_bulk(accounts.map(&:id))
|
|
queued += accounts.size
|
|
end
|
|
|
|
say("Scheduled refreshment of #{queued} accounts", :green, true)
|
|
elsif username.present?
|
|
username, domain = username.split('@')
|
|
account = Account.find_local(username)
|
|
|
|
if account.nil?
|
|
say('No such account', :red)
|
|
exit(1)
|
|
end
|
|
|
|
Maintenance::RedownloadAccountMediaWorker.perform_async(account.id)
|
|
say('OK', :green)
|
|
else
|
|
say('No account(s) given', :red)
|
|
exit(1)
|
|
end
|
|
end
|
|
|
|
desc 'follow ACCT', 'Make all local accounts follow account specified by ACCT'
|
|
long_desc <<-LONG_DESC
|
|
Make all local accounts follow an account specified by ACCT. ACCT can be
|
|
a simple username, in case of a local user. It can also be in the format
|
|
username@domain, in case of a remote user.
|
|
LONG_DESC
|
|
def follow(acct)
|
|
target_account = ResolveAccountService.new.call(acct)
|
|
processed = 0
|
|
failed = 0
|
|
|
|
if target_account.nil?
|
|
say("Target account (#{acct}) could not be resolved", :red)
|
|
exit(1)
|
|
end
|
|
|
|
Account.local.without_suspended.find_each do |account|
|
|
begin
|
|
FollowService.new.call(account, target_account)
|
|
processed += 1
|
|
say('.', :green, false)
|
|
rescue StandardError
|
|
failed += 1
|
|
say('.', :red, false)
|
|
end
|
|
end
|
|
|
|
say("OK, followed target from #{processed} accounts, skipped #{failed}", :green)
|
|
end
|
|
|
|
option :number, type: :numeric, aliases: [:n]
|
|
desc 'unfollow ACCT', 'Make all local accounts unfollow account specified by ACCT'
|
|
long_desc <<-LONG_DESC
|
|
Make all local accounts unfollow an account specified by ACCT. ACCT can be
|
|
a simple username, in case of a local user. It can also be in the format
|
|
username@domain, in case of a remote user.
|
|
LONG_DESC
|
|
def unfollow(acct)
|
|
target_account = nil
|
|
followers = nil
|
|
|
|
if options[:number]
|
|
target_account = Account.find(acct)
|
|
followers = target_account.followers.local.limit(options[:number])
|
|
else
|
|
say('Error. Specify a limit', :red)
|
|
exit(1)
|
|
end
|
|
|
|
processed = 0
|
|
failed = 0
|
|
|
|
if target_account.nil?
|
|
say("Target account (#{acct}) was not found", :red)
|
|
exit(1)
|
|
end
|
|
|
|
followers.find_each do |account|
|
|
begin
|
|
UnfollowService.new.call(account, target_account)
|
|
processed += 1
|
|
say('.', :green, false)
|
|
rescue StandardError
|
|
failed += 1
|
|
say('.', :red, false)
|
|
end
|
|
end
|
|
|
|
say("OK, unfollowed target from #{processed} accounts, skipped #{failed}", :green)
|
|
end
|
|
|
|
option :follows, type: :boolean, default: false
|
|
option :followers, type: :boolean, default: false
|
|
desc 'reset-relationships USERNAME', 'Reset all follows and/or followers for a user'
|
|
long_desc <<-LONG_DESC
|
|
Reset all follows and/or followers for a user specified by USERNAME.
|
|
|
|
With the --follows option, the command unfollows everyone that the account follows,
|
|
and then re-follows the users that would be followed by a brand new account.
|
|
|
|
With the --followers option, the command removes all followers of the account.
|
|
LONG_DESC
|
|
def reset_relationships(username)
|
|
unless options[:follows] || options[:followers]
|
|
say('Please specify either --follows or --followers, or both', :red)
|
|
exit(1)
|
|
end
|
|
|
|
account = Account.find_local(username)
|
|
|
|
if account.nil?
|
|
say('No user with such username', :red)
|
|
exit(1)
|
|
end
|
|
|
|
if options[:follows]
|
|
processed = 0
|
|
failed = 0
|
|
|
|
say("Unfollowing #{account.username}'s followees, this might take a while...")
|
|
|
|
Account.where(id: ::Follow.where(account: account).select(:target_account_id)).find_each do |target_account|
|
|
begin
|
|
UnfollowService.new.call(account, target_account)
|
|
processed += 1
|
|
say('.', :green, false)
|
|
rescue StandardError
|
|
failed += 1
|
|
say('.', :red, false)
|
|
end
|
|
end
|
|
|
|
BootstrapTimelineWorker.perform_async(account.id)
|
|
|
|
say("OK, unfollowed #{processed} followees, skipped #{failed}", :green)
|
|
end
|
|
|
|
if options[:followers]
|
|
processed = 0
|
|
failed = 0
|
|
|
|
say("Removing #{account.username}'s followers, this might take a while...")
|
|
|
|
Account.where(id: ::Follow.where(target_account: account).select(:account_id)).find_each do |target_account|
|
|
begin
|
|
UnfollowService.new.call(target_account, account)
|
|
processed += 1
|
|
say('.', :green, false)
|
|
rescue StandardError
|
|
failed += 1
|
|
say('.', :red, false)
|
|
end
|
|
end
|
|
|
|
say("OK, removed #{processed} followers, skipped #{failed}", :green)
|
|
end
|
|
end
|
|
|
|
desc 'suspend USERNAME', 'Suspend a user'
|
|
long_desc <<-LONG_DESC
|
|
Suspend the user account with the given USERNAME (can be undone).
|
|
LONG_DESC
|
|
def suspend(username)
|
|
account = Account.find_local(username)
|
|
|
|
if account.nil?
|
|
say('No user with such username', :red)
|
|
exit(1)
|
|
end
|
|
|
|
say("Suspending user and removing #{account.statuses_count} statuses...")
|
|
SuspendAccountService.new.call(account, including_user: false)
|
|
say('OK', :green)
|
|
end
|
|
|
|
option :number, type: :numeric, aliases: [:n]
|
|
option :all, type: :boolean
|
|
desc 'approve [USERNAME]', 'Approve pending accounts'
|
|
long_desc <<~LONG_DESC
|
|
When registrations require review from staff, approve pending accounts,
|
|
either all of them with the --all option, or a specific number of them
|
|
specified with the --number (-n) option, or only a single specific
|
|
account identified by its username.
|
|
LONG_DESC
|
|
def approve(username = nil)
|
|
if options[:all]
|
|
User.pending.find_each(&:approve!)
|
|
say('OK', :green)
|
|
elsif options[:number]
|
|
User.pending.limit(options[:number]).each(&:approve!)
|
|
say('OK', :green)
|
|
elsif username.present?
|
|
account = Account.find_local(username)
|
|
|
|
if account.nil?
|
|
say('No such account', :red)
|
|
exit(1)
|
|
end
|
|
|
|
account.user&.approve!
|
|
say('OK', :green)
|
|
else
|
|
exit(1)
|
|
end
|
|
end
|
|
|
|
desc 'update_account_stats ACCT', 'Update account stats.'
|
|
long_desc <<-LONG_DESC
|
|
Update account stats (follow count, status count, following count).
|
|
LONG_DESC
|
|
def update_account_stats(acct)
|
|
target_account = Account.find(acct)
|
|
if target_account.nil?
|
|
say("Target account (#{acct}) was not found", :red)
|
|
exit(1)
|
|
end
|
|
|
|
AccountStat.where(account: target_account).update(
|
|
followers_count: target_account.local_followers_count,
|
|
following_count: target_account.local_following_count,
|
|
statuses_count: target_account.statuses.count,
|
|
)
|
|
|
|
say('OK', :green)
|
|
end
|
|
|
|
private
|
|
|
|
end
|
|
end
|