mastodon/lib/mastodon/preview_cards_cli.rb

# 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