2017-05-23 13:31:20 -04:00
|
|
|
require "final_destination"
|
2017-06-13 07:27:05 -04:00
|
|
|
require "mini_mime"
|
|
|
|
require "open-uri"
|
2014-04-22 11:11:06 -04:00
|
|
|
|
2014-04-14 16:55:57 -04:00
|
|
|
class FileHelper
|
|
|
|
|
2017-09-27 19:00:13 -04:00
|
|
|
def self.log(log_level, message)
|
2017-09-27 19:07:43 -04:00
|
|
|
Rails.logger.public_send(
|
|
|
|
log_level,
|
|
|
|
"#{RailsMultisite::ConnectionManagement.current_db}: #{message}"
|
|
|
|
)
|
|
|
|
end
|
2017-09-27 19:00:13 -04:00
|
|
|
|
2018-09-09 22:22:45 -04:00
|
|
|
def self.is_supported_image?(filename)
|
|
|
|
filename =~ supported_images_regexp
|
2014-04-14 16:55:57 -04:00
|
|
|
end
|
|
|
|
|
2017-09-28 02:35:27 -04:00
|
|
|
class FakeIO
|
|
|
|
attr_accessor :status
|
|
|
|
end
|
|
|
|
|
2017-05-24 13:46:57 -04:00
|
|
|
def self.download(url,
|
|
|
|
max_file_size:,
|
|
|
|
tmp_file_name:,
|
|
|
|
follow_redirect: false,
|
|
|
|
read_timeout: 5,
|
2017-09-27 21:32:26 -04:00
|
|
|
skip_rate_limit: false,
|
2018-08-17 04:52:55 -04:00
|
|
|
verbose: false,
|
|
|
|
retain_on_max_file_size_exceeded: false)
|
2017-05-24 13:46:57 -04:00
|
|
|
|
2017-05-15 15:32:55 -04:00
|
|
|
url = "https:" + url if url.start_with?("//")
|
2014-05-12 10:57:52 -04:00
|
|
|
raise Discourse::InvalidParameters.new(:url) unless url =~ /^https?:\/\//
|
2014-04-14 16:55:57 -04:00
|
|
|
|
2018-02-24 06:35:57 -05:00
|
|
|
tmp = nil
|
|
|
|
|
|
|
|
fd = FinalDestination.new(
|
2017-05-24 13:46:57 -04:00
|
|
|
url,
|
|
|
|
max_redirects: follow_redirect ? 5 : 1,
|
2017-10-31 12:03:03 -04:00
|
|
|
skip_rate_limit: skip_rate_limit,
|
|
|
|
verbose: verbose
|
2017-09-28 02:35:27 -04:00
|
|
|
)
|
2017-06-13 07:27:05 -04:00
|
|
|
|
2018-02-24 06:35:57 -05:00
|
|
|
fd.get do |response, chunk, uri|
|
|
|
|
if tmp.nil?
|
|
|
|
# error handling
|
|
|
|
if uri.blank?
|
|
|
|
if response.code.to_i >= 400
|
|
|
|
# attempt error API compatibility
|
|
|
|
io = FakeIO.new
|
|
|
|
io.status = [response.code, ""]
|
2018-09-13 03:43:58 -04:00
|
|
|
raise OpenURI::HTTPError.new("#{response.code} Error: #{response.body}", io)
|
2018-02-24 06:35:57 -05:00
|
|
|
else
|
|
|
|
log(:error, "FinalDestination did not work for: #{url}") if verbose
|
|
|
|
throw :done
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2018-07-29 22:48:44 -04:00
|
|
|
if response.content_type.present?
|
2018-02-24 06:35:57 -05:00
|
|
|
ext = MiniMime.lookup_by_content_type(response.content_type)&.extension
|
|
|
|
ext = "jpg" if ext == "jpe"
|
|
|
|
tmp_file_ext = "." + ext if ext.present?
|
|
|
|
end
|
|
|
|
|
2018-07-29 22:48:44 -04:00
|
|
|
tmp_file_ext ||= File.extname(uri.path)
|
2018-02-24 06:35:57 -05:00
|
|
|
tmp = Tempfile.new([tmp_file_name, tmp_file_ext])
|
|
|
|
tmp.binmode
|
|
|
|
end
|
2018-02-22 12:15:42 -05:00
|
|
|
|
2018-02-24 06:35:57 -05:00
|
|
|
tmp.write(chunk)
|
2014-04-14 16:55:57 -04:00
|
|
|
|
2018-08-17 04:17:58 -04:00
|
|
|
if tmp.size > max_file_size
|
2018-08-17 04:52:55 -04:00
|
|
|
unless retain_on_max_file_size_exceeded
|
|
|
|
tmp.close
|
|
|
|
tmp = nil
|
|
|
|
end
|
|
|
|
|
2018-08-17 04:17:58 -04:00
|
|
|
throw :done
|
|
|
|
end
|
2014-04-14 16:55:57 -04:00
|
|
|
end
|
|
|
|
|
2018-02-24 06:35:57 -05:00
|
|
|
tmp&.rewind
|
2014-04-14 16:55:57 -04:00
|
|
|
tmp
|
|
|
|
end
|
|
|
|
|
2019-01-02 01:19:52 -05:00
|
|
|
def self.optimize_image!(filename, allow_pngquant: false)
|
2019-01-08 20:28:06 -05:00
|
|
|
image_optim(
|
|
|
|
allow_pngquant: allow_pngquant,
|
|
|
|
strip_image_metadata: SiteSetting.strip_image_metadata
|
|
|
|
).optimize_image!(filename)
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.image_optim(allow_pngquant: false, strip_image_metadata: true)
|
|
|
|
# memoization is critical, initializing an ImageOptim object is very expensive
|
|
|
|
# sometimes up to 200ms searching for binaries and looking at versions
|
|
|
|
memoize("image_optim", allow_pngquant, strip_image_metadata) do
|
|
|
|
pngquant_options = false
|
|
|
|
if allow_pngquant
|
|
|
|
pngquant_options = { allow_lossy: true }
|
|
|
|
end
|
|
|
|
|
|
|
|
ImageOptim.new(
|
|
|
|
# GLOBAL
|
|
|
|
timeout: 15,
|
|
|
|
skip_missing_workers: true,
|
|
|
|
# PNG
|
|
|
|
optipng: { level: 2, strip: strip_image_metadata },
|
|
|
|
advpng: false,
|
|
|
|
pngcrush: false,
|
|
|
|
pngout: false,
|
|
|
|
pngquant: pngquant_options,
|
|
|
|
# JPG
|
|
|
|
jpegoptim: { strip: strip_image_metadata ? "all" : "none" },
|
|
|
|
jpegtran: false,
|
|
|
|
jpegrecompress: false,
|
|
|
|
)
|
2019-01-02 01:19:52 -05:00
|
|
|
end
|
2019-01-08 20:28:06 -05:00
|
|
|
end
|
2019-01-02 01:19:52 -05:00
|
|
|
|
2019-01-08 20:28:06 -05:00
|
|
|
def self.memoize(*args)
|
|
|
|
(@memoized ||= {})[args] ||= yield
|
2017-07-25 05:48:39 -04:00
|
|
|
end
|
|
|
|
|
2018-09-09 22:03:44 -04:00
|
|
|
def self.supported_images
|
2018-09-09 22:49:01 -04:00
|
|
|
@@supported_images ||= Set.new %w{jpg jpeg png gif svg ico}
|
2018-06-07 01:28:18 -04:00
|
|
|
end
|
2014-04-14 16:55:57 -04:00
|
|
|
|
2018-09-09 22:22:45 -04:00
|
|
|
def self.supported_images_regexp
|
|
|
|
@@supported_images_regexp ||= /\.(#{supported_images.to_a.join("|")})$/i
|
2018-06-07 01:28:18 -04:00
|
|
|
end
|
2014-04-14 16:55:57 -04:00
|
|
|
|
|
|
|
end
|