87 lines
2.6 KiB
Ruby
87 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Routing
|
|
module PseudonymizationHelper
|
|
class MaskHelper
|
|
QUERY_PARAMS_TO_NOT_MASK = %w[].freeze
|
|
|
|
def initialize(request_object, group, project)
|
|
@request = request_object
|
|
@group = group
|
|
@project = project
|
|
end
|
|
|
|
def mask_params
|
|
return default_root_url + @request.original_fullpath unless has_maskable_params?
|
|
|
|
masked_params = @request.path_parameters.to_h do |key, value|
|
|
case key
|
|
when :project_id
|
|
[key, "project#{@project&.id}"]
|
|
when :namespace_id, :group_id
|
|
namespace = @group || @project&.namespace
|
|
[key, "namespace#{namespace&.id}"]
|
|
when :id
|
|
[key, mask_id(value)]
|
|
else
|
|
[key, value]
|
|
end
|
|
end
|
|
|
|
Gitlab::Routing.url_helpers.url_for(masked_params.merge(params: masked_query_params))
|
|
end
|
|
|
|
private
|
|
|
|
def mask_id(value)
|
|
if @request.path_parameters[:controller] == 'projects/blob'
|
|
':repository_path'
|
|
elsif @request.path_parameters[:controller] == 'projects'
|
|
"project#{@project&.id}"
|
|
elsif @request.path_parameters[:controller] == 'groups'
|
|
"namespace#{@group&.id}"
|
|
else
|
|
value
|
|
end
|
|
end
|
|
|
|
def has_maskable_params?
|
|
request_params = @request.path_parameters.to_h
|
|
request_params.key?(:namespace_id) || request_params.key?(:group_id) || request_params.key?(:project_id) || request_params.key?(:id) || @request.query_string.present?
|
|
end
|
|
|
|
def masked_query_params
|
|
return {} unless @request.query_string.present?
|
|
|
|
query_string_hash = Rack::Utils.parse_nested_query(@request.query_string)
|
|
|
|
query_string_hash.keys.each do |key|
|
|
next if QUERY_PARAMS_TO_NOT_MASK.include?(key)
|
|
|
|
query_string_hash[key] = "masked_#{key}"
|
|
end
|
|
|
|
query_string_hash
|
|
end
|
|
|
|
def default_root_url
|
|
Gitlab::Routing.url_helpers.root_url(only_path: false)
|
|
end
|
|
end
|
|
|
|
def masked_page_url
|
|
return unless Feature.enabled?(:mask_page_urls, type: :ops)
|
|
|
|
current_group = group if defined?(group)
|
|
current_project = project if defined?(project)
|
|
mask_helper = MaskHelper.new(request, current_group, current_project)
|
|
mask_helper.mask_params
|
|
|
|
# We rescue all exception for time being till we test this helper extensively.
|
|
# Check https://gitlab.com/gitlab-org/gitlab/-/merge_requests/72864#note_711515501
|
|
rescue => e # rubocop:disable Style/RescueStandardError
|
|
Gitlab::ErrorTracking.track_exception(e, url: request.original_fullpath)
|
|
nil
|
|
end
|
|
end
|
|
end
|