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, store, source, pb=None):
51
"""Create a new missing objects iterator.
55
self._object_store = store
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()
129
self.source_store = BazaarObjectStore(self.source, self.mapping)
132
def _get_repo_format_to_test():
135
def copy_content(self, revision_id=None, pb=None):
136
"""See InterRepository.copy_content."""
137
self.fetch(revision_id, pb, find_ghosts=False)
139
def fetch(self, revision_id=None, pb=None, find_ghosts=False,
141
raise NoPushSupport()
144
class InterToLocalGitRepository(InterToGitRepository):
146
def missing_revisions(self, stop_revisions, check_revid):
148
pb = ui.ui_factory.nested_progress_bar()
150
graph = self.source.get_graph()
151
for revid, _ in graph.iter_ancestry(stop_revisions):
152
pb.update("determining revisions to fetch", len(missing))
153
if not check_revid(revid):
154
missing.append(revid)
155
return graph.iter_topo_order(missing)
159
def dfetch_refs(self, refs):
161
revidmap, gitidmap = self.dfetch(refs.values())
162
for name, revid in refs.iteritems():
163
if revid in gitidmap:
164
gitid = gitidmap[revid]
166
gitid = self.source_store._lookup_revision_sha1(revid)
167
self.target._git.refs[name] = gitid
168
new_refs[name] = gitid
169
return revidmap, new_refs
171
def dfetch(self, stop_revisions):
172
"""Import the gist of the ancestry of a particular revision."""
175
self.source.lock_read()
177
target_store = self.target._git.object_store
178
def check_revid(revid):
179
if revid == NULL_REVISION:
181
return (self.source_store._lookup_revision_sha1(revid) in target_store)
182
todo = list(self.missing_revisions(stop_revisions, check_revid))
183
pb = ui.ui_factory.nested_progress_bar()
185
object_generator = MissingObjectsIterator(self.source_store, self.source, pb)
186
for old_bzr_revid, git_commit in object_generator.import_revisions(
188
new_bzr_revid = self.mapping.revision_id_foreign_to_bzr(git_commit)
189
revidmap[old_bzr_revid] = new_bzr_revid
190
gitidmap[old_bzr_revid] = git_commit
191
target_store.add_objects(object_generator)
196
return revidmap, gitidmap
199
def is_compatible(source, target):
200
"""Be compatible with GitRepository."""
201
return (not isinstance(source, GitRepository) and
202
isinstance(target, LocalGitRepository))
205
class InterToRemoteGitRepository(InterToGitRepository):
207
def dfetch_refs(self, new_refs):
208
"""Import the gist of the ancestry of a particular revision."""
210
def determine_wants(refs):
212
for name, revid in new_refs.iteritems():
213
ret[name] = self.source_store._lookup_revision_sha1(revid)
215
self.source.lock_read()
217
new_refs = self.target.send_pack(determine_wants,
218
self.source_store.generate_pack_contents)
221
return revidmap, new_refs
224
def is_compatible(source, target):
225
"""Be compatible with GitRepository."""
226
return (not isinstance(source, GitRepository) and
227
isinstance(target, RemoteGitRepository))
added
removed
push.py
