/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/urlutils.py

  • Committer: Robert Collins
  • Date: 2010-05-06 23:41:35 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506234135-yivbzczw1sejxnxc
Lock methods on ``Tree``, ``Branch`` and ``Repository`` are now
expected to return an object which can be used to unlock them. This reduces
duplicate code when using cleanups. The previous 'tokens's returned by
``Branch.lock_write`` and ``Repository.lock_write`` are now attributes
on the result of the lock_write. ``repository.RepositoryWriteLockResult``
and ``branch.BranchWriteLockResult`` document this. (Robert Collins)

``log._get_info_for_log_files`` now takes an add_cleanup callable.
(Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
"""A collection of function for handling URL operations."""
18
18
 
19
 
from __future__ import absolute_import
20
 
 
21
19
import os
22
20
import re
23
21
import sys
24
22
 
25
 
try:
26
 
    import urlparse
27
 
except ImportError:
28
 
    from urllib import parse as urlparse
 
23
from bzrlib.lazy_import import lazy_import
 
24
lazy_import(globals(), """
 
25
from posixpath import split as _posix_split, normpath as _posix_normpath
 
26
import urllib
 
27
import urlparse
29
28
 
30
 
from . import (
 
29
from bzrlib import (
31
30
    errors,
32
31
    osutils,
33
32
    )
34
 
 
35
 
from .lazy_import import lazy_import
36
 
lazy_import(globals(), """
37
 
from posixpath import split as _posix_split
38
33
""")
39
34
 
40
 
from .sixish import (
41
 
    int2byte,
42
 
    PY3,
43
 
    text_type,
44
 
    unichr,
45
 
    )
46
 
 
47
 
 
48
 
class InvalidURL(errors.PathError):
49
 
 
50
 
    _fmt = 'Invalid url supplied to transport: "%(path)s"%(extra)s'
51
 
 
52
 
 
53
 
class InvalidURLJoin(errors.PathError):
54
 
 
55
 
    _fmt = "Invalid URL join request: %(reason)s: %(base)r + %(join_args)r"
56
 
 
57
 
    def __init__(self, reason, base, join_args):
58
 
        self.reason = reason
59
 
        self.base = base
60
 
        self.join_args = join_args
61
 
        errors.PathError.__init__(self, base, reason)
62
 
 
63
 
 
64
 
class InvalidRebaseURLs(errors.PathError):
65
 
 
66
 
    _fmt = "URLs differ by more than path: %(from_)r and %(to)r"
67
 
 
68
 
    def __init__(self, from_, to):
69
 
        self.from_ = from_
70
 
        self.to = to
71
 
        errors.PathError.__init__(
72
 
            self, from_, 'URLs differ by more than path.')
73
 
 
74
35
 
75
36
def basename(url, exclude_trailing_slash=True):
76
37
    """Return the last component of a URL.
99
60
    return split(url, exclude_trailing_slash=exclude_trailing_slash)[0]
100
61
 
101
62
 
102
 
if PY3:
103
 
    quote_from_bytes = urlparse.quote_from_bytes
104
 
    quote = urlparse.quote
105
 
    unquote_to_bytes = urlparse.unquote_to_bytes
106
 
else:
107
 
    # Private copies of quote and unquote, copied from Python's urllib module
108
 
    # because urllib unconditionally imports socket, which imports ssl.
109
 
 
110
 
    always_safe = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
111
 
                   'abcdefghijklmnopqrstuvwxyz'
112
 
                   '0123456789' '_.-')
113
 
    _safe_map = {}
114
 
    for i, c in zip(range(256), ''.join(map(chr, range(256)))):
115
 
        _safe_map[c] = c if (
116
 
            i < 128 and c in always_safe) else '%{0:02X}'.format(i)
117
 
    _safe_quoters = {}
118
 
 
119
 
    def quote_from_bytes(s, safe='/'):
120
 
        """quote('abc def') -> 'abc%20def'
121
 
 
122
 
        Each part of a URL, e.g. the path info, the query, etc., has a
123
 
        different set of reserved characters that must be quoted.
124
 
 
125
 
        RFC 2396 Uniform Resource Identifiers (URI): Generic Syntax lists
126
 
        the following reserved characters.
127
 
 
128
 
        reserved    = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" |
129
 
                      "$" | ","
130
 
 
131
 
        Each of these characters is reserved in some component of a URL,
132
 
        but not necessarily in all of them.
133
 
 
134
 
        By default, the quote function is intended for quoting the path
135
 
        section of a URL.  Thus, it will not encode '/'.  This character
136
 
        is reserved, but in typical usage the quote function is being
137
 
        called on a path where the existing slash characters are used as
138
 
        reserved characters.
139
 
        """
140
 
        # fastpath
141
 
        if not s:
142
 
            if s is None:
143
 
                raise TypeError('None object cannot be quoted')
144
 
            return s
145
 
        cachekey = (safe, always_safe)
146
 
        try:
147
 
            (quoter, safe) = _safe_quoters[cachekey]
148
 
        except KeyError:
149
 
            safe_map = _safe_map.copy()
150
 
            safe_map.update([(c, c) for c in safe])
151
 
            quoter = safe_map.__getitem__
152
 
            safe = always_safe + safe
153
 
            _safe_quoters[cachekey] = (quoter, safe)
154
 
        if not s.rstrip(safe):
155
 
            return s
156
 
        return ''.join(map(quoter, s))
157
 
 
158
 
    quote = quote_from_bytes
159
 
    unquote_to_bytes = urlparse.unquote
160
 
 
161
 
 
162
 
unquote = urlparse.unquote
163
 
 
164
 
 
165
 
def escape(relpath, safe='/~'):
 
63
def escape(relpath):
166
64
    """Escape relpath to be a valid url."""
167
 
    if not isinstance(relpath, str) and sys.version_info[0] == 2:
168
 
        # GZ 2019-06-16: Should use _fs_enc instead here really?
 
65
    if isinstance(relpath, unicode):
169
66
        relpath = relpath.encode('utf-8')
170
 
    return quote(relpath, safe=safe)
 
67
    # After quoting and encoding, the path should be perfectly
 
68
    # safe as a plain ASCII string, str() just enforces this
 
69
    return str(urllib.quote(relpath, safe='/~'))
171
70
 
172
71
 
173
72
def file_relpath(base, path):
177
76
    """
178
77
    if len(base) < MIN_ABS_FILEURL_LENGTH:
179
78
        raise ValueError('Length of base (%r) must equal or'
180
 
                         ' exceed the platform minimum url length (which is %d)' %
181
 
                         (base, MIN_ABS_FILEURL_LENGTH))
182
 
    base = osutils.normpath(local_path_from_url(base))
183
 
    path = osutils.normpath(local_path_from_url(path))
 
79
            ' exceed the platform minimum url length (which is %d)' %
 
80
            (base, MIN_ABS_FILEURL_LENGTH))
 
81
    base = local_path_from_url(base)
 
82
    path = local_path_from_url(path)
184
83
    return escape(osutils.relpath(base, path))
185
84
 
186
85
 
202
101
    first_path_slash = path.find('/')
203
102
    if first_path_slash == -1:
204
103
        return len(scheme), None
205
 
    return len(scheme), first_path_slash + m.start('path')
206
 
 
207
 
 
208
 
def is_url(url):
209
 
    """Tests whether a URL is in actual fact a URL."""
210
 
    return _url_scheme_re.match(url) is not None
 
104
    return len(scheme), first_path_slash+len(scheme)+3
211
105
 
212
106
 
213
107
def join(base, *args):
224
118
    """
225
119
    if not args:
226
120
        return base
227
 
    scheme_end, path_start = _find_scheme_and_separator(base)
228
 
    if scheme_end is None and path_start is None:
229
 
        path_start = 0
230
 
    elif path_start is None:
231
 
        path_start = len(base)
232
 
    path = base[path_start:]
 
121
    match = _url_scheme_re.match(base)
 
122
    scheme = None
 
123
    if match:
 
124
        scheme = match.group('scheme')
 
125
        path = match.group('path').split('/')
 
126
        if path[-1:] == ['']:
 
127
            # Strip off a trailing slash
 
128
            # This helps both when we are at the root, and when
 
129
            # 'base' has an extra slash at the end
 
130
            path = path[:-1]
 
131
    else:
 
132
        path = base.split('/')
 
133
 
 
134
    if scheme is not None and len(path) >= 1:
 
135
        host = path[:1]
 
136
        # the path should be represented as an abs path.
 
137
        # we know this must be absolute because of the presence of a URL scheme.
 
138
        remove_root = True
 
139
        path = [''] + path[1:]
 
140
    else:
 
141
        # create an empty host, but dont alter the path - this might be a
 
142
        # relative url fragment.
 
143
        host = []
 
144
        remove_root = False
 
145
 
233
146
    for arg in args:
234
 
        arg_scheme_end, arg_path_start = _find_scheme_and_separator(arg)
235
 
        if arg_scheme_end is None and arg_path_start is None:
236
 
            arg_path_start = 0
237
 
        elif arg_path_start is None:
238
 
            arg_path_start = len(arg)
239
 
        if arg_scheme_end is not None:
240
 
            base = arg
241
 
            path = arg[arg_path_start:]
242
 
            scheme_end = arg_scheme_end
243
 
            path_start = arg_path_start
 
147
        match = _url_scheme_re.match(arg)
 
148
        if match:
 
149
            # Absolute URL
 
150
            scheme = match.group('scheme')
 
151
            # this skips .. normalisation, making http://host/../../..
 
152
            # be rather strange.
 
153
            path = match.group('path').split('/')
 
154
            # set the host and path according to new absolute URL, discarding
 
155
            # any previous values.
 
156
            # XXX: duplicates mess from earlier in this function.  This URL
 
157
            # manipulation code needs some cleaning up.
 
158
            if scheme is not None and len(path) >= 1:
 
159
                host = path[:1]
 
160
                path = path[1:]
 
161
                # url scheme implies absolute path.
 
162
                path = [''] + path
 
163
            else:
 
164
                # no url scheme we take the path as is.
 
165
                host = []
244
166
        else:
 
167
            path = '/'.join(path)
245
168
            path = joinpath(path, arg)
246
 
    return base[:path_start] + path
 
169
            path = path.split('/')
 
170
    if remove_root and path[0:1] == ['']:
 
171
        del path[0]
 
172
    if host:
 
173
        # Remove the leading slash from the path, so long as it isn't also the
 
174
        # trailing slash, which we want to keep if present.
 
175
        if path and path[0] == '' and len(path) > 1:
 
176
            del path[0]
 
177
        path = host + path
 
178
 
 
179
    if scheme is None:
 
180
        return '/'.join(path)
 
181
    return scheme + '://' + '/'.join(path)
247
182
 
248
183
 
249
184
def joinpath(base, *args):
258
193
    """
259
194
    path = base.split('/')
260
195
    if len(path) > 1 and path[-1] == '':
261
 
        # If the path ends in a trailing /, remove it.
 
196
        #If the path ends in a trailing /, remove it.
262
197
        path.pop()
263
198
    for arg in args:
264
199
        if arg.startswith('/'):
268
203
                continue
269
204
            elif chunk == '..':
270
205
                if path == ['']:
271
 
                    raise InvalidURLJoin('Cannot go above root',
272
 
                                         base, args)
 
206
                    raise errors.InvalidURLJoin('Cannot go above root',
 
207
                            base, args)
273
208
                path.pop()
274
209
            else:
275
210
                path.append(chunk)
282
217
# jam 20060502 Sorted to 'l' because the final target is 'local_path_from_url'
283
218
def _posix_local_path_from_url(url):
284
219
    """Convert a url like file:///path/to/foo into /path/to/foo"""
285
 
    url = strip_segment_parameters(url)
286
220
    file_localhost_prefix = 'file://localhost/'
287
221
    if url.startswith(file_localhost_prefix):
288
222
        path = url[len(file_localhost_prefix) - 1:]
289
223
    elif not url.startswith('file:///'):
290
 
        raise InvalidURL(
 
224
        raise errors.InvalidURL(
291
225
            url, 'local urls must start with file:/// or file://localhost/')
292
226
    else:
293
227
        path = url[len('file://'):]
302
236
    """
303
237
    # importing directly from posixpath allows us to test this
304
238
    # on non-posix platforms
305
 
    return 'file://' + escape(osutils._posix_abspath(path))
 
239
    return 'file://' + escape(_posix_normpath(
 
240
        osutils._posix_abspath(path)))
306
241
 
307
242
 
308
243
def _win32_local_path_from_url(url):
309
244
    """Convert a url like file:///C:/path/to/foo into C:/path/to/foo"""
310
245
    if not url.startswith('file://'):
311
 
        raise InvalidURL(url, 'local urls must start with file:///, '
312
 
                         'UNC path urls must start with file://')
313
 
    url = strip_segment_parameters(url)
 
246
        raise errors.InvalidURL(url, 'local urls must start with file:///, '
 
247
                                     'UNC path urls must start with file://')
314
248
    # We strip off all 3 slashes
315
249
    win32_url = url[len('file:'):]
316
250
    # check for UNC path: //HOST/path
317
251
    if not win32_url.startswith('///'):
318
252
        if (win32_url[2] == '/'
319
 
                or win32_url[3] in '|:'):
320
 
            raise InvalidURL(url, 'Win32 UNC path urls'
321
 
                             ' have form file://HOST/path')
 
253
            or win32_url[3] in '|:'):
 
254
            raise errors.InvalidURL(url, 'Win32 UNC path urls'
 
255
                ' have form file://HOST/path')
322
256
        return unescape(win32_url)
323
257
 
324
258
    # allow empty paths so we can serve all roots
326
260
        return '/'
327
261
 
328
262
    # usual local path with drive letter
329
 
    if (len(win32_url) < 6
330
 
        or win32_url[3] not in ('abcdefghijklmnopqrstuvwxyz'
331
 
                                'ABCDEFGHIJKLMNOPQRSTUVWXYZ') or
332
 
        win32_url[4] not in '|:'
333
 
            or win32_url[5] != '/'):
334
 
        raise InvalidURL(url, 'Win32 file urls start with'
335
 
                         ' file:///x:/, where x is a valid drive letter')
 
263
    if (win32_url[3] not in ('abcdefghijklmnopqrstuvwxyz'
 
264
                             'ABCDEFGHIJKLMNOPQRSTUVWXYZ')
 
265
        or win32_url[4] not in  '|:'
 
266
        or win32_url[5] != '/'):
 
267
        raise errors.InvalidURL(url, 'Win32 file urls start with'
 
268
                ' file:///x:/, where x is a valid drive letter')
336
269
    return win32_url[3].upper() + u':' + unescape(win32_url[5:])
337
270
 
338
271
 
345
278
    # on non-win32 platform
346
279
    # FIXME: It turns out that on nt, ntpath.abspath uses nt._getfullpathname
347
280
    #       which actually strips trailing space characters.
348
 
    #       The worst part is that on linux ntpath.abspath has different
 
281
    #       The worst part is that under linux ntpath.abspath has different
349
282
    #       semantics, since 'nt' is not an available module.
350
283
    if path == '/':
351
284
        return 'file:///'
355
288
    if win32_path.startswith('//'):
356
289
        return 'file:' + escape(win32_path)
357
290
    return ('file:///' + str(win32_path[0].upper()) + ':' +
358
 
            escape(win32_path[2:]))
 
291
        escape(win32_path[2:]))
359
292
 
360
293
 
361
294
local_path_to_url = _posix_local_path_to_url
370
303
    MIN_ABS_FILEURL_LENGTH = WIN32_MIN_ABS_FILEURL_LENGTH
371
304
 
372
305
 
373
 
_url_scheme_re = re.compile('^(?P<scheme>[^:/]{2,}):(//)?(?P<path>.*)$')
374
 
_url_hex_escapes_re = re.compile('(%[0-9a-fA-F]{2})')
 
306
_url_scheme_re = re.compile(r'^(?P<scheme>[^:/]{2,})://(?P<path>.*)$')
 
307
_url_hex_escapes_re = re.compile(r'(%[0-9a-fA-F]{2})')
375
308
 
376
309
 
377
310
def _unescape_safe_chars(matchobj):
406
339
    :param url: Either a hybrid URL or a local path
407
340
    :return: A normalized URL which only includes 7-bit ASCII characters.
408
341
    """
409
 
    scheme_end, path_start = _find_scheme_and_separator(url)
410
 
    if scheme_end is None:
 
342
    m = _url_scheme_re.match(url)
 
343
    if not m:
411
344
        return local_path_to_url(url)
412
 
    prefix = url[:path_start]
413
 
    path = url[path_start:]
414
 
    if not isinstance(url, text_type):
 
345
    scheme = m.group('scheme')
 
346
    path = m.group('path')
 
347
    if not isinstance(url, unicode):
415
348
        for c in url:
416
349
            if c not in _url_safe_characters:
417
 
                raise InvalidURL(url, 'URLs can only contain specific'
418
 
                                 ' safe characters (not %r)' % c)
 
350
                raise errors.InvalidURL(url, 'URLs can only contain specific'
 
351
                                            ' safe characters (not %r)' % c)
419
352
        path = _url_hex_escapes_re.sub(_unescape_safe_chars, path)
420
 
        return str(prefix + ''.join(path))
 
353
        return str(scheme + '://' + ''.join(path))
421
354
 
422
355
    # We have a unicode (hybrid) url
423
356
    path_chars = list(path)
424
357
 
425
 
    for i in range(len(path_chars)):
 
358
    for i in xrange(len(path_chars)):
426
359
        if path_chars[i] not in _url_safe_characters:
 
360
            chars = path_chars[i].encode('utf-8')
427
361
            path_chars[i] = ''.join(
428
 
                ['%%%02X' % c for c in bytearray(path_chars[i].encode('utf-8'))])
 
362
                ['%%%02X' % ord(c) for c in path_chars[i].encode('utf-8')])
429
363
    path = ''.join(path_chars)
430
364
    path = _url_hex_escapes_re.sub(_unescape_safe_chars, path)
431
 
    return str(prefix + path)
 
365
    return str(scheme + '://' + path)
432
366
 
433
367
 
434
368
def relative_url(base, other):
451
385
    if base_scheme != other_scheme:
452
386
        return other
453
387
    elif sys.platform == 'win32' and base_scheme == 'file://':
454
 
        base_drive = base[base_first_slash + 1:base_first_slash + 3]
455
 
        other_drive = other[other_first_slash + 1:other_first_slash + 3]
 
388
        base_drive = base[base_first_slash+1:base_first_slash+3]
 
389
        other_drive = other[other_first_slash+1:other_first_slash+3]
456
390
        if base_drive != other_drive:
457
391
            return other
458
392
 
459
 
    base_path = base[base_first_slash + 1:]
460
 
    other_path = other[other_first_slash + 1:]
 
393
    base_path = base[base_first_slash+1:]
 
394
    other_path = other[other_first_slash+1:]
461
395
 
462
396
    if base_path.endswith('/'):
463
397
        base_path = base_path[:-1]
487
421
    """On win32 the drive letter needs to be added to the url base."""
488
422
    # Strip off the drive letter
489
423
    # path is currently /C:/foo
490
 
    if len(path) < 4 or path[2] not in ':|' or path[3] != '/':
491
 
        raise InvalidURL(url_base + path,
492
 
                         'win32 file:/// paths need a drive letter')
493
 
    url_base += path[0:3]  # file:// + /C:
494
 
    path = path[3:]  # /foo
 
424
    if len(path) < 3 or path[2] not in ':|' or path[3] != '/':
 
425
        raise errors.InvalidURL(url_base + path,
 
426
            'win32 file:/// paths need a drive letter')
 
427
    url_base += path[0:3] # file:// + /C:
 
428
    path = path[3:] # /foo
495
429
    return url_base, path
496
430
 
497
431
 
502
436
    :param exclude_trailing_slash: Strip off a final '/' if it is part
503
437
        of the path (but not if it is part of the protocol specification)
504
438
 
505
 
    :return: (parent_url, child_dir).  child_dir may be the empty string if
506
 
        we're at the root.
 
439
    :return: (parent_url, child_dir).  child_dir may be the empty string if we're at
 
440
        the root.
507
441
    """
508
442
    scheme_loc, first_path_slash = _find_scheme_and_separator(url)
509
443
 
519
453
            return url, ''
520
454
 
521
455
    # We have a fully defined path
522
 
    url_base = url[:first_path_slash]  # http://host, file://
523
 
    path = url[first_path_slash:]  # /file/foo
 
456
    url_base = url[:first_path_slash] # http://host, file://
 
457
    path = url[first_path_slash:] # /file/foo
524
458
 
525
459
    if sys.platform == 'win32' and url.startswith('file:///'):
526
460
        # Strip off the drive letter
535
469
    return url_base + head, tail
536
470
 
537
471
 
538
 
def split_segment_parameters_raw(url):
539
 
    """Split the subsegment of the last segment of a URL.
540
 
 
541
 
    :param url: A relative or absolute URL
542
 
    :return: (url, subsegments)
543
 
    """
544
 
    # GZ 2011-11-18: Dodgy removing the terminal slash like this, function
545
 
    #                operates on urls not url+segments, and Transport classes
546
 
    #                should not be blindly adding slashes in the first place.
547
 
    lurl = strip_trailing_slash(url)
548
 
    # Segments begin at first comma after last forward slash, if one exists
549
 
    segment_start = lurl.find(",", lurl.rfind("/") + 1)
550
 
    if segment_start == -1:
551
 
        return (url, [])
552
 
    return (lurl[:segment_start],
553
 
            [str(s) for s in lurl[segment_start + 1:].split(",")])
554
 
 
555
 
 
556
 
def split_segment_parameters(url):
557
 
    """Split the segment parameters of the last segment of a URL.
558
 
 
559
 
    :param url: A relative or absolute URL
560
 
    :return: (url, segment_parameters)
561
 
    """
562
 
    (base_url, subsegments) = split_segment_parameters_raw(url)
563
 
    parameters = {}
564
 
    for subsegment in subsegments:
565
 
        try:
566
 
            (key, value) = subsegment.split("=", 1)
567
 
        except ValueError:
568
 
            raise InvalidURL(url, "missing = in subsegment")
569
 
        if not isinstance(key, str):
570
 
            raise TypeError(key)
571
 
        if not isinstance(value, str):
572
 
            raise TypeError(value)
573
 
        parameters[key] = value
574
 
    return (base_url, parameters)
575
 
 
576
 
 
577
 
def strip_segment_parameters(url):
578
 
    """Strip the segment parameters from a URL.
579
 
 
580
 
    :param url: A relative or absolute URL
581
 
    :return: url
582
 
    """
583
 
    base_url, subsegments = split_segment_parameters_raw(url)
584
 
    return base_url
585
 
 
586
 
 
587
 
def join_segment_parameters_raw(base, *subsegments):
588
 
    """Create a new URL by adding subsegments to an existing one.
589
 
 
590
 
    This adds the specified subsegments to the last path in the specified
591
 
    base URL. The subsegments should be bytestrings.
592
 
 
593
 
    :note: You probably want to use join_segment_parameters instead.
594
 
    """
595
 
    if not subsegments:
596
 
        return base
597
 
    for subsegment in subsegments:
598
 
        if not isinstance(subsegment, str):
599
 
            raise TypeError("Subsegment %r is not a bytestring" % subsegment)
600
 
        if "," in subsegment:
601
 
            raise InvalidURLJoin(", exists in subsegments",
602
 
                                 base, subsegments)
603
 
    return ",".join((base,) + subsegments)
604
 
 
605
 
 
606
 
def join_segment_parameters(url, parameters):
607
 
    """Create a new URL by adding segment parameters to an existing one.
608
 
 
609
 
    The parameters of the last segment in the URL will be updated; if a
610
 
    parameter with the same key already exists it will be overwritten.
611
 
 
612
 
    :param url: A URL, as string
613
 
    :param parameters: Dictionary of parameters, keys and values as bytestrings
614
 
    """
615
 
    (base, existing_parameters) = split_segment_parameters(url)
616
 
    new_parameters = {}
617
 
    new_parameters.update(existing_parameters)
618
 
    for key, value in parameters.items():
619
 
        if not isinstance(key, str):
620
 
            raise TypeError("parameter key %r is not a str" % key)
621
 
        if not isinstance(value, str):
622
 
            raise TypeError("parameter value %r for %r is not a str" %
623
 
                            (value, key))
624
 
        if "=" in key:
625
 
            raise InvalidURLJoin("= exists in parameter key", url,
626
 
                                 parameters)
627
 
        new_parameters[key] = value
628
 
    return join_segment_parameters_raw(
629
 
        base, *["%s=%s" % item for item in sorted(new_parameters.items())])
630
 
 
631
 
 
632
472
def _win32_strip_local_trailing_slash(url):
633
473
    """Strip slashes after the drive letter"""
634
474
    if len(url) > WIN32_MIN_ABS_FILEURL_LENGTH:
670
510
        # so just chop off the last character
671
511
        return url[:-1]
672
512
 
673
 
    if first_path_slash is None or first_path_slash == len(url) - 1:
 
513
    if first_path_slash is None or first_path_slash == len(url)-1:
674
514
        # Don't chop off anything if the only slash is the path
675
515
        # separating slash
676
516
        return url
684
524
    This returns a Unicode path from a URL
685
525
    """
686
526
    # jam 20060427 URLs are supposed to be ASCII only strings
687
 
    #       If they are passed in as unicode, unquote
 
527
    #       If they are passed in as unicode, urllib.unquote
688
528
    #       will return a UNICODE string, which actually contains
689
529
    #       utf-8 bytes. So we have to ensure that they are
690
530
    #       plain ASCII strings, or the final .decode will
691
531
    #       try to encode the UNICODE => ASCII, and then decode
692
532
    #       it into utf-8.
 
533
    try:
 
534
        url = str(url)
 
535
    except UnicodeError, e:
 
536
        raise errors.InvalidURL(url, 'URL was not a plain ASCII url: %s' % (e,))
693
537
 
694
 
    if PY3:
695
 
        if isinstance(url, text_type):
696
 
            try:
697
 
                url.encode("ascii")
698
 
            except UnicodeError as e:
699
 
                raise InvalidURL(
700
 
                    url, 'URL was not a plain ASCII url: %s' % (e,))
701
 
        return urlparse.unquote(url)
702
 
    else:
703
 
        if isinstance(url, text_type):
704
 
            try:
705
 
                url = url.encode("ascii")
706
 
            except UnicodeError as e:
707
 
                raise InvalidURL(
708
 
                    url, 'URL was not a plain ASCII url: %s' % (e,))
709
 
        unquoted = unquote(url)
710
 
        try:
711
 
            unicode_path = unquoted.decode('utf-8')
712
 
        except UnicodeError as e:
713
 
            raise InvalidURL(
714
 
                url, 'Unable to encode the URL as utf-8: %s' % (e,))
715
 
        return unicode_path
 
538
    unquoted = urllib.unquote(url)
 
539
    try:
 
540
        unicode_path = unquoted.decode('utf-8')
 
541
    except UnicodeError, e:
 
542
        raise errors.InvalidURL(url, 'Unable to encode the URL as utf-8: %s' % (e,))
 
543
    return unicode_path
716
544
 
717
545
 
718
546
# These are characters that if escaped, should stay that way
719
547
_no_decode_chars = ';/?:@&=+$,#'
720
548
_no_decode_ords = [ord(c) for c in _no_decode_chars]
721
549
_no_decode_hex = (['%02x' % o for o in _no_decode_ords]
722
 
                  + ['%02X' % o for o in _no_decode_ords])
723
 
_hex_display_map = dict(([('%02x' % o, int2byte(o)) for o in range(256)]
724
 
                         + [('%02X' % o, int2byte(o)) for o in range(256)]))
725
 
# These entries get mapped to themselves
726
 
_hex_display_map.update((hex, b'%' + hex.encode('ascii'))
727
 
                        for hex in _no_decode_hex)
 
550
                + ['%02X' % o for o in _no_decode_ords])
 
551
_hex_display_map = dict(([('%02x' % o, chr(o)) for o in range(256)]
 
552
                    + [('%02X' % o, chr(o)) for o in range(256)]))
 
553
#These entries get mapped to themselves
 
554
_hex_display_map.update((hex,'%'+hex) for hex in _no_decode_hex)
728
555
 
729
556
# These characters shouldn't be percent-encoded, and it's always safe to
730
557
# unencode them if they are.
731
558
_url_dont_escape_characters = set(
732
 
    "abcdefghijklmnopqrstuvwxyz"  # Lowercase alpha
733
 
    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"  # Uppercase alpha
734
 
    "0123456789"  # Numbers
735
 
    "-._~"  # Unreserved characters
 
559
   "abcdefghijklmnopqrstuvwxyz" # Lowercase alpha
 
560
   "ABCDEFGHIJKLMNOPQRSTUVWXYZ" # Uppercase alpha
 
561
   "0123456789" # Numbers
 
562
   "-._~"  # Unreserved characters
736
563
)
737
564
 
738
565
# These characters should not be escaped
739
566
_url_safe_characters = set(
740
 
    "abcdefghijklmnopqrstuvwxyz"  # Lowercase alpha
741
 
    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"  # Uppercase alpha
742
 
    "0123456789"  # Numbers
743
 
    "_.-!~*'()"  # Unreserved characters
744
 
    "/;?:@&=+$,"  # Reserved characters
745
 
    "%#"         # Extra reserved characters
 
567
   "abcdefghijklmnopqrstuvwxyz" # Lowercase alpha
 
568
   "ABCDEFGHIJKLMNOPQRSTUVWXYZ" # Uppercase alpha
 
569
   "0123456789" # Numbers
 
570
   "_.-!~*'()"  # Unreserved characters
 
571
   "/;?:@&=+$," # Reserved characters
 
572
   "%#"         # Extra reserved characters
746
573
)
747
574
 
748
 
 
749
 
def _unescape_segment_for_display(segment, encoding):
750
 
    """Unescape a segment for display.
751
 
 
752
 
    Helper for unescape_for_display
753
 
 
754
 
    :param url: A 7-bit ASCII URL
755
 
    :param encoding: The final output encoding
756
 
 
757
 
    :return: A unicode string which can be safely encoded into the
758
 
         specified encoding.
759
 
    """
760
 
    escaped_chunks = segment.split('%')
761
 
    escaped_chunks[0] = escaped_chunks[0].encode('utf-8')
762
 
    for j in range(1, len(escaped_chunks)):
763
 
        item = escaped_chunks[j]
764
 
        try:
765
 
            escaped_chunks[j] = _hex_display_map[item[:2]]
766
 
        except KeyError:
767
 
            # Put back the percent symbol
768
 
            escaped_chunks[j] = b'%' + \
769
 
                (item[:2].encode('utf-8') if PY3 else item[:2])
770
 
        except UnicodeDecodeError:
771
 
            escaped_chunks[j] = unichr(int(item[:2], 16)).encode('utf-8')
772
 
        escaped_chunks[j] += (item[2:].encode('utf-8') if PY3 else item[2:])
773
 
    unescaped = b''.join(escaped_chunks)
774
 
    try:
775
 
        decoded = unescaped.decode('utf-8')
776
 
    except UnicodeDecodeError:
777
 
        # If this path segment cannot be properly utf-8 decoded
778
 
        # after doing unescaping we will just leave it alone
779
 
        return segment
780
 
    else:
781
 
        try:
782
 
            decoded.encode(encoding)
783
 
        except UnicodeEncodeError:
784
 
            # If this chunk cannot be encoded in the local
785
 
            # encoding, then we should leave it alone
786
 
            return segment
787
 
        else:
788
 
            # Otherwise take the url decoded one
789
 
            return decoded
790
 
 
791
 
 
792
575
def unescape_for_display(url, encoding):
793
576
    """Decode what you can for a URL, so that we get a nice looking path.
794
577
 
816
599
 
817
600
    # Split into sections to try to decode utf-8
818
601
    res = url.split('/')
819
 
    for i in range(1, len(res)):
820
 
        res[i] = _unescape_segment_for_display(res[i], encoding)
 
602
    for i in xrange(1, len(res)):
 
603
        escaped_chunks = res[i].split('%')
 
604
        for j in xrange(1, len(escaped_chunks)):
 
605
            item = escaped_chunks[j]
 
606
            try:
 
607
                escaped_chunks[j] = _hex_display_map[item[:2]] + item[2:]
 
608
            except KeyError:
 
609
                # Put back the percent symbol
 
610
                escaped_chunks[j] = '%' + item
 
611
            except UnicodeDecodeError:
 
612
                escaped_chunks[j] = unichr(int(item[:2], 16)) + item[2:]
 
613
        unescaped = ''.join(escaped_chunks)
 
614
        try:
 
615
            decoded = unescaped.decode('utf-8')
 
616
        except UnicodeDecodeError:
 
617
            # If this path segment cannot be properly utf-8 decoded
 
618
            # after doing unescaping we will just leave it alone
 
619
            pass
 
620
        else:
 
621
            try:
 
622
                decoded.encode(encoding)
 
623
            except UnicodeEncodeError:
 
624
                # If this chunk cannot be encoded in the local
 
625
                # encoding, then we should leave it alone
 
626
                pass
 
627
            else:
 
628
                # Otherwise take the url decoded one
 
629
                res[i] = decoded
821
630
    return u'/'.join(res)
822
631
 
823
632
 
832
641
    is used without a path, e.g. c:foo-bar => foo-bar.
833
642
    If no /, path separator or : is found, the from_location is returned.
834
643
    """
835
 
    from_location = strip_segment_parameters(from_location)
836
644
    if from_location.find("/") >= 0 or from_location.find(os.sep) >= 0:
837
645
        return os.path.basename(from_location.rstrip("/\\"))
838
646
    else:
839
647
        sep = from_location.find(":")
840
648
        if sep > 0:
841
 
            return from_location[sep + 1:]
 
649
            return from_location[sep+1:]
842
650
        else:
843
651
            return from_location
844
652
 
861
669
    old_parsed = urlparse.urlparse(old_base)
862
670
    new_parsed = urlparse.urlparse(new_base)
863
671
    if (old_parsed[:2]) != (new_parsed[:2]):
864
 
        raise InvalidRebaseURLs(old_base, new_base)
 
672
        raise errors.InvalidRebaseURLs(old_base, new_base)
865
673
    return determine_relative_path(new_parsed[2],
866
674
                                   join(old_parsed[2], url))
867
675
 
872
680
    to_segments = osutils.splitpath(to_path)
873
681
    count = -1
874
682
    for count, (from_element, to_element) in enumerate(zip(from_segments,
875
 
                                                           to_segments)):
 
683
                                                       to_segments)):
876
684
        if from_element != to_element:
877
685
            break
878
686
    else:
885
693
    return osutils.pathjoin(*segments)
886
694
 
887
695
 
888
 
class URL(object):
889
 
    """Parsed URL."""
890
 
 
891
 
    def __init__(self, scheme, quoted_user, quoted_password, quoted_host,
892
 
                 port, quoted_path):
893
 
        self.scheme = scheme
894
 
        self.quoted_host = quoted_host
895
 
        self.host = unquote(self.quoted_host)
896
 
        self.quoted_user = quoted_user
897
 
        if self.quoted_user is not None:
898
 
            self.user = unquote(self.quoted_user)
899
 
        else:
900
 
            self.user = None
901
 
        self.quoted_password = quoted_password
902
 
        if self.quoted_password is not None:
903
 
            self.password = unquote(self.quoted_password)
904
 
        else:
905
 
            self.password = None
906
 
        self.port = port
907
 
        self.quoted_path = _url_hex_escapes_re.sub(
908
 
            _unescape_safe_chars, quoted_path)
909
 
        self.path = unquote(self.quoted_path)
910
 
 
911
 
    def __eq__(self, other):
912
 
        return (isinstance(other, self.__class__) and
913
 
                self.scheme == other.scheme and
914
 
                self.host == other.host and
915
 
                self.user == other.user and
916
 
                self.password == other.password and
917
 
                self.path == other.path)
918
 
 
919
 
    def __repr__(self):
920
 
        return "<%s(%r, %r, %r, %r, %r, %r)>" % (
921
 
            self.__class__.__name__,
922
 
            self.scheme, self.quoted_user, self.quoted_password,
923
 
            self.quoted_host, self.port, self.quoted_path)
924
 
 
925
 
    @classmethod
926
 
    def from_string(cls, url):
927
 
        """Create a URL object from a string.
928
 
 
929
 
        :param url: URL as bytestring
930
 
        """
931
 
        # GZ 2017-06-09: Actually validate ascii-ness
932
 
        # pad.lv/1696545: For the moment, accept both native strings and
933
 
        # unicode.
934
 
        if isinstance(url, str):
935
 
            pass
936
 
        elif isinstance(url, text_type):
937
 
            try:
938
 
                url = url.encode()
939
 
            except UnicodeEncodeError:
940
 
                raise InvalidURL(url)
941
 
        else:
942
 
            raise InvalidURL(url)
943
 
        (scheme, netloc, path, params,
944
 
         query, fragment) = urlparse.urlparse(url, allow_fragments=False)
945
 
        user = password = host = port = None
946
 
        if '@' in netloc:
947
 
            user, host = netloc.rsplit('@', 1)
948
 
            if ':' in user:
949
 
                user, password = user.split(':', 1)
950
 
        else:
951
 
            host = netloc
952
 
 
953
 
        if ':' in host and not (host[0] == '[' and host[-1] == ']'):
954
 
            # there *is* port
955
 
            host, port = host.rsplit(':', 1)
956
 
            if port:
957
 
                try:
958
 
                    port = int(port)
959
 
                except ValueError:
960
 
                    raise InvalidURL('invalid port number %s in url:\n%s' %
961
 
                                     (port, url))
962
 
            else:
963
 
                port = None
964
 
        if host != "" and host[0] == '[' and host[-1] == ']':  # IPv6
965
 
            host = host[1:-1]
966
 
 
967
 
        return cls(scheme, user, password, host, port, path)
968
 
 
969
 
    def __str__(self):
970
 
        netloc = self.quoted_host
971
 
        if ":" in netloc:
972
 
            netloc = "[%s]" % netloc
973
 
        if self.quoted_user is not None:
974
 
            # Note that we don't put the password back even if we
975
 
            # have one so that it doesn't get accidentally
976
 
            # exposed.
977
 
            netloc = '%s@%s' % (self.quoted_user, netloc)
978
 
        if self.port is not None:
979
 
            netloc = '%s:%d' % (netloc, self.port)
980
 
        return urlparse.urlunparse(
981
 
            (self.scheme, netloc, self.quoted_path, None, None, None))
982
 
 
983
 
    @staticmethod
984
 
    def _combine_paths(base_path, relpath):
985
 
        """Transform a Transport-relative path to a remote absolute path.
986
 
 
987
 
        This does not handle substitution of ~ but does handle '..' and '.'
988
 
        components.
989
 
 
990
 
        Examples::
991
 
 
992
 
            t._combine_paths('/home/sarah', 'project/foo')
993
 
                => '/home/sarah/project/foo'
994
 
            t._combine_paths('/home/sarah', '../../etc')
995
 
                => '/etc'
996
 
            t._combine_paths('/home/sarah', '/etc')
997
 
                => '/etc'
998
 
 
999
 
        :param base_path: base path
1000
 
        :param relpath: relative url string for relative part of remote path.
1001
 
        :return: urlencoded string for final path.
1002
 
        """
1003
 
        # pad.lv/1696545: For the moment, accept both native strings and
1004
 
        # unicode.
1005
 
        if isinstance(relpath, str):
1006
 
            pass
1007
 
        elif isinstance(relpath, text_type):
1008
 
            try:
1009
 
                relpath = relpath.encode()
1010
 
            except UnicodeEncodeError:
1011
 
                raise InvalidURL(relpath)
1012
 
        else:
1013
 
            raise InvalidURL(relpath)
1014
 
        relpath = _url_hex_escapes_re.sub(_unescape_safe_chars, relpath)
1015
 
        if relpath.startswith('/'):
1016
 
            base_parts = []
1017
 
        else:
1018
 
            base_parts = base_path.split('/')
1019
 
        if len(base_parts) > 0 and base_parts[-1] == '':
1020
 
            base_parts = base_parts[:-1]
1021
 
        for p in relpath.split('/'):
1022
 
            if p == '..':
1023
 
                if len(base_parts) == 0:
1024
 
                    # In most filesystems, a request for the parent
1025
 
                    # of root, just returns root.
1026
 
                    continue
1027
 
                base_parts.pop()
1028
 
            elif p == '.':
1029
 
                continue  # No-op
1030
 
            elif p != '':
1031
 
                base_parts.append(p)
1032
 
        path = '/'.join(base_parts)
1033
 
        if not path.startswith('/'):
1034
 
            path = '/' + path
1035
 
        return path
1036
 
 
1037
 
    def clone(self, offset=None):
1038
 
        """Return a new URL for a path relative to this URL.
1039
 
 
1040
 
        :param offset: A relative path, already urlencoded
1041
 
        :return: `URL` instance
1042
 
        """
1043
 
        if offset is not None:
1044
 
            relative = unescape(offset)
1045
 
            if sys.version_info[0] == 2:
1046
 
                relative = relative.encode('utf-8')
1047
 
            path = self._combine_paths(self.path, relative)
1048
 
            path = quote(path, safe="/~")
1049
 
        else:
1050
 
            path = self.quoted_path
1051
 
        return self.__class__(self.scheme, self.quoted_user,
1052
 
                              self.quoted_password, self.quoted_host, self.port,
1053
 
                              path)
1054
 
 
1055
696
 
1056
697
def parse_url(url):
1057
698
    """Extract the server address, the credentials and the path from the url.
1060
701
    chars.
1061
702
 
1062
703
    :param url: an quoted url
 
704
 
1063
705
    :return: (scheme, user, password, host, port, path) tuple, all fields
1064
706
        are unquoted.
1065
707
    """
1066
 
    parsed_url = URL.from_string(url)
1067
 
    return (parsed_url.scheme, parsed_url.user, parsed_url.password,
1068
 
            parsed_url.host, parsed_url.port, parsed_url.path)
 
708
    if isinstance(url, unicode):
 
709
        raise errors.InvalidURL('should be ascii:\n%r' % url)
 
710
    url = url.encode('utf-8')
 
711
    (scheme, netloc, path, params,
 
712
     query, fragment) = urlparse.urlparse(url, allow_fragments=False)
 
713
    user = password = host = port = None
 
714
    if '@' in netloc:
 
715
        user, host = netloc.rsplit('@', 1)
 
716
        if ':' in user:
 
717
            user, password = user.split(':', 1)
 
718
            password = urllib.unquote(password)
 
719
        user = urllib.unquote(user)
 
720
    else:
 
721
        host = netloc
 
722
 
 
723
    if ':' in host and not (host[0] == '[' and host[-1] == ']'): #there *is* port
 
724
        host, port = host.rsplit(':',1)
 
725
        try:
 
726
            port = int(port)
 
727
        except ValueError:
 
728
            raise errors.InvalidURL('invalid port number %s in url:\n%s' %
 
729
                                    (port, url))
 
730
    if host != "" and host[0] == '[' and host[-1] == ']': #IPv6
 
731
        host = host[1:-1]
 
732
 
 
733
    host = urllib.unquote(host)
 
734
    path = urllib.unquote(path)
 
735
 
 
736
    return (scheme, user, password, host, port, path)