discourse/lib/file_helper.rb

132 lines
3.4 KiB
Ruby
Raw Normal View History

require "final_destination"
require "mini_mime"
require "open-uri"
2014-04-15 04:55:57 +08:00
class FileHelper
2017-09-28 07:00:13 +08:00
def self.log(log_level, message)
2017-09-28 07:07:43 +08:00
Rails.logger.public_send(
log_level,
"#{RailsMultisite::ConnectionManagement.current_db}: #{message}"
)
end
2017-09-28 07:00:13 +08:00
def self.is_supported_image?(filename)
filename =~ supported_images_regexp
2014-04-15 04:55:57 +08:00
end
class FakeIO
attr_accessor :status
end
def self.download(url,
max_file_size:,
tmp_file_name:,
follow_redirect: false,
read_timeout: 5,
skip_rate_limit: false,
2018-08-17 16:52:55 +08:00
verbose: false,
retain_on_max_file_size_exceeded: false)
url = "https:" + url if url.start_with?("//")
raise Discourse::InvalidParameters.new(:url) unless url =~ /^https?:\/\//
2014-04-15 04:55:57 +08:00
tmp = nil
fd = FinalDestination.new(
url,
max_redirects: follow_redirect ? 5 : 1,
skip_rate_limit: skip_rate_limit,
verbose: verbose
)
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, ""]
raise OpenURI::HTTPError.new("#{response.code} Error: #{response.body}", io)
else
log(:error, "FinalDestination did not work for: #{url}") if verbose
throw :done
end
end
if response.content_type.present?
ext = MiniMime.lookup_by_content_type(response.content_type)&.extension
ext = "jpg" if ext == "jpe"
tmp_file_ext = "." + ext if ext.present?
end
tmp_file_ext ||= File.extname(uri.path)
tmp = Tempfile.new([tmp_file_name, tmp_file_ext])
tmp.binmode
end
tmp.write(chunk)
2014-04-15 04:55:57 +08:00
if tmp.size > max_file_size
2018-08-17 16:52:55 +08:00
unless retain_on_max_file_size_exceeded
tmp.close
tmp = nil
end
throw :done
end
2014-04-15 04:55:57 +08:00
end
tmp&.rewind
2014-04-15 04:55:57 +08:00
tmp
end
def self.optimize_image!(filename, allow_pngquant: false)
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,
)
end
end
def self.memoize(*args)
(@memoized ||= {})[args] ||= yield
end
def self.supported_images
@@supported_images ||= Set.new %w{jpg jpeg png gif svg ico}
2018-06-07 13:28:18 +08:00
end
2014-04-15 04:55:57 +08:00
def self.supported_images_regexp
@@supported_images_regexp ||= /\.(#{supported_images.to_a.join("|")})$/i
2018-06-07 13:28:18 +08:00
end
2014-04-15 04:55:57 +08:00
end