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.mapping import (
33
extract_unusual_modes,
35
from bzrlib.plugins.git.object_store import (
38
from bzrlib.plugins.git.repository import (
43
from bzrlib.plugins.git.remote import (
48
class MissingObjectsIterator(object):
49
"""Iterate over git objects that are missing from a target repository.
53
def __init__(self, store, source, pb=None):
54
"""Create a new missing objects iterator.
58
self._object_store = store
60
self._sent_shas = set()
64
def import_revisions(self, revids):
65
self._revids.update(revids)
66
for i, revid in enumerate(revids):
68
self.pb.update("pushing revisions", i, len(revids))
69
git_commit = self.import_revision(revid)
70
yield (revid, git_commit)
72
def need_sha(self, sha):
73
if sha in self._sent_shas:
75
(type, (fileid, revid)) = self._object_store._idmap.lookup_git_sha(sha)
76
assert type in ("blob", "tree")
77
if revid in self._revids:
78
# Not sent yet, and part of the set of revisions to send
80
# Not changed in the revisions to send, so either not necessary
81
# or already present remotely (as git doesn't do ghosts)
84
def queue(self, sha, obj, path, ie=None, inv=None):
87
self._pending.append((obj, path))
88
self._sent_shas.add(sha)
90
def import_revision(self, revid):
91
"""Import the gist of a revision into this Git repository.
94
inv = self.source.get_inventory(revid)
95
rev = self.source.get_revision(revid)
96
unusual_modes = extract_unusual_modes(rev)
101
(sha, object) = self._object_store._get_ie_object_or_sha1(ie, inv, unusual_modes)
102
if ie.parent_id is None:
104
if not self.need_sha(sha):
106
self.queue(sha, object, inv.id2path(ie.file_id), ie, inv)
107
if ie.kind == "directory":
108
todo.extend(ie.children.values())
109
assert tree_sha is not None
110
commit = self._object_store._get_commit(rev, tree_sha)
111
self.queue(commit.id, commit, None)
115
return len(self._pending)
118
for i, (object, path) in enumerate(self._pending):
120
self.pb.update("writing pack objects", i, len(self))
121
if isinstance(object, tuple):
122
object = self._object_store._get_ie_object(*object)
126
class InterToGitRepository(InterRepository):
127
"""InterRepository that copies into a Git repository."""
129
_matching_repo_format = GitRepositoryFormat()
131
def __init__(self, source, target):
132
super(InterToGitRepository, self).__init__(source, target)
133
self.mapping = self.target.get_mapping()
134
self.source_store = BazaarObjectStore(self.source, self.mapping)
137
def _get_repo_format_to_test():
140
def copy_content(self, revision_id=None, pb=None):
141
"""See InterRepository.copy_content."""
142
self.fetch(revision_id, pb, find_ghosts=False)
144
def fetch(self, revision_id=None, pb=None, find_ghosts=False,
146
raise NoPushSupport()
149
class InterToLocalGitRepository(InterToGitRepository):
151
def missing_revisions(self, stop_revisions, check_revid):
153
pb = ui.ui_factory.nested_progress_bar()
155
graph = self.source.get_graph()
156
for revid, _ in graph.iter_ancestry(stop_revisions):
157
pb.update("determining revisions to fetch", len(missing))
158
if not check_revid(revid):
159
missing.append(revid)
160
return graph.iter_topo_order(missing)
164
def dfetch_refs(self, refs):
166
revidmap, gitidmap = self.dfetch(refs.values())
167
for name, revid in refs.iteritems():
168
if revid in gitidmap:
169
gitid = gitidmap[revid]
171
gitid = self.source_store._lookup_revision_sha1(revid)
172
self.target._git.refs[name] = gitid
173
new_refs[name] = gitid
174
return revidmap, new_refs
176
def dfetch(self, stop_revisions):
177
"""Import the gist of the ancestry of a particular revision."""
180
self.source.lock_read()
182
target_store = self.target._git.object_store
183
def check_revid(revid):
184
if revid == NULL_REVISION:
186
return (self.source_store._lookup_revision_sha1(revid) in target_store)
187
todo = list(self.missing_revisions(stop_revisions, check_revid))
188
pb = ui.ui_factory.nested_progress_bar()
190
object_generator = MissingObjectsIterator(self.source_store, self.source, 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
target_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 determine_wants(refs):
217
for name, revid in new_refs.iteritems():
218
ret[name] = self.source_store._lookup_revision_sha1(revid)
220
self.source.lock_read()
222
new_refs = self.target.send_pack(determine_wants,
223
self.source_store.generate_pack_contents)
226
return revidmap, new_refs
229
def is_compatible(source, target):
230
"""Be compatible with GitRepository."""
231
return (not isinstance(source, GitRepository) and
232
isinstance(target, RemoteGitRepository))
added
removed
push.py
