35
35
from bzrlib.decorators import (
38
from bzrlib.revision import (
38
41
from bzrlib.trace import (
43
from bzrlib.plugins.git import (
46
46
from bzrlib.plugins.git.config import (
63
62
class GitPullResult(branch.PullResult):
63
"""Result of a pull from a Git branch."""
65
65
def _lookup_revno(self, revid):
66
66
assert isinstance(revid, str), "was %r" % revid
111
111
for name in extra:
112
112
if name.startswith("refs/tags/"):
113
113
del self.repository._git[name]
115
115
def set_tag(self, name, revid):
116
116
self.repository._git.refs[tag_name_to_ref(name)], _ = \
117
117
self.branch.mapping.revision_id_bzr_to_foreign(revid)
182
182
"""Return the most suitable metadir for a checkout of this branch.
183
183
Weaves are used if this branch's repository uses weaves.
185
return get_rich_root_format()
185
return bzrdir.format_registry.make_bzrdir("default")
187
187
def get_child_submit_format(self):
188
188
"""Return the preferred format of submissions to this branch."""
254
254
overwrite, stop_revision)
256
256
def lookup_foreign_revision_id(self, foreign_revid):
257
return self.repository.lookup_foreign_revision_id(foreign_revid,
257
return self.repository.lookup_foreign_revision_id(foreign_revid,
262
262
"""A local Git branch."""
264
264
def __init__(self, bzrdir, repository, name, lockfiles, tagsdict=None):
265
super(LocalGitBranch, self).__init__(bzrdir, repository, name,
265
super(LocalGitBranch, self).__init__(bzrdir, repository, name,
266
266
lockfiles, tagsdict)
267
267
refs = repository._git.get_refs()
268
268
if not (name in refs.keys() or "HEAD" in refs.keys()):
294
294
:return: WorkingTree object of checkout.
296
296
checkout_branch = bzrdir.BzrDir.create_branch_convenience(
297
to_location, force_new_tree=False, format=get_rich_root_format())
297
to_location, force_new_tree=False)
298
298
checkout = checkout_branch.bzrdir
299
299
checkout_branch.bind(self)
300
300
# pull up to the specified revision_id to set the initial
601
601
interrepo = repository.InterRepository.get(self.source.repository,
602
602
self.target.repository)
603
603
if stop_revision is None:
604
refs = interrepo.fetch_refs(branches=["HEAD"])
604
refs = interrepo.fetch(branches=["HEAD"])
605
605
stop_revision = self.target.lookup_foreign_revision_id(refs["HEAD"])
607
refs = interrepo.fetch_refs(revision_id=stop_revision)
607
refs = interrepo.fetch(revision_id=stop_revision)
608
608
return refs, stop_revision
610
610
def pull(self, stop_revision=None, overwrite=False,
626
class InterToGitBranch(branch.InterBranch):
626
class InterToGitBranch(branch.GenericInterBranch):
627
627
"""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,
630
635
def _get_branch_formats_to_test():
634
639
def is_compatible(self, source, target):
641
646
def _get_new_refs(self, stop_revision=None):
642
647
if stop_revision is None:
643
648
stop_revision = self.source.last_revision()
649
assert type(stop_revision) is str
644
650
main_ref = self.target.ref or "refs/heads/master"
645
refs = { main_ref: stop_revision }
651
refs = { main_ref: (None, stop_revision) }
646
652
for name, revid in self.source.tags.get_tag_dict().iteritems():
647
653
if self.source.repository.has_revision(revid):
648
refs[tag_name_to_ref(name)] = revid
654
refs[tag_name_to_ref(name)] = (None, revid)
649
655
return refs, main_ref
651
657
def pull(self, overwrite=False, stop_revision=None, local=False,
654
660
result = GitBranchPullResult()
655
661
result.source_branch = self.source
656
662
result.target_branch = self.target
657
# FIXME: Check for diverged branches
658
old_refs = self.target.repository._git.get_refs()
659
refs = dict(old_refs)
660
663
new_refs, main_ref = self._get_new_refs(stop_revision)
661
refs.update(new_refs)
662
self.target.repository.fetch_refs(self.source.repository, refs)
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)
663
670
result.old_revid = self.target.lookup_foreign_revision_id(
664
671
old_refs.get(main_ref, ZERO_SHA))
665
result.new_revid = refs[main_ref]
672
result.new_revid = new_refs[main_ref]
668
675
def push(self, overwrite=False, stop_revision=None,
671
678
result = GitBranchPushResult()
672
679
result.source_branch = self.source
673
680
result.target_branch = self.target
674
# FIXME: Check for diverged branches
675
old_refs = self.target.repository._git.get_refs()
676
refs = dict(old_refs)
677
681
new_refs, main_ref = self._get_new_refs(stop_revision)
678
refs.update(new_refs)
679
self.target.repository.fetch_refs(self.source.repository, refs)
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)
680
688
result.old_revid = self.target.lookup_foreign_revision_id(
681
689
old_refs.get(main_ref, ZERO_SHA))
682
result.new_revid = refs[main_ref]
690
result.new_revid = new_refs[main_ref]
685
693
def lossy_push(self, stop_revision=None):
686
from dulwich.protocol import ZERO_SHA
687
694
result = GitBranchPushResult()
688
695
result.source_branch = self.source
689
696
result.target_branch = self.target
690
# FIXME: Check for diverged branches
691
refs, main_ref = self._get_new_refs(stop_revision)
692
result.revidmap, old_refs, new_refs = self.target.repository.dfetch_refs(
693
self.source.repository, refs)
694
result.old_revid = self.target.lookup_foreign_revision_id(
695
old_refs.get(self.target.ref, ZERO_SHA))
696
result.new_revid = self.target.lookup_foreign_revision_id(
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]