gitlab-org--gitlab-foss/config/initializers/direct_upload_support.rb

38 lines
1.2 KiB
Ruby
Raw Normal View History

2018-06-04 08:06:07 -04:00
class DirectUploadsValidator
SUPPORTED_DIRECT_UPLOAD_PROVIDERS = %w(Google AWS).freeze
2018-05-09 11:27:38 -04:00
2018-06-04 08:06:07 -04:00
ValidationError = Class.new(StandardError)
2018-05-09 11:27:38 -04:00
def verify!(uploader_type, object_store)
2018-06-04 08:06:07 -04:00
return unless object_store.enabled
return unless object_store.direct_upload
raise ValidationError, "Object storage is configured for '#{uploader_type}', but the 'connection' section is missing" unless object_store.key?('connection')
provider = object_store.connection&.provider.to_s
raise ValidationError, "No provider configured for '#{uploader_type}'. #{supported_provider_text}" if provider.blank?
return if SUPPORTED_DIRECT_UPLOAD_PROVIDERS.include?(provider)
raise ValidationError, "Object storage provider '#{provider}' is not supported " \
"when 'direct_upload' is used for '#{uploader_type}'. #{supported_provider_text}"
end
def supported_provider_text
"Only #{SUPPORTED_DIRECT_UPLOAD_PROVIDERS.join(', ')} are supported."
2018-06-04 08:06:07 -04:00
end
2018-05-09 11:27:38 -04:00
end
2018-06-04 08:06:07 -04:00
DirectUploadsValidator.new.tap do |validator|
CONFIGS = {
artifacts: Gitlab.config.artifacts,
uploads: Gitlab.config.uploads,
lfs: Gitlab.config.lfs
}.freeze
CONFIGS.each do |uploader_type, uploader|
validator.verify!(uploader_type, uploader.object_store)
2018-06-04 08:06:07 -04:00
end
end