about summary refs log tree commit diff
path: root/app/models/concerns/remotable.rb
diff options
context:
space:
mode:
authorYamagishi Kazutoshi <ykzts@desire.sh>2017-05-18 22:43:10 +0900
committerEugen Rochko <eugen@zeonfederated.com>2017-05-18 15:43:10 +0200
commit79ef8b3653a6ff7edbc1af0741dff36522262c07 (patch)
tree161a1bec18bed8475ae780dd8e28344d0dcfbb1d /app/models/concerns/remotable.rb
parentb11c4326d289d8b0307f5d1466171ac147a07767 (diff)
Fetch remote image using http.rb (#3114)
Diffstat (limited to 'app/models/concerns/remotable.rb')
-rw-r--r--app/models/concerns/remotable.rb35
1 files changed, 35 insertions, 0 deletions
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