blob: a44aaceba5828204cbcd3af63188cc17924c357c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
# frozen_string_literal: true
##
# The Uri handles rubygems source URIs.
#
class Gem::Uri
##
# Parses and redacts uri
def self.redact(uri)
new(uri).redacted
end
##
# Parses uri, raising if it's invalid
def self.parse!(uri)
require_relative "vendor/uri/lib/uri"
raise Gem::URI::InvalidURIError unless uri
return uri unless uri.is_a?(String)
# Always escape URI's to deal with potential spaces and such
# It should also be considered that source_uri may already be
# a valid URI with escaped characters. e.g. "{DESede}" is encoded
# as "%7BDESede%7D". If this is escaped again the percentage
# symbols will be escaped.
begin
Gem::URI.parse(uri)
rescue Gem::URI::InvalidURIError
Gem::URI.parse(Gem::URI::DEFAULT_PARSER.escape(uri))
end
end
##
# Parses uri, returning the original uri if it's invalid
def self.parse(uri)
parse!(uri)
rescue Gem::URI::InvalidURIError
uri
end
def initialize(source_uri)
@parsed_uri = parse(source_uri)
end
def redacted
return self unless valid_uri?
if token? || oauth_basic?
with_redacted_user
elsif password?
with_redacted_password
else
self
end
end
def to_s
@parsed_uri.to_s
end
def redact_credentials_from(text)
return text unless valid_uri? && password? && text.include?(to_s)
text.sub(password, "REDACTED")
end
def method_missing(method_name, *args, &blk)
if @parsed_uri.respond_to?(method_name)
@parsed_uri.send(method_name, *args, &blk)
else
super
end
end
def respond_to_missing?(method_name, include_private = false)
@parsed_uri.respond_to?(method_name, include_private) || super
end
protected
# Add a protected reader for the cloned instance to access the original object's parsed uri
attr_reader :parsed_uri
private
def parse!(uri)
self.class.parse!(uri)
end
def parse(uri)
self.class.parse(uri)
end
def with_redacted_user
clone.tap {|uri| uri.user = "REDACTED" }
end
def with_redacted_password
clone.tap {|uri| uri.password = "REDACTED" }
end
def valid_uri?
!@parsed_uri.is_a?(String)
end
def password?
!!password
end
def oauth_basic?
password == "x-oauth-basic"
end
def token?
!user.nil? && password.nil?
end
def initialize_copy(original)
@parsed_uri = original.parsed_uri.clone
end
end
|
