feeds_cli.rb 1.78 KB
Newer Older
1 2 3 4 5 6 7 8
# frozen_string_literal: true

require_relative '../../config/boot'
require_relative '../../config/environment'
require_relative 'cli_helper'

module Mastodon
  class FeedsCLI < Thor
9 10
    include CLIHelper

11 12 13
    def self.exit_on_failure?
      true
    end
14

15
    option :all, type: :boolean, default: false
16 17
    option :concurrency, type: :numeric, default: 5, aliases: [:c]
    option :verbose, type: :boolean, aliases: [:v]
18 19 20 21 22 23 24 25 26 27 28 29
    option :dry_run, type: :boolean, default: false
    desc 'build [USERNAME]', 'Build home and list feeds for one or all users'
    long_desc <<-LONG_DESC
      Build home and list feeds that are stored in Redis from the database.

      With the --all option, all active users will be processed.
      Otherwise, a single user specified by USERNAME.
    LONG_DESC
    def build(username = nil)
      dry_run = options[:dry_run] ? '(DRY RUN)' : ''

      if options[:all] || username.nil?
30 31
        processed, = parallelize_with_progress(Account.joins(:user).merge(User.active)) do |account|
          PrecomputeFeedService.new.call(account) unless options[:dry_run]
32 33
        end

34
        say("Regenerated feeds for #{processed} accounts #{dry_run}", :green, true)
35 36 37
      elsif username.present?
        account = Account.find_local(username)

38
        if account.nil?
39
          say('No such account', :red)
40 41 42
          exit(1)
        end

43
        PrecomputeFeedService.new.call(account) unless options[:dry_run]
44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63

        say("OK #{dry_run}", :green, true)
      else
        say('No account(s) given', :red)
        exit(1)
      end
    end

    desc 'clear', 'Remove all home and list feeds from Redis'
    def clear
      keys = Redis.current.keys('feed:*')

      Redis.current.pipelined do
        keys.each { |key| Redis.current.del(key) }
      end

      say('OK', :green)
    end
  end
end