Update existing data requests

db/data_migrations/20241024131255_migrate_data_request_areas.rb

@@ -0,0 +1,48 @@
+# 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]
+      "prison"
+    when *DataRequest::PROBATION_DATA_REQUEST_TYPES + %w[court]
+      "probation"
+    when "other"
+      "other_department"
+    end
+  end
+
+  def up
+    # get DataRequests for only Offender and OffenderComplaint
+    data_requests = DataRequest.joins(:offender_sar_case)
+                               .where(offender_sar_case: { type: %w[Case::SAR::Offender Case::SAR::OffenderComplaint] })
+
+    # loop through the DataRequest records
+    data_requests.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.create!(
+        user_id: request.user_id,
+        case_id: request.case_id,
+        data_request_area_type:,
+        contact_id: request.contact_id,
+        location: request.location.presence,
+      )
+
+      # 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}"
+    end
+  end
+end