namespace :after_party do
  desc 'Deployment task: fix_cloned_revisions'
  task fix_cloned_revisions: :environment do
    puts "Running deploy task 'fix_cloned_revisions'"

    Procedure.with_discarded.where(aasm_state: :brouillon).where.not(published_revision_id: nil).update_all(published_revision_id: nil)

    types_de_champ = TypeDeChamp.joins(:revision).where('types_de_champ.procedure_id != procedure_revisions.procedure_id')
    progress = ProgressReport.new(types_de_champ.count)

    types_de_champ.find_each do |type_de_champ|
      procedure = type_de_champ.procedure ? type_de_champ.procedure : Procedure.with_discarded.find(type_de_champ.procedure_id)
      revision_id = procedure.published_revision_id || procedure.draft_revision_id
      type_de_champ.update_column(:revision_id, revision_id)
      progress.inc
    end

    progress.finish

    # Update task as completed.  If you remove the line below, the task will
    # run with every deploy (or every time you call after_party:run).
    AfterParty::TaskRecord
      .create version: AfterParty::TaskRecorder.new(__FILE__).timestamp
  end
end