discourse/app/jobs/onceoff/migrate_badge_image_to_uploads.rb

# frozen_string_literal: true
require 'uri'

module Jobs
  class MigrateBadgeImageToUploads < ::Jobs::Onceoff
    def execute_onceoff(args)
      column_exists = DB.exec(<<~SQL) == 1
        SELECT 1
        FROM INFORMATION_SCHEMA.COLUMNS
        WHERE
          table_schema = 'public' AND
          table_name = 'badges' AND
          column_name = 'image_upload_id'
      SQL
      return unless column_exists

      Badge.where.not(image: nil).select(:id, :image_upload_id, :image).each do |badge|
        if badge.image_upload.present?
          DB.exec("UPDATE badges SET image = NULL WHERE id = ?", badge.id)
          next
        end

        image_url = badge[:image]
        next if image_url.blank? || image_url !~ URI.regexp

        count = 0
        file = nil
        sleep_interval = 5

        loop do
          url = UrlHelper.absolute_without_cdn(image_url)

          begin
            file = FileHelper.download(
              url,
              max_file_size: [
                SiteSetting.max_image_size_kb.kilobytes,
                20.megabytes
              ].max,
              tmp_file_name: 'tmp_badge_image_upload',
              skip_rate_limit: true,
              follow_redirect: true
            )
          rescue OpenURI::HTTPError,
                 OpenSSL::SSL::SSLError,
                 Net::OpenTimeout,
                 Net::ReadTimeout,
                 Errno::ECONNREFUSED,
                 EOFError,
                 SocketError,
                 Discourse::InvalidParameters => e

            logger.error(
              "Error encountered when trying to download from URL '#{image_url}' " +
              "for badge '#{badge[:id]}'.\n#{e.class}: #{e.message}\n#{e.backtrace.join("\n")}"
            )
          end

          count += 1
          break if file

          logger.warn(
            "Failed to download image from #{url} for badge '#{badge[:id]}'. Retrying (#{count}/3)..."
          )
          break if count >= 3
          sleep(count * sleep_interval)
        end

        next if file.blank?

        upload = UploadCreator.new(
          file,
          "image_for_badge_#{badge[:id]}",
          origin: UrlHelper.absolute(image_url)
        ).create_for(Discourse.system_user.id)

        if upload.errors.count > 0 || upload&.id.blank?
          logger.error("Failed to create an upload for the image of badge '#{badge[:id]}'. Error: #{upload.errors.full_messages}")
        else
          DB.exec("UPDATE badges SET image = NULL, image_upload_id = ? WHERE id = ?", upload.id, badge[:id])
        end
      end
    end

    private

    def logger
      Rails.logger
    end
  end
end