namespace :gitlab do
  namespace :uploads do
    namespace :migrate do
      desc "GitLab | Uploads | Migrate all uploaded files to object storage"
      task all: :environment do
        Gitlab::Uploads::MigrationHelper::CATEGORIES.each do |args|
          Rake::Task["gitlab:uploads:migrate"].invoke(*args)
          Rake::Task["gitlab:uploads:migrate"].reenable
        end
      end
    end

    # The following is the actual rake task that migrates uploads of specified
    # category to object storage
    desc 'GitLab | Uploads | Migrate the uploaded files of specified type to object storage'
    task :migrate, [:uploader_class, :model_class, :mounted_as] => :environment do |_t, args|
      Gitlab::Uploads::MigrationHelper.new(args, Logger.new(STDOUT)).migrate_to_remote_storage
    end

    namespace :migrate_to_local do
      desc "GitLab | Uploads | Migrate all uploaded files to local storage"
      task all: :environment do
        Gitlab::Uploads::MigrationHelper::CATEGORIES.each do |args|
          Rake::Task["gitlab:uploads:migrate_to_local"].invoke(*args)
          Rake::Task["gitlab:uploads:migrate_to_local"].reenable
        end
      end
    end

    desc 'GitLab | Uploads | Migrate the uploaded files of specified type to local storage'
    task :migrate_to_local, [:uploader_class, :model_class, :mounted_as] => :environment do |_t, args|
      Gitlab::Uploads::MigrationHelper.new(args, Logger.new(STDOUT)).migrate_to_local_storage
    end
  end
end