86
89
def get_tag_dict(self):
88
for k,v in extract_tags(self.repository._git.get_refs()).iteritems():
90
obj = self.repository._git[v]
92
mutter("Tag %s points at unknown object %s, ignoring", v, obj)
91
for k,v in self.repository._git.refs.as_dict("refs/tags").iteritems():
92
obj = self.repository._git.get_object(v)
94
93
while isinstance(obj, Tag):
96
obj = self.repository._git[v]
95
obj = self.repository._git.get_object(v)
97
96
if not isinstance(obj, Commit):
98
97
mutter("Tag %s points at object %r that is not a commit, "
99
98
"ignoring", k, obj)
101
ret[k] = self.branch.lookup_foreign_revision_id(v)
100
ret[k] = self.branch.mapping.revision_id_foreign_to_bzr(v)
104
def _set_tag_dict(self, to_dict):
105
extra = set(self.repository._git.get_refs().keys())
106
for k, revid in to_dict.iteritems():
107
name = tag_name_to_ref(k)
110
self.set_tag(k, revid)
112
if name.startswith("refs/tags/"):
113
del self.repository._git[name]
115
103
def set_tag(self, name, revid):
116
self.repository._git.refs[tag_name_to_ref(name)], _ = \
104
self.repository._git.refs["refs/tags/%s" % name], _ = \
117
105
self.branch.mapping.revision_id_bzr_to_foreign(revid)
120
108
class DictTagDict(LocalGitTagDict):
122
111
def __init__(self, branch, tags):
123
112
super(DictTagDict, self).__init__(branch)
124
113
self._tags = tags
150
133
return LocalGitTagDict(branch)
153
class GitReadLock(object):
155
def __init__(self, unlock):
159
class GitWriteLock(object):
161
def __init__(self, unlock):
165
136
class GitBranch(ForeignBranch):
166
137
"""An adapter to git repositories for bzr Branch objects."""
168
def __init__(self, bzrdir, repository, ref, lockfiles, tagsdict=None):
139
def __init__(self, bzrdir, repository, name, lockfiles, tagsdict=None):
169
140
self.repository = repository
170
141
self._format = GitBranchFormat()
171
142
self.control_files = lockfiles
247
214
# perhaps should escape this ?
248
215
if self.head is None:
249
216
return revision.NULL_REVISION
250
return self.lookup_foreign_revision_id(self.head)
217
return self.mapping.revision_id_foreign_to_bzr(self.head)
252
219
def _basic_push(self, target, overwrite=False, stop_revision=None):
253
220
return branch.InterBranch.get(self, target)._basic_push(
254
221
overwrite, stop_revision)
256
def lookup_foreign_revision_id(self, foreign_revid):
257
return self.repository.lookup_foreign_revision_id(foreign_revid,
261
224
class LocalGitBranch(GitBranch):
262
225
"""A local Git branch."""
264
def __init__(self, bzrdir, repository, name, lockfiles, tagsdict=None):
265
super(LocalGitBranch, self).__init__(bzrdir, repository, name,
267
refs = repository._git.get_refs()
268
if not (name in refs.keys() or "HEAD" in refs.keys()):
269
raise errors.NotBranchError(self.base)
271
227
def create_checkout(self, to_location, revision_id=None, lightweight=False,
272
228
accelerator_tree=None, hardlink=False):
276
232
format = self._get_checkout_format()
277
233
checkout = format.initialize_on_transport(t)
278
from_branch = branch.BranchReferenceFormat().initialize(checkout,
234
from_branch = branch.BranchReferenceFormat().initialize(checkout,
280
236
tree = checkout.create_workingtree(revision_id,
281
237
from_branch=from_branch, hardlink=hardlink)
284
240
return self._create_heavyweight_checkout(to_location, revision_id,
287
def _create_heavyweight_checkout(self, to_location, revision_id=None,
243
def _create_heavyweight_checkout(self, to_location, revision_id=None,
289
245
"""Create a new heavyweight checkout of this branch.
350
307
class GitBranchPullResult(branch.PullResult):
353
super(GitBranchPullResult, self).__init__()
354
self.new_git_head = None
355
self._old_revno = None
356
self._new_revno = None
358
309
def report(self, to_file):
359
310
if not is_quiet():
360
311
if self.old_revid == self.new_revid:
361
312
to_file.write('No revisions to pull.\n')
362
elif self.new_git_head is not None:
363
to_file.write('Now on revision %d (git sha: %s).\n' %
314
to_file.write('Now on revision %d (git sha: %s).\n' %
364
315
(self.new_revno, self.new_git_head))
366
to_file.write('Now on revision %d.\n' % (self.new_revno,))
367
316
self._show_tag_conficts(to_file)
369
def _lookup_revno(self, revid):
370
assert isinstance(revid, str), "was %r" % revid
371
# Try in source branch first, it'll be faster
373
return self.source_branch.revision_id_to_revno(revid)
374
except errors.NoSuchRevision:
375
# FIXME: Check using graph.find_distance_to_null() ?
376
return self.target_branch.revision_id_to_revno(revid)
378
def _get_old_revno(self):
379
if self._old_revno is not None:
380
return self._old_revno
381
return self._lookup_revno(self.old_revid)
383
def _set_old_revno(self, revno):
384
self._old_revno = revno
386
old_revno = property(_get_old_revno, _set_old_revno)
388
def _get_new_revno(self):
389
if self._new_revno is not None:
390
return self._new_revno
391
return self._lookup_revno(self.new_revid)
393
def _set_new_revno(self, revno):
394
self._new_revno = revno
396
new_revno = property(_get_new_revno, _set_new_revno)
399
319
class GitBranchPushResult(branch.BranchPushResult):
420
340
"""InterBranch implementation that pulls from Git into bzr."""
423
def _get_interrepo(self, source, target):
424
return repository.InterRepository.get(source.repository,
428
def is_compatible(cls, source, target):
429
return (isinstance(source, GitBranch) and
430
not isinstance(target, GitBranch) and
431
(getattr(cls._get_interrepo(source, target), "fetch_objects", None) is not None))
433
def _update_revisions(self, stop_revision=None, overwrite=False,
434
graph=None, limit=None):
435
"""Like InterBranch.update_revisions(), but with additions.
437
Compared to the `update_revisions()` below, this function takes a
438
`limit` argument that limits how many git commits will be converted
439
and returns the new git head.
441
interrepo = self._get_interrepo(self.source, self.target)
343
def is_compatible(self, source, target):
344
return (isinstance(source, GitBranch) and
345
not isinstance(target, GitBranch))
347
def update_revisions(self, stop_revision=None, overwrite=False,
349
"""See InterBranch.update_revisions()."""
350
interrepo = repository.InterRepository.get(self.source.repository,
351
self.target.repository)
353
self._last_revid = None
442
354
def determine_wants(heads):
443
if self.source.ref is not None and not self.source.ref in heads:
444
raise NoSuchRef(self.source.ref, heads.keys())
355
if not self.source.name in heads:
356
raise NoSuchRef(self.source.name, heads.keys())
445
357
if stop_revision is not None:
446
358
self._last_revid = stop_revision
447
head, mapping = self.source.repository.lookup_bzr_revision_id(
359
self._head, mapping = self.source.repository.lookup_git_revid(
450
if self.source.ref is not None:
451
head = heads[self.source.ref]
454
self._last_revid = self.source.lookup_foreign_revision_id(head)
362
self._head = heads[self.source.name]
364
self.source.mapping.revision_id_foreign_to_bzr(self._head)
455
365
if self.target.repository.has_revision(self._last_revid):
458
pack_hint, head = interrepo.fetch_objects(
459
determine_wants, self.source.mapping, limit=limit)
460
if (pack_hint is not None and
461
self.target.repository._format.pack_compresses):
462
self.target.repository.pack(hint=pack_hint)
464
self._last_revid = self.source.lookup_foreign_revision_id(head)
368
interrepo.fetch_objects(determine_wants, self.source.mapping)
466
370
prev_last_revid = None
468
372
prev_last_revid = self.target.last_revision()
469
self.target.generate_revision_history(self._last_revid,
473
def update_revisions(self, stop_revision=None, overwrite=False,
475
"""See InterBranch.update_revisions()."""
476
self._update_revisions(stop_revision, overwrite, graph)
373
self.target.generate_revision_history(self._last_revid, prev_last_revid)
478
375
def pull(self, overwrite=False, stop_revision=None,
479
376
possible_transports=None, _hook_master=None, run_hooks=True,
480
_override_hook_target=None, local=False, limit=None):
377
_override_hook_target=None, local=False):
481
378
"""See Branch.pull.
483
380
:param _hook_master: Private parameter - set the branch to
504
399
# We assume that during 'pull' the target repository is closer than
505
400
# the source one.
506
401
graph = self.target.repository.get_graph(self.source.repository)
507
(result.old_revno, result.old_revid) = \
402
result.old_revno, result.old_revid = \
508
403
self.target.last_revision_info()
509
result.new_git_head = self._update_revisions(
510
stop_revision, overwrite=overwrite, graph=graph, limit=limit)
404
self.update_revisions(stop_revision, overwrite=overwrite,
406
result.new_git_head = self._head
511
407
result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
513
(result.new_revno, result.new_revid) = \
514
self.target.last_revision_info()
409
result.new_revno, result.new_revid = self.target.last_revision_info()
516
411
result.master_branch = _hook_master
517
412
result.local_branch = result.target_branch
561
456
stop_revision = self.source.last_revision()
562
457
# FIXME: Check for diverged branches
563
458
def get_changed_refs(old_refs):
564
result.old_revid = self.target.lookup_foreign_revision_id(old_refs.get(self.target.ref, ZERO_SHA))
565
refs = { self.target.ref: self.source.repository.lookup_bzr_revision_id(stop_revision)[0] }
459
result.old_revid = self.target.mapping.revision_id_foreign_to_bzr(old_refs.get("refs/heads/master", "0" * 40))
460
refs = { "refs/heads/master": self.source.repository.lookup_git_revid(stop_revision)[0] }
566
461
result.new_revid = stop_revision
567
462
for name, sha in self.source.repository._git.refs.as_dict("refs/tags").iteritems():
568
refs[tag_name_to_ref(name)] = sha
463
refs["refs/tags/%s" % name] = sha
570
self.target.repository.send_pack(get_changed_refs,
571
self.source.repository._git.object_store.generate_pack_contents)
465
self.target.repository.send_pack(get_changed_refs,
466
self.source.repository._git.object_store.generate_pack_contents)
595
490
def update_tags(self, refs):
596
for name, v in extract_tags(refs).iteritems():
597
revid = self.target.lookup_foreign_revision_id(v)
491
for name, revid in extract_tags(refs, self.target.mapping).iteritems():
598
492
self.target.tags.set_tag(name, revid)
600
494
def update_refs(self, stop_revision=None):
601
interrepo = repository.InterRepository.get(self.source.repository,
495
interrepo = repository.InterRepository.get(self.source.repository,
602
496
self.target.repository)
603
497
if stop_revision is None:
604
refs = interrepo.fetch(branches=["HEAD"])
605
stop_revision = self.target.lookup_foreign_revision_id(refs["HEAD"])
498
refs = interrepo.fetch_refs(branches=["HEAD"])
499
stop_revision = self.target.mapping.revision_id_foreign_to_bzr(refs["HEAD"])
607
refs = interrepo.fetch(revision_id=stop_revision)
501
refs = interrepo.fetch_refs(revision_id=stop_revision)
608
502
return refs, stop_revision
610
def pull(self, stop_revision=None, overwrite=False,
611
possible_transports=None, run_hooks=True,local=False):
504
def pull(self, stop_revision=None, overwrite=False,
505
possible_transports=None, local=False):
612
506
# This type of branch can't be bound.
614
508
raise errors.LocalRequiresBoundBranch()
622
516
result.new_revid = self.target.last_revision()
626
class InterToGitBranch(branch.GenericInterBranch):
520
class InterToGitBranch(branch.InterBranch):
627
521
"""InterBranch implementation that pulls from Git into bzr."""
629
def __init__(self, source, target):
630
super(InterToGitBranch, self).__init__(source, target)
631
self.interrepo = repository.InterRepository.get(source.repository,
635
def _get_branch_formats_to_test():
639
524
def is_compatible(self, source, target):
640
return (not isinstance(source, GitBranch) and
525
return (not isinstance(source, GitBranch) and
641
526
isinstance(target, GitBranch))
643
528
def update_revisions(self, *args, **kwargs):
644
529
raise NoPushSupport()
646
def _get_new_refs(self, stop_revision=None):
531
def push(self, overwrite=True, stop_revision=None,
532
_override_hook_source_branch=None):
533
raise NoPushSupport()
535
def lossy_push(self, stop_revision=None):
536
result = GitBranchPushResult()
537
result.source_branch = self.source
538
result.target_branch = self.target
539
result.old_revid = self.target.last_revision()
647
540
if stop_revision is None:
648
541
stop_revision = self.source.last_revision()
649
assert type(stop_revision) is str
650
main_ref = self.target.ref or "refs/heads/master"
651
refs = { main_ref: (None, stop_revision) }
542
# FIXME: Check for diverged branches
543
refs = { "refs/heads/master": stop_revision }
652
544
for name, revid in self.source.tags.get_tag_dict().iteritems():
653
545
if self.source.repository.has_revision(revid):
654
refs[tag_name_to_ref(name)] = (None, revid)
655
return refs, main_ref
657
def pull(self, overwrite=False, stop_revision=None, local=False,
658
possible_transports=None):
659
from dulwich.protocol import ZERO_SHA
660
result = GitBranchPullResult()
661
result.source_branch = self.source
662
result.target_branch = self.target
663
new_refs, main_ref = self._get_new_refs(stop_revision)
664
def update_refs(old_refs):
665
refs = dict(old_refs)
666
# FIXME: Check for diverged branches
667
refs.update(new_refs)
669
old_refs, new_refs = self.interrepo.fetch_refs(update_refs)
670
result.old_revid = self.target.lookup_foreign_revision_id(
671
old_refs.get(main_ref, ZERO_SHA))
672
result.new_revid = new_refs[main_ref]
675
def push(self, overwrite=False, stop_revision=None,
676
_override_hook_source_branch=None):
677
from dulwich.protocol import ZERO_SHA
678
result = GitBranchPushResult()
679
result.source_branch = self.source
680
result.target_branch = self.target
681
new_refs, main_ref = self._get_new_refs(stop_revision)
682
def update_refs(old_refs):
683
refs = dict(old_refs)
684
# FIXME: Check for diverged branches
685
refs.update(new_refs)
687
old_refs, new_refs = self.interrepo.fetch_refs(update_refs)
688
result.old_revid = self.target.lookup_foreign_revision_id(
689
old_refs.get(main_ref, ZERO_SHA))
690
result.new_revid = new_refs[main_ref]
693
def lossy_push(self, stop_revision=None):
694
result = GitBranchPushResult()
695
result.source_branch = self.source
696
result.target_branch = self.target
697
new_refs, main_ref = self._get_new_refs(stop_revision)
698
def update_refs(old_refs):
699
refs = dict(old_refs)
700
# FIXME: Check for diverged branches
701
refs.update(new_refs)
703
result.revidmap, old_refs, new_refs = self.interrepo.dfetch_refs(
705
result.old_revid = old_refs.get(self.target.ref, (None, NULL_REVISION))[1]
706
result.new_revid = new_refs[main_ref][1]
546
refs["refs/tags/%s" % name] = revid
547
revidmap, new_refs = self.target.repository.dfetch_refs(
548
self.source.repository, refs)
550
self.target.generate_revision_history(revidmap[stop_revision])
551
result.new_revid = revidmap[stop_revision]
553
result.new_revid = result.old_revid
554
result.revidmap = revidmap