~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revision.py

merge merge tweaks from aaron, which includes latest .dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
2
 
#
 
1
# (C) 2005 Canonical
 
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
# TODO: Some kind of command-line display of revision properties:
18
 
# perhaps show them in log -v and allow them as options to the commit command.
19
 
 
20
 
 
21
 
from bzrlib.lazy_import import lazy_import
22
 
lazy_import(globals(), """
23
 
from bzrlib import deprecated_graph
24
 
from bzrlib import bugtracker
25
 
""")
26
 
from bzrlib import (
27
 
    errors,
28
 
    symbol_versioning,
29
 
    )
30
 
from bzrlib.osutils import contains_whitespace
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
import bzrlib.errors
 
19
from bzrlib.graph import node_distances, select_farthest, all_descendants
31
20
 
32
21
NULL_REVISION="null:"
33
 
CURRENT_REVISION="current:"
 
22
 
 
23
class RevisionReference(object):
 
24
    """
 
25
    Reference to a stored revision.
 
26
 
 
27
    Includes the revision_id and revision_sha1.
 
28
    """
 
29
    revision_id = None
 
30
    revision_sha1 = None
 
31
    def __init__(self, revision_id, revision_sha1=None):
 
32
        if revision_id == None \
 
33
           or isinstance(revision_id, basestring):
 
34
            self.revision_id = revision_id
 
35
        else:
 
36
            raise ValueError('bad revision_id %r' % revision_id)
 
37
 
 
38
        if revision_sha1 != None:
 
39
            if isinstance(revision_sha1, basestring) \
 
40
               and len(revision_sha1) == 40:
 
41
                self.revision_sha1 = revision_sha1
 
42
            else:
 
43
                raise ValueError('bad revision_sha1 %r' % revision_sha1)
 
44
                
34
45
 
35
46
 
36
47
class Revision(object):
42
53
 
43
54
    After bzr 0.0.5 revisions are allowed to have multiple parents.
44
55
 
45
 
    parent_ids
46
 
        List of parent revision_ids
47
 
 
48
 
    properties
49
 
        Dictionary of revision properties.  These are attached to the
50
 
        revision as extra metadata.  The name must be a single
51
 
        word; the value can be an arbitrary string.
 
56
    parents
 
57
        List of parent revisions, each is a RevisionReference.
52
58
    """
53
 
 
54
 
    def __init__(self, revision_id, properties=None, **args):
55
 
        self.revision_id = revision_id
56
 
        if properties is None:
57
 
            self.properties = {}
58
 
        else:
59
 
            self.properties = properties
60
 
            self._check_properties()
61
 
        self.committer = None
62
 
        self.parent_ids = []
63
 
        self.parent_sha1s = []
64
 
        """Not used anymore - legacy from for 4."""
 
59
    inventory_id = None
 
60
    inventory_sha1 = None
 
61
    revision_id = None
 
62
    timestamp = None
 
63
    message = None
 
64
    timezone = None
 
65
    committer = None
 
66
    
 
67
    def __init__(self, **args):
65
68
        self.__dict__.update(args)
 
69
        self.parents = []
 
70
 
66
71
 
67
72
    def __repr__(self):
68
73
        return "<Revision id %s>" % self.revision_id
70
75
    def __eq__(self, other):
71
76
        if not isinstance(other, Revision):
72
77
            return False
73
 
        return (
74
 
                self.inventory_sha1 == other.inventory_sha1
 
78
        return (self.inventory_id == other.inventory_id
 
79
                and self.inventory_sha1 == other.inventory_sha1
75
80
                and self.revision_id == other.revision_id
76
81
                and self.timestamp == other.timestamp
77
82
                and self.message == other.message
78
83
                and self.timezone == other.timezone
79
 
                and self.committer == other.committer
80
 
                and self.properties == other.properties
81
 
                and self.parent_ids == other.parent_ids)
 
84
                and self.committer == other.committer)
82
85
 
83
86
    def __ne__(self, other):
84
87
        return not self.__eq__(other)
85
88
 
86
 
    def _check_properties(self):
87
 
        """Verify that all revision properties are OK."""
88
 
        for name, value in self.properties.iteritems():
89
 
            if not isinstance(name, basestring) or contains_whitespace(name):
90
 
                raise ValueError("invalid property name %r" % name)
91
 
            if not isinstance(value, basestring):
92
 
                raise ValueError("invalid property value %r for %r" %
93
 
                                 (value, name))
94
 
 
95
 
    def get_history(self, repository):
96
 
        """Return the canonical line-of-history for this revision.
97
 
 
98
 
        If ghosts are present this may differ in result from a ghost-free
99
 
        repository.
100
 
        """
101
 
        current_revision = self
102
 
        reversed_result = []
103
 
        while current_revision is not None:
104
 
            reversed_result.append(current_revision.revision_id)
105
 
            if not len (current_revision.parent_ids):
106
 
                reversed_result.append(None)
107
 
                current_revision = None
108
 
            else:
109
 
                next_revision_id = current_revision.parent_ids[0]
110
 
                current_revision = repository.get_revision(next_revision_id)
111
 
        reversed_result.reverse()
112
 
        return reversed_result
113
 
 
114
 
    def get_summary(self):
115
 
        """Get the first line of the log message for this revision.
116
 
 
117
 
        Return an empty string if message is None.
118
 
        """
119
 
        if self.message:
120
 
            return self.message.lstrip().split('\n', 1)[0]
121
 
        else:
122
 
            return ''
123
 
 
124
 
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((1, 13, 0)))
125
 
    def get_apparent_author(self):
126
 
        """Return the apparent author of this revision.
127
 
 
128
 
        This method is deprecated in favour of get_apparent_authors.
129
 
 
130
 
        If the revision properties contain any author names,
131
 
        return the first. Otherwise return the committer name.
132
 
        """
133
 
        authors = self.get_apparent_authors()
134
 
        if authors:
135
 
            return authors[0]
136
 
        else:
137
 
            return None
138
 
 
139
 
    def get_apparent_authors(self):
140
 
        """Return the apparent authors of this revision.
141
 
 
142
 
        If the revision properties contain the names of the authors,
143
 
        return them. Otherwise return the committer name.
144
 
 
145
 
        The return value will be a list containing at least one element.
146
 
        """
147
 
        authors = self.properties.get('authors', None)
148
 
        if authors is None:
149
 
            author = self.properties.get('author', self.committer)
150
 
            if author is None:
151
 
                return []
152
 
            return [author]
153
 
        else:
154
 
            return authors.split("\n")
155
 
 
156
 
    def iter_bugs(self):
157
 
        """Iterate over the bugs associated with this revision."""
158
 
        bug_property = self.properties.get('bugs', None)
159
 
        if bug_property is None:
160
 
            return
161
 
        for line in bug_property.splitlines():
162
 
            try:
163
 
                url, status = line.split(None, 2)
164
 
            except ValueError:
165
 
                raise errors.InvalidLineInBugsProperty(line)
166
 
            if status not in bugtracker.ALLOWED_BUG_STATUSES:
167
 
                raise errors.InvalidBugStatus(status)
168
 
            yield url, status
169
 
 
 
89
        
 
90
 
 
91
REVISION_ID_RE = None
 
92
 
 
93
def validate_revision_id(rid):
 
94
    """Check rid is syntactically valid for a revision id."""
 
95
    global REVISION_ID_RE
 
96
    if not REVISION_ID_RE:
 
97
        import re
 
98
        REVISION_ID_RE = re.compile('[\w.-]+@[\w.-]+--?\d+--?[0-9a-f]+\Z')
 
99
 
 
100
    if not REVISION_ID_RE.match(rid):
 
101
        raise ValueError("malformed revision-id %r" % rid)
 
102
 
 
103
def is_ancestor(revision_id, candidate_id, revision_source):
 
104
    """Return true if candidate_id is an ancestor of revision_id.
 
105
    A false negative will be returned if any intermediate descendent of
 
106
    candidate_id is not present in any of the revision_sources.
 
107
    
 
108
    revisions_source is an object supporting a get_revision operation that
 
109
    behaves like Branch's.
 
110
    """
 
111
    if candidate_id is None:
 
112
        return True
 
113
    for ancestor_id, distance in iter_ancestors(revision_id, revision_source):
 
114
        if ancestor_id == candidate_id:
 
115
            return True
 
116
    return False
170
117
 
171
118
def iter_ancestors(revision_id, revision_source, only_present=False):
172
119
    ancestors = (revision_id,)
178
125
                yield ancestor, distance
179
126
            try:
180
127
                revision = revision_source.get_revision(ancestor)
181
 
            except errors.NoSuchRevision, e:
 
128
            except bzrlib.errors.NoSuchRevision, e:
182
129
                if e.revision == revision_id:
183
 
                    raise
 
130
                    raise 
184
131
                else:
185
132
                    continue
186
133
            if only_present:
187
134
                yield ancestor, distance
188
 
            new_ancestors.extend(revision.parent_ids)
 
135
            new_ancestors.extend([p.revision_id for p in revision.parents])
189
136
        ancestors = new_ancestors
190
137
        distance += 1
191
138
 
194
141
    """Return the ancestors of a revision present in a branch.
195
142
 
196
143
    It's possible that a branch won't have the complete ancestry of
197
 
    one of its revisions.
 
144
    one of its revisions.  
198
145
 
199
146
    """
200
147
    found_ancestors = {}
201
148
    anc_iter = enumerate(iter_ancestors(revision_id, revision_source,
202
149
                         only_present=True))
203
150
    for anc_order, (anc_id, anc_distance) in anc_iter:
204
 
        if anc_id not in found_ancestors:
 
151
        if not found_ancestors.has_key(anc_id):
205
152
            found_ancestors[anc_id] = (anc_order, anc_distance)
206
153
    return found_ancestors
207
 
 
 
154
    
208
155
 
209
156
def __get_closest(intersection):
210
157
    intersection.sort()
211
 
    matches = []
 
158
    matches = [] 
212
159
    for entry in intersection:
213
160
        if entry[0] == intersection[0][0]:
214
161
            matches.append(entry[2])
215
162
    return matches
216
163
 
217
164
 
218
 
def is_reserved_id(revision_id):
219
 
    """Determine whether a revision id is reserved
220
 
 
221
 
    :return: True if the revision is reserved, False otherwise
 
165
def old_common_ancestor(revision_a, revision_b, revision_source):
 
166
    """Find the ancestor common to both revisions that is closest to both.
222
167
    """
223
 
    return isinstance(revision_id, basestring) and revision_id.endswith(':')
224
 
 
225
 
 
226
 
def check_not_reserved_id(revision_id):
227
 
    """Raise ReservedId if the supplied revision_id is reserved"""
228
 
    if is_reserved_id(revision_id):
229
 
        raise errors.ReservedId(revision_id)
230
 
 
231
 
 
232
 
def ensure_null(revision_id):
233
 
    """Ensure only NULL_REVISION is used to represent the null revision"""
234
 
    if revision_id is None:
235
 
        symbol_versioning.warn('NULL_REVISION should be used for the null'
236
 
            ' revision instead of None, as of bzr 0.91.',
237
 
            DeprecationWarning, stacklevel=2)
238
 
        return NULL_REVISION
 
168
    from bzrlib.trace import mutter
 
169
    a_ancestors = find_present_ancestors(revision_a, revision_source)
 
170
    b_ancestors = find_present_ancestors(revision_b, revision_source)
 
171
    a_intersection = []
 
172
    b_intersection = []
 
173
    # a_order is used as a tie-breaker when two equally-good bases are found
 
174
    for revision, (a_order, a_distance) in a_ancestors.iteritems():
 
175
        if b_ancestors.has_key(revision):
 
176
            a_intersection.append((a_distance, a_order, revision))
 
177
            b_intersection.append((b_ancestors[revision][1], a_order, revision))
 
178
    mutter("a intersection: %r" % a_intersection)
 
179
    mutter("b intersection: %r" % b_intersection)
 
180
 
 
181
    a_closest = __get_closest(a_intersection)
 
182
    if len(a_closest) == 0:
 
183
        return None
 
184
    b_closest = __get_closest(b_intersection)
 
185
    assert len(b_closest) != 0
 
186
    mutter ("a_closest %r" % a_closest)
 
187
    mutter ("b_closest %r" % b_closest)
 
188
    if a_closest[0] in b_closest:
 
189
        return a_closest[0]
 
190
    elif b_closest[0] in a_closest:
 
191
        return b_closest[0]
239
192
    else:
240
 
        return revision_id
241
 
 
242
 
 
243
 
def is_null(revision_id):
244
 
    if revision_id is None:
245
 
        symbol_versioning.warn('NULL_REVISION should be used for the null'
246
 
            ' revision instead of None, as of bzr 0.90.',
247
 
            DeprecationWarning, stacklevel=2)
248
 
    return revision_id in (None, NULL_REVISION)
 
193
        raise bzrlib.errors.AmbiguousBase((a_closest[0], b_closest[0]))
 
194
    return a_closest[0]
 
195
 
 
196
def revision_graph(revision, revision_source):
 
197
    """Produce a graph of the ancestry of the specified revision.
 
198
    Return root, ancestors map, descendants map
 
199
 
 
200
    TODO: Produce graphs with the NULL revision as root, so that we can find
 
201
    a common even when trees are not branches don't represent a single line
 
202
    of descent.
 
203
    """
 
204
    ancestors = {}
 
205
    descendants = {}
 
206
    lines = [revision]
 
207
    root = None
 
208
    descendants[revision] = {}
 
209
    while len(lines) > 0:
 
210
        new_lines = set()
 
211
        for line in lines:
 
212
            if line == NULL_REVISION:
 
213
                parents = []
 
214
                root = NULL_REVISION
 
215
            else:
 
216
                try:
 
217
                    rev = revision_source.get_revision(line)
 
218
                    parents = [p.revision_id for p in rev.parents]
 
219
                    if len(parents) == 0:
 
220
                        parents = [NULL_REVISION]
 
221
                except bzrlib.errors.NoSuchRevision:
 
222
                    if line == revision:
 
223
                        raise
 
224
                    parents = None
 
225
            if parents is not None:
 
226
                for parent in parents:
 
227
                    if parent not in ancestors:
 
228
                        new_lines.add(parent)
 
229
                    if parent not in descendants:
 
230
                        descendants[parent] = {}
 
231
                    descendants[parent][line] = 1
 
232
            if parents is not None:
 
233
                ancestors[line] = set(parents)
 
234
        lines = new_lines
 
235
    assert root not in descendants[root]
 
236
    assert root not in ancestors[root]
 
237
    return root, ancestors, descendants
 
238
 
 
239
def combined_graph(revision_a, revision_b, revision_source):
 
240
    """Produce a combined ancestry graph.
 
241
    Return graph root, ancestors map, descendants map, set of common nodes"""
 
242
    root, ancestors, descendants = revision_graph(revision_a, revision_source)
 
243
    root_b, ancestors_b, descendants_b = revision_graph(revision_b, 
 
244
                                                        revision_source)
 
245
    if root != root_b:
 
246
        raise bzrlib.errors.NoCommonRoot(revision_a, revision_b)
 
247
    common = set()
 
248
    for node, node_anc in ancestors_b.iteritems():
 
249
        if node in ancestors:
 
250
            common.add(node)
 
251
        else:
 
252
            ancestors[node] = set()
 
253
        ancestors[node].update(node_anc)
 
254
    for node, node_dec in descendants_b.iteritems():
 
255
        if node not in descendants:
 
256
            descendants[node] = set()
 
257
        descendants[node].update(node_dec)
 
258
    return root, ancestors, descendants, common
 
259
 
 
260
def common_ancestor(revision_a, revision_b, revision_source):
 
261
    try:
 
262
        root, ancestors, descendants, common = \
 
263
            combined_graph(revision_a, revision_b, revision_source)
 
264
    except bzrlib.errors.NoCommonRoot:
 
265
        raise bzrlib.errors.NoCommonAncestor(revision_a, revision_b)
 
266
        
 
267
    distances = node_distances (descendants, ancestors, root)
 
268
    farthest = select_farthest(distances, common)
 
269
    if farthest is None or farthest == NULL_REVISION:
 
270
        raise bzrlib.errors.NoCommonAncestor(revision_a, revision_b)
 
271
    return farthest
 
272
 
 
273
class MultipleRevisionSources(object):
 
274
    """Proxy that looks in multiple branches for revisions."""
 
275
    def __init__(self, *args):
 
276
        object.__init__(self)
 
277
        assert len(args) != 0
 
278
        self._revision_sources = args
 
279
 
 
280
    def get_revision(self, revision_id):
 
281
        for source in self._revision_sources:
 
282
            try:
 
283
                return source.get_revision(revision_id)
 
284
            except bzrlib.errors.NoSuchRevision, e:
 
285
                pass
 
286
        raise e
 
287
 
 
288
def get_intervening_revisions(ancestor_id, rev_id, rev_source, 
 
289
                              revision_history=None):
 
290
    """Find the longest line of descent from maybe_ancestor to revision.
 
291
    Revision history is followed where possible.
 
292
 
 
293
    If ancestor_id == rev_id, list will be empty.
 
294
    Otherwise, rev_id will be the last entry.  ancestor_id will never appear.
 
295
    If ancestor_id is not an ancestor, NotAncestor will be thrown
 
296
    """
 
297
    root, ancestors, descendants = revision_graph(rev_id, rev_source)
 
298
    if len(descendants) == 0:
 
299
        raise NoSuchRevision(rev_source, rev_id)
 
300
    if ancestor_id not in descendants:
 
301
        rev_source.get_revision(ancestor_id)
 
302
        raise bzrlib.errors.NotAncestor(rev_id, ancestor_id)
 
303
    root_descendants = all_descendants(descendants, ancestor_id)
 
304
    root_descendants.add(ancestor_id)
 
305
    if rev_id not in root_descendants:
 
306
        raise bzrlib.errors.NotAncestor(rev_id, ancestor_id)
 
307
    distances = node_distances(descendants, ancestors, ancestor_id,
 
308
                               root_descendants=root_descendants)
 
309
 
 
310
    def best_ancestor(rev_id):
 
311
        best = None
 
312
        for anc_id in ancestors[rev_id]:
 
313
            try:
 
314
                distance = distances[anc_id]
 
315
            except KeyError:
 
316
                continue
 
317
            if revision_history is not None and anc_id in revision_history:
 
318
                return anc_id
 
319
            elif best is None or distance > best[1]:
 
320
                best = (anc_id, distance)
 
321
        return best[0]
 
322
 
 
323
    next = rev_id
 
324
    path = []
 
325
    while next != ancestor_id:
 
326
        path.append(next)
 
327
        next = best_ancestor(next)
 
328
    path.reverse()
 
329
    return path