Merge pull request #2590 from alphagov/add-rake-tasks-to-fix-first-made-live-at

Add rake tasks to fix first_made_live_at for copied forms

Author: stephencdaly

lib/tasks/fix_first_made_live_at.rake

@@ -0,0 +1,62 @@
+namespace :fix_first_made_live_at do
+  desc "Updates forms to add first_made_live_at"
+  task :update_date, [] => :environment do |_, args|
+    Rails.logger.info "data_migration:update_date"
+
+    update_forms_first_made_live_at(args.to_a)
+  end
+
+  desc "Dry run updates forms to add first_made_live_at"
+  task :update_date_dry_run, [] => :environment do |_, args|
+    ActiveRecord::Base.transaction do
+      Rails.logger.info "data_migration:update_date_dry_run"
+
+      update_forms_first_made_live_at(args.to_a)
+
+      raise ActiveRecord::Rollback
+    end
+  end
+
+  desc "Clear first_made_live_at for draft forms"
+  task clear_for_draft: :environment do
+    forms = Form.draft.where.not(first_made_live_at: nil)
+    Rails.logger.info "Found #{forms.count} draft forms with first_made_live_at set"
+    forms.update_all(first_made_live_at: nil)
+    Rails.logger.info "Finished clearing first_made_live_at for draft forms"
+  end
+
+  desc "Clear first_made_live_at for draft forms"
+  task clear_for_draft_dry_run: :environment do
+    forms = Form.draft.where.not(first_made_live_at: nil)
+    Rails.logger.info "Found #{forms.count} draft forms with first_made_live_at set"
+    forms.find_each do |form|
+      Rails.logger.info "Would clear first_made_live_at for form with ID #{form.id}"
+    end
+  end
+end
+
+def update_forms_first_made_live_at(forms_and_dates)
+  Rails.logger.info "Updating #{forms_and_dates.length} forms"
+
+  forms_and_dates.each do |form_and_date|
+    form_id, first_made_live_at = form_and_date.split(":", 2)
+    first_made_live_at = Time.zone.parse(first_made_live_at)
+
+    Rails.logger.info "Updating form #{form_id} to have first_made_live_at #{first_made_live_at}"
+
+    form = Form.find(form_id)
+    form.first_made_live_at = first_made_live_at
+
+    begin
+      form.save!
+
+      form.form_documents.each do |form_document|
+        form_document.content[:first_made_live_at] = first_made_live_at
+
+        form_document.save!
+      end
+    rescue StandardError => e
+      Rails.logger.info "Failed to update form #{form_id}: #{e.message}"
+    end
+  end
+end

lib/tasks/forms.rake

@@ -153,12 +153,12 @@ namespace :forms do
   desc "Check selection options in Page DraftQuestion and FormDocument"
   task check_selection_options: :environment do
     pages_with_selection_options_with_no_value = Page.where(answer_type: "selection")
-        .where("jsonb_path_exists(answer_settings, '$.selection_options[*] ? (!exists(@.value))')")
-        .count
+                                                     .where("jsonb_path_exists(answer_settings, '$.selection_options[*] ? (!exists(@.value))')")
+                                                     .count
 
     draft_questions_with_selection_options_with_no_value = DraftQuestion.where(answer_type: "selection")
-        .where("jsonb_path_exists(answer_settings, '$.selection_options[*] ? (!exists(@.value))')")
-        .count
+                                                                        .where("jsonb_path_exists(answer_settings, '$.selection_options[*] ? (!exists(@.value))')")
+                                                                        .count
 
     Rails.logger.info "Pages with selection options with no value: #{pages_with_selection_options_with_no_value}"
     Rails.logger.info "DraftQuestions with selection options with no value: #{draft_questions_with_selection_options_with_no_value}"