Retrospring/Rakefile

# frozen_string_literal: true

# Add your own tasks in files placed in lib/tasks ending in .rake,
# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.

require File.expand_path("config/application", __dir__)

Rails.application.load_tasks

namespace :justask do
  desc "Upload to AWS"
  task paperclaws: :environment do
    throw "Needs fog (AWS) to be defined in justask.yml" if APP_CONFIG["fog"]["credentials"].nil? || (APP_CONFIG["fog"]["credentials"]["provider"] != "AWS")

    format = "%t (%c/%C) [%b>%i] %e"
    root = Rails.root.join("/public/system")
    files = Dir["#{root}/**/*.*"]
    progress = ProgressBar.create(title: "Processing files", format:, starting_at: 0, total: files.length)

    # weird voodoo, something is causing just using "APP_CONFIG["fog"]["credentials"]" as Fog::Storage.new to cause an exception
    # TODO: Programmatically copy?
    credentials = {
      provider:              "AWS",
      aws_access_key_id:     APP_CONFIG["fog"]["credentials"]["aws_access_key_id"],
      aws_secret_access_key: APP_CONFIG["fog"]["credentials"]["aws_secret_access_key"],
      region:                APP_CONFIG["fog"]["credentials"]["region"]
    }

    fog = Fog::Storage.new credentials
    bucket = fog.directories.get APP_CONFIG["fog"]["directory"]

    files.each do |file|
      bucket.files.create key: file[root.length + 1...file.length], body: File.open(file), public: true
      progress.increment
    end

    puts "hopefully uploaded #{files.length} files"
  end

  desc "Recount everything!"
  task recount: :environment do
    format = "%t (%c/%C) [%b>%i] %e"
    total = User.count
    progress = ProgressBar.create(title: "Processing users", format:, starting_at: 0, total:)
    User.all.each do |user|
      begin
        user.answered_count = Answer.where(user:).count
        user.asked_count = Question.where(user:).where(author_is_anonymous: false).count
        user.commented_count = Comment.where(user:).count
        user.smiled_count = Appendable::Reaction.where(user:).count
        user.save!
      end
      progress.increment
    end

    total = Question.count
    progress = ProgressBar.create(title: "Processing questions", format:, starting_at: 0, total:)
    Question.all.each do |question|
      begin
        answers = Answer.where(question:).count
        question.answer_count = answers
        question.save!
      end
      progress.increment
    end

    total = Answer.count
    progress = ProgressBar.create(title: "Processing answers", format:, starting_at: 0, total:)
    Answer.all.each do |answer|
      begin
        answer.comment_count = Comment.where(answer:).count
        answer.smile_count = Appendable::Reaction.where(answer:).count
        answer.save!
      end
      progress.increment
    end
  end

  desc "Gives admin status to a user."
  task :admin, [:screen_name] => :environment do |_t, args|
    abort "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    abort "user #{args[:screen_name]} not found" if user.nil?

    user.add_role :administrator
    puts "#{user.screen_name} is now an administrator."
  end

  desc "Removes admin status from a user."
  task :deadmin, [:screen_name] => :environment do |_t, args|
    abort "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    abort "user #{args[:screen_name]} not found" if user.nil?

    user.remove_role :administrator
    puts "#{user.screen_name} is no longer an administrator."
  end

  desc "Gives moderator status to a user."
  task :mod, [:screen_name] => :environment do |_t, args|
    abort "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    abort "user #{args[:screen_name]} not found" if user.nil?

    user.add_role :moderator
    puts "#{user.screen_name} is now a moderator."
  end

  desc "Removes moderator status from a user."
  task :demod, [:screen_name] => :environment do |_t, args|
    abort "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    abort "user #{args[:screen_name]} not found" if user.nil?

    user.remove_role :moderator
    puts "#{user.screen_name} is no longer a moderator."
  end

  desc "Hits an user with the banhammer."
  task :permanently_ban, %i[screen_name reason] => :environment do |_t, args|
    raise "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    raise "user #{args[:screen_name]} not found" if user.nil?

    UseCase::User::Ban.call(
      target_user_id: user.id,
      expiry:         nil,
      reason:         args[:reason]
    )
    puts "#{user.screen_name} got hit by\033[5m YE OLDE BANHAMMER\033[0m!!1!"
  end

  desc "Hits an user with the banhammer for one day."
  task :ban, %i[screen_name reason] => :environment do |_t, args|
    raise "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    UseCase::User::Ban.call(
      target_user_id: user.id,
      expiry:         DateTime.current + 1,
      reason:         args[:reason]
    )
    puts "#{user.screen_name} got hit by\033[5m YE OLDE BANHAMMER\033[0m!!1!"
  end

  desc "Hits an user with the banhammer for one week."
  task :week_ban, %i[screen_name reason] => :environment do |_t, args|
    raise "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    UseCase::User::Ban.call(
      target_user_id: user.id,
      expiry:         DateTime.current + 7,
      reason:         args[:reason]
    )
    puts "#{user.screen_name} got hit by\033[5m YE OLDE BANHAMMER\033[0m!!1!"
  end

  desc "Hits an user with the banhammer for one month."
  task :month_ban, %i[screen_name reason] => :environment do |_t, args|
    raise "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    UseCase::User::Ban.call(
      target_user_id: user.id,
      expiry:         DateTime.current + 30,
      reason:         args[:reason]
    )
    puts "#{user.screen_name} got hit by\033[5m YE OLDE BANHAMMER\033[0m!!1!"
  end

  desc "Hits an user with the banhammer for one year."
  task :year_ban, %i[screen_name reason] => :environment do |_t, args|
    raise "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    UseCase::User::Ban.call(
      target_user_id: user.id,
      expiry:         DateTime.current + 365,
      reason:         args[:reason]
    )
    puts "#{user.screen_name} got hit by\033[5m YE OLDE BANHAMMER\033[0m!!1!"
  end

  desc "Hits an user with the banhammer for one aeon."
  task :aeon_ban, %i[screen_name reason] => :environment do |_t, args|
    raise "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    UseCase::User::Ban.call(
      target_user_id: user.id,
      expiry:         DateTime.current + 365_000_000_000,
      reason:         args[:reason]
    )
    puts "#{user.screen_name} got hit by\033[5m YE OLDE BANHAMMER\033[0m!!1!"
  end

  desc "Removes banned status from an user."
  task :unban, [:screen_name] => :environment do |_t, args|
    raise "screen name required" if args[:screen_name].nil?

    user = User.find_by(screen_name: args[:screen_name])
    raise "user #{args[:screen_name]} not found" if user.nil?

    UseCase::User::Unban.call(user.id)
    puts "#{user.screen_name} is no longer banned."
  end

  desc "Lists all users."
  task lusers: :environment do
    User.all.each do |u|
      puts "#{format '%3d', u.id}. #{u.screen_name}"
    end
  end

  desc "Removes users whose accounts haven't been verified for over 3 months."
  task remove_stale: :environment do
    puts "Removing stale users…"
    removed = User.where(confirmed_at: nil)
                  .where("confirmation_sent_at < ?", DateTime.now.utc - 3.months)
                  .destroy_all.count
    puts "Removed #{removed} users"
  end

  desc "Prints lonely people."
  task loners: :environment do
    people = {}
    Question.all.each do |q|
      next unless q.author_is_anonymous && %w[justask retrospring_exporter].exclude?(q.author_identifier)

      q.answers.each do |a|
        next unless q.user == a.user

        people[q.user.screen_name] ||= 0
        people[q.user.screen_name] += 1
        puts "#{q.user.screen_name} -- answer id #{a.id}"
      end
    end

    max = 0
    res = []
    people.each do |k, v|
      max = v if v > max
      res << k if v == max
    end
    if res.length.zero?
      puts "No one?  I hope you're just on the development session."
    else
      puts res.length == 1 ? "And the winner is..." : "And the winners are..."
      print "\033[5;31m"
      res.each { |name| puts " - #{name}" }
      print "\033[0m"
    end
  end
end

namespace :db do
  namespace :schema do
    task create_timestampid_function: :environment do
      conn = ActiveRecord::Base.connection
      have_func = conn.execute("SELECT EXISTS(SELECT * FROM pg_proc WHERE proname = 'gen_timestamp_id');").values.first.first
      next if have_func

      statement = File.read(File.join(__dir__, "db/migrate/20200704163504_use_timestamped_ids.rb")).match(/<~SQL(?<stmt>.+)SQL$/m)[:stmt]
      conn.execute(statement)
    end

    task create_id_sequences: :environment do
      conn = ActiveRecord::Base.connection

      # required for the timestampid function to work properly
      %i[questions answers comments users mute_rules].each do |tbl|
        conn.execute("CREATE SEQUENCE IF NOT EXISTS #{tbl}_id_seq")
      end
    end
  end

  # create timestampid before loading schema
  Rake::Task["db:schema:load"].enhance ["db:schema:create_timestampid_function"]
  Rake::Task["db:test:load_schema"].enhance ["db:schema:create_timestampid_function"]
  # create id_sequences after loading schema
  Rake::Task["db:schema:load"].enhance do
    Rake::Task["db:schema:create_id_sequences"].invoke
  end
  Rake::Task["db:test:load_schema"].enhance do
    Rake::Task["db:schema:create_id_sequences"].invoke
  end
end