51 lines
1.6 KiB
Ruby
51 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class DirectUploadsValidator
|
|
SUPPORTED_DIRECT_UPLOAD_PROVIDERS = [ObjectStorage::Config::GOOGLE_PROVIDER,
|
|
ObjectStorage::Config::AWS_PROVIDER,
|
|
ObjectStorage::Config::AZURE_PROVIDER].freeze
|
|
|
|
ValidationError = Class.new(StandardError)
|
|
|
|
def verify!(uploader_type, object_store)
|
|
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 provider_loaded?(provider)
|
|
|
|
raise ValidationError, "Object storage provider '#{provider}' is not supported " \
|
|
"when 'direct_upload' is used for '#{uploader_type}'. #{supported_provider_text}"
|
|
end
|
|
|
|
private
|
|
|
|
def provider_loaded?(provider)
|
|
return false unless SUPPORTED_DIRECT_UPLOAD_PROVIDERS.include?(provider)
|
|
|
|
require 'fog/azurerm' if provider == ObjectStorage::Config::AZURE_PROVIDER
|
|
|
|
true
|
|
end
|
|
|
|
def supported_provider_text
|
|
"Only #{SUPPORTED_DIRECT_UPLOAD_PROVIDERS.to_sentence} are supported."
|
|
end
|
|
end
|
|
|
|
DirectUploadsValidator.new.tap do |validator|
|
|
CONFIGS = {
|
|
artifacts: Gitlab.config.artifacts,
|
|
lfs: Gitlab.config.lfs,
|
|
uploads: Gitlab.config.uploads
|
|
}.freeze
|
|
|
|
CONFIGS.each do |uploader_type, uploader|
|
|
validator.verify!(uploader_type, uploader.object_store)
|
|
end
|
|
end
|