1
# Copyright (C) 2009 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."""
19
from dulwich.client import (
20
SimpleFetchGraphWalker,
22
from dulwich.repo import (
29
from bzrlib.repository import (
32
from bzrlib.revision import (
36
from bzrlib.plugins.git.converter import (
39
from bzrlib.plugins.git.errors import (
42
from bzrlib.plugins.git.repository import (
47
from bzrlib.plugins.git.remote import (
52
class MissingObjectsIterator(object):
53
"""Iterate over git objects that are missing from a target repository.
57
def __init__(self, source, mapping, pb=None):
58
"""Create a new missing objects iterator.
62
self._object_store = BazaarObjectStore(self.source, mapping)
64
self._sent_shas = set()
68
def import_revisions(self, revids):
69
self._revids.update(revids)
70
for i, revid in enumerate(revids):
72
self.pb.update("pushing revisions", i, len(revids))
73
git_commit = self.import_revision(revid)
74
yield (revid, git_commit)
76
def need_sha(self, sha):
77
if sha in self._sent_shas:
79
(type, (fileid, revid)) = self._object_store._idmap.lookup_git_sha(sha)
80
assert type in ("blob", "tree")
81
if revid in self._revids:
82
# Not sent yet, and part of the set of revisions to send
84
# Not changed in the revisions to send, so either not necessary
85
# or already present remotely (as git doesn't do ghosts)
88
def queue(self, sha, obj, path, ie=None, inv=None):
91
self._pending.append((obj, path))
92
self._sent_shas.add(sha)
94
def import_revision(self, revid):
95
"""Import the gist of a revision into this Git repository.
98
inv = self.source.get_inventory(revid)
103
(sha, object) = self._object_store._get_ie_object_or_sha1(ie, inv)
104
if ie.parent_id is None:
106
if not self.need_sha(sha):
108
self.queue(sha, object, inv.id2path(ie.file_id), ie, inv)
109
if ie.kind == "directory":
110
todo.extend(ie.children.values())
111
assert tree_sha is not None
112
commit = self._object_store._get_commit(revid, tree_sha)
113
self.queue(commit.id, commit, None)
117
return len(self._pending)
120
for i, (object, path) in enumerate(self._pending):
122
self.pb.update("writing pack objects", i, len(self))
123
if isinstance(object, tuple):
124
object = self._object_store._get_ie_object(*object)
128
class InterToGitRepository(InterRepository):
129
"""InterRepository that copies into a Git repository."""
131
_matching_repo_format = GitRepositoryFormat()
133
def __init__(self, source, target):
134
super(InterToGitRepository, self).__init__(source, target)
135
self.mapping = self.target.get_mapping()
138
def _get_repo_format_to_test():
141
def copy_content(self, revision_id=None, pb=None):
142
"""See InterRepository.copy_content."""
143
self.fetch(revision_id, pb, find_ghosts=False)
145
def fetch(self, revision_id=None, pb=None, find_ghosts=False,
147
raise NoPushSupport()
150
class InterToLocalGitRepository(InterToGitRepository):
152
def missing_revisions(self, stop_revision):
153
if stop_revision is None:
154
raise NotImplementedError
156
pb = ui.ui_factory.nested_progress_bar()
158
graph = self.source.get_graph()
159
for revid, _ in graph.iter_ancestry([stop_revision]):
160
pb.update("determining revisions to fetch", len(missing))
161
if not self.target.has_revision(revid):
162
missing.append(revid)
163
return graph.iter_topo_order(missing)
167
def dfetch_refs(self, refs):
170
for name, revid in refs.iteritems():
171
newrevidmap, newgitidmap = self.dfetch(revid)
172
revidmap.update(newrevidmap)
173
if revid in newgitidmap:
174
gitid = newgitidmap[revid]
176
gitid, _ = self.mapping.revision_id_bzr_to_foreign(revid)
177
self.target._git.set_ref(name, gitid)
178
new_refs[name] = gitid
179
return revidmap, new_refs
181
def dfetch(self, stop_revision=None):
182
"""Import the gist of the ancestry of a particular revision."""
185
self.source.lock_read()
187
todo = [revid for revid in self.missing_revisions(stop_revision) if revid != NULL_REVISION]
188
pb = ui.ui_factory.nested_progress_bar()
190
object_generator = MissingObjectsIterator(self.source, self.mapping, pb)
191
for old_bzr_revid, git_commit in object_generator.import_revisions(
193
new_bzr_revid = self.mapping.revision_id_foreign_to_bzr(git_commit)
194
revidmap[old_bzr_revid] = new_bzr_revid
195
gitidmap[old_bzr_revid] = git_commit
196
self.target._git.object_store.add_objects(object_generator)
201
return revidmap, gitidmap
204
def is_compatible(source, target):
205
"""Be compatible with GitRepository."""
206
return (not isinstance(source, GitRepository) and
207
isinstance(target, LocalGitRepository))
210
class InterToRemoteGitRepository(InterToGitRepository):
212
def dfetch_refs(self, new_refs):
213
"""Import the gist of the ancestry of a particular revision."""
215
def get_changed_refs(refs):
217
for name, revid in new_refs.iteritems():
218
ret[name] = store._lookup_revision_sha1(revid)
220
self.source.lock_read()
222
store = BazaarObjectStore(self.source, self.mapping)
223
def generate_blob_contents(have, want):
224
graphwalker = SimpleFetchGraphWalker(have, store.get_parents)
225
objfinder = MissingObjectFinder(store, want, graphwalker)
226
return store.iter_shas(iter(objfinder.next, None))
227
new_refs = self.target.send_pack(get_changed_refs, generate_blob_contents)
230
return revidmap, new_refs
233
def is_compatible(source, target):
234
"""Be compatible with GitRepository."""
235
return (not isinstance(source, GitRepository) and
236
isinstance(target, RemoteGitRepository))
added
removed
push.py
