/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 remote.py

  • Committer: Jelmer Vernooij
  • Date: 2018-04-02 14:59:43 UTC
  • mto: (0.200.1913 work)
  • mto: This revision was merged to the branch mainline in revision 6960.
  • Revision ID: jelmer@jelmer.uk-20180402145943-s5jmpbvvf1x42pao
Just don't touch the URL if it's already a valid URL.

Show diffs side-by-side

added added

removed removed

Lines of Context:
remote.py
 
1
# Copyright (C) 2007-2018 Jelmer Vernooij <jelmer@jelmer.uk>
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
"""Remote dirs, repositories and branches."""
 
18
 
 
19
from __future__ import absolute_import
 
20
 
 
21
import re
 
22
 
 
23
from ... import (
 
24
    config,
 
25
    debug,
 
26
    errors,
 
27
    trace,
 
28
    ui,
 
29
    urlutils,
 
30
    )
 
31
from ...push import (
 
32
    PushResult,
 
33
    )
 
34
from ...errors import (
 
35
    AlreadyBranchError,
 
36
    BzrError,
 
37
    DivergedBranches,
 
38
    InProcessTransport,
 
39
    InvalidRevisionId,
 
40
    NoSuchFile,
 
41
    NoSuchRevision,
 
42
    NoSuchTag,
 
43
    NotBranchError,
 
44
    NotLocalUrl,
 
45
    NoWorkingTree,
 
46
    UninitializableFormat,
 
47
    )
 
48
from ...transport import (
 
49
    Transport,
 
50
    )
 
51
 
 
52
from . import (
 
53
    lazy_check_versions,
 
54
    )
 
55
lazy_check_versions()
 
56
 
 
57
from .branch import (
 
58
    GitBranch,
 
59
    GitBranchFormat,
 
60
    GitBranchPushResult,
 
61
    GitTags,
 
62
    _quick_lookup_revno,
 
63
    )
 
64
from .dir import (
 
65
    GitControlDirFormat,
 
66
    GitDir,
 
67
    BareLocalGitControlDirFormat,
 
68
    )
 
69
from .errors import (
 
70
    GitSmartRemoteNotSupported,
 
71
    NoSuchRef,
 
72
    )
 
73
from .mapping import (
 
74
    mapping_registry,
 
75
    )
 
76
from .object_store import (
 
77
    get_object_store,
 
78
    )
 
79
from .push import (
 
80
    remote_divergence,
 
81
    )
 
82
from .repository import (
 
83
    GitRepository,
 
84
    )
 
85
from .refs import (
 
86
    branch_name_to_ref,
 
87
    is_peeled,
 
88
    ref_to_tag_name,
 
89
    tag_name_to_ref,
 
90
    )
 
91
 
 
92
import dulwich
 
93
import dulwich.client
 
94
from dulwich.errors import (
 
95
    GitProtocolError,
 
96
    )
 
97
from dulwich.pack import (
 
98
    Pack,
 
99
    pack_objects_to_data,
 
100
    )
 
101
from dulwich.protocol import ZERO_SHA
 
102
from dulwich.refs import SYMREF
 
103
from dulwich.repo import DictRefsContainer
 
104
import os
 
105
import select
 
106
import tempfile
 
107
import urllib
 
108
import urlparse
 
109
 
 
110
# urlparse only supports a limited number of schemes by default
 
111
 
 
112
urlparse.uses_netloc.extend(['git', 'git+ssh'])
 
113
 
 
114
from dulwich.pack import load_pack_index
 
115
 
 
116
 
 
117
class GitPushResult(PushResult):
 
118
 
 
119
    def _lookup_revno(self, revid):
 
120
        try:
 
121
            return _quick_lookup_revno(self.source_branch, self.target_branch,
 
122
                revid)
 
123
        except GitSmartRemoteNotSupported:
 
124
            return None
 
125
 
 
126
    @property
 
127
    def old_revno(self):
 
128
        return self._lookup_revno(self.old_revid)
 
129
 
 
130
    @property
 
131
    def new_revno(self):
 
132
        return self._lookup_revno(self.new_revid)
 
133
 
 
134
 
 
135
# Don't run any tests on GitSmartTransport as it is not intended to be
 
136
# a full implementation of Transport
 
137
def get_test_permutations():
 
138
    return []
 
139
 
 
140
 
 
141
def split_git_url(url):
 
142
    """Split a Git URL.
 
143
 
 
144
    :param url: Git URL
 
145
    :return: Tuple with host, port, username, path.
 
146
    """
 
147
    (scheme, netloc, loc, _, _) = urlparse.urlsplit(url)
 
148
    path = urllib.unquote(loc)
 
149
    if path.startswith("/~"):
 
150
        path = path[1:]
 
151
    (username, hostport) = urllib.splituser(netloc)
 
152
    (host, port) = urllib.splitnport(hostport, None)
 
153
    return (host, port, username, path)
 
154
 
 
155
 
 
156
class RemoteGitError(BzrError):
 
157
 
 
158
    _fmt = "Remote server error: %(msg)s"
 
159
 
 
160
 
 
161
def parse_git_error(url, message):
 
162
    """Parse a remote git server error and return a bzr exception.
 
163
 
 
164
    :param url: URL of the remote repository
 
165
    :param message: Message sent by the remote git server
 
166
    """
 
167
    message = str(message).strip()
 
168
    if message.startswith("Could not find Repository "):
 
169
        return NotBranchError(url, message)
 
170
    if message == "HEAD failed to update":
 
171
        base_url, _ = urlutils.split_segment_parameters(url)
 
172
        raise BzrError(
 
173
            ("Unable to update remote HEAD branch. To update the master "
 
174
             "branch, specify the URL %s,branch=master.") % base_url)
 
175
    # Don't know, just return it to the user as-is
 
176
    return RemoteGitError(message)
 
177
 
 
178
 
 
179
class GitSmartTransport(Transport):
 
180
 
 
181
    def __init__(self, url, _client=None):
 
182
        Transport.__init__(self, url)
 
183
        (self._host, self._port, self._username, self._path) = \
 
184
            split_git_url(url)
 
185
        if 'transport' in debug.debug_flags:
 
186
            trace.mutter('host: %r, user: %r, port: %r, path: %r',
 
187
                         self._host, self._username, self._port, self._path)
 
188
        self._client = _client
 
189
        self._stripped_path = self._path.rsplit(",", 1)[0]
 
190
 
 
191
    def external_url(self):
 
192
        return self.base
 
193
 
 
194
    def has(self, relpath):
 
195
        return False
 
196
 
 
197
    def _get_client(self):
 
198
        raise NotImplementedError(self._get_client)
 
199
 
 
200
    def _get_path(self):
 
201
        return self._stripped_path
 
202
 
 
203
    def get(self, path):
 
204
        raise NoSuchFile(path)
 
205
 
 
206
    def abspath(self, relpath):
 
207
        return urlutils.join(self.base, relpath)
 
208
 
 
209
    def clone(self, offset=None):
 
210
        """See Transport.clone()."""
 
211
        if offset is None:
 
212
            newurl = self.base
 
213
        else:
 
214
            newurl = urlutils.join(self.base, offset)
 
215
 
 
216
        return self.__class__(newurl, self._client)
 
217
 
 
218
 
 
219
class TCPGitSmartTransport(GitSmartTransport):
 
220
 
 
221
    _scheme = 'git'
 
222
 
 
223
    def _get_client(self):
 
224
        if self._client is not None:
 
225
            ret = self._client
 
226
            self._client = None
 
227
            return ret
 
228
        if self._host == '':
 
229
            # return dulwich.client.LocalGitClient()
 
230
            return dulwich.client.SubprocessGitClient()
 
231
        return dulwich.client.TCPGitClient(self._host, self._port,
 
232
            report_activity=self._report_activity)
 
233
 
 
234
 
 
235
class SSHSocketWrapper(object):
 
236
 
 
237
    def __init__(self, sock):
 
238
        self.sock = sock
 
239
 
 
240
    def read(self, len=None):
 
241
        return self.sock.recv(len)
 
242
 
 
243
    def write(self, data):
 
244
        return self.sock.write(data)
 
245
 
 
246
    def can_read(self):
 
247
        return len(select.select([self.sock.fileno()], [], [], 0)[0]) > 0
 
248
 
 
249
 
 
250
class DulwichSSHVendor(dulwich.client.SSHVendor):
 
251
 
 
252
    def __init__(self):
 
253
        from ...transport import ssh
 
254
        self.bzr_ssh_vendor = ssh._get_ssh_vendor()
 
255
 
 
256
    def run_command(self, host, command, username=None, port=None):
 
257
        connection = self.bzr_ssh_vendor.connect_ssh(username=username,
 
258
            password=None, port=port, host=host, command=command)
 
259
        (kind, io_object) = connection.get_sock_or_pipes()
 
260
        if kind == 'socket':
 
261
            return SSHSocketWrapper(io_object)
 
262
        else:
 
263
            raise AssertionError("Unknown io object kind %r'" % kind)
 
264
 
 
265
 
 
266
#dulwich.client.get_ssh_vendor = DulwichSSHVendor
 
267
 
 
268
 
 
269
class SSHGitSmartTransport(GitSmartTransport):
 
270
 
 
271
    _scheme = 'git+ssh'
 
272
 
 
273
    def _get_path(self):
 
274
        path = self._stripped_path
 
275
        if path.startswith("/~/"):
 
276
            return path[3:]
 
277
        return path
 
278
 
 
279
    def _get_client(self):
 
280
        if self._client is not None:
 
281
            ret = self._client
 
282
            self._client = None
 
283
            return ret
 
284
        location_config = config.LocationConfig(self.base)
 
285
        client = dulwich.client.SSHGitClient(self._host, self._port, self._username,
 
286
            report_activity=self._report_activity)
 
287
        # Set up alternate pack program paths
 
288
        upload_pack = location_config.get_user_option('git_upload_pack')
 
289
        if upload_pack:
 
290
            client.alternative_paths["upload-pack"] = upload_pack
 
291
        receive_pack = location_config.get_user_option('git_receive_pack')
 
292
        if receive_pack:
 
293
            client.alternative_paths["receive-pack"] = receive_pack
 
294
        return client
 
295
 
 
296
 
 
297
class RemoteGitBranchFormat(GitBranchFormat):
 
298
 
 
299
    def get_format_description(self):
 
300
        return 'Remote Git Branch'
 
301
 
 
302
    @property
 
303
    def _matchingcontroldir(self):
 
304
        return RemoteGitControlDirFormat()
 
305
 
 
306
    def initialize(self, a_controldir, name=None, repository=None,
 
307
                   append_revisions_only=None):
 
308
        raise UninitializableFormat(self)
 
309
 
 
310
 
 
311
class DefaultProgressReporter(object):
 
312
 
 
313
    _GIT_PROGRESS_PARTIAL_RE = re.compile(r"(.*?): +(\d+)% \((\d+)/(\d+)\)")
 
314
    _GIT_PROGRESS_TOTAL_RE = re.compile(r"(.*?): (\d+)")
 
315
 
 
316
    def __init__(self, pb):
 
317
        self.pb = pb
 
318
 
 
319
    def progress(self, text):
 
320
        text = text.rstrip("\r\n")
 
321
        if text.startswith('error: '):
 
322
            trace.show_error('git: %s', text[len('error: '):])
 
323
        else:
 
324
            trace.mutter("git: %s", text)
 
325
            g = self._GIT_PROGRESS_PARTIAL_RE.match(text)
 
326
            if g is not None:
 
327
                (text, pct, current, total) = g.groups()
 
328
                self.pb.update(text, int(current), int(total))
 
329
            else:
 
330
                g = self._GIT_PROGRESS_TOTAL_RE.match(text)
 
331
                if g is not None:
 
332
                    (text, total) = g.groups()
 
333
                    self.pb.update(text, None, int(total))
 
334
                else:
 
335
                    trace.note("%s", text)
 
336
 
 
337
 
 
338
class RemoteGitDir(GitDir):
 
339
 
 
340
    def __init__(self, transport, format, client, client_path):
 
341
        self._format = format
 
342
        self.root_transport = transport
 
343
        self.transport = transport
 
344
        self._mode_check_done = None
 
345
        self._client = client
 
346
        self._client_path = client_path
 
347
        self.base = self.root_transport.base
 
348
        self._refs = None
 
349
 
 
350
    @property
 
351
    def _gitrepository_class(self):
 
352
        return RemoteGitRepository
 
353
 
 
354
    def fetch_pack(self, determine_wants, graph_walker, pack_data, progress=None):
 
355
        if progress is None:
 
356
            pb = ui.ui_factory.nested_progress_bar()
 
357
            progress = DefaultProgressReporter(pb).progress
 
358
        else:
 
359
            pb = None
 
360
        try:
 
361
            result = self._client.fetch_pack(self._client_path, determine_wants,
 
362
                graph_walker, pack_data, progress)
 
363
            if result.refs is None:
 
364
                result.refs = {}
 
365
            self._refs = remote_refs_dict_to_container(result.refs, result.symrefs)
 
366
            return result
 
367
        except GitProtocolError, e:
 
368
            raise parse_git_error(self.transport.external_url(), e)
 
369
        finally:
 
370
            if pb is not None:
 
371
                pb.finished()
 
372
 
 
373
    def send_pack(self, get_changed_refs, generate_pack_data, progress=None):
 
374
        if progress is None:
 
375
            pb = ui.ui_factory.nested_progress_bar()
 
376
            progress = DefaultProgressReporter(pb).progress
 
377
        else:
 
378
            pb = None
 
379
        try:
 
380
            return self._client.send_pack(self._client_path, get_changed_refs,
 
381
                generate_pack_data, progress)
 
382
        except GitProtocolError, e:
 
383
            raise parse_git_error(self.transport.external_url(), e)
 
384
        finally:
 
385
            if pb is not None:
 
386
                pb.finished()
 
387
 
 
388
    def create_branch(self, name=None, repository=None,
 
389
                      append_revisions_only=None, ref=None):
 
390
        refname = self._get_selected_ref(name, ref)
 
391
        if refname != b'HEAD' and refname in self.get_refs_container():
 
392
            raise AlreadyBranchError(self.user_url)
 
393
        if refname in self.get_refs_container():
 
394
            ref_chain, unused_sha = self.get_refs_container().follow(self._get_selected_ref(None))
 
395
            if ref_chain[0] == b'HEAD':
 
396
                refname = ref_chain[1]
 
397
        repo = self.open_repository()
 
398
        return RemoteGitBranch(self, repo, refname)
 
399
 
 
400
    def destroy_branch(self, name=None):
 
401
        refname = self._get_selected_ref(name)
 
402
        def get_changed_refs(old_refs):
 
403
            ret = dict(old_refs)
 
404
            if not refname in ret:
 
405
                raise NotBranchError(self.user_url)
 
406
            ret[refname] = dulwich.client.ZERO_SHA
 
407
            return ret
 
408
        def generate_pack_data(have, want, ofs_delta=False):
 
409
            return pack_objects_to_data([])
 
410
        self.send_pack(get_changed_refs, generate_pack_data)
 
411
 
 
412
    @property
 
413
    def user_url(self):
 
414
        return self.control_url
 
415
 
 
416
    @property
 
417
    def user_transport(self):
 
418
        return self.root_transport
 
419
 
 
420
    @property
 
421
    def control_url(self):
 
422
        return self.control_transport.base
 
423
 
 
424
    @property
 
425
    def control_transport(self):
 
426
        return self.root_transport
 
427
 
 
428
    def open_repository(self):
 
429
        return RemoteGitRepository(self)
 
430
 
 
431
    def open_branch(self, name=None, unsupported=False,
 
432
            ignore_fallbacks=False, ref=None, possible_transports=None,
 
433
            nascent_ok=False):
 
434
        repo = self.open_repository()
 
435
        ref = self._get_selected_ref(name, ref)
 
436
        if not nascent_ok and ref not in self.get_refs_container():
 
437
            raise NotBranchError(self.root_transport.base,
 
438
                    controldir=self)
 
439
        ref_chain, unused_sha = self.get_refs_container().follow(ref)
 
440
        return RemoteGitBranch(self, repo, ref_chain[-1])
 
441
 
 
442
    def open_workingtree(self, recommend_upgrade=False):
 
443
        raise NotLocalUrl(self.transport.base)
 
444
 
 
445
    def has_workingtree(self):
 
446
        return False
 
447
 
 
448
    def get_peeled(self, name):
 
449
        return self.get_refs_container().get_peeled(name)
 
450
 
 
451
    def get_refs_container(self):
 
452
        if self._refs is not None:
 
453
            return self._refs
 
454
        result = self.fetch_pack(lambda x: None, None,
 
455
            lambda x: None, lambda x: trace.mutter("git: %s" % x))
 
456
        self._refs = remote_refs_dict_to_container(
 
457
                result.refs, result.symrefs)
 
458
        return self._refs
 
459
 
 
460
    def push_branch(self, source, revision_id=None, overwrite=False,
 
461
                    remember=False, create_prefix=False, lossy=False,
 
462
                    name=None):
 
463
        """Push the source branch into this ControlDir."""
 
464
        if revision_id is None:
 
465
            # No revision supplied by the user, default to the branch
 
466
            # revision
 
467
            revision_id = source.last_revision()
 
468
 
 
469
        push_result = GitPushResult()
 
470
        push_result.workingtree_updated = None
 
471
        push_result.master_branch = None
 
472
        push_result.source_branch = source
 
473
        push_result.stacked_on = None
 
474
        push_result.branch_push_result = None
 
475
        repo = self.find_repository()
 
476
        refname = self._get_selected_ref(name)
 
477
        if isinstance(source, GitBranch) and lossy:
 
478
            raise errors.LossyPushToSameVCS(source.controldir, self)
 
479
        source_store = get_object_store(source.repository)
 
480
        with source_store.lock_read():
 
481
            def get_changed_refs(refs):
 
482
                self._refs = remote_refs_dict_to_container(refs)
 
483
                ret = dict(refs)
 
484
                # TODO(jelmer): Unpeel if necessary
 
485
                push_result.new_original_revid = revision_id
 
486
                if lossy:
 
487
                    new_sha = source_store._lookup_revision_sha1(revision_id)
 
488
                else:
 
489
                    new_sha = repo.lookup_bzr_revision_id(revision_id)[0]
 
490
                if not overwrite:
 
491
                    if remote_divergence(ret.get(refname), new_sha, source_store):
 
492
                        raise DivergedBranches(
 
493
                                source, self.open_branch(name, nascent_ok=True))
 
494
                ret[refname] = new_sha
 
495
                return ret
 
496
            if lossy:
 
497
                generate_pack_data = source_store.generate_lossy_pack_data
 
498
            else:
 
499
                generate_pack_data = source_store.generate_pack_data
 
500
            new_refs = self.send_pack(get_changed_refs, generate_pack_data)
 
501
        push_result.new_revid = repo.lookup_foreign_revision_id(
 
502
                new_refs[refname])
 
503
        try:
 
504
            old_remote = self._refs[refname]
 
505
        except KeyError:
 
506
            old_remote = ZERO_SHA
 
507
        push_result.old_revid = repo.lookup_foreign_revision_id(old_remote)
 
508
        self._refs = remote_refs_dict_to_container(new_refs)
 
509
        push_result.target_branch = self.open_branch(name)
 
510
        if old_remote != ZERO_SHA:
 
511
            push_result.branch_push_result = GitBranchPushResult()
 
512
            push_result.branch_push_result.source_branch = source
 
513
            push_result.branch_push_result.target_branch = push_result.target_branch
 
514
            push_result.branch_push_result.local_branch = None
 
515
            push_result.branch_push_result.master_branch = push_result.target_branch
 
516
            push_result.branch_push_result.old_revid = push_result.old_revid
 
517
            push_result.branch_push_result.new_revid = push_result.new_revid
 
518
            push_result.branch_push_result.new_original_revid = push_result.new_original_revid
 
519
        if source.get_push_location() is None or remember:
 
520
            source.set_push_location(push_result.target_branch.base)
 
521
        return push_result
 
522
 
 
523
 
 
524
class EmptyObjectStoreIterator(dict):
 
525
 
 
526
    def iterobjects(self):
 
527
        return []
 
528
 
 
529
 
 
530
class TemporaryPackIterator(Pack):
 
531
 
 
532
    def __init__(self, path, resolve_ext_ref):
 
533
        super(TemporaryPackIterator, self).__init__(
 
534
            path, resolve_ext_ref=resolve_ext_ref)
 
535
        self._idx_load = lambda: self._idx_load_or_generate(self._idx_path)
 
536
 
 
537
    def _idx_load_or_generate(self, path):
 
538
        if not os.path.exists(path):
 
539
            pb = ui.ui_factory.nested_progress_bar()
 
540
            try:
 
541
                def report_progress(cur, total):
 
542
                    pb.update("generating index", cur, total)
 
543
                self.data.create_index(path,
 
544
                    progress=report_progress)
 
545
            finally:
 
546
                pb.finished()
 
547
        return load_pack_index(path)
 
548
 
 
549
    def __del__(self):
 
550
        if self._idx is not None:
 
551
            self._idx.close()
 
552
            os.remove(self._idx_path)
 
553
        if self._data is not None:
 
554
            self._data.close()
 
555
            os.remove(self._data_path)
 
556
 
 
557
 
 
558
class BzrGitHttpClient(dulwich.client.HttpGitClient):
 
559
 
 
560
    def __init__(self, transport, *args, **kwargs):
 
561
        self.transport = transport
 
562
        super(BzrGitHttpClient, self).__init__(transport.external_url(), *args, **kwargs)
 
563
        import urllib2
 
564
        self._http_perform = getattr(self.transport, "_perform", urllib2.urlopen)
 
565
 
 
566
    def _perform(self, req):
 
567
        req.accepted_errors = (200, 404)
 
568
        req.follow_redirections = True
 
569
        req.redirected_to = None
 
570
        return self._http_perform(req)
 
571
 
 
572
 
 
573
class RemoteGitControlDirFormat(GitControlDirFormat):
 
574
    """The .git directory control format."""
 
575
 
 
576
    supports_workingtrees = False
 
577
 
 
578
    @classmethod
 
579
    def _known_formats(self):
 
580
        return set([RemoteGitControlDirFormat()])
 
581
 
 
582
    def get_branch_format(self):
 
583
        return RemoteGitBranchFormat()
 
584
 
 
585
    def is_initializable(self):
 
586
        return False
 
587
 
 
588
    def is_supported(self):
 
589
        return True
 
590
 
 
591
    def open(self, transport, _found=None):
 
592
        """Open this directory.
 
593
 
 
594
        """
 
595
        # we dont grok readonly - git isn't integrated with transport.
 
596
        url = transport.base
 
597
        if url.startswith('readonly+'):
 
598
            url = url[len('readonly+'):]
 
599
        scheme = urlparse.urlsplit(transport.external_url())[0]
 
600
        if isinstance(transport, GitSmartTransport):
 
601
            client = transport._get_client()
 
602
            client_path = transport._get_path()
 
603
        elif scheme in ("http", "https"):
 
604
            client = BzrGitHttpClient(transport)
 
605
            client_path, _ = urlutils.split_segment_parameters(transport._path)
 
606
        elif scheme == 'file':
 
607
            client = dulwich.client.LocalGitClient()
 
608
            client_path = transport.local_abspath('.')
 
609
        else:
 
610
            raise NotBranchError(transport.base)
 
611
        if not _found:
 
612
            pass # TODO(jelmer): Actually probe for something
 
613
        return RemoteGitDir(transport, self, client, client_path)
 
614
 
 
615
    def get_format_description(self):
 
616
        return "Remote Git Repository"
 
617
 
 
618
    def initialize_on_transport(self, transport):
 
619
        raise UninitializableFormat(self)
 
620
 
 
621
    def supports_transport(self, transport):
 
622
        try:
 
623
            external_url = transport.external_url()
 
624
        except InProcessTransport:
 
625
            raise NotBranchError(path=transport.base)
 
626
        return (external_url.startswith("http:") or
 
627
                external_url.startswith("https:") or
 
628
                external_url.startswith("git+") or
 
629
                external_url.startswith("git:"))
 
630
 
 
631
 
 
632
class RemoteGitRepository(GitRepository):
 
633
 
 
634
    @property
 
635
    def user_url(self):
 
636
        return self.control_url
 
637
 
 
638
    def get_parent_map(self, revids):
 
639
        raise GitSmartRemoteNotSupported(self.get_parent_map, self)
 
640
 
 
641
    def fetch_pack(self, determine_wants, graph_walker, pack_data,
 
642
                   progress=None):
 
643
        return self.controldir.fetch_pack(determine_wants, graph_walker,
 
644
                                          pack_data, progress)
 
645
 
 
646
    def send_pack(self, get_changed_refs, generate_pack_data):
 
647
        return self.controldir.send_pack(get_changed_refs, generate_pack_data)
 
648
 
 
649
    def fetch_objects(self, determine_wants, graph_walker, resolve_ext_ref,
 
650
                      progress=None):
 
651
        fd, path = tempfile.mkstemp(suffix=".pack")
 
652
        try:
 
653
            self.fetch_pack(determine_wants, graph_walker,
 
654
                lambda x: os.write(fd, x), progress)
 
655
        finally:
 
656
            os.close(fd)
 
657
        if os.path.getsize(path) == 0:
 
658
            return EmptyObjectStoreIterator()
 
659
        return TemporaryPackIterator(path[:-len(".pack")], resolve_ext_ref)
 
660
 
 
661
    def lookup_bzr_revision_id(self, bzr_revid, mapping=None):
 
662
        # This won't work for any round-tripped bzr revisions, but it's a start..
 
663
        try:
 
664
            return mapping_registry.revision_id_bzr_to_foreign(bzr_revid)
 
665
        except InvalidRevisionId:
 
666
            raise NoSuchRevision(self, bzr_revid)
 
667
 
 
668
    def lookup_foreign_revision_id(self, foreign_revid, mapping=None):
 
669
        """Lookup a revision id.
 
670
 
 
671
        """
 
672
        if mapping is None:
 
673
            mapping = self.get_mapping()
 
674
        # Not really an easy way to parse foreign revids here..
 
675
        return mapping.revision_id_foreign_to_bzr(foreign_revid)
 
676
 
 
677
    def revision_tree(self, revid):
 
678
        raise GitSmartRemoteNotSupported(self.revision_tree, self)
 
679
 
 
680
    def get_revisions(self, revids):
 
681
        raise GitSmartRemoteNotSupported(self.get_revisions, self)
 
682
 
 
683
    def has_revisions(self, revids):
 
684
        raise GitSmartRemoteNotSupported(self.get_revisions, self)
 
685
 
 
686
 
 
687
class RemoteGitTagDict(GitTags):
 
688
 
 
689
    def set_tag(self, name, revid):
 
690
        sha = self.branch.lookup_bzr_revision_id(revid)[0]
 
691
        self._set_ref(name, sha)
 
692
 
 
693
    def delete_tag(self, name):
 
694
        self._set_ref(name, dulwich.client.ZERO_SHA)
 
695
 
 
696
    def _set_ref(self, name, sha):
 
697
        ref = tag_name_to_ref(name)
 
698
        def get_changed_refs(old_refs):
 
699
            ret = dict(old_refs)
 
700
            if sha == dulwich.client.ZERO_SHA and ref not in ret:
 
701
                raise NoSuchTag(name)
 
702
            ret[ref] = sha
 
703
            return ret
 
704
        def generate_pack_data(have, want, ofs_delta=False):
 
705
            return pack_objects_to_data([])
 
706
        self.repository.send_pack(get_changed_refs, generate_pack_data)
 
707
 
 
708
 
 
709
class RemoteGitBranch(GitBranch):
 
710
 
 
711
    def __init__(self, controldir, repository, name):
 
712
        self._sha = None
 
713
        super(RemoteGitBranch, self).__init__(controldir, repository, name,
 
714
                RemoteGitBranchFormat())
 
715
 
 
716
    def last_revision_info(self):
 
717
        raise GitSmartRemoteNotSupported(self.last_revision_info, self)
 
718
 
 
719
    @property
 
720
    def user_url(self):
 
721
        return self.control_url
 
722
 
 
723
    @property
 
724
    def control_url(self):
 
725
        return self.base
 
726
 
 
727
    def revision_id_to_revno(self, revision_id):
 
728
        raise GitSmartRemoteNotSupported(self.revision_id_to_revno, self)
 
729
 
 
730
    def last_revision(self):
 
731
        return self.lookup_foreign_revision_id(self.head)
 
732
 
 
733
    @property
 
734
    def head(self):
 
735
        if self._sha is not None:
 
736
            return self._sha
 
737
        refs = self.controldir.get_refs_container()
 
738
        name = branch_name_to_ref(self.name)
 
739
        try:
 
740
            self._sha = refs[name]
 
741
        except KeyError:
 
742
            raise NoSuchRef(name, self.repository.user_url, refs)
 
743
        return self._sha
 
744
 
 
745
    def _synchronize_history(self, destination, revision_id):
 
746
        """See Branch._synchronize_history()."""
 
747
        destination.generate_revision_history(self.last_revision())
 
748
 
 
749
    def _get_parent_location(self):
 
750
        return None
 
751
 
 
752
    def get_push_location(self):
 
753
        return None
 
754
 
 
755
    def set_push_location(self, url):
 
756
        pass
 
757
 
 
758
    def _iter_tag_refs(self):
 
759
        """Iterate over the tag refs.
 
760
 
 
761
        :param refs: Refs dictionary (name -> git sha1)
 
762
        :return: iterator over (ref_name, tag_name, peeled_sha1, unpeeled_sha1)
 
763
        """
 
764
        refs = self.controldir.get_refs_container()
 
765
        for ref_name, unpeeled in refs.as_dict().iteritems():
 
766
            try:
 
767
                tag_name = ref_to_tag_name(ref_name)
 
768
            except (ValueError, UnicodeDecodeError):
 
769
                continue
 
770
            peeled = refs.get_peeled(ref_name)
 
771
            if peeled is None:
 
772
                try:
 
773
                    peeled = refs.peel_sha(unpeeled).id
 
774
                except KeyError:
 
775
                    # Let's just hope it's a commit
 
776
                    peeled = unpeeled
 
777
            if type(tag_name) is not unicode:
 
778
                raise TypeError(tag_name)
 
779
            yield (ref_name, tag_name, peeled, unpeeled)
 
780
 
 
781
 
 
782
def remote_refs_dict_to_container(refs_dict, symrefs_dict={}):
 
783
    base = {}
 
784
    peeled = {}
 
785
    for k, v in refs_dict.iteritems():
 
786
        if is_peeled(k):
 
787
            peeled[k[:-3]] = v
 
788
        else:
 
789
            base[k] = v
 
790
            peeled[k] = v
 
791
    for name, target in symrefs_dict.iteritems():
 
792
        base[name] = SYMREF + target
 
793
    ret = DictRefsContainer(base)
 
794
    ret._peeled = peeled
 
795
    return ret