SpeciesFileGroup/taxonworks

namespace :tw do

  # Tasks that auto-curate data
  namespace :maintenance do

    namespace :images do

      # TODO: Capture raises on frequency stats vi Sqed::Error hopefully.
      # rake tw:maintenance:images:recalculate_sqed_boundaries project_id=20 id_start=1 id_end=999999
      desc 'Re-calculate the boundaries for Sqed depictions'
      task recalculate_sqed_boundaries: [:environment, :project_id, :id_start, :id_end] do |t|
        project_id = ENV['project_id']

        all = ENV['tw_sqed_calculate_all'] == 'true' ? true : false

        a = SqedDepiction.where(project_id: project_id).order(:id)

        a = a.where(result_boundary_coordinates: nil) unless all

        id_min = a.first.try(:id)
        id_max = a.last.try(:id)

        puts Rainbow("id range (all: #{all}) for project #{project_id} is #{id_min}-#{id_max}.").blue
        records = a.where('id > ?', @args[:id_start] - 1).where('id < ?', @args[:id_end] + 1)
        puts Rainbow("Processing #{records.count} sqed_depictions.").blue

        i = 0

        records.order(:id).in_groups_of(3, false) do |group|
          ApplicationRecord.transaction_with_retry do
            begin
              print Rainbow("Writing\n").bold
              group.each do |o|
                print " id: #{o.id}\n"
                o.preprocess
                i += 1
              end
              print Rainbow("...saved.\n").bold

            rescue Magick::ImageMagickError => e
              puts Rainbow("Error: #{e}. ImageMagick processing error. Current batch of 3 records not written, skipping.").red.bold
              next
            rescue RuntimeError => e
              puts Rainbow("Error: #{e}. Current batch of 3 records not written, skipping.").red.bold
              next
            end
          end
        end
      end

      desc 'Rotate images'
      task rotate_images: [:environment, :project_id, :user_id, :id_start, :id_end] do |t|
        raise TaxonWorks::Error, Rainbow('Specify rotate, like rotate=180').yellow if ENV['rotate'].blank?
        rotate = ENV['rotate'].to_i

        a = Image.where(project_id: @args[:project_id]).order(:id)

        records = a.where('id > ?', @args[:id_start] - 1).where('id < ?', @args[:id_end] + 1)

        puts Rainbow("Processing #{records.count} images.").blue

        i = 0

        begin
          print Rainbow("Rotating\n").bold
          records.order(:id).find_each do |i|
            ApplicationRecord.transaction do
              original_file = i.image_file.path(:original)
              puts Rainbow(i.id.to_s + ': ' + original_file).purple

              temp_original = original_file + '.tmp'

              FileUtils.cp(original_file, temp_original)

              begin
                j = Magick::Image::read(original_file).first

                jr = j.rotate(rotate)

                j.destroy! # free memory/cache

                File.delete(original_file)
                jr.write(original_file)
                jr.destroy! # free memory/cache
              rescue
                FileUtils.cp(temp_original, original_file)
                raise
              ensure
                File.delete(temp_original) if File.exist?(temp_original)
              end

              i.image_file.reprocess!
            rescue
              raise
            end
          end
        end
      end
    end
  end
end