From 87e427f4335df91479e2a74196893e54e40485be Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sen=C3=A9n=20Rodero=20Rodr=C3=ADguez?=
 <senenrodero@gmail.com>
Date: Fri, 17 Sep 2021 15:00:30 +0200
Subject: [PATCH] Allow remote storages in Active Storage migration

So forks that connected Paperclip with S3 buckets can migrate files too.
---
 lib/tasks/active_storage.rake | 36 ++++++++++++++++++++++++-----------
 1 file changed, 25 insertions(+), 11 deletions(-)

diff --git a/lib/tasks/active_storage.rake b/lib/tasks/active_storage.rake
index ecf3d0388..d548ab21d 100644
--- a/lib/tasks/active_storage.rake
+++ b/lib/tasks/active_storage.rake
@@ -23,6 +23,7 @@ namespace :active_storage do
 
     Rails.application.eager_load!
     models = ActiveRecord::Base.descendants.reject(&:abstract_class?)
+    paperclip_storage = Paperclip::Attachment.default_options[:storage]
 
     ActiveRecord::Base.transaction do
       models.each do |model|
@@ -42,11 +43,19 @@ namespace :active_storage do
           attachments.each do |attachment|
             next if instance.send(:"storage_#{attachment}").attached?
 
-            source = instance.send(attachment).path
+            source = if paperclip_storage == :filesystem
+                       instance.send(attachment).path
+                     else
+                       instance.send(attachment).url
+                     end
 
-            next if source.blank?
+            next if source.blank? || source == "/images/original/missing.png"
 
-            file = File.read(source) if File.exist?(source)
+            file = if paperclip_storage == :filesystem
+                     File.read(source) if File.exist?(source)
+                   else
+                     Net::HTTP.get(URI(source))
+                   end
 
             connection.exec_prepared(
               statement_name, [
@@ -67,18 +76,23 @@ namespace :active_storage do
     end
 
     ActiveStorage::Attachment.find_each do |attachment|
-      dest = ActiveStorage::Blob.service.path_for(attachment.blob.key)
+      blob = attachment.blob
 
-      next if File.exist?(dest) || !attachment.record
+      next if blob.service.exist?(blob.key) || !attachment.record
 
       name = attachment.name.delete_prefix("storage_")
-      source = attachment.record.send(name).path
+      paperclip_attachment = attachment.record.send(name)
 
-      if source && File.exist?(source)
-        FileUtils.mkdir_p(File.dirname(dest))
-        logger.info "Copying #{source} to #{dest}"
-        FileUtils.cp(source, dest)
-      end
+      next unless paperclip_attachment.exists?
+
+      source_file = if paperclip_storage == :filesystem
+                      paperclip_attachment.path
+                    else
+                      URI.open(paperclip_attachment.url, &:read)
+                    end
+
+      logger.info "Copying #{paperclip_attachment.url} to active storage"
+      blob.service.upload(blob.key, source_file)
     end
   end
 end