/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 roundtrip.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
roundtrip.py
1
 
# Copyright (C) 2010 Jelmer Vernooij <jelmer@samba.org>
2
 
#
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.
7
 
#
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.
12
 
#
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
16
 
 
17
 
"""Roundtripping support."""
18
 
 
19
 
 
20
 
from cStringIO import StringIO
21
 
 
22
 
 
23
 
class BzrGitRevisionMetadata(object):
24
 
    """Metadata for a Bazaar revision roundtripped into Git.
25
 
 
26
 
    :ivar revision_id: Revision id, as string
27
 
    :ivar properties: Revision properties, as dictionary
28
 
    :ivar explicit_parent_ids: Parent ids (needed if there are ghosts)
29
 
    :ivar verifiers: Verifier information
30
 
    """
31
 
 
32
 
    revision_id = None
33
 
 
34
 
    explicit_parent_ids = None
35
 
 
36
 
    verifiers = {}
37
 
 
38
 
    def __init__(self):
39
 
        self.properties = {}
40
 
 
41
 
    def __nonzero__(self):
42
 
        return bool(self.revision_id or self.properties)
43
 
 
44
 
 
45
 
def parse_roundtripping_metadata(text):
46
 
    """Parse Bazaar roundtripping metadata."""
47
 
    ret = BzrGitRevisionMetadata()
48
 
    f = StringIO(text)
49
 
    for l in f.readlines():
50
 
        (key, value) = l.split(":", 1)
51
 
        if key == "revision-id":
52
 
            ret.revision_id = value.strip()
53
 
        elif key == "parent-ids":
54
 
            ret.explicit_parent_ids = tuple(value.strip().split(" "))
55
 
        elif key == "testament3-sha1":
56
 
            ret.verifiers["testament3-sha1"] = value.strip()
57
 
        elif key.startswith("property-"):
58
 
            ret.properties[key[len("property-"):]] = value[1:].rstrip("\n")
59
 
        else:
60
 
            raise ValueError
61
 
    return ret
62
 
 
63
 
 
64
 
def generate_roundtripping_metadata(metadata, encoding):
65
 
    """Serialize the roundtripping metadata.
66
 
 
67
 
    :param metadata: A `BzrGitRevisionMetadata` instance
68
 
    :return: String with revision metadata
69
 
    """
70
 
    lines = []
71
 
    if metadata.revision_id:
72
 
        lines.append("revision-id: %s\n" % metadata.revision_id)
73
 
    if metadata.explicit_parent_ids:
74
 
        lines.append("parent-ids: %s\n" % " ".join(metadata.explicit_parent_ids))
75
 
    for key in sorted(metadata.properties.keys()):
76
 
        lines.append("property-%s: %s\n" % (key.encode(encoding), metadata.properties[key].encode(encoding)))
77
 
    if "testament3-sha1" in metadata.verifiers:
78
 
        lines.append("testament3-sha1: %s\n" %
79
 
                     metadata.verifiers["testament3-sha1"])
80
 
    return "".join(lines)
81
 
 
82
 
 
83
 
def extract_bzr_metadata(message):
84
 
    """Extract Bazaar metadata from a commit message.
85
 
 
86
 
    :param message: Commit message to extract from
87
 
    :return: Tuple with original commit message and metadata object
88
 
    """
89
 
    split = message.split("\n--BZR--\n", 1)
90
 
    if len(split) != 2:
91
 
        return message, None
92
 
    return split[0], parse_roundtripping_metadata(split[1])
93
 
 
94
 
 
95
 
def inject_bzr_metadata(message, metadata, encoding):
96
 
    if not metadata:
97
 
        return message
98
 
    rt_data = generate_roundtripping_metadata(metadata, encoding)
99
 
    if not rt_data:
100
 
        return message
101
 
    assert type(rt_data) == str
102
 
    return message + "\n--BZR--\n" + rt_data
103
 
 
104
 
 
105
 
def serialize_fileid_map(file_ids):
106
 
    """Serialize a file id map."""
107
 
    lines = []
108
 
    for path in sorted(file_ids.keys()):
109
 
        lines.append("%s\0%s\n" % (path, file_ids[path]))
110
 
    return lines
111
 
 
112
 
 
113
 
def deserialize_fileid_map(filetext):
114
 
    """Deserialize a file id map."""
115
 
    ret = {}
116
 
    f = StringIO(filetext)
117
 
    lines = f.readlines()
118
 
    for l in lines:
119
 
        (path, file_id) = l.rstrip("\n").split("\0")
120
 
        ret[path] = file_id
121
 
    return ret