102
107
def all_revision_ids(self):
103
108
ret = set([revision.NULL_REVISION])
104
if self._git.heads() == []:
109
heads = self._git.heads()
106
bzr_heads = [self.get_mapping().revision_id_foreign_to_bzr(h) for h in self._git.heads()]
112
bzr_heads = [self.get_mapping().revision_id_foreign_to_bzr(h) for h in heads.itervalues()]
107
113
ret = set(bzr_heads)
108
114
graph = self.get_graph()
109
115
for rev, parents in graph.iter_ancestry(bzr_heads):
139
145
"""See Repository.get_ancestry().
141
147
if revision_id is None:
142
return self._all_revision_ids()
148
return [None, revision.NULL_REVISION] + self._all_revision_ids()
143
149
assert isinstance(revision_id, str)
145
151
graph = self.get_graph()
146
152
for rev, parents in graph.iter_ancestry([revision_id]):
147
if rev == revision.NULL_REVISION:
149
153
ancestry.append(rev)
150
154
ancestry.reverse()
153
def import_revision_gist(self, source, revid):
155
return [None] + ancestry
157
def import_revision_gist(self, source, revid, parent_lookup):
158
"""Import the gist of a revision into this Git repository.
162
rev = source.get_revision(revid)
163
for sha, object, path in inventory_to_tree_and_blobs(source, None, revid):
166
objects.append((object, path))
167
commit = revision_to_commit(rev, tree_sha, parent_lookup)
168
objects.append((commit, None))
169
self._git.object_store.add_objects(objects)
170
return commit.sha().hexdigest()
156
172
def dfetch(self, source, stop_revision):
173
"""Import the gist of the ancestry of a particular revision."""
157
174
if stop_revision is None:
158
175
raise NotImplementedError
163
181
graph = source.get_graph()
164
for revid, parents in graph.iter_ancestry([stop_revision]):
182
ancestry = [x for x in source.get_ancestry(stop_revision) if x is not None]
183
for revid in graph.iter_topo_order(ancestry):
165
184
if not self.has_revision(revid):
166
185
todo.append(revid)
167
186
pb = ui.ui_factory.nested_progress_bar()
169
for i, revid in enumerate(reversed(todo)):
188
for i, revid in enumerate(todo):
170
189
pb.update("pushing revisions", i, len(todo))
171
revidmap[revid] = self.import_revision_gist(source, revid)
190
git_commit = self.import_revision_gist(source, revid, gitidmap.__getitem__)
191
gitidmap[revid] = git_commit
192
git_revid = self.get_mapping().revision_id_foreign_to_bzr(git_commit)
193
revidmap[revid] = git_revid
196
source.fetch(self, revision_id=revidmap[stop_revision])
239
262
def set_make_working_trees(self, trees):
242
def fetch_objects(self, determine_wants, graph_walker, progress=None):
265
def fetch_objects(self, determine_wants, graph_walker, resolve_ext_ref, progress=None):
243
266
return self._git.fetch_objects(determine_wants, graph_walker, progress)