redux-scraper/Rakefile

# 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 "rake/testtask"
require_relative "config/application"

Rails.application.load_tasks
Dir.glob(Rails.root.join("rake", "*.rake")).each { |rake_file| load rake_file }

task set_ar_stdout: :environment do
  ActiveRecord::Base.logger = Logger.new($stdout)
end

task set_logger_stdout: :environment do
  Rails.logger = Logger.new($stdout)
  Rails.logger.formatter =
    proc { |severity, datetime, progname, msg| "#{severity}: #{msg}\n" }
  ActiveRecord::Base.logger = nil
  ActiveJob::Base.logger = nil
  GoodJob.logger = Rails.logger
end

task periodic_tasks: %i[environment set_logger_stdout] do
  Thread.new do
    loop do
      Rake::Task["pghero:capture_space_stats"].execute
      puts "logged space stats"
      sleep 6.hours
    end
  end

  Thread.new do
    loop do
      Rake::Task["pghero:capture_query_stats"].execute
      puts "logged query stats"
      sleep 5.minutes
    end
  end

  loop { sleep 10 }
end

namespace :db_sampler do
  task export: :environment do
    url_names = ENV["url_names"] || raise("need 'url_names' (comma-separated)")
    outfile = $stdout
    DbSampler.new(outfile).export(url_names.split(","))
  ensure
    outfile.close if outfile
  end

  task import: [:environment] do
    infile = $stdin
    DbSampler.new(infile).import
  ensure
    infile.close if infile
  end
end

task good_job: %i[environment set_ar_stdout set_logger_stdout] do
  env_hash = {
    "RAILS_ENV" => "worker",
    "GOOD_JOB_POLL_INTERVAL" => "5",
    "GOOD_JOB_MAX_CACHE" => "10000",
    "GOOD_JOB_QUEUE_SELECT_LIMIT" => "4096",
    "GOOD_JOB_MAX_THREADS" => "4",
    "GOOD_JOB_ENABLE_CRON" => "1",
    "GOOD_JOB_QUEUES" =>
      ENV["GOOD_JOB_QUEUES"] ||
        %w[manual:4 fa_post,e621:2 *:6].reject(&:nil?).join(";"),
  }

  env_hash.each do |key, value|
    ENV[key] = value
    puts "$> #{key.light_black.bold} = #{value.bold}"
  end

  cmd = "bundle exec good_job"
  puts "$> #{cmd.bold}"
  exec(cmd)
end

task :reverse_csv do
  file = ENV["file"] || raise("need 'file' (file path)")
  in_csv = CSV.parse(File.open(file, "r+"), headers: true)
  out_csv =
    CSV.new(
      File.open("rev_" + file, "w"),
      write_headers: true,
      headers: in_csv.headers,
    )
  in_csv.reverse_each { |row| out_csv << row.map(&:second) }
  out_csv.close
end

task migrate_to_domain: :environment do
  only_user = ENV["only_user"]
  allowed_domains = %w[e621 fa ib]
  only_domains = (ENV["only_domains"] || "").split(",")
  only_domains = allowed_domains if only_domains.empty?
  if (only_domains - allowed_domains).any?
    raise "only_domains must be a subset of #{allowed_domains.join(", ")}"
  end

  migrator = Domain::MigrateToDomain.new

  if only_domains.include?("e621")
    migrator.migrate_e621_users(only_user: only_user)
    migrator.migrate_e621_posts(only_user: only_user)
    migrator.migrate_e621_users_favs(only_user: only_user)
  end

  if only_domains.include?("fa")
    migrator.migrate_fa_users(only_user: only_user)
    migrator.migrate_fa_posts(only_user: only_user)
    migrator.migrate_fa_users_favs(only_user: only_user)
    migrator.migrate_fa_users_followed_users(only_user: only_user)
  end

  if only_domains.include?("ib")
    migrator.migrate_inkbunny_users(only_user: only_user)
    migrator.migrate_inkbunny_posts(only_user: only_user)
    migrator.migrate_inkbunny_pools(only_user: nil) if only_user.nil?
  end
end

task infer_last_submission_log_entries: :environment do
  only_fa_id = ENV["only_fa_id"]
  start = ENV["start_at"]&.to_i || nil

  if only_fa_id
    relation = Domain::Fa::Post.where(fa_id: only_fa_id)
  else
    relation =
      Domain::Fa::Post
        .where(state: :ok)
        .where(last_submission_page_id: nil)
        .or(Domain::Fa::Post.where(state: :ok).where(posted_at: nil))
  end

  relation.find_each(batch_size: 10, start:) do |post|
    parts = ["[id: #{post.id}]", "[fa_id: #{post.fa_id}]"]

    log_entry = post.guess_last_submission_page
    unless log_entry
      parts << "[no log entry]"
      next
    end

    contents = log_entry.response&.contents
    unless contents
      parts << "[no contents]"
      next
    end

    parser = Domain::Fa::Parser::Page.new(contents)
    if parser.submission_not_found?
      parts << "[removed]"
      post.state = :removed
    else
      posted_at = parser.submission.posted_date
      post.posted_at ||= posted_at
      parts << "[posted at: #{posted_at}]"
    end

    if post.last_submission_page_id.present? &&
         log_entry.id != post.last_submission_page_id
      parts << "[overwrite]"
    end
    post.last_submission_page_id = log_entry.id

    parts << "[log entry: #{log_entry.id}]"
    parts << "[uri: #{log_entry.uri.to_s}]"
    post.save!
  rescue => e
    parts << "[error: #{e.message}]"
  ensure
    puts parts.join(" ")
  end
end