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
"""Map from Git sha's to Bazaar objects."""
19
from dulwich.objects import (
30
from bzrlib.plugins.git.mapping import (
35
from bzrlib.plugins.git.shamap import (
40
class BazaarObjectStore(object):
41
"""A Git-style object store backed onto a Bazaar repository."""
43
def __init__(self, repository, mapping=None):
44
self.repository = repository
46
self.mapping = self.repository.get_mapping()
48
self.mapping = mapping
49
self._idmap = SqliteGitShaMap.from_repository(repository)
51
def _update_sha_map(self):
52
all_revids = self.repository.all_revision_ids()
53
graph = self.repository.get_graph()
54
present_revids = set(self._idmap.revids())
55
missing_revids = [revid for revid in graph.iter_topo_order(all_revids) if revid not in present_revids]
56
pb = ui.ui_factory.nested_progress_bar()
58
for i, revid in enumerate(missing_revids):
59
pb.update("updating git map", i, len(missing_revids))
60
self._update_sha_map_revision(revid)
66
self._update_sha_map()
67
return iter(self._idmap.sha1s())
69
def _update_sha_map_revision(self, revid):
70
inv = self.repository.get_inventory(revid)
71
tree_sha = self._get_ie_sha1(inv.root, inv)
72
rev = self.repository.get_revision(revid)
73
commit_obj = revision_to_commit(rev, tree_sha,
74
self._idmap._parent_lookup)
76
foreign_revid, mapping = mapping_registry.parse_revision_id(revid)
77
except errors.InvalidRevisionId:
80
if foreign_revid != commit_obj.id:
81
raise AssertionError("recreated git commit had different sha1: expected %s, got %s" % (foreign_revid, commit_obj.id))
82
self._idmap.add_entry(commit_obj.id, "commit", (revid, tree_sha))
84
def _check_expected_sha(self, expected_sha, object):
85
if expected_sha is None:
87
if expected_sha != object.id:
88
raise AssertionError("Invalid sha for %r: %s" % (object, expected_sha))
90
def _get_ie_object(self, entry, inv):
91
if entry.kind == "directory":
92
return self._get_tree(entry.file_id, inv.revision_id, inv=inv)
94
return self._get_blob(entry.file_id, entry.revision)
96
def _get_ie_object_or_sha1(self, entry, inv):
97
if entry.kind == "directory":
99
return self._idmap.lookup_tree(entry.file_id, inv.revision_id), None
101
ret = self._get_ie_object(entry, inv)
102
self._idmap.add_entry(ret.id, "tree", (entry.file_id, inv.revision_id))
106
return self._idmap.lookup_blob(entry.file_id, entry.revision), None
108
ret = self._get_ie_object(entry, inv)
109
self._idmap.add_entry(ret.id, "blob", (entry.file_id, entry.revision))
112
def _get_ie_sha1(self, entry, inv):
113
return self._get_ie_object_or_sha1(entry, inv)[0]
115
def _get_blob(self, fileid, revision, expected_sha=None):
116
"""Return a Git Blob object from a fileid and revision stored in bzr.
118
:param fileid: File id of the text
119
:param revision: Revision of the text
121
text = self.repository.texts.get_record_stream([(fileid, revision)],
122
"unordered", True).next().get_bytes_as("fulltext")
125
self._check_expected_sha(expected_sha, blob)
128
def _get_tree(self, fileid, revid, inv=None, expected_sha=None):
129
"""Return a Git Tree object from a file id and a revision stored in bzr.
131
:param fileid: fileid in the tree.
132
:param revision: Revision of the tree.
135
inv = self.repository.get_inventory(revid)
136
tree = directory_to_tree(inv[fileid], lambda ie: self._get_ie_sha1(ie, inv))
137
self._check_expected_sha(expected_sha, tree)
140
def _get_commit(self, revid, tree_sha, expected_sha=None):
141
rev = self.repository.get_revision(revid)
142
commit = revision_to_commit(rev, tree_sha, self._lookup_revision_sha1)
143
self._check_expected_sha(expected_sha, commit)
146
def _lookup_revision_sha1(self, revid):
148
return self._idmap._parent_lookup(revid)
150
inv = self.repository.get_inventory(revid)
151
tree_sha = self._get_ie_sha1(inv.root, inv)
152
ret = self._get_commit(revid, tree_sha).id
153
self._idmap.add_entry(ret, "commit", (revid, tree_sha))
156
def get_raw(self, sha):
157
return self[sha].as_raw_string()
159
def __getitem__(self, sha):
160
# See if sha is in map
162
(type, type_data) = self._idmap.lookup_git_sha(sha)
164
# if not, see if there are any unconverted revisions and add them
165
# to the map, search for sha in map again
166
self._update_sha_map()
167
(type, type_data) = self._idmap.lookup_git_sha(sha)
168
# convert object to git object
170
return self._get_commit(type_data[0], type_data[1],
173
return self._get_blob(type_data[0], type_data[1], expected_sha=sha)
175
return self._get_tree(type_data[0], type_data[1], expected_sha=sha)
177
raise AssertionError("Unknown object type '%s'" % type)
added
removed
converter.py
