forked from treehouse/mastodon
95 lines
3.4 KiB
Ruby
95 lines
3.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'tty-prompt'
|
|
require_relative '../../config/boot'
|
|
require_relative '../../config/environment'
|
|
require_relative 'cli_helper'
|
|
|
|
module Mastodon
|
|
class PreviewCardsCLI < Thor
|
|
include ActionView::Helpers::NumberHelper
|
|
|
|
def self.exit_on_failure?
|
|
true
|
|
end
|
|
|
|
option :days, type: :numeric, default: 180
|
|
option :background, type: :boolean, default: false
|
|
option :verbose, type: :boolean, default: false
|
|
option :dry_run, type: :boolean, default: false
|
|
option :link, type: :boolean, default: false
|
|
desc 'remove', 'Remove preview cards'
|
|
long_desc <<-DESC
|
|
Removes locally thumbnails for previews.
|
|
|
|
The --days option specifies how old preview cards have to be before
|
|
they are removed. It defaults to 180 days.
|
|
|
|
With the --background option, instead of deleting the files sequentially,
|
|
they will be queued into Sidekiq and the command will exit as soon as
|
|
possible. In Sidekiq they will be processed with higher concurrency, but
|
|
it may impact other operations of the Mastodon server, and it may overload
|
|
the underlying file storage.
|
|
|
|
With the --dry-run option, no work will be done.
|
|
|
|
With the --verbose option, when preview cards are processed sequentially in the
|
|
foreground, the IDs of the preview cards will be printed.
|
|
|
|
With the --link option, delete only link-type preview cards.
|
|
DESC
|
|
def remove
|
|
prompt = TTY::Prompt.new
|
|
time_ago = options[:days].days.ago
|
|
queued = 0
|
|
processed = 0
|
|
size = 0
|
|
dry_run = options[:dry_run] ? '(DRY RUN)' : ''
|
|
link = options[:link] ? 'link-type ' : ''
|
|
scope = PreviewCard.where.not(image_file_name: nil)
|
|
scope = scope.where.not(image_file_name: '')
|
|
scope = scope.where(type: :link) if options[:link]
|
|
scope = scope.where('updated_at < ?', time_ago)
|
|
|
|
if time_ago > 2.weeks.ago
|
|
prompt.say "\n"
|
|
prompt.say('The preview cards less than the past two weeks will not be re-acquired even when needed.')
|
|
prompt.say "\n"
|
|
|
|
unless prompt.yes?('Are you sure you want to delete the preview cards?', default: false)
|
|
prompt.say "\n"
|
|
prompt.warn 'Nothing execute. Bye!'
|
|
prompt.say "\n"
|
|
exit(1)
|
|
end
|
|
end
|
|
|
|
if options[:background]
|
|
scope.select(:id, :image_file_size).reorder(nil).find_in_batches do |preview_cards|
|
|
queued += preview_cards.size
|
|
size += preview_cards.reduce(0) { |sum, p| sum + (p.image_file_size || 0) }
|
|
Maintenance::UncachePreviewWorker.push_bulk(preview_cards.map(&:id)) unless options[:dry_run]
|
|
end
|
|
|
|
else
|
|
scope.select(:id, :image_file_size).reorder(nil).find_in_batches do |preview_cards|
|
|
preview_cards.each do |p|
|
|
size += p.image_file_size || 0
|
|
Maintenance::UncachePreviewWorker.new.perform(p.id) unless options[:dry_run]
|
|
options[:verbose] ? say(p.id) : say('.', :green, false)
|
|
processed += 1
|
|
end
|
|
end
|
|
end
|
|
|
|
say
|
|
|
|
if options[:background]
|
|
say("Scheduled the deletion of #{queued} #{link}preview cards (approx. #{number_to_human_size(size)}) #{dry_run}", :green, true)
|
|
else
|
|
say("Removed #{processed} #{link}preview cards (approx. #{number_to_human_size(size)}) #{dry_run}", :green, true)
|
|
end
|
|
end
|
|
end
|
|
end
|