discourse/lib/tasks/topics.rake

# frozen_string_literal: true

def print_status_with_label(label, current, max)
  print "\r%s%9d / %d (%5.1f%%)" % [label, current, max, ((current.to_f / max.to_f) * 100).round(1)]
end

def close_old_topics(category)
  topics = Topic.where(closed: false, category_id: category.id)

  if category.auto_close_based_on_last_post
    topics = topics.where('last_posted_at < ?', category.auto_close_hours.hours.ago)
  else
    topics = topics.where('created_at < ?', category.auto_close_hours.hours.ago)
  end

  topics_closed = 0
  total = topics.count

  if total == 0
    print "    all old topics are closed"
    return
  end

  topics.find_each do |topic|
    topic.update_status("closed", true, Discourse.system_user)
    print_status_with_label("    closing old topics: ", topics_closed += 1, total)
  end
end

def apply_auto_close(category)
  topics = Topic.where(closed: false, category_id: category.id)
    .where(<<-SQL, TopicTimer.types[:close])
      NOT EXISTS (
        SELECT 1
        FROM topic_timers
        WHERE topic_timers.topic_id = topics.id
          AND topic_timers.status_type = ?
          AND topic_timers.deleted_at IS NULL
      )
    SQL

  topics_closed = 0
  total = topics.count

  if total == 0
    print "    all topics have auto-close applied"
    return
  end

  topics.find_each do |topic|
    topic.inherit_auto_close_from_category
    print_status_with_label("    applying auto-close to topics: ", topics_closed += 1, total)
  end
end

task "topics:apply_autoclose" => :environment do
  categories = Category.where("auto_close_hours > 0")

  categories.find_each do |category|
    puts "", "Applying auto-close to category '#{category.name}' ..."
    close_old_topics(category)
    puts ""
    apply_auto_close(category)
    puts ""
  end

  puts "", "Done"
end

task "topics:watch_all_replied_topics" => :environment do
  puts "Setting all topics to Watching on which a user has posted at least once..."
  topics = Topic.where("archetype != ?", Archetype.private_message)
  total = topics.count
  count = 0

  topics.find_each do |t|
    t.topic_users.where(posted: true).find_each do |tp|
      tp.update!(notification_level: TopicUser.notification_levels[:watching], notifications_reason_id: TopicUser.notification_reasons[:created_post])
    end
    print_status(count += 1, total)
  end

  puts "", "Done"
end

task "topics:update_fancy_titles" => :environment do
  if !SiteSetting.title_fancy_entities?
    puts "fancy topic titles are disabled"
    return
  end

  DB.exec("UPDATE topics SET fancy_title = NULL")

  total = Topic.count
  count = 0

  Topic.find_each do |topic|
    topic.fancy_title
    print_status(count += 1, total)
  end

  puts "", "Done"
end

def print_status(current, max)
  print "\r%9d / %d (%5.1f%%)" % [current, max, ((current.to_f / max.to_f) * 100).round(1)]
end