From a9e1bcb92ded7929fc5e7a13ea79a5fc09f262aa Mon Sep 17 00:00:00 2001
From: syed-ali-tw <syed.ali@digital.cabinet-office.gov.uk>
Date: Wed, 1 Nov 2023 14:58:26 +0000
Subject: [PATCH] Inorder to be able to move worldwide organisations to use
 FeaturedImageData we need to be able to have migration setup so that we can
 migrate default_news_image. This commit creates a placeholder for
 FeaturedImageData in WWO and migration uses that to create the associations.

---
 app/models/worldwide_organisation.rb          |  1 +
 .../migrate_worldwide_organisations.rake      | 74 +++++++++++++++++++
 2 files changed, 75 insertions(+)
 create mode 100644 lib/tasks/migrate_worldwide_organisations.rake

diff --git a/app/models/worldwide_organisation.rb b/app/models/worldwide_organisation.rb
index b6ab3598358..df431d64142 100644
--- a/app/models/worldwide_organisation.rb
+++ b/app/models/worldwide_organisation.rb
@@ -20,6 +20,7 @@ class WorldwideOrganisation < ApplicationRecord
   has_many :editions, through: :edition_worldwide_organisations
 
   belongs_to :default_news_image, class_name: "DefaultNewsOrganisationImageData", foreign_key: :default_news_organisation_image_data_id
+  has_one :default_news_image_new, class_name: "FeaturedImageData", as: :featured_imageable, inverse_of: :featured_imageable
 
   accepts_nested_attributes_for :default_news_image, reject_if: :all_blank
 
diff --git a/lib/tasks/migrate_worldwide_organisations.rake b/lib/tasks/migrate_worldwide_organisations.rake
new file mode 100644
index 00000000000..bb3aef11765
--- /dev/null
+++ b/lib/tasks/migrate_worldwide_organisations.rake
@@ -0,0 +1,74 @@
+desc "Migrate worldwide organisations to use FeaturedImageData with assets"
+task :migrate_worldwide_organisations, %i[start_id end_id] => :environment do |_, args|
+  puts "Migrating worldwide organisations start!"
+  start_id = args[:start_id]
+  end_id = args[:end_id]
+
+  worldwide_organisations = WorldwideOrganisation.where(id: start_id..end_id).where("default_news_organisation_image_data_id is not null")
+  puts "Number of worldwide organisations found: #{worldwide_organisations.count}"
+  puts "Creating Asset for worldwide organisations records from #{start_id} to #{end_id}"
+
+  count = 0
+  asset_counter = 0
+  assetable_type = FeaturedImageData.to_s
+
+  worldwide_organisations.each do |worldwide_organisation|
+    next if worldwide_organisation.default_news_image_new.present?
+
+    all_variants = %i[original s960 s712 s630 s465 s300 s216]
+    assetable = FeaturedImageData.create!(
+      carrierwave_image: worldwide_organisation.default_news_image.carrierwave_image,
+      featured_imageable_type: WorldwideOrganisation.to_s,
+      featured_imageable_id: worldwide_organisation.id,
+    )
+
+    all_variants.each do |variant|
+      path = variant == :original ? worldwide_organisation.default_news_image.file.path : worldwide_organisation.default_news_image.file.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} worldwide organisations"
+  puts "Created asset counter #{asset_counter}"
+  puts "Migrating worldwide organisations 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