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."""
22
from bzrlib.repository import (
25
from bzrlib.revision import (
29
from bzrlib.plugins.git.errors import (
32
from bzrlib.plugins.git.object_store import (
35
from bzrlib.plugins.git.repository import (
40
from bzrlib.plugins.git.remote import (
45
class MissingObjectsIterator(object):
46
"""Iterate over git objects that are missing from a target repository.
50
def __init__(self, source, mapping, pb=None):
51
"""Create a new missing objects iterator.
55
self._object_store = BazaarObjectStore(self.source, mapping)
57
self._sent_shas = set()
61
def import_revisions(self, revids):
62
self._revids.update(revids)
63
for i, revid in enumerate(revids):
65
self.pb.update("pushing revisions", i, len(revids))
66
git_commit = self.import_revision(revid)
67
yield (revid, git_commit)
69
def need_sha(self, sha):
70
if sha in self._sent_shas:
72
(type, (fileid, revid)) = self._object_store._idmap.lookup_git_sha(sha)
73
assert type in ("blob", "tree")
74
if revid in self._revids:
75
# Not sent yet, and part of the set of revisions to send
77
# Not changed in the revisions to send, so either not necessary
78
# or already present remotely (as git doesn't do ghosts)
81
def queue(self, sha, obj, path, ie=None, inv=None):
84
self._pending.append((obj, path))
85
self._sent_shas.add(sha)
87
def import_revision(self, revid):
88
"""Import the gist of a revision into this Git repository.
91
inv = self.source.get_inventory(revid)
96
(sha, object) = self._object_store._get_ie_object_or_sha1(ie, inv)
97
if ie.parent_id is None:
99
if not self.need_sha(sha):
101
self.queue(sha, object, inv.id2path(ie.file_id), ie, inv)
102
if ie.kind == "directory":
103
todo.extend(ie.children.values())
104
assert tree_sha is not None
105
commit = self._object_store._get_commit(revid, tree_sha)
106
self.queue(commit.id, commit, None)
110
return len(self._pending)
113
for i, (object, path) in enumerate(self._pending):
115
self.pb.update("writing pack objects", i, len(self))
116
if isinstance(object, tuple):
117
object = self._object_store._get_ie_object(*object)
121
class InterToGitRepository(InterRepository):
122
"""InterRepository that copies into a Git repository."""
124
_matching_repo_format = GitRepositoryFormat()
126
def __init__(self, source, target):
127
super(InterToGitRepository, self).__init__(source, target)
128
self.mapping = self.target.get_mapping()
131
def _get_repo_format_to_test():
134
def copy_content(self, revision_id=None, pb=None):
135
"""See InterRepository.copy_content."""
136
self.fetch(revision_id, pb, find_ghosts=False)
138
def fetch(self, revision_id=None, pb=None, find_ghosts=False,
140
raise NoPushSupport()
143
class InterToLocalGitRepository(InterToGitRepository):
145
def missing_revisions(self, stop_revision):
146
if stop_revision is None:
147
raise NotImplementedError
149
pb = ui.ui_factory.nested_progress_bar()
151
graph = self.source.get_graph()
152
for revid, _ in graph.iter_ancestry([stop_revision]):
153
pb.update("determining revisions to fetch", len(missing))
154
if not self.target.has_revision(revid):
155
missing.append(revid)
156
return graph.iter_topo_order(missing)
160
def dfetch_refs(self, refs):
163
for name, revid in refs.iteritems():
164
newrevidmap, newgitidmap = self.dfetch(revid)
165
revidmap.update(newrevidmap)
166
if revid in newgitidmap:
167
gitid = newgitidmap[revid]
169
gitid, _ = self.mapping.revision_id_bzr_to_foreign(revid)
170
self.target._git.refs[name] = gitid
171
new_refs[name] = gitid
172
return revidmap, new_refs
174
def dfetch(self, stop_revision=None):
175
"""Import the gist of the ancestry of a particular revision."""
178
self.source.lock_read()
180
todo = [revid for revid in self.missing_revisions(stop_revision) if revid != NULL_REVISION]
181
pb = ui.ui_factory.nested_progress_bar()
183
object_generator = MissingObjectsIterator(self.source, self.mapping, pb)
184
for old_bzr_revid, git_commit in object_generator.import_revisions(
186
new_bzr_revid = self.mapping.revision_id_foreign_to_bzr(git_commit)
187
revidmap[old_bzr_revid] = new_bzr_revid
188
gitidmap[old_bzr_revid] = git_commit
189
self.target._git.object_store.add_objects(object_generator)
194
return revidmap, gitidmap
197
def is_compatible(source, target):
198
"""Be compatible with GitRepository."""
199
return (not isinstance(source, GitRepository) and
200
isinstance(target, LocalGitRepository))
203
class InterToRemoteGitRepository(InterToGitRepository):
205
def dfetch_refs(self, new_refs):
206
"""Import the gist of the ancestry of a particular revision."""
208
def determine_wants(refs):
210
for name, revid in new_refs.iteritems():
211
ret[name] = store._lookup_revision_sha1(revid)
213
self.source.lock_read()
215
store = BazaarObjectStore(self.source, self.mapping)
216
def generate_blob_contents(have, want):
217
return store.iter_shas(store.find_missing_objects(have, want))
218
new_refs = self.target.send_pack(determine_wants,
219
generate_blob_contents)
222
return revidmap, new_refs
225
def is_compatible(source, target):
226
"""Be compatible with GitRepository."""
227
return (not isinstance(source, GitRepository) and
228
isinstance(target, RemoteGitRepository))
added
removed
push.py
