require 'logger'
require 'resolv-replace'

desc "GitLab | Migrate files for artifacts to comply with new storage format"
namespace :gitlab do
  namespace :artifacts do
    task migrate: :environment do
      logger = Logger.new(STDOUT)
      logger.info('Starting transfer of artifacts')

      Ci::Build.joins(:project)
        .with_artifacts_stored_locally
        .find_each(batch_size: 10) do |build|

        build.artifacts_file.migrate!(ObjectStorage::Store::REMOTE)
        build.artifacts_metadata.migrate!(ObjectStorage::Store::REMOTE)

        logger.info("Transferred artifact ID #{build.id} with size #{build.artifacts_size} to object storage")
      rescue => e
        logger.error("Failed to transfer artifacts of #{build.id} with error: #{e.message}")
      end
    end
  end
end