Migrate attestation files to active_storage

lib/tasks/2019_08_22_migrate_attestation_files.rake

@@ -0,0 +1,97 @@
+namespace :'2019_08_22_migrate_attestation_files' do
+  task migrate_attestation_pdf: :environment do
+    attestations = Attestation.where
+      .not(pdf: nil)
+      .left_joins(:pdf_active_storage_attachment)
+      .where('active_storage_attachments.id IS NULL')
+      .order(:created_at)
+
+    limit = ENV['LIMIT']
+    if limit
+      attestations.limit!(limit.to_i)
+    end
+
+    progress = ProgressReport.new(attestations.count)
+    attestations.find_each do |attestation|
+      if attestation.pdf.present?
+        uri = URI.parse(URI.escape(attestation.pdf_url))
+        response = Typhoeus.get(uri)
+        if response.success?
+          filename = attestation.pdf.filename || attestation.pdf_identifier
+          attestation.pdf_active_storage.attach(
+            io: StringIO.new(response.body),
+            filename: filename,
+            content_type: attestation.pdf.content_type,
+            metadata: { virus_scan_result: ActiveStorage::VirusScanner::SAFE }
+          )
+        end
+      end
+      progress.inc
+    end
+    progress.finish
+  end
+
+  task migrate_attestation_template_logo: :environment do
+    attestation_templates = AttestationTemplate.where
+      .not(logo: nil)
+      .left_joins(:logo_active_storage_attachment)
+      .where('active_storage_attachments.id IS NULL')
+      .order(:created_at)
+
+    limit = ENV['LIMIT']
+    if limit
+      attestation_templates.limit!(limit.to_i)
+    end
+
+    progress = ProgressReport.new(attestation_templates.count)
+    attestation_templates.find_each do |attestation_template|
+      if attestation_template.logo.present?
+        uri = URI.parse(URI.escape(attestation_template.logo_url))
+        response = Typhoeus.get(uri)
+        if response.success?
+          filename = attestation_template.logo.filename || attestation_template.logo_identifier
+          attestation_template.logo_active_storage.attach(
+            io: StringIO.new(response.body),
+            filename: filename,
+            content_type: attestation_template.logo.content_type,
+            metadata: { virus_scan_result: ActiveStorage::VirusScanner::SAFE }
+          )
+        end
+      end
+      progress.inc
+    end
+    progress.finish
+  end
+
+  task migrate_attestation_template_signature: :environment do
+    attestation_templates = AttestationTemplate.where
+      .not(signature: nil)
+      .left_joins(:signature_active_storage_attachment)
+      .where('active_storage_attachments.id IS NULL')
+      .order(:created_at)
+
+    limit = ENV['LIMIT']
+    if limit
+      attestation_templates.limit!(limit.to_i)
+    end
+
+    progress = ProgressReport.new(attestation_templates.count)
+    attestation_templates.find_each do |attestation_template|
+      if attestation_template.signature.present?
+        uri = URI.parse(URI.escape(attestation_template.signature_url))
+        response = Typhoeus.get(uri)
+        if response.success?
+          filename = attestation_template.signature.filename || attestation_template.signature_identifier
+          attestation_template.signature_active_storage.attach(
+            io: StringIO.new(response.body),
+            filename: filename,
+            content_type: attestation_template.signature.content_type,
+            metadata: { virus_scan_result: ActiveStorage::VirusScanner::SAFE }
+          )
+        end
+      end
+      progress.inc
+    end
+    progress.finish
+  end
+end