debian-mirror-gitlab/app/uploaders/gitlab_uploader.rb

93 lines
2 KiB
Ruby
Raw Normal View History

2017-08-17 22:00:37 +05:30
class GitlabUploader < CarrierWave::Uploader::Base
2018-03-17 18:26:18 +05:30
class_attribute :options
2017-08-17 22:00:37 +05:30
2018-03-17 18:26:18 +05:30
class << self
# DSL setter
def storage_options(options)
self.options = options
end
2017-08-17 22:00:37 +05:30
2018-03-17 18:26:18 +05:30
def root
options.storage_path
end
2017-08-17 22:00:37 +05:30
2018-03-17 18:26:18 +05:30
# represent the directory namespacing at the class level
def base_dir
options.fetch('base_dir', '')
end
2017-08-17 22:00:37 +05:30
2018-03-17 18:26:18 +05:30
def file_storage?
storage == CarrierWave::Storage::File
end
def absolute_path(upload_record)
File.join(root, upload_record.path)
end
2017-08-17 22:00:37 +05:30
end
2018-03-17 18:26:18 +05:30
storage_options Gitlab.config.uploads
2017-08-17 22:00:37 +05:30
delegate :base_dir, :file_storage?, to: :class
2018-03-17 18:26:18 +05:30
def initialize(model, mounted_as = nil, **uploader_context)
super(model, mounted_as)
end
2017-09-10 17:25:29 +05:30
def file_cache_storage?
cache_storage.is_a?(CarrierWave::Storage::File)
end
2017-08-17 22:00:37 +05:30
# Reduce disk IO
def move_to_cache
2018-03-17 18:26:18 +05:30
file_storage?
2017-08-17 22:00:37 +05:30
end
# Reduce disk IO
def move_to_store
2018-03-17 18:26:18 +05:30
file_storage?
2017-08-17 22:00:37 +05:30
end
2018-03-17 18:26:18 +05:30
def exists?
file.present?
2017-08-17 22:00:37 +05:30
end
2018-03-17 18:26:18 +05:30
def store_dir
File.join(base_dir, dynamic_segment)
end
def cache_dir
File.join(root, base_dir, 'tmp/cache')
2017-08-17 22:00:37 +05:30
end
2017-09-10 17:25:29 +05:30
def work_dir
2018-03-17 18:26:18 +05:30
File.join(root, base_dir, 'tmp/work')
2017-09-10 17:25:29 +05:30
end
def filename
super || file&.filename
end
2018-03-17 18:26:18 +05:30
def model_valid?
!!model
end
2017-09-10 17:25:29 +05:30
private
2018-03-17 18:26:18 +05:30
# Designed to be overridden by child uploaders that have a dynamic path
# segment -- that is, a path that changes based on mutable attributes of its
# associated model
def dynamic_segment
raise(NotImplementedError)
end
2017-09-10 17:25:29 +05:30
# To prevent files from moving across filesystems, override the default
# implementation:
# http://github.com/carrierwaveuploader/carrierwave/blob/v1.0.0/lib/carrierwave/uploader/cache.rb#L181-L183
def workfile_path(for_file = original_filename)
# To be safe, keep this directory outside of the the cache directory
# because calling CarrierWave.clean_cache_files! will remove any files in
# the cache directory.
2018-03-17 18:26:18 +05:30
File.join(work_dir, cache_id, version_name.to_s, for_file)
2017-09-10 17:25:29 +05:30
end
2017-08-17 22:00:37 +05:30
end