20
20
from dulwich.objects import (
25
from bzrlib.revision import (
28
26
from bzrlib.versionedfile import (
29
27
AbsentContentFactory,
30
FulltextContentFactory,
28
ChunkedContentFactory,
34
from bzrlib.plugins.git.converter import (
33
class GitRevisions(VersionedFiles):
35
def __init__(self, repository, object_store):
36
self.repository = repository
37
self.object_store = object_store
39
def check(self, progressbar=None):
43
for sha in self.object_store:
44
if type(sha) == Commit:
48
return list(self.iterkeys())
50
def add_mpdiffs(self, records):
51
raise NotImplementedError(self.add_mpdiffs)
53
def get_record_stream(self, keys, ordering, include_delta_closure):
56
(commit_id, mapping) = self.repository.lookup_bzr_revision_id(revid)
58
commit = self.object_store[commit_id]
60
yield AbsentContentFactory(key)
62
yield ChunkedContentFactory(key,
63
tuple([(self.repository.lookup_foreign_revision_id(p, mapping),) for p in commit.parents]), None,
64
commit.as_raw_chunks())
66
def get_parent_map(self, keys):
69
(commit_id, mapping) = self.repository.lookup_bzr_revision_id(revid)
71
ret[(revid,)] = [(self.repository.lookup_foreign_revision_id(p, mapping),) for p in self.object_store[commit_id].parents]
38
77
class GitTexts(VersionedFiles):
39
78
"""A texts VersionedFiles instance that is backed onto a Git object store."""
51
90
def get_record_stream(self, keys, ordering, include_delta_closure):
53
92
(fileid, revid) = key
54
(foreign_revid, mapping) = self.repository.lookup_git_revid(revid)
55
root_tree = self.object_store[foreign_revid].tree
93
(commit_id, mapping) = self.repository.lookup_bzr_revision_id(revid)
94
root_tree = self.object_store[commit_id].tree
56
95
path = mapping.parse_file_id(fileid)
58
obj = tree_lookup_path(self.object_store.__getitem__,
97
obj = tree_lookup_path(
98
self.object_store.__getitem__, root_tree, path)
99
if isinstance(obj, tuple):
100
(mode, item_id) = obj
101
obj = self.object_store[item_id]
61
103
yield AbsentContentFactory(key)
63
105
if isinstance(obj, Tree):
64
yield FulltextContentFactory(key, None, None, "")
106
yield ChunkedContentFactory(key, None, None, [])
65
107
elif isinstance(obj, Blob):
66
yield FulltextContentFactory(key, None, None, obj._text)
108
yield ChunkedContentFactory(key, None, None, obj.chunked)
68
yield AbsentContentFactory(key)
110
raise AssertionError("file text resolved to %r" % obj)
70
112
def get_parent_map(self, keys):
71
113
raise NotImplementedError(self.get_parent_map)