gitlab-org--gitlab-foss/app/uploaders/file_uploader.rb

49 lines
1 KiB
Ruby
Raw Normal View History

# encoding: utf-8
class FileUploader < CarrierWave::Uploader::Base
storage :file
attr_accessor :project, :secret
2015-02-16 13:58:40 -05:00
def initialize(project, secret = self.class.generate_secret)
@project = project
@secret = secret
end
def base_dir
"uploads"
end
def store_dir
2015-02-16 13:58:40 -05:00
File.join(base_dir, @project.path_with_namespace, @secret)
end
def cache_dir
2015-02-16 13:58:40 -05:00
File.join(base_dir, 'tmp', @project.path_with_namespace, @secret)
end
2015-02-16 13:58:40 -05:00
def self.generate_secret
2015-02-16 16:08:55 -05:00
SecureRandom.hex
end
2015-02-20 09:56:12 -05:00
def secure_url
File.join(Gitlab.config.gitlab.url, @project.path_with_namespace, "uploads", @secret, file.filename)
end
2015-02-16 13:58:40 -05:00
def file_storage?
self.class.storage == CarrierWave::Storage::File
end
def image?
img_ext = %w(png jpg jpeg gif bmp tiff)
if file.respond_to?(:extension)
img_ext.include?(file.extension.downcase)
else
# Not all CarrierWave storages respond to :extension
ext = file.path.split('.').last.downcase
img_ext.include?(ext)
end
rescue
false
end
end