/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 bzrlib/revision.py

  • Committer: Aaron Bentley
  • Date: 2005-09-12 02:55:34 UTC
  • mto: (1185.3.4)
  • mto: This revision was merged to the branch mainline in revision 1390.
  • Revision ID: aaron.bentley@utoronto.ca-20050912025534-43d7275dd948e4ad
Made pull work after remote branch has merged latest revision

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# (C) 2005 Canonical
 
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
 
 
18
import bzrlib.errors
 
19
from bzrlib.graph import farthest_nodes, node_distances, all_descendants
 
20
 
 
21
class RevisionReference(object):
 
22
    """
 
23
    Reference to a stored revision.
 
24
 
 
25
    Includes the revision_id and revision_sha1.
 
26
    """
 
27
    revision_id = None
 
28
    revision_sha1 = None
 
29
    def __init__(self, revision_id, revision_sha1=None):
 
30
        if revision_id == None \
 
31
           or isinstance(revision_id, basestring):
 
32
            self.revision_id = revision_id
 
33
        else:
 
34
            raise ValueError('bad revision_id %r' % revision_id)
 
35
 
 
36
        if revision_sha1 != None:
 
37
            if isinstance(revision_sha1, basestring) \
 
38
               and len(revision_sha1) == 40:
 
39
                self.revision_sha1 = revision_sha1
 
40
            else:
 
41
                raise ValueError('bad revision_sha1 %r' % revision_sha1)
 
42
                
 
43
 
 
44
 
 
45
class Revision(object):
 
46
    """Single revision on a branch.
 
47
 
 
48
    Revisions may know their revision_hash, but only once they've been
 
49
    written out.  This is not stored because you cannot write the hash
 
50
    into the file it describes.
 
51
 
 
52
    After bzr 0.0.5 revisions are allowed to have multiple parents.
 
53
 
 
54
    parents
 
55
        List of parent revisions, each is a RevisionReference.
 
56
    """
 
57
    inventory_id = None
 
58
    inventory_sha1 = None
 
59
    revision_id = None
 
60
    timestamp = None
 
61
    message = None
 
62
    timezone = None
 
63
    committer = None
 
64
    
 
65
    def __init__(self, **args):
 
66
        self.__dict__.update(args)
 
67
        self.parents = []
 
68
 
 
69
 
 
70
    def __repr__(self):
 
71
        return "<Revision id %s>" % self.revision_id
 
72
 
 
73
        
 
74
    def to_element(self):
 
75
        from bzrlib.xml import Element, SubElement
 
76
        
 
77
        root = Element('revision',
 
78
                       committer = self.committer,
 
79
                       timestamp = '%.9f' % self.timestamp,
 
80
                       revision_id = self.revision_id,
 
81
                       inventory_id = self.inventory_id,
 
82
                       inventory_sha1 = self.inventory_sha1,
 
83
                       )
 
84
        if self.timezone:
 
85
            root.set('timezone', str(self.timezone))
 
86
        root.text = '\n'
 
87
        
 
88
        msg = SubElement(root, 'message')
 
89
        msg.text = self.message
 
90
        msg.tail = '\n'
 
91
 
 
92
        if self.parents:
 
93
            pelts = SubElement(root, 'parents')
 
94
            pelts.tail = pelts.text = '\n'
 
95
            for rr in self.parents:
 
96
                assert isinstance(rr, RevisionReference)
 
97
                p = SubElement(pelts, 'revision_ref')
 
98
                p.tail = '\n'
 
99
                assert rr.revision_id
 
100
                p.set('revision_id', rr.revision_id)
 
101
                if rr.revision_sha1:
 
102
                    p.set('revision_sha1', rr.revision_sha1)
 
103
 
 
104
        return root
 
105
 
 
106
 
 
107
    def from_element(cls, elt):
 
108
        return unpack_revision(elt)
 
109
 
 
110
    from_element = classmethod(from_element)
 
111
 
 
112
 
 
113
 
 
114
def unpack_revision(elt):
 
115
    """Convert XML element into Revision object."""
 
116
    # <changeset> is deprecated...
 
117
    if elt.tag not in ('revision', 'changeset'):
 
118
        raise bzrlib.errors.BzrError("unexpected tag in revision file: %r" % elt)
 
119
 
 
120
    rev = Revision(committer = elt.get('committer'),
 
121
                   timestamp = float(elt.get('timestamp')),
 
122
                   revision_id = elt.get('revision_id'),
 
123
                   inventory_id = elt.get('inventory_id'),
 
124
                   inventory_sha1 = elt.get('inventory_sha1')
 
125
                   )
 
126
 
 
127
    precursor = elt.get('precursor')
 
128
    precursor_sha1 = elt.get('precursor_sha1')
 
129
 
 
130
    pelts = elt.find('parents')
 
131
 
 
132
    if pelts:
 
133
        for p in pelts:
 
134
            assert p.tag == 'revision_ref', \
 
135
                   "bad parent node tag %r" % p.tag
 
136
            rev_ref = RevisionReference(p.get('revision_id'),
 
137
                                        p.get('revision_sha1'))
 
138
            rev.parents.append(rev_ref)
 
139
 
 
140
        if precursor:
 
141
            # must be consistent
 
142
            prec_parent = rev.parents[0].revision_id
 
143
            assert prec_parent == precursor
 
144
    elif precursor:
 
145
        # revisions written prior to 0.0.5 have a single precursor
 
146
        # give as an attribute
 
147
        rev_ref = RevisionReference(precursor, precursor_sha1)
 
148
        rev.parents.append(rev_ref)
 
149
 
 
150
    v = elt.get('timezone')
 
151
    rev.timezone = v and int(v)
 
152
 
 
153
    rev.message = elt.findtext('message') # text of <message>
 
154
    return rev
 
155
 
 
156
 
 
157
 
 
158
REVISION_ID_RE = None
 
159
 
 
160
def validate_revision_id(rid):
 
161
    """Check rid is syntactically valid for a revision id."""
 
162
    global REVISION_ID_RE
 
163
    if not REVISION_ID_RE:
 
164
        import re
 
165
        REVISION_ID_RE = re.compile('[\w.-]+@[\w.-]+--?\d+--?[0-9a-f]+\Z')
 
166
 
 
167
    if not REVISION_ID_RE.match(rid):
 
168
        raise ValueError("malformed revision-id %r" % rid)
 
169
 
 
170
def is_ancestor(revision_id, candidate_id, revision_source):
 
171
    """Return true if candidate_id is an ancestor of revision_id.
 
172
    A false negative will be returned if any intermediate descendent of
 
173
    candidate_id is not present in any of the revision_sources.
 
174
    
 
175
    revisions_source is an object supporting a get_revision operation that
 
176
    behaves like Branch's.
 
177
    """
 
178
 
 
179
    for ancestor_id, distance in iter_ancestors(revision_id, revision_source):
 
180
        if ancestor_id == candidate_id:
 
181
            return True
 
182
    return False
 
183
 
 
184
def iter_ancestors(revision_id, revision_source, only_present=False):
 
185
    ancestors = (revision_id,)
 
186
    distance = 0
 
187
    while len(ancestors) > 0:
 
188
        new_ancestors = []
 
189
        for ancestor in ancestors:
 
190
            if not only_present:
 
191
                yield ancestor, distance
 
192
            try:
 
193
                revision = revision_source.get_revision(ancestor)
 
194
            except bzrlib.errors.NoSuchRevision, e:
 
195
                if e.revision == revision_id:
 
196
                    raise 
 
197
                else:
 
198
                    continue
 
199
            if only_present:
 
200
                yield ancestor, distance
 
201
            new_ancestors.extend([p.revision_id for p in revision.parents])
 
202
        ancestors = new_ancestors
 
203
        distance += 1
 
204
 
 
205
 
 
206
def find_present_ancestors(revision_id, revision_source):
 
207
    """Return the ancestors of a revision present in a branch.
 
208
 
 
209
    It's possible that a branch won't have the complete ancestry of
 
210
    one of its revisions.  
 
211
 
 
212
    """
 
213
    found_ancestors = {}
 
214
    anc_iter = enumerate(iter_ancestors(revision_id, revision_source,
 
215
                         only_present=True))
 
216
    for anc_order, (anc_id, anc_distance) in anc_iter:
 
217
        if not found_ancestors.has_key(anc_id):
 
218
            found_ancestors[anc_id] = (anc_order, anc_distance)
 
219
    return found_ancestors
 
220
    
 
221
 
 
222
def __get_closest(intersection):
 
223
    intersection.sort()
 
224
    matches = [] 
 
225
    for entry in intersection:
 
226
        if entry[0] == intersection[0][0]:
 
227
            matches.append(entry[2])
 
228
    return matches
 
229
 
 
230
 
 
231
def old_common_ancestor(revision_a, revision_b, revision_source):
 
232
    """Find the ancestor common to both revisions that is closest to both.
 
233
    """
 
234
    from bzrlib.trace import mutter
 
235
    a_ancestors = find_present_ancestors(revision_a, revision_source)
 
236
    b_ancestors = find_present_ancestors(revision_b, revision_source)
 
237
    a_intersection = []
 
238
    b_intersection = []
 
239
    # a_order is used as a tie-breaker when two equally-good bases are found
 
240
    for revision, (a_order, a_distance) in a_ancestors.iteritems():
 
241
        if b_ancestors.has_key(revision):
 
242
            a_intersection.append((a_distance, a_order, revision))
 
243
            b_intersection.append((b_ancestors[revision][1], a_order, revision))
 
244
    mutter("a intersection: %r" % a_intersection)
 
245
    mutter("b intersection: %r" % b_intersection)
 
246
 
 
247
    a_closest = __get_closest(a_intersection)
 
248
    if len(a_closest) == 0:
 
249
        return None
 
250
    b_closest = __get_closest(b_intersection)
 
251
    assert len(b_closest) != 0
 
252
    mutter ("a_closest %r" % a_closest)
 
253
    mutter ("b_closest %r" % b_closest)
 
254
    if a_closest[0] in b_closest:
 
255
        return a_closest[0]
 
256
    elif b_closest[0] in a_closest:
 
257
        return b_closest[0]
 
258
    else:
 
259
        raise bzrlib.errors.AmbiguousBase((a_closest[0], b_closest[0]))
 
260
    return a_closest[0]
 
261
 
 
262
def revision_graph(revision, revision_source):
 
263
    """Produce a graph of the ancestry of the specified revision.
 
264
    Return root, ancestors map, descendants map
 
265
 
 
266
    TODO: Produce graphs with the NULL revision as root, so that we can find
 
267
    a common even when trees are not branches don't represent a single line
 
268
    of descent.
 
269
    """
 
270
    ancestors = {}
 
271
    descendants = {}
 
272
    lines = [revision]
 
273
    root = None
 
274
    descendants[revision] = {}
 
275
    while len(lines) > 0:
 
276
        new_lines = set()
 
277
        for line in lines:
 
278
            try:
 
279
                rev = revision_source.get_revision(line)
 
280
                parents = [p.revision_id for p in rev.parents]
 
281
                if len(parents) == 0:
 
282
                    root = line
 
283
            except bzrlib.errors.NoSuchRevision:
 
284
                if line == revision:
 
285
                    raise
 
286
                parents = None
 
287
            if parents is not None:
 
288
                for parent in parents:
 
289
                    if parent not in ancestors:
 
290
                        new_lines.add(parent)
 
291
                    if parent not in descendants:
 
292
                        descendants[parent] = {}
 
293
                    descendants[parent][line] = 1
 
294
            if parents is not None:
 
295
                ancestors[line] = set(parents)
 
296
        lines = new_lines
 
297
    assert root not in descendants[root]
 
298
    assert root not in ancestors[root]
 
299
    return root, ancestors, descendants
 
300
 
 
301
def combined_graph(revision_a, revision_b, revision_source):
 
302
    """Produce a combined ancestry graph.
 
303
    Return graph root, ancestors map, descendants map, set of common nodes"""
 
304
    root, ancestors, descendants = revision_graph(revision_a, revision_source)
 
305
    root_b, ancestors_b, descendants_b = revision_graph(revision_b, 
 
306
                                                        revision_source)
 
307
    assert root == root_b
 
308
    common = set()
 
309
    for node, node_anc in ancestors_b.iteritems():
 
310
        if node in ancestors:
 
311
            common.add(node)
 
312
        else:
 
313
            ancestors[node] = set()
 
314
        ancestors[node].update(node_anc)
 
315
    for node, node_dec in descendants_b.iteritems():
 
316
        if node not in descendants:
 
317
            descendants[node] = set()
 
318
        descendants[node].update(node_dec)
 
319
    return root, ancestors, descendants, common
 
320
 
 
321
def common_ancestor(revision_a, revision_b, revision_source):
 
322
    root, ancestors, descendants, common = \
 
323
        combined_graph(revision_a, revision_b, revision_source)
 
324
    nodes = farthest_nodes(descendants, ancestors, root)
 
325
    for node in nodes:
 
326
        if node in common:
 
327
            return node
 
328
 
 
329
class MultipleRevisionSources(object):
 
330
    """Proxy that looks in multiple branches for revisions."""
 
331
    def __init__(self, *args):
 
332
        object.__init__(self)
 
333
        assert len(args) != 0
 
334
        self._revision_sources = args
 
335
 
 
336
    def get_revision(self, revision_id):
 
337
        for source in self._revision_sources:
 
338
            try:
 
339
                return source.get_revision(revision_id)
 
340
            except bzrlib.errors.NoSuchRevision, e:
 
341
                pass
 
342
        raise e
 
343
 
 
344
def get_intervening_revisions(ancestor_id, rev_id, rev_source, 
 
345
                              revision_history=None):
 
346
    """Find the longest line of descent from maybe_ancestor to revision.
 
347
    Revision history is followed where possible.
 
348
 
 
349
    If ancestor_id == rev_id, list will be empty.
 
350
    Otherwise, rev_id will be the last entry.  ancestor_id will never appear.
 
351
    If ancestor_id is not an ancestor, NotAncestor will be thrown
 
352
    """
 
353
    root, ancestors, descendants = revision_graph(rev_id, rev_source)
 
354
    if len(descendants) == 0:
 
355
        raise NoSuchRevision(rev_source, rev_id)
 
356
    if ancestor_id not in descendants:
 
357
        rev_source.get_revision(ancestor_id)
 
358
        raise bzrlib.errors.NotAncestor(rev_id, ancestor_id)
 
359
    root_descendants = all_descendants(descendants, ancestor_id)
 
360
    root_descendants.add(ancestor_id)
 
361
    if rev_id not in root_descendants:
 
362
        raise bzrlib.errors.NotAncestor(rev_id, ancestor_id)
 
363
    distances = node_distances(descendants, ancestors, ancestor_id,
 
364
                               root_descendants=root_descendants)
 
365
 
 
366
    def best_ancestor(rev_id):
 
367
        best = None
 
368
        for anc_id in ancestors[rev_id]:
 
369
            try:
 
370
                distance = distances[anc_id]
 
371
            except KeyError:
 
372
                continue
 
373
            if revision_history is not None and anc_id in revision_history:
 
374
                return anc_id
 
375
            elif best is None or distance > best[1]:
 
376
                best = (anc_id, distance)
 
377
        return best[0]
 
378
 
 
379
    next = rev_id
 
380
    path = []
 
381
    while next != ancestor_id:
 
382
        path.append(next)
 
383
        next = best_ancestor(next)
 
384
    path.reverse()
 
385
    return path