/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to mapping.py

Add description of git-v1 mapping.

Show diffs side-by-side

added added

removed removed

Lines of Context:
mapping.py
1
 
# Copyright (C) 2007-2008 Canonical Ltd
 
1
# Copyright (C) 2007 Canonical Ltd
 
2
# Copyright (C) 2008-2009 Jelmer Vernooij <jelmer@samba.org>
 
3
# Copyright (C) 2008 John Carr
2
4
#
3
5
# This program is free software; you can redistribute it and/or modify
4
6
# it under the terms of the GNU General Public License as published by
16
18
 
17
19
"""Converters, etc for going between Bazaar and Git ids."""
18
20
 
19
 
from bzrlib import errors, foreign, urlutils
20
 
from bzrlib.inventory import ROOT_ID
 
21
from bzrlib import (
 
22
    errors,
 
23
    foreign,
 
24
    urlutils,
 
25
    )
 
26
from bzrlib.inventory import (
 
27
    ROOT_ID,
 
28
    )
21
29
from bzrlib.foreign import (
22
 
        ForeignVcs, 
23
 
        VcsMappingRegistry, 
24
 
        ForeignRevision,
25
 
        )
 
30
    ForeignVcs, 
 
31
    VcsMappingRegistry, 
 
32
    ForeignRevision,
 
33
    )
 
34
from bzrlib.xml_serializer import (
 
35
    escape_invalid_chars,
 
36
    )
 
37
 
26
38
 
27
39
def escape_file_id(file_id):
28
40
    return file_id.replace('_', '__').replace(' ', '_s')
55
67
        return bzr_rev_id[len(cls.revid_prefix)+1:], cls()
56
68
 
57
69
    def generate_file_id(self, path):
 
70
        # Git paths are just bytestrings
 
71
        # We must just hope they are valid UTF-8..
 
72
        assert isinstance(path, str)
58
73
        if path == "":
59
74
            return ROOT_ID
60
 
        return escape_file_id(path.encode('utf-8'))
 
75
        return escape_file_id(path)
 
76
 
 
77
    def parse_file_id(self, file_id):
 
78
        if file_id == ROOT_ID:
 
79
            return ""
 
80
        return unescape_file_id(file_id)
61
81
 
62
82
    def import_commit(self, commit):
63
83
        """Convert a git commit to a bzr revision.
68
88
            raise AssertionError("Commit object can't be None")
69
89
        rev = ForeignRevision(commit.id, self, self.revision_id_foreign_to_bzr(commit.id))
70
90
        rev.parent_ids = tuple([self.revision_id_foreign_to_bzr(p) for p in commit.parents])
71
 
        rev.message = commit.message.decode("utf-8", "replace")
72
 
        rev.committer = str(commit.committer).decode("utf-8", "replace")
 
91
        rev.message = escape_invalid_chars(commit.message.decode("utf-8", "replace"))[0]
 
92
        rev.committer = escape_invalid_chars(str(commit.committer).decode("utf-8", "replace"))[0]
73
93
        if commit.committer != commit.author:
74
 
            rev.properties['author'] = str(commit.author).decode("utf-8", "replace")
 
94
            rev.properties['author'] = escape_invalid_chars(str(commit.author).decode("utf-8", "replace"))[0]
75
95
        rev.timestamp = commit.commit_time
76
96
        rev.timezone = 0
77
97
        return rev
121
141
default_mapping = BzrGitMappingv1()
122
142
 
123
143
 
 
144
def text_to_blob(text):
 
145
    from dulwich.objects import Blob
 
146
    blob = Blob()
 
147
    blob._text = text
 
148
    return blob
 
149
 
 
150
 
124
151
def inventory_to_tree_and_blobs(repo, mapping, revision_id):
125
 
    from dulwich.objects import Tree, Blob
 
152
    from dulwich.objects import Tree
126
153
    from bzrlib.inventory import InventoryDirectory, InventoryFile
127
154
    import stat
128
155
    stack = []
133
160
 
134
161
    # stack contains the set of trees that we haven't 
135
162
    # finished constructing
136
 
 
137
163
    for path, entry in inv.iter_entries():
138
164
        while stack and not path.startswith(cur):
139
165
            tree.serialize()
140
 
            sha = tree.sha().hexdigest()
 
166
            sha = tree.id
141
167
            yield sha, tree, cur
142
168
            t = (stat.S_IFDIR, urlutils.basename(cur).encode('UTF-8'), sha)
143
169
            cur, tree = stack.pop()
151
177
        if type(entry) == InventoryFile:
152
178
            #FIXME: We can make potentially make this Lazy to avoid shaing lots of stuff
153
179
            # and having all these objects in memory at once
154
 
            blob = Blob()
155
 
            _, blob._text = repo.iter_files_bytes([(entry.file_id, entry.revision, path)]).next()
156
 
            sha = blob.sha().hexdigest()
 
180
            text = repo.texts.get_record_stream([(entry.file_id, entry.revision)], 'unordered', True).next().get_bytes_as('fulltext')
 
181
            blob = text_to_blob(text)
 
182
            sha = blob.id
157
183
            yield sha, blob, path
158
184
 
159
185
            name = urlutils.basename(path).encode("utf-8")
164
190
 
165
191
    while len(stack) > 1:
166
192
        tree.serialize()
167
 
        sha = tree.sha().hexdigest()
 
193
        sha = tree.id
168
194
        yield sha, tree, cur
169
195
        t = (stat.S_IFDIR, urlutils.basename(cur).encode('UTF-8'), sha)
170
196
        cur, tree = stack.pop()
171
197
        tree.add(*t)
172
198
 
173
199
    tree.serialize()
174
 
    yield tree.sha().hexdigest(), tree, cur
 
200
    yield tree.id, tree, cur
175
201
 
176
202
 
177
203
def revision_to_commit(rev, tree_sha, parent_lookup):
187
213
    for p in rev.parent_ids:
188
214
        git_p = parent_lookup(p)
189
215
        if git_p is not None:
 
216
            assert len(git_p) == 40, "unexpected length for %r" % git_p
190
217
            commit._parents.append(git_p)
191
218
    commit._message = rev.message.encode("utf-8")
192
219
    commit._committer = rev.committer.encode("utf-8")
193
 
    commit._author = rev.get_apparent_author().encode("utf-8")
 
220
    commit._author = rev.get_apparent_authors()[0].encode("utf-8")
194
221
    commit._commit_time = long(rev.timestamp)
195
222
    commit.serialize()
196
223
    return commit