discourse/lib/file_helper.rb

50 lines
1.2 KiB
Ruby
Raw Normal View History

require "open-uri"
require "addressable/uri"
class URI::Parser
# HACK to support underscores in URLs
def split(url)
a = Addressable::URI::parse(url)
[a.scheme, a.userinfo, a.host, a.port, nil, a.path, nil, a.query, a.fragment]
end
end
2014-04-14 16:55:57 -04:00
class FileHelper
def self.is_image?(filename)
filename =~ images_regexp
end
def self.download(url, max_file_size, tmp_file_name, follow_redirect=false)
raise Discourse::InvalidParameters.new(:url) unless url =~ /^https?:\/\//
2014-04-14 16:55:57 -04:00
uri = URI.parse(url)
extension = File.extname(uri.path)
2014-04-14 16:55:57 -04:00
tmp = Tempfile.new([tmp_file_name, extension])
File.open(tmp.path, "wb") do |f|
downloaded = uri.open("rb", read_timeout: 5, redirect: follow_redirect)
2014-04-15 07:04:14 -04:00
while f.size <= max_file_size && data = downloaded.read(max_file_size)
2014-04-14 16:55:57 -04:00
f.write(data)
end
# tiny files are StringIO, no close! on them
downloaded.close! if downloaded.respond_to? :close!
2014-04-14 16:55:57 -04:00
end
tmp
end
private
def self.images
@@images ||= Set.new ["jpg", "jpeg", "png", "gif", "tif", "tiff", "bmp", "svg", "webp"]
2014-04-14 16:55:57 -04:00
end
def self.images_regexp
@@images_regexp ||= /\.(#{images.to_a.join("|")})$/i
2014-04-14 16:55:57 -04:00
end
end