Merge pull request #8492 from alphagov/prepare-take-page-to-use-featu…

…red-image-data-with-assets

Prepare migration task for TakePartPage

app/models/take_part_page.rb

@@ -14,6 +14,7 @@ class TakePartPage < ApplicationRecord
   include PublishesToPublishingApi
 
   mount_uploader :image, FeaturedImageUploader, mount_on: :carrierwave_image
+  has_one :image_new, class_name: "FeaturedImageData", as: :featured_imageable, inverse_of: :featured_imageable
 
   validates :image, presence: true, on: :create
   validates :image_alt_text, presence: true, allow_blank: true, length: { maximum: 255 }, on: :create

lib/tasks/migrate_take_part_page.rake

@@ -0,0 +1,75 @@
+desc "Migrate take part page to use FeaturedImageData with assets"
+task :migrate_take_part_page, %i[start_id end_id] => :environment do |_, args|
+  puts "Migrating take part page start!"
+  start_id = args[:start_id]
+  end_id = args[:end_id]
+
+  take_part_pages = TakePartPage.where(id: start_id..end_id).where("carrierwave_image is not null")
+  puts "Number of take part pages found: #{take_part_pages.count}"
+  puts "Creating Asset for take part pages from #{start_id} to #{end_id}"
+
+  count = 0
+  asset_counter = 0
+  assetable_type = FeaturedImageData.to_s
+
+  take_part_pages.each do |take_part_page|
+    # Ensure we do not replay the same migration
+    next if take_part_page.image_new
+
+    all_variants = take_part_page.image.versions.keys.push(:original)
+    assetable = FeaturedImageData.create!(
+      carrierwave_image: take_part_page.carrierwave_image,
+      featured_imageable_type: TakePartPage.to_s,
+      featured_imageable_id: take_part_page.id,
+    )
+
+    all_variants.each do |variant|
+      path = variant == :original ? take_part_page.image.path : take_part_page.image.versions[variant].path
+      puts "Creating Asset for path #{path}"
+      asset_counter += 1 if save_asset(assetable, assetable_type, variant, path)
+    end
+
+    count += 1
+  end
+
+  puts "Created assets for #{count} take part pages"
+  puts "Created asset counter #{asset_counter}"
+  puts "Migrating take part page finish!"
+end
+
+def save_asset(assetable, assetable_type, variant, path)
+  asset_info = get_asset_data(path)
+  save_asset_id_to_assets(assetable.id, assetable_type, Asset.variants[variant], asset_info[:asset_manager_id], asset_info[:filename])
+rescue GdsApi::HTTPNotFound
+  puts "#{assetable_type} of id##{assetable.id} - could not find asset variant :#{variant} at path #{path}"
+
+  false
+end
+
+def asset_manager
+  Services.asset_manager
+end
+
+def get_asset_data(legacy_url_path)
+  asset_info = {}
+  path = legacy_url_path.sub(/^\//, "")
+  response_hash = asset_manager.whitehall_asset(path).to_hash
+  asset_info[:asset_manager_id] = get_asset_id(response_hash)
+  asset_info[:filename] = get_filename(response_hash)
+
+  asset_info
+end
+
+def get_filename(response)
+  response["name"]
+end
+
+def get_asset_id(response)
+  url = response["id"]
+  url[/\/assets\/(.*)/, 1]
+end
+
+def save_asset_id_to_assets(assetable_id, assetable_type, variant, asset_manager_id, filename)
+  asset = Asset.new(asset_manager_id:, assetable_type:, assetable_id:, variant:, filename:)
+  asset.save!
+end