namespace :legal_doc do

  desc 'Legal documents duplicates fix'
  task all: :environment do
    Rake::Task['legal_doc:generate_hash'].invoke
    Rake::Task['legal_doc:remove_duplicates'].invoke
  end

  desc 'Generate hash'
  task generate_hash: :environment do
    start = Time.zone.now.to_f
    puts '-----> Generating unique hash for legal documents'
    count = 0

    LegalDocument.where(checksum: [nil, ""]).find_each do |x|
      if File.exist?(x.path)
        x.checksum = x.calc_checksum
        x.save
        count += 1
      end

    end
    puts "-----> Hash generated for #{count} rows in #{(Time.zone.now.to_f - start).round(2)} seconds"
  end

  desc 'Remove duplicates'
  task remove_duplicates: :environment do

    start = Time.zone.now.to_f
    puts '-----> Removing legal documents duplicates'
    count = 0
    modified = Array.new

    LegalDocument.where.not(checksum: [nil, ""]).find_each do |x|
      next if modified.include?(x.checksum)
      next if !File.exist?(x.path)
      modified.push(x.checksum)


      LegalDocument.where(checksum: x.checksum).where.not(id: x.id) do |y|
        unless modified.include?(x.id)
          File.delete(y.path) if File.exist?(y.path)
          y.update(path: x.path)
          count += 1
        end
      end
    end
    puts "-----> Duplicates fixed for #{count} rows in #{(Time.zone.now.to_f - start).round(2)} seconds"

  end

end