Fetch remote image using http.rb (#3114)

6 files changed, 40 insertions, 26 deletions

diff --git a/app/models/account.rb b/app/models/account.rb
index bd47ce8a9..03e7db398 100644
--- a/app/models/account.rb
+++ b/app/models/account.rb
@@ -44,6 +44,7 @@ class Account < ApplicationRecord
   include AccountAvatar
   include AccountHeader
   include Attachmentable
+  include Remotable
   include Targetable
 
   # Local users
diff --git a/app/models/concerns/account_avatar.rb b/app/models/concerns/account_avatar.rb
index 8b9b72659..c664366ef 100644
--- a/app/models/concerns/account_avatar.rb
+++ b/app/models/concerns/account_avatar.rb
@@ -26,16 +26,5 @@ module AccountAvatar
     def avatar_static_url
       avatar_content_type == 'image/gif' ? avatar.url(:static) : avatar_original_url
     end
-
-    def avatar_remote_url=(url)
-      parsed_url = Addressable::URI.parse(url).normalize
-
-      return if !%w(http https).include?(parsed_url.scheme) || parsed_url.host.empty? || self[:avatar_remote_url] == url
-
-      self.avatar              = URI.parse(parsed_url.to_s)
-      self[:avatar_remote_url] = url
-    rescue OpenURI::HTTPError, OpenSSL::SSL::SSLError, Paperclip::Errors::NotIdentifiedByImageMagickError => e
-      Rails.logger.debug "Error fetching remote avatar: #{e}"
-    end
   end
 end
diff --git a/app/models/concerns/account_header.rb b/app/models/concerns/account_header.rb
index 42f556a46..f1b0883ee 100644
--- a/app/models/concerns/account_header.rb
+++ b/app/models/concerns/account_header.rb
@@ -26,16 +26,5 @@ module AccountHeader
     def header_static_url
       header_content_type == 'image/gif' ? header.url(:static) : header_original_url
     end
-
-    def header_remote_url=(url)
-      parsed_url = Addressable::URI.parse(url).normalize
-
-      return if !%w(http https).include?(parsed_url.scheme) || parsed_url.host.empty? || self[:header_remote_url] == url
-
-      self.header              = URI.parse(parsed_url.to_s)
-      self[:header_remote_url] = url
-    rescue OpenURI::HTTPError, OpenSSL::SSL::SSLError, Paperclip::Errors::NotIdentifiedByImageMagickError => e
-      Rails.logger.debug "Error fetching remote header: #{e}"
-    end
   end
 end
diff --git a/app/models/concerns/remotable.rb b/app/models/concerns/remotable.rb
new file mode 100644
index 000000000..4a412ee3d
--- /dev/null
+++ b/app/models/concerns/remotable.rb
@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Remotable
+  include HttpHelper
+  extend ActiveSupport::Concern
+
+  included do
+    attachment_definitions.each_key do |attachment_name|
+      attribute_name = "#{attachment_name}_remote_url".to_sym
+      method_name = "#{attribute_name}=".to_sym
+
+      define_method method_name do |url|
+        parsed_url = Addressable::URI.parse(url).normalize
+
+        return if !%w(http https).include?(parsed_url.scheme) || parsed_url.host.empty? || self[attribute_name] == url
+
+        begin
+          response = http_client.get(url)
+
+          return if response.code != 200
+
+          matches  = response.headers['content-disposition']&.match(/filename="([^"]*)"/)
+          filename = matches.nil? ? parsed_url.path.split('/').last : matches[1]
+
+          send("#{attachment_name}=", StringIO.new(response.to_s))
+          send("#{attachment_name}_file_name=", filename)
+
+          self[attribute_name] = url if has_attribute?(attribute_name)
+        rescue HTTP::TimeoutError, OpenSSL::SSL::SSLError, Paperclip::Errors::NotIdentifiedByImageMagickError => e
+          Rails.logger.debug "Error fetching remote #{attachment_name}: #{e}"
+        end
+      end
+    end
+  end
+end
diff --git a/app/models/media_attachment.rb b/app/models/media_attachment.rb
index 3277eb227..c29a66eb2 100644
--- a/app/models/media_attachment.rb
+++ b/app/models/media_attachment.rb
@@ -46,6 +46,9 @@ class MediaAttachment < ApplicationRecord
                     styles: ->(f) { file_styles f },
                     processors: ->(f) { file_processors f },
                     convert_options: { all: '-quality 90 -strip' }
+
+  include Remotable
+
   validates_attachment_content_type :file, content_type: IMAGE_MIME_TYPES + VIDEO_MIME_TYPES
   validates_attachment_size :file, less_than: 8.megabytes
 
@@ -59,10 +62,6 @@ class MediaAttachment < ApplicationRecord
     remote_url.blank?
   end
 
-  def file_remote_url=(url)
-    self.file = URI.parse(Addressable::URI.parse(url).normalize.to_s)
-  end
-
   def to_param
     shortcode
   end
diff --git a/app/models/preview_card.rb b/app/models/preview_card.rb
index a57f1252c..c334c48aa 100644
--- a/app/models/preview_card.rb
+++ b/app/models/preview_card.rb
@@ -36,6 +36,7 @@ class PreviewCard < ApplicationRecord
   has_attached_file :image, styles: { original: '120x120#' }, convert_options: { all: '-quality 80 -strip' }
 
   include Attachmentable
+  include Remotable
 
   validates :url, presence: true
   validates_attachment_content_type :image, content_type: IMAGE_MIME_TYPES