1
# Copyright (C) 2009-2010 Jelmer Vernooij <jelmer@samba.org>
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""Push implementation that simply prints message saying push is not supported."""
23
from bzrlib.repository import (
26
from bzrlib.revision import (
30
from bzrlib.plugins.git.errors import (
33
from bzrlib.plugins.git.object_store import (
36
from bzrlib.plugins.git.repository import (
41
from bzrlib.plugins.git.remote import (
44
from bzrlib.plugins.git.unpeel_map import (
49
class MissingObjectsIterator(object):
50
"""Iterate over git objects that are missing from a target repository.
54
def __init__(self, store, source, pb=None):
55
"""Create a new missing objects iterator.
59
self._object_store = store
63
def import_revisions(self, revids, roundtrip):
64
"""Import a set of revisions into this git repository.
66
:param revids: Revision ids of revisions to import
67
:param roundtrip: Whether to roundtrip bzr metadata
69
for i, revid in enumerate(revids):
71
self.pb.update("pushing revisions", i, len(revids))
72
git_commit = self.import_revision(revid, roundtrip)
73
yield (revid, git_commit)
75
def import_revision(self, revid, roundtrip):
76
"""Import a revision into this Git repository.
78
:param revid: Revision id of the revision
79
:param roundtrip: Whether to roundtrip bzr metadata
81
tree = self._object_store.tree_cache.revision_tree(revid)
82
rev = self.source.get_revision(revid)
84
for path, obj, ie in self._object_store._revision_to_objects(rev, tree,
86
if obj.type_name == "commit":
88
self._pending.append((obj, path))
92
return len(self._pending)
95
return iter(self._pending)
98
class InterToGitRepository(InterRepository):
99
"""InterRepository that copies into a Git repository."""
101
_matching_repo_format = GitRepositoryFormat()
103
def __init__(self, source, target):
104
super(InterToGitRepository, self).__init__(source, target)
105
self.mapping = self.target.get_mapping()
106
self.source_store = get_object_store(self.source, self.mapping)
109
def _get_repo_format_to_test():
112
def copy_content(self, revision_id=None, pb=None):
113
"""See InterRepository.copy_content."""
114
self.fetch(revision_id, pb, find_ghosts=False)
116
def fetch_refs(self, update_refs, lossy):
117
"""Fetch possibly roundtripped revisions into the target repository
120
:param update_refs: Generate refs to fetch. Receives dictionary
121
with old refs (git shas), returns dictionary of new names to
123
:param lossy: Whether to roundtrip
124
:return: old refs, new refs
126
raise NotImplementedError(self.fetch_refs)
129
class InterToLocalGitRepository(InterToGitRepository):
130
"""InterBranch implementation between a Bazaar and a Git repository."""
132
def __init__(self, source, target):
133
super(InterToLocalGitRepository, self).__init__(source, target)
134
self.target_store = self.target._git.object_store
135
self.target_refs = self.target._git.refs
137
def _revision_needs_fetching(self, sha_id, revid):
138
if revid == NULL_REVISION:
142
sha_id = self.source_store._lookup_revision_sha1(revid)
146
return (sha_id not in self.target_store)
147
except errors.NoSuchRevision:
151
def missing_revisions(self, stop_revisions):
152
"""Find the revisions that are missing from the target repository.
154
:param stop_revisions: Revisions to check for (tuples with
156
:return: sequence of missing revisions, in topological order
157
:raise: NoSuchRevision if the stop_revisions are not present in
163
for (sha1, revid) in stop_revisions:
164
if sha1 is not None and revid is not None:
165
revid_sha_map[revid] = sha1
166
elif sha1 is not None:
167
stop_sha1s.append(sha1)
169
assert revid is not None
170
stop_revids.append(revid)
172
graph = self.source.get_graph()
173
pb = ui.ui_factory.nested_progress_bar()
175
for revid, _ in graph.iter_ancestry(stop_revids):
176
assert type(revid) is str
177
pb.update("determining revisions to fetch", len(missing))
178
sha1 = revid_sha_map.get(revid)
179
if self._revision_needs_fetching(sha1, revid):
180
missing.append(revid)
183
for sha1 in stop_sha1s:
185
for (kind, (revid, tree_sha, verifiers)) in self.source_store.lookup_git_sha(sha1):
186
missing.append(revid)
187
revid_sha_map[revid] = sha1
190
return graph.iter_topo_order(missing)
192
def _get_target_bzr_refs(self):
193
"""Return a dictionary with references.
195
:return: Dictionary with reference names as keys and tuples
196
with Git SHA, Bazaar revid as values.
200
for k in self.target._git.refs.allkeys():
201
v = self.target._git.refs.read_ref(k)
203
for (kind, type_data) in self.source_store.lookup_git_sha(v):
204
if kind == "commit" and self.source.has_revision(type_data[0]):
211
bzr_refs[k] = (v, revid)
214
def fetch_refs(self, update_refs, lossy):
215
self.source_store.lock_read()
217
old_refs = self._get_target_bzr_refs()
218
new_refs = update_refs(old_refs)
219
revidmap = self.fetch_objects(new_refs.values(), roundtrip=not lossy)
220
for name, (gitid, revid) in new_refs.iteritems():
223
gitid = revidmap[revid][0]
225
gitid = self.source_store._lookup_revision_sha1(revid)
226
assert len(gitid) == 40
227
self.target_refs[name] = gitid
229
self.source_store.unlock()
230
return revidmap, old_refs, new_refs
232
def fetch_objects(self, revs, roundtrip):
233
todo = list(self.missing_revisions(revs))
235
pb = ui.ui_factory.nested_progress_bar()
237
object_generator = self._get_missing_objects_iterator(pb)
238
for (old_revid, git_sha) in object_generator.import_revisions(
239
todo, roundtrip=roundtrip):
241
self.mapping.revision_id_bzr_to_foreign(old_revid)
242
except errors.InvalidRevisionId:
243
self.target_refs[self.mapping.revid_as_refname(old_revid)] = git_sha
245
new_revid = self.mapping.revision_id_foreign_to_bzr(git_sha)
247
new_revid = old_revid
248
revidmap[old_revid] = (git_sha, new_revid)
249
self.target_store.add_objects(object_generator)
254
def _get_missing_objects_iterator(self, pb):
255
return MissingObjectsIterator(self.source_store, self.source, pb)
257
def dfetch(self, stop_revisions):
258
"""Import the gist of the ancestry of a particular revision."""
260
def fetch(self, revision_id=None, pb=None, find_ghosts=False,
261
fetch_spec=None, mapped_refs=None):
262
if not self.mapping.roundtripping:
263
raise NoPushSupport()
264
self.source_store.lock_read()
266
if mapped_refs is not None:
267
stop_revisions = mapped_refs
268
elif revision_id is not None:
269
stop_revisions = [(None, revision_id)]
270
elif fetch_spec is not None:
271
stop_revisions = [(None, revid) for revid in fetch_spec.heads]
273
stop_revisions = [(None, revid) for revid in self.source.all_revision_ids()]
274
self.fetch_objects(stop_revisions, roundtrip=True)
276
self.source_store.unlock()
279
def is_compatible(source, target):
280
"""Be compatible with GitRepository."""
281
return (not isinstance(source, GitRepository) and
282
isinstance(target, LocalGitRepository))
285
class InterToRemoteGitRepository(InterToGitRepository):
287
def fetch_refs(self, update_refs, lossy):
288
"""Import the gist of the ancestry of a particular revision."""
290
raise NoPushSupport()
291
unpeel_map = UnpeelMap.from_repository(self.source)
293
def determine_wants(old_refs):
295
self.old_refs = dict([(k, (v, None)) for (k, v) in old_refs.iteritems()])
296
self.new_refs = update_refs(self.old_refs)
297
for name, (gitid, revid) in self.new_refs.iteritems():
299
git_sha = self.source_store._lookup_revision_sha1(revid)
300
ret[name] = unpeel_map.re_unpeel_tag(git_sha, old_refs.get(name))
304
self.source_store.lock_read()
306
new_refs = self.target.send_pack(determine_wants,
307
self.source_store.generate_lossy_pack_contents)
309
self.source_store.unlock()
311
return revidmap, self.old_refs, self.new_refs
314
def is_compatible(source, target):
315
"""Be compatible with GitRepository."""
316
return (not isinstance(source, GitRepository) and
317
isinstance(target, RemoteGitRepository))
added
removed
push.py
