debian-mirror-gitlab/lib/api/helpers/packages/dependency_proxy_helpers.rb
2021-10-27 15:23:28 +05:30

52 lines
1.6 KiB
Ruby

# frozen_string_literal: true
module API
module Helpers
module Packages
module DependencyProxyHelpers
REGISTRY_BASE_URLS = {
npm: 'https://registry.npmjs.org/',
pypi: 'https://pypi.org/simple/'
}.freeze
APPLICATION_SETTING_NAMES = {
npm: 'npm_package_requests_forwarding',
pypi: 'pypi_package_requests_forwarding'
}.freeze
def redirect_registry_request(forward_to_registry, package_type, options)
if forward_to_registry && redirect_registry_request_available?(package_type)
::Gitlab::Tracking.event(self.options[:for].name, "#{package_type}_request_forward")
redirect(registry_url(package_type, options))
else
yield
end
end
def registry_url(package_type, options)
base_url = REGISTRY_BASE_URLS[package_type]
raise ArgumentError, "Can't build registry_url for package_type #{package_type}" unless base_url
case package_type
when :npm
"#{base_url}#{options[:package_name]}"
when :pypi
"#{base_url}#{options[:package_name]}/"
end
end
def redirect_registry_request_available?(package_type)
application_setting_name = APPLICATION_SETTING_NAMES[package_type]
raise ArgumentError, "Can't find application setting for package_type #{package_type}" unless application_setting_name
::Gitlab::CurrentSettings
.current_application_settings
.attributes
.fetch(application_setting_name, false)
end
end
end
end
end