Update existing data requests

db/data_migrations/20241024131255_migrate_data_request_areas.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+class MigrateDataRequestAreas < ActiveRecord::DataMigration
+  def data_request_area_type_for(request_type)
+    case request_type
+    when *DataRequest::BRANSTON_DATA_REQUEST_TYPES
+      "branston"
+    when *DataRequest::BRANSTON_REGISTRY_DATA_REQUEST_TYPES
+      "branston_registry"
+    when *DataRequest::MAPPA_DATA_REQUEST_TYPES
+      "mappa"
+    when *DataRequest::PRISON_DATA_REQUEST_TYPES + %w[cctv_and_bwcf] - %w[other]
+      "prison"
+    when *DataRequest::PROBATION_DATA_REQUEST_TYPES + %w[court] - %w[other]
+      "probation"
+    when "other"
+      "other_department"
+    end
+  end
+
+  def up
+    # loop through the DataRequest records
+    DataRequest.find_each do |request|
+      # Get the data_request_area_type for this request
+      data_request_area_type = data_request_area_type_for(request.request_type)
+
+      Rails.logger.debug "Finding DataRequestArea with user_id: #{request.user_id}, case_id: #{request.case_id}, contact_id: #{request.contact_id.presence}"
+
+      data_request_area = DataRequestArea.new(
+        user_id: request.user_id,
+        case_id: request.case_id,
+        data_request_area_type:,
+        contact_id: request.contact_id,
+        location: request.location.presence,
+        )
+
+      # Prevents existing data request commissioning documents being overwritten during the migration
+      if request.commissioning_document.present?
+        data_request_area.commissioning_document = request.commissioning_document
+      else
+        template_name = data_request_area_type == "mappa" ? "mappa" : "standard"
+        data_request_area.build_commissioning_document(template_name: template_name)
+      end
+
+      # Ensure that any uploaded attachments are kept
+      if request.commissioning_document&.attachment
+        data_request_area.commissioning_document.attachment = request.commissioning_document.attachment
+      end
+
+      data_request_area.save!
+
+      # Update DataRequest with the correct data_request_area_id
+      request.update!(data_request_area_id: data_request_area.id)
+
+      # TEMP LOGGING INFO
+      Rails.logger.debug "Updated DataRequest ##{request.id} - #{request.request_type}, with data_request_area_id: #{data_request_area.id} (area_type: #{data_request_area_type}, contact_id: #{data_request_area.contact_id}, data_request_area.user_id: #{data_request_area.user_id}, data_request_area.case_id: #{data_request_area.case_id}, data_request_area.commissioning_document: #{data_request_area.commissioning_document})"
+    end
+  end
+end