Move self destruct CLI command definition code to module (#28131)
This commit is contained in:
Matt Jankowski
GitHub
parent
6e55ff9643
commit
b696ca6b1a
74
lib/mastodon/cli/federation.rb
Normal file
74
lib/mastodon/cli/federation.rb
Normal file
@ -0,0 +1,74 @@
|
|||||||
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
require 'tty-prompt'
|
||||||
|
|
||||||
|
module Mastodon::CLI
|
||||||
|
module Federation
|
||||||
|
extend ActiveSupport::Concern
|
||||||
|
|
||||||
|
included do
|
||||||
|
desc 'self-destruct', 'Erase the server from the federation'
|
||||||
|
long_desc <<~LONG_DESC
|
||||||
|
Erase the server from the federation by broadcasting account delete
|
||||||
|
activities to all known other servers. This allows a "clean exit" from
|
||||||
|
running a Mastodon server, as it leaves next to no cache behind on
|
||||||
|
other servers.
|
||||||
|
|
||||||
|
This command is always interactive and requires confirmation twice.
|
||||||
|
|
||||||
|
No local data is actually deleted, because emptying the
|
||||||
|
database or removing files is much faster through other, external
|
||||||
|
means, such as e.g. deleting the entire VPS. However, because other
|
||||||
|
servers will delete data about local users, but no local data will be
|
||||||
|
updated (such as e.g. followers), there will be a state mismatch
|
||||||
|
that will lead to glitches and issues if you then continue to run and use
|
||||||
|
the server.
|
||||||
|
|
||||||
|
So either you know exactly what you are doing, or you are starting
|
||||||
|
from a blank slate afterwards by manually clearing out all the local
|
||||||
|
data!
|
||||||
|
LONG_DESC
|
||||||
|
def self_destruct
|
||||||
|
if SelfDestructHelper.self_destruct?
|
||||||
|
prompt.ok('Self-destruct mode is already enabled for this Mastodon server')
|
||||||
|
|
||||||
|
pending_accounts = Account.local.without_suspended.count + Account.local.suspended.joins(:deletion_request).count
|
||||||
|
sidekiq_stats = Sidekiq::Stats.new
|
||||||
|
|
||||||
|
if pending_accounts.positive?
|
||||||
|
prompt.warn("#{pending_accounts} accounts are still pending deletion.")
|
||||||
|
elsif sidekiq_stats.enqueued.positive?
|
||||||
|
prompt.warn('Deletion notices are still being processed')
|
||||||
|
elsif sidekiq_stats.retry_size.positive?
|
||||||
|
prompt.warn('At least one delivery attempt for each deletion notice has been made, but some have failed and are scheduled for retry')
|
||||||
|
else
|
||||||
|
prompt.ok('Every deletion notice has been sent! You can safely delete all data and decomission your servers!')
|
||||||
|
end
|
||||||
|
|
||||||
|
exit(0)
|
||||||
|
end
|
||||||
|
|
||||||
|
exit(1) unless prompt.ask('Type in the domain of the server to confirm:', required: true) == Rails.configuration.x.local_domain
|
||||||
|
|
||||||
|
prompt.warn('This operation WILL NOT be reversible.')
|
||||||
|
prompt.warn('While the data won\'t be erased locally, the server will be in a BROKEN STATE afterwards.')
|
||||||
|
prompt.warn('The deletion process itself may take a long time, and will be handled by Sidekiq, so do not shut it down until it has finished (you will be able to re-run this command to see the state of the self-destruct process).')
|
||||||
|
|
||||||
|
exit(1) if prompt.no?('Are you sure you want to proceed?')
|
||||||
|
|
||||||
|
self_destruct_value = Rails.application.message_verifier('self-destruct').generate(Rails.configuration.x.local_domain)
|
||||||
|
prompt.ok('To switch Mastodon to self-destruct mode, add the following variable to your evironment (e.g. by adding a line to your `.env.production`) and restart all Mastodon processes:')
|
||||||
|
prompt.ok(" SELF_DESTRUCT=#{self_destruct_value}")
|
||||||
|
prompt.ok("\nYou can re-run this command to see the state of the self-destruct process.")
|
||||||
|
rescue TTY::Reader::InputInterrupt
|
||||||
|
exit(1)
|
||||||
|
end
|
||||||
|
|
||||||
|
private
|
||||||
|
|
||||||
|
def prompt
|
||||||
|
@prompt ||= TTY::Prompt.new
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
@ -8,6 +8,7 @@ require_relative 'canonical_email_blocks'
|
|||||||
require_relative 'domains'
|
require_relative 'domains'
|
||||||
require_relative 'email_domain_blocks'
|
require_relative 'email_domain_blocks'
|
||||||
require_relative 'emoji'
|
require_relative 'emoji'
|
||||||
|
require_relative 'federation'
|
||||||
require_relative 'feeds'
|
require_relative 'feeds'
|
||||||
require_relative 'ip_blocks'
|
require_relative 'ip_blocks'
|
||||||
require_relative 'maintenance'
|
require_relative 'maintenance'
|
||||||
@ -65,66 +66,7 @@ module Mastodon::CLI
|
|||||||
desc 'maintenance SUBCOMMAND ...ARGS', 'Various maintenance utilities'
|
desc 'maintenance SUBCOMMAND ...ARGS', 'Various maintenance utilities'
|
||||||
subcommand 'maintenance', Maintenance
|
subcommand 'maintenance', Maintenance
|
||||||
|
|
||||||
desc 'self-destruct', 'Erase the server from the federation'
|
include Federation
|
||||||
long_desc <<~LONG_DESC
|
|
||||||
Erase the server from the federation by broadcasting account delete
|
|
||||||
activities to all known other servers. This allows a "clean exit" from
|
|
||||||
running a Mastodon server, as it leaves next to no cache behind on
|
|
||||||
other servers.
|
|
||||||
|
|
||||||
This command is always interactive and requires confirmation twice.
|
|
||||||
|
|
||||||
No local data is actually deleted, because emptying the
|
|
||||||
database or removing files is much faster through other, external
|
|
||||||
means, such as e.g. deleting the entire VPS. However, because other
|
|
||||||
servers will delete data about local users, but no local data will be
|
|
||||||
updated (such as e.g. followers), there will be a state mismatch
|
|
||||||
that will lead to glitches and issues if you then continue to run and use
|
|
||||||
the server.
|
|
||||||
|
|
||||||
So either you know exactly what you are doing, or you are starting
|
|
||||||
from a blank slate afterwards by manually clearing out all the local
|
|
||||||
data!
|
|
||||||
LONG_DESC
|
|
||||||
def self_destruct
|
|
||||||
require 'tty-prompt'
|
|
||||||
|
|
||||||
prompt = TTY::Prompt.new
|
|
||||||
|
|
||||||
if SelfDestructHelper.self_destruct?
|
|
||||||
prompt.ok('Self-destruct mode is already enabled for this Mastodon server')
|
|
||||||
|
|
||||||
pending_accounts = Account.local.without_suspended.count + Account.local.suspended.joins(:deletion_request).count
|
|
||||||
sidekiq_stats = Sidekiq::Stats.new
|
|
||||||
|
|
||||||
if pending_accounts.positive?
|
|
||||||
prompt.warn("#{pending_accounts} accounts are still pending deletion.")
|
|
||||||
elsif sidekiq_stats.enqueued.positive?
|
|
||||||
prompt.warn('Deletion notices are still being processed')
|
|
||||||
elsif sidekiq_stats.retry_size.positive?
|
|
||||||
prompt.warn('At least one delivery attempt for each deletion notice has been made, but some have failed and are scheduled for retry')
|
|
||||||
else
|
|
||||||
prompt.ok('Every deletion notice has been sent! You can safely delete all data and decomission your servers!')
|
|
||||||
end
|
|
||||||
|
|
||||||
exit(0)
|
|
||||||
end
|
|
||||||
|
|
||||||
exit(1) unless prompt.ask('Type in the domain of the server to confirm:', required: true) == Rails.configuration.x.local_domain
|
|
||||||
|
|
||||||
prompt.warn('This operation WILL NOT be reversible.')
|
|
||||||
prompt.warn('While the data won\'t be erased locally, the server will be in a BROKEN STATE afterwards.')
|
|
||||||
prompt.warn('The deletion process itself may take a long time, and will be handled by Sidekiq, so do not shut it down until it has finished (you will be able to re-run this command to see the state of the self-destruct process).')
|
|
||||||
|
|
||||||
exit(1) if prompt.no?('Are you sure you want to proceed?')
|
|
||||||
|
|
||||||
self_destruct_value = Rails.application.message_verifier('self-destruct').generate(Rails.configuration.x.local_domain)
|
|
||||||
prompt.ok('To switch Mastodon to self-destruct mode, add the following variable to your evironment (e.g. by adding a line to your `.env.production`) and restart all Mastodon processes:')
|
|
||||||
prompt.ok(" SELF_DESTRUCT=#{self_destruct_value}")
|
|
||||||
prompt.ok("\nYou can re-run this command to see the state of the self-destruct process.")
|
|
||||||
rescue TTY::Reader::InputInterrupt
|
|
||||||
exit(1)
|
|
||||||
end
|
|
||||||
|
|
||||||
map %w(--version -v) => :version
|
map %w(--version -v) => :version
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user