91
88
interrepo = repository.InterRepository.get(source, self)
92
89
return interrepo.dfetch(stop_revision)
91
def dfetch_refs(self, source, stop_revision):
92
interrepo = repository.InterRepository.get(source, self)
93
return interrepo.dfetch_refs(stop_revision)
95
96
class LocalGitRepository(GitRepository):
96
97
"""Git repository on the file system."""
98
99
def __init__(self, gitdir, lockfiles):
100
# FIXME: This also caches negatives. Need to be more careful
101
# about this once we start writing to git
102
self._parents_provider = graph.CachingParentsProvider(self)
99
103
GitRepository.__init__(self, gitdir, lockfiles)
100
104
self.base = gitdir.root_transport.base
101
105
self._git = gitdir._git
104
108
self.inventories = None
105
109
self.texts = GitTexts(self)
107
def _iter_revision_ids(self):
108
mapping = self.get_mapping()
109
for sha in self._git.object_store:
110
o = self._git.object_store[sha]
111
if not isinstance(o, Commit):
113
rev, roundtrip_revid, verifiers = mapping.import_commit(o,
114
self.lookup_foreign_revision_id)
115
yield o.id, rev.revision_id, roundtrip_revid
117
111
def all_revision_ids(self):
119
for git_sha, revid, roundtrip_revid in self._iter_revision_ids():
122
ret.add(roundtrip_revid)
113
heads = self._git.refs.as_dict('refs/heads')
116
bzr_heads = [self.get_mapping().revision_id_foreign_to_bzr(h) for h in heads.itervalues()]
118
graph = self.get_graph()
119
for rev, parents in graph.iter_ancestry(bzr_heads):
123
def _make_parents_provider(self):
124
"""See Repository._make_parents_provider()."""
125
return self._parents_provider
125
127
def get_parent_map(self, revids):
127
129
for revision_id in revids:
132
134
hexsha, mapping = self.lookup_bzr_revision_id(revision_id)
134
commit = self._git[hexsha]
136
commit = self._git.commit(hexsha)
137
parent_map[revision_id] = [
138
self.lookup_foreign_revision_id(p, mapping)
139
for p in commit.parents]
140
# Older versions of Dulwich used to return None rather than
144
parent_map[revision_id] = [mapping.revision_id_foreign_to_bzr(p) for p in commit.parents]
140
145
return parent_map
142
147
def get_ancestry(self, revision_id, topo_sorted=True):
155
160
def get_signature_text(self, revision_id):
156
161
raise errors.NoSuchRevision(self, revision_id)
158
def pack(self, hint=None, clean_obsolete_packs=False):
159
self._git.object_store.pack_loose_objects()
161
163
def lookup_foreign_revision_id(self, foreign_revid, mapping=None):
162
164
"""Lookup a revision id.
166
:param revid: Bazaar revision id.
167
:return: Tuple with git revisionid and mapping.
165
assert type(foreign_revid) is str
166
169
if mapping is None:
167
170
mapping = self.get_mapping()
168
from dulwich.protocol import (
171
if foreign_revid == ZERO_SHA:
172
return revision.NULL_REVISION
173
commit = self._git[foreign_revid]
174
rev, roundtrip_revid, verifiers = mapping.import_commit(commit,
176
# FIXME: check testament before doing this?
178
return roundtrip_revid
180
return rev.revision_id
171
return mapping.revision_id_foreign_to_bzr(foreign_revid)
182
173
def has_signature_for_revision_id(self, revision_id):
185
def lookup_bzr_revision_id(self, bzr_revid, mapping=None):
176
def lookup_bzr_revision_id(self, bzr_revid):
187
178
return mapping_registry.revision_id_bzr_to_foreign(bzr_revid)
188
179
except errors.InvalidRevisionId:
190
mapping = self.get_mapping()
192
return (self._git.refs[mapping.revid_as_refname(bzr_revid)],
195
# Update refs from Git commit objects
196
# FIXME: Hitting this a lot will be very inefficient...
197
for git_sha, revid, roundtrip_revid in self._iter_revision_ids():
198
if not roundtrip_revid:
200
refname = mapping.revid_as_refname(roundtrip_revid)
201
self._git.refs[refname] = git_sha
202
if roundtrip_revid == bzr_revid:
203
return git_sha, mapping
204
raise errors.NoSuchRevision(self, bzr_revid)
180
raise errors.NoSuchRevision(self, bzr_revid)
206
182
def get_revision(self, revision_id):
207
183
git_commit_id, mapping = self.lookup_bzr_revision_id(revision_id)
209
commit = self._git[git_commit_id]
185
commit = self._git.commit(git_commit_id)
211
187
raise errors.NoSuchRevision(self, revision_id)
212
revision, roundtrip_revid, verifiers = mapping.import_commit(
213
commit, self.lookup_foreign_revision_id)
188
# print "fetched revision:", git_commit_id
189
revision = mapping.import_commit(commit)
214
190
assert revision is not None
215
# FIXME: check verifiers ?
217
revision.revision_id = roundtrip_revid
220
193
def has_revision(self, revision_id):
222
git_commit_id, mapping = self.lookup_bzr_revision_id(revision_id)
195
self.get_revision(revision_id)
223
196
except errors.NoSuchRevision:
225
return (git_commit_id in self._git)
227
def has_revisions(self, revision_ids):
228
return set(filter(self.has_revision, revision_ids))
230
201
def get_revisions(self, revids):
231
202
return [self.get_revision(r) for r in revids]