1
# Copyright (C) 2009-2018 Jelmer Vernooij <jelmer@jelmer.uk>
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.
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.
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
17
"""Basic push implementation."""
23
GitSmartRemoteNotSupported,
27
class GitPushResult(PushResult):
29
def _lookup_revno(self, revid):
30
from .branch import _quick_lookup_revno
32
return _quick_lookup_revno(self.source_branch, self.target_branch,
34
except GitSmartRemoteNotSupported:
39
return self._lookup_revno(self.old_revid)
43
return self._lookup_revno(self.new_revid)
46
class MissingObjectsIterator(object):
47
"""Iterate over git objects that are missing from a target repository.
51
def __init__(self, store, source, pb=None):
52
"""Create a new missing objects iterator.
56
self._object_store = store
60
def import_revisions(self, revids, lossy):
61
"""Import a set of revisions into this git repository.
63
:param revids: Revision ids of revisions to import
64
:param lossy: Whether to not roundtrip bzr metadata
66
for i, revid in enumerate(revids):
68
self.pb.update("pushing revisions", i, len(revids))
69
git_commit = self.import_revision(revid, lossy)
70
yield (revid, git_commit)
72
def import_revision(self, revid, lossy):
73
"""Import a revision into this Git repository.
75
:param revid: Revision id of the revision
76
:param roundtrip: Whether to roundtrip bzr metadata
78
tree = self._object_store.tree_cache.revision_tree(revid)
79
rev = self.source.get_revision(revid)
81
for path, obj in self._object_store._revision_to_objects(
83
if obj.type_name == b"commit":
85
self._pending.append((obj, path))
87
raise AssertionError("no commit object generated for revision %s" %
92
return len(self._pending)
95
return iter(self._pending)
98
class ObjectStoreParentsProvider(object):
100
def __init__(self, store):
103
def get_parent_map(self, shas):
110
parents = self._store[sha].parents
117
def remote_divergence(old_sha, new_sha, store):
120
if not isinstance(old_sha, bytes):
121
raise TypeError(old_sha)
122
if not isinstance(new_sha, bytes):
123
raise TypeError(new_sha)
124
from breezy.graph import Graph
125
graph = Graph(ObjectStoreParentsProvider(store))
126
return not graph.is_ancestor(old_sha, new_sha)