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

  • Committer: Andrew Bennetts
  • Date: 2008-10-30 01:05:59 UTC
  • mto: This revision was merged to the branch mainline in revision 3813.
  • Revision ID: andrew.bennetts@canonical.com-20081030010559-tumoefnsmhg4snxo
Add contrib/bzr_ssh_path_limiter.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2008 Canonical Ltd
2
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
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
 
 
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
import errno
 
19
from itertools import chain
 
20
import os
17
21
import warnings
18
22
 
19
 
from bzrlib.lazy_import import lazy_import
20
 
lazy_import(globals(), """
21
23
from bzrlib import (
22
 
    branch as _mod_branch,
23
 
    conflicts as _mod_conflicts,
24
24
    debug,
25
25
    errors,
26
26
    graph as _mod_graph,
27
 
    merge3,
28
27
    osutils,
29
28
    patiencediff,
 
29
    registry,
30
30
    revision as _mod_revision,
31
 
    textfile,
32
 
    trace,
33
 
    transform,
34
31
    tree as _mod_tree,
35
32
    tsort,
36
 
    ui,
37
 
    versionedfile,
38
 
    )
39
 
from bzrlib.cleanup import OperationWithCleanups
40
 
""")
41
 
from bzrlib import (
42
 
    decorators,
43
 
    hooks,
44
 
    )
45
 
from bzrlib.symbol_versioning import (
46
 
    deprecated_in,
47
 
    deprecated_method,
48
 
    )
 
33
    )
 
34
from bzrlib.branch import Branch
 
35
from bzrlib.conflicts import ConflictList, Conflict
 
36
from bzrlib.errors import (BzrCommandError,
 
37
                           BzrError,
 
38
                           NoCommonAncestor,
 
39
                           NoCommits,
 
40
                           NoSuchRevision,
 
41
                           NoSuchFile,
 
42
                           NotBranchError,
 
43
                           NotVersionedError,
 
44
                           UnrelatedBranches,
 
45
                           UnsupportedOperation,
 
46
                           WorkingTreeNotRevision,
 
47
                           BinaryFile,
 
48
                           )
 
49
from bzrlib.graph import Graph
 
50
from bzrlib.merge3 import Merge3
 
51
from bzrlib.osutils import rename, pathjoin
 
52
from progress import DummyProgress, ProgressPhase
 
53
from bzrlib.revision import (NULL_REVISION, ensure_null)
 
54
from bzrlib.textfile import check_text_lines
 
55
from bzrlib.trace import mutter, warning, note, is_quiet
 
56
from bzrlib.transform import (TransformPreview, TreeTransform,
 
57
                              resolve_conflicts, cook_conflicts,
 
58
                              conflict_pass, FinalPaths, create_from_tree,
 
59
                              unique_add, ROOT_PARENT)
 
60
from bzrlib.versionedfile import PlanWeaveMerge
 
61
from bzrlib import ui
 
62
 
49
63
# TODO: Report back as changes are merged in
50
64
 
51
65
 
52
66
def transform_tree(from_tree, to_tree, interesting_ids=None):
53
 
    from_tree.lock_tree_write()
54
 
    operation = OperationWithCleanups(merge_inner)
55
 
    operation.add_cleanup(from_tree.unlock)
56
 
    operation.run_simple(from_tree.branch, to_tree, from_tree,
57
 
        ignore_zero=True, interesting_ids=interesting_ids, this_tree=from_tree)
58
 
 
59
 
 
60
 
class MergeHooks(hooks.Hooks):
61
 
 
62
 
    def __init__(self):
63
 
        hooks.Hooks.__init__(self)
64
 
        self.create_hook(hooks.HookPoint('merge_file_content',
65
 
            "Called with a bzrlib.merge.Merger object to create a per file "
66
 
            "merge object when starting a merge. "
67
 
            "Should return either None or a subclass of "
68
 
            "``bzrlib.merge.AbstractPerFileMerger``. "
69
 
            "Such objects will then be called per file "
70
 
            "that needs to be merged (including when one "
71
 
            "side has deleted the file and the other has changed it). "
72
 
            "See the AbstractPerFileMerger API docs for details on how it is "
73
 
            "used by merge.",
74
 
            (2, 1), None))
75
 
 
76
 
 
77
 
class AbstractPerFileMerger(object):
78
 
    """PerFileMerger objects are used by plugins extending merge for bzrlib.
79
 
 
80
 
    See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
81
 
    
82
 
    :ivar merger: The Merge3Merger performing the merge.
83
 
    """
84
 
 
85
 
    def __init__(self, merger):
86
 
        """Create a PerFileMerger for use with merger."""
87
 
        self.merger = merger
88
 
 
89
 
    def merge_contents(self, merge_params):
90
 
        """Attempt to merge the contents of a single file.
91
 
        
92
 
        :param merge_params: A bzrlib.merge.MergeHookParams
93
 
        :return : A tuple of (status, chunks), where status is one of
94
 
            'not_applicable', 'success', 'conflicted', or 'delete'.  If status
95
 
            is 'success' or 'conflicted', then chunks should be an iterable of
96
 
            strings for the new file contents.
97
 
        """
98
 
        return ('not applicable', None)
99
 
 
100
 
 
101
 
class PerFileMerger(AbstractPerFileMerger):
102
 
    """Merge individual files when self.file_matches returns True.
103
 
 
104
 
    This class is intended to be subclassed.  The file_matches and
105
 
    merge_matching methods should be overridden with concrete implementations.
106
 
    """
107
 
 
108
 
    def file_matches(self, params):
109
 
        """Return True if merge_matching should be called on this file.
110
 
 
111
 
        Only called with merges of plain files with no clear winner.
112
 
 
113
 
        Subclasses must override this.
114
 
        """
115
 
        raise NotImplementedError(self.file_matches)
116
 
 
117
 
    def get_filename(self, params, tree):
118
 
        """Lookup the filename (i.e. basename, not path), given a Tree (e.g.
119
 
        self.merger.this_tree) and a MergeHookParams.
120
 
        """
121
 
        return osutils.basename(tree.id2path(params.file_id))
122
 
 
123
 
    def get_filepath(self, params, tree):
124
 
        """Calculate the path to the file in a tree.
125
 
 
126
 
        :param params: A MergeHookParams describing the file to merge
127
 
        :param tree: a Tree, e.g. self.merger.this_tree.
128
 
        """
129
 
        return tree.id2path(params.file_id)
130
 
 
131
 
    def merge_contents(self, params):
132
 
        """Merge the contents of a single file."""
133
 
        # Check whether this custom merge logic should be used.
134
 
        if (
135
 
            # OTHER is a straight winner, rely on default merge.
136
 
            params.winner == 'other' or
137
 
            # THIS and OTHER aren't both files.
138
 
            not params.is_file_merge() or
139
 
            # The filename doesn't match *.xml
140
 
            not self.file_matches(params)):
141
 
            return 'not_applicable', None
142
 
        return self.merge_matching(params)
143
 
 
144
 
    def merge_matching(self, params):
145
 
        """Merge the contents of a single file that has matched the criteria
146
 
        in PerFileMerger.merge_contents (is a conflict, is a file,
147
 
        self.file_matches is True).
148
 
 
149
 
        Subclasses must override this.
150
 
        """
151
 
        raise NotImplementedError(self.merge_matching)
152
 
 
153
 
 
154
 
class ConfigurableFileMerger(PerFileMerger):
155
 
    """Merge individual files when configured via a .conf file.
156
 
 
157
 
    This is a base class for concrete custom file merging logic. Concrete
158
 
    classes should implement ``merge_text``.
159
 
 
160
 
    See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
161
 
    
162
 
    :ivar affected_files: The configured file paths to merge.
163
 
 
164
 
    :cvar name_prefix: The prefix to use when looking up configuration
165
 
        details. <name_prefix>_merge_files describes the files targeted by the
166
 
        hook for example.
167
 
        
168
 
    :cvar default_files: The default file paths to merge when no configuration
169
 
        is present.
170
 
    """
171
 
 
172
 
    name_prefix = None
173
 
    default_files = None
174
 
 
175
 
    def __init__(self, merger):
176
 
        super(ConfigurableFileMerger, self).__init__(merger)
177
 
        self.affected_files = None
178
 
        self.default_files = self.__class__.default_files or []
179
 
        self.name_prefix = self.__class__.name_prefix
180
 
        if self.name_prefix is None:
181
 
            raise ValueError("name_prefix must be set.")
182
 
 
183
 
    def file_matches(self, params):
184
 
        """Check whether the file should call the merge hook.
185
 
 
186
 
        <name_prefix>_merge_files configuration variable is a list of files
187
 
        that should use the hook.
188
 
        """
189
 
        affected_files = self.affected_files
190
 
        if affected_files is None:
191
 
            config = self.merger.this_branch.get_config()
192
 
            # Until bzr provides a better policy for caching the config, we
193
 
            # just add the part we're interested in to the params to avoid
194
 
            # reading the config files repeatedly (bazaar.conf, location.conf,
195
 
            # branch.conf).
196
 
            config_key = self.name_prefix + '_merge_files'
197
 
            affected_files = config.get_user_option_as_list(config_key)
198
 
            if affected_files is None:
199
 
                # If nothing was specified in the config, use the default.
200
 
                affected_files = self.default_files
201
 
            self.affected_files = affected_files
202
 
        if affected_files:
203
 
            filepath = self.get_filepath(params, self.merger.this_tree)
204
 
            if filepath in affected_files:
205
 
                return True
206
 
        return False
207
 
 
208
 
    def merge_matching(self, params):
209
 
        return self.merge_text(params)
210
 
 
211
 
    def merge_text(self, params):
212
 
        """Merge the byte contents of a single file.
213
 
 
214
 
        This is called after checking that the merge should be performed in
215
 
        merge_contents, and it should behave as per
216
 
        ``bzrlib.merge.AbstractPerFileMerger.merge_contents``.
217
 
        """
218
 
        raise NotImplementedError(self.merge_text)
219
 
 
220
 
 
221
 
class MergeHookParams(object):
222
 
    """Object holding parameters passed to merge_file_content hooks.
223
 
 
224
 
    There are some fields hooks can access:
225
 
 
226
 
    :ivar file_id: the file ID of the file being merged
227
 
    :ivar trans_id: the transform ID for the merge of this file
228
 
    :ivar this_kind: kind of file_id in 'this' tree
229
 
    :ivar other_kind: kind of file_id in 'other' tree
230
 
    :ivar winner: one of 'this', 'other', 'conflict'
231
 
    """
232
 
 
233
 
    def __init__(self, merger, file_id, trans_id, this_kind, other_kind,
234
 
            winner):
235
 
        self._merger = merger
236
 
        self.file_id = file_id
237
 
        self.trans_id = trans_id
238
 
        self.this_kind = this_kind
239
 
        self.other_kind = other_kind
240
 
        self.winner = winner
241
 
 
242
 
    def is_file_merge(self):
243
 
        """True if this_kind and other_kind are both 'file'."""
244
 
        return self.this_kind == 'file' and self.other_kind == 'file'
245
 
 
246
 
    @decorators.cachedproperty
247
 
    def base_lines(self):
248
 
        """The lines of the 'base' version of the file."""
249
 
        return self._merger.get_lines(self._merger.base_tree, self.file_id)
250
 
 
251
 
    @decorators.cachedproperty
252
 
    def this_lines(self):
253
 
        """The lines of the 'this' version of the file."""
254
 
        return self._merger.get_lines(self._merger.this_tree, self.file_id)
255
 
 
256
 
    @decorators.cachedproperty
257
 
    def other_lines(self):
258
 
        """The lines of the 'other' version of the file."""
259
 
        return self._merger.get_lines(self._merger.other_tree, self.file_id)
 
67
    merge_inner(from_tree.branch, to_tree, from_tree, ignore_zero=True,
 
68
                interesting_ids=interesting_ids, this_tree=from_tree)
260
69
 
261
70
 
262
71
class Merger(object):
263
 
 
264
 
    hooks = MergeHooks()
265
 
 
266
72
    def __init__(self, this_branch, other_tree=None, base_tree=None,
267
 
                 this_tree=None, pb=None, change_reporter=None,
 
73
                 this_tree=None, pb=DummyProgress(), change_reporter=None,
268
74
                 recurse='down', revision_graph=None):
269
75
        object.__init__(self)
270
76
        self.this_branch = this_branch
283
89
        self.interesting_files = None
284
90
        self.show_base = False
285
91
        self.reprocess = False
286
 
        if pb is not None:
287
 
            warnings.warn("pb parameter to Merger() is deprecated and ignored")
 
92
        self._pb = pb
288
93
        self.pp = None
289
94
        self.recurse = recurse
290
95
        self.change_reporter = change_reporter
295
100
        self._is_criss_cross = None
296
101
        self._lca_trees = None
297
102
 
298
 
    def cache_trees_with_revision_ids(self, trees):
299
 
        """Cache any tree in trees if it has a revision_id."""
300
 
        for maybe_tree in trees:
301
 
            if maybe_tree is None:
302
 
                continue
303
 
            try:
304
 
                rev_id = maybe_tree.get_revision_id()
305
 
            except AttributeError:
306
 
                continue
307
 
            self._cached_trees[rev_id] = maybe_tree
308
 
 
309
103
    @property
310
104
    def revision_graph(self):
311
105
        if self._revision_graph is None:
338
132
                                      _set_base_is_other_ancestor)
339
133
 
340
134
    @staticmethod
341
 
    def from_uncommitted(tree, other_tree, pb=None, base_tree=None):
 
135
    def from_uncommitted(tree, other_tree, pb, base_tree=None):
342
136
        """Return a Merger for uncommitted changes in other_tree.
343
137
 
344
138
        :param tree: The tree to merge into
373
167
                base_revision_id, tree.branch.last_revision())):
374
168
                base_revision_id = None
375
169
            else:
376
 
                trace.warning('Performing cherrypick')
 
170
                warning('Performing cherrypick')
377
171
        merger = klass.from_revision_ids(pb, tree, other_revision_id,
378
172
                                         base_revision_id, revision_graph=
379
173
                                         revision_graph)
431
225
        if revno is None:
432
226
            tree = workingtree.WorkingTree.open_containing(location)[0]
433
227
            return tree.branch, tree
434
 
        branch = _mod_branch.Branch.open_containing(
435
 
            location, possible_transports)[0]
 
228
        branch = Branch.open_containing(location, possible_transports)[0]
436
229
        if revno == -1:
437
230
            revision_id = branch.last_revision()
438
231
        else:
439
232
            revision_id = branch.get_rev_id(revno)
440
 
        revision_id = _mod_revision.ensure_null(revision_id)
 
233
        revision_id = ensure_null(revision_id)
441
234
        return branch, self.revision_tree(revision_id, branch)
442
235
 
443
 
    @deprecated_method(deprecated_in((2, 1, 0)))
444
236
    def ensure_revision_trees(self):
445
237
        if self.this_revision_tree is None:
446
238
            self.this_basis_tree = self.revision_tree(self.this_basis)
449
241
 
450
242
        if self.other_rev_id is None:
451
243
            other_basis_tree = self.revision_tree(self.other_basis)
452
 
            if other_basis_tree.has_changes(self.other_tree):
453
 
                raise errors.WorkingTreeNotRevision(self.this_tree)
 
244
            changes = other_basis_tree.changes_from(self.other_tree)
 
245
            if changes.has_changed():
 
246
                raise WorkingTreeNotRevision(self.this_tree)
454
247
            other_rev_id = self.other_basis
455
248
            self.other_tree = other_basis_tree
456
249
 
457
 
    @deprecated_method(deprecated_in((2, 1, 0)))
458
250
    def file_revisions(self, file_id):
459
251
        self.ensure_revision_trees()
460
252
        def get_id(tree, file_id):
463
255
        if self.this_rev_id is None:
464
256
            if self.this_basis_tree.get_file_sha1(file_id) != \
465
257
                self.this_tree.get_file_sha1(file_id):
466
 
                raise errors.WorkingTreeNotRevision(self.this_tree)
 
258
                raise WorkingTreeNotRevision(self.this_tree)
467
259
 
468
260
        trees = (self.this_basis_tree, self.other_tree)
469
261
        return [get_id(tree, file_id) for tree in trees]
470
262
 
471
 
    @deprecated_method(deprecated_in((2, 1, 0)))
472
263
    def check_basis(self, check_clean, require_commits=True):
473
264
        if self.this_basis is None and require_commits is True:
474
 
            raise errors.BzrCommandError(
475
 
                "This branch has no commits."
476
 
                " (perhaps you would prefer 'bzr pull')")
 
265
            raise BzrCommandError("This branch has no commits."
 
266
                                  " (perhaps you would prefer 'bzr pull')")
477
267
        if check_clean:
478
268
            self.compare_basis()
479
269
            if self.this_basis != self.this_rev_id:
480
270
                raise errors.UncommittedChanges(self.this_tree)
481
271
 
482
 
    @deprecated_method(deprecated_in((2, 1, 0)))
483
272
    def compare_basis(self):
484
273
        try:
485
274
            basis_tree = self.revision_tree(self.this_tree.last_revision())
486
275
        except errors.NoSuchRevision:
487
276
            basis_tree = self.this_tree.basis_tree()
488
 
        if not self.this_tree.has_changes(basis_tree):
 
277
        changes = self.this_tree.changes_from(basis_tree)
 
278
        if not changes.has_changed():
489
279
            self.this_rev_id = self.this_basis
490
280
 
491
281
    def set_interesting_files(self, file_list):
492
282
        self.interesting_files = file_list
493
283
 
494
284
    def set_pending(self):
495
 
        if (not self.base_is_ancestor or not self.base_is_other_ancestor
496
 
            or self.other_rev_id is None):
 
285
        if not self.base_is_ancestor or not self.base_is_other_ancestor or self.other_rev_id is None:
497
286
            return
498
287
        self._add_parent()
499
288
 
500
289
    def _add_parent(self):
501
290
        new_parents = self.this_tree.get_parent_ids() + [self.other_rev_id]
502
291
        new_parent_trees = []
503
 
        operation = OperationWithCleanups(self.this_tree.set_parent_trees)
504
292
        for revision_id in new_parents:
505
293
            try:
506
294
                tree = self.revision_tree(revision_id)
508
296
                tree = None
509
297
            else:
510
298
                tree.lock_read()
511
 
                operation.add_cleanup(tree.unlock)
512
299
            new_parent_trees.append((revision_id, tree))
513
 
        operation.run_simple(new_parent_trees, allow_leftmost_as_ghost=True)
 
300
        try:
 
301
            self.this_tree.set_parent_trees(new_parent_trees,
 
302
                                            allow_leftmost_as_ghost=True)
 
303
        finally:
 
304
            for _revision_id, tree in new_parent_trees:
 
305
                if tree is not None:
 
306
                    tree.unlock()
514
307
 
515
308
    def set_other(self, other_revision, possible_transports=None):
516
309
        """Set the revision and tree to merge from.
525
318
            self.other_rev_id = _mod_revision.ensure_null(
526
319
                self.other_branch.last_revision())
527
320
            if _mod_revision.is_null(self.other_rev_id):
528
 
                raise errors.NoCommits(self.other_branch)
 
321
                raise NoCommits(self.other_branch)
529
322
            self.other_basis = self.other_rev_id
530
323
        elif other_revision[1] is not None:
531
324
            self.other_rev_id = self.other_branch.get_rev_id(other_revision[1])
534
327
            self.other_rev_id = None
535
328
            self.other_basis = self.other_branch.last_revision()
536
329
            if self.other_basis is None:
537
 
                raise errors.NoCommits(self.other_branch)
 
330
                raise NoCommits(self.other_branch)
538
331
        if self.other_rev_id is not None:
539
332
            self._cached_trees[self.other_rev_id] = self.other_tree
540
333
        self._maybe_fetch(self.other_branch,self.this_branch, self.other_basis)
567
360
            target.fetch(source, revision_id)
568
361
 
569
362
    def find_base(self):
570
 
        revisions = [_mod_revision.ensure_null(self.this_basis),
571
 
                     _mod_revision.ensure_null(self.other_basis)]
572
 
        if _mod_revision.NULL_REVISION in revisions:
573
 
            self.base_rev_id = _mod_revision.NULL_REVISION
 
363
        revisions = [ensure_null(self.this_basis),
 
364
                     ensure_null(self.other_basis)]
 
365
        if NULL_REVISION in revisions:
 
366
            self.base_rev_id = NULL_REVISION
574
367
            self.base_tree = self.revision_tree(self.base_rev_id)
575
368
            self._is_criss_cross = False
576
369
        else:
577
370
            lcas = self.revision_graph.find_lca(revisions[0], revisions[1])
578
371
            self._is_criss_cross = False
579
372
            if len(lcas) == 0:
580
 
                self.base_rev_id = _mod_revision.NULL_REVISION
 
373
                self.base_rev_id = NULL_REVISION
581
374
            elif len(lcas) == 1:
582
375
                self.base_rev_id = list(lcas)[0]
583
376
            else: # len(lcas) > 1
592
385
                    self.base_rev_id = self.revision_graph.find_unique_lca(
593
386
                                            *lcas)
594
387
                self._is_criss_cross = True
595
 
            if self.base_rev_id == _mod_revision.NULL_REVISION:
596
 
                raise errors.UnrelatedBranches()
 
388
            if self.base_rev_id == NULL_REVISION:
 
389
                raise UnrelatedBranches()
597
390
            if self._is_criss_cross:
598
 
                trace.warning('Warning: criss-cross merge encountered.  See bzr'
599
 
                              ' help criss-cross.')
600
 
                trace.mutter('Criss-cross lcas: %r' % lcas)
 
391
                warning('Warning: criss-cross merge encountered.  See bzr'
 
392
                        ' help criss-cross.')
601
393
                interesting_revision_ids = [self.base_rev_id]
602
394
                interesting_revision_ids.extend(lcas)
603
395
                interesting_trees = dict((t.get_revision_id(), t)
613
405
                self.base_tree = self.revision_tree(self.base_rev_id)
614
406
        self.base_is_ancestor = True
615
407
        self.base_is_other_ancestor = True
616
 
        trace.mutter('Base revid: %r' % self.base_rev_id)
617
408
 
618
409
    def set_base(self, base_revision):
619
410
        """Set the base revision to use for the merge.
620
411
 
621
412
        :param base_revision: A 2-list containing a path and revision number.
622
413
        """
623
 
        trace.mutter("doing merge() with no base_revision specified")
 
414
        mutter("doing merge() with no base_revision specified")
624
415
        if base_revision == [None, None]:
625
416
            self.find_base()
626
417
        else:
639
430
                  'other_tree': self.other_tree,
640
431
                  'interesting_ids': self.interesting_ids,
641
432
                  'interesting_files': self.interesting_files,
642
 
                  'this_branch': self.this_branch,
 
433
                  'pp': self.pp,
643
434
                  'do_merge': False}
644
435
        if self.merge_type.requires_base:
645
436
            kwargs['base_tree'] = self.base_tree
646
437
        if self.merge_type.supports_reprocess:
647
438
            kwargs['reprocess'] = self.reprocess
648
439
        elif self.reprocess:
649
 
            raise errors.BzrError(
650
 
                "Conflict reduction is not supported for merge"
651
 
                " type %s." % self.merge_type)
 
440
            raise BzrError("Conflict reduction is not supported for merge"
 
441
                                  " type %s." % self.merge_type)
652
442
        if self.merge_type.supports_show_base:
653
443
            kwargs['show_base'] = self.show_base
654
444
        elif self.show_base:
655
 
            raise errors.BzrError("Showing base is not supported for this"
656
 
                                  " merge type. %s" % self.merge_type)
 
445
            raise BzrError("Showing base is not supported for this"
 
446
                           " merge type. %s" % self.merge_type)
657
447
        if (not getattr(self.merge_type, 'supports_reverse_cherrypick', True)
658
448
            and not self.base_is_other_ancestor):
659
449
            raise errors.CannotReverseCherrypick()
663
453
        if self._is_criss_cross and getattr(self.merge_type,
664
454
                                            'supports_lca_trees', False):
665
455
            kwargs['lca_trees'] = self._lca_trees
666
 
        return self.merge_type(pb=None,
 
456
        return self.merge_type(pb=self._pb,
667
457
                               change_reporter=self.change_reporter,
668
458
                               **kwargs)
669
459
 
670
 
    def _do_merge_to(self):
671
 
        merge = self.make_merger()
672
 
        if self.other_branch is not None:
673
 
            self.other_branch.update_references(self.this_branch)
 
460
    def _do_merge_to(self, merge):
674
461
        merge.do_merge()
675
462
        if self.recurse == 'down':
676
463
            for relpath, file_id in self.this_tree.iter_references():
688
475
                    sub_tree.branch.repository.revision_tree(base_revision)
689
476
                sub_merge.base_rev_id = base_revision
690
477
                sub_merge.do_merge()
691
 
        return merge
692
 
 
 
478
        
693
479
    def do_merge(self):
694
 
        operation = OperationWithCleanups(self._do_merge_to)
695
480
        self.this_tree.lock_tree_write()
696
 
        operation.add_cleanup(self.this_tree.unlock)
697
 
        if self.base_tree is not None:
698
 
            self.base_tree.lock_read()
699
 
            operation.add_cleanup(self.base_tree.unlock)
700
 
        if self.other_tree is not None:
701
 
            self.other_tree.lock_read()
702
 
            operation.add_cleanup(self.other_tree.unlock)
703
 
        merge = operation.run_simple()
 
481
        try:
 
482
            if self.base_tree is not None:
 
483
                self.base_tree.lock_read()
 
484
            try:
 
485
                if self.other_tree is not None:
 
486
                    self.other_tree.lock_read()
 
487
                try:
 
488
                    merge = self.make_merger()
 
489
                    self._do_merge_to(merge)
 
490
                finally:
 
491
                    if self.other_tree is not None:
 
492
                        self.other_tree.unlock()
 
493
            finally:
 
494
                if self.base_tree is not None:
 
495
                    self.base_tree.unlock()
 
496
        finally:
 
497
            self.this_tree.unlock()
704
498
        if len(merge.cooked_conflicts) == 0:
705
 
            if not self.ignore_zero and not trace.is_quiet():
706
 
                trace.note("All changes applied successfully.")
 
499
            if not self.ignore_zero and not is_quiet():
 
500
                note("All changes applied successfully.")
707
501
        else:
708
 
            trace.note("%d conflicts encountered."
709
 
                       % len(merge.cooked_conflicts))
 
502
            note("%d conflicts encountered." % len(merge.cooked_conflicts))
710
503
 
711
504
        return len(merge.cooked_conflicts)
712
505
 
739
532
    winner_idx = {"this": 2, "other": 1, "conflict": 1}
740
533
    supports_lca_trees = True
741
534
 
742
 
    def __init__(self, working_tree, this_tree, base_tree, other_tree,
 
535
    def __init__(self, working_tree, this_tree, base_tree, other_tree, 
743
536
                 interesting_ids=None, reprocess=False, show_base=False,
744
 
                 pb=None, pp=None, change_reporter=None,
 
537
                 pb=DummyProgress(), pp=None, change_reporter=None,
745
538
                 interesting_files=None, do_merge=True,
746
 
                 cherrypick=False, lca_trees=None, this_branch=None):
 
539
                 cherrypick=False, lca_trees=None):
747
540
        """Initialize the merger object and perform the merge.
748
541
 
749
542
        :param working_tree: The working tree to apply the merge to
750
543
        :param this_tree: The local tree in the merge operation
751
544
        :param base_tree: The common tree in the merge operation
752
 
        :param other_tree: The other tree to merge changes from
753
 
        :param this_branch: The branch associated with this_tree.  Defaults to
754
 
            this_tree.branch if not supplied.
 
545
        :param other_tree: The other other tree to merge changes from
755
546
        :param interesting_ids: The file_ids of files that should be
756
547
            participate in the merge.  May not be combined with
757
548
            interesting_files.
758
549
        :param: reprocess If True, perform conflict-reduction processing.
759
550
        :param show_base: If True, show the base revision in text conflicts.
760
551
            (incompatible with reprocess)
761
 
        :param pb: ignored
 
552
        :param pb: A Progress bar
762
553
        :param pp: A ProgressPhase object
763
554
        :param change_reporter: An object that should report changes made
764
555
        :param interesting_files: The tree-relative paths of files that should
775
566
        if interesting_files is not None and interesting_ids is not None:
776
567
            raise ValueError(
777
568
                'specify either interesting_ids or interesting_files')
778
 
        if this_branch is None:
779
 
            this_branch = this_tree.branch
780
569
        self.interesting_ids = interesting_ids
781
570
        self.interesting_files = interesting_files
782
571
        self.this_tree = working_tree
783
572
        self.base_tree = base_tree
784
573
        self.other_tree = other_tree
785
 
        self.this_branch = this_branch
786
574
        self._raw_conflicts = []
787
575
        self.cooked_conflicts = []
788
576
        self.reprocess = reprocess
793
581
        # making sure we haven't missed any corner cases.
794
582
        # if lca_trees is None:
795
583
        #     self._lca_trees = [self.base_tree]
 
584
        self.pb = pb
 
585
        self.pp = pp
796
586
        self.change_reporter = change_reporter
797
587
        self.cherrypick = cherrypick
 
588
        if self.pp is None:
 
589
            self.pp = ProgressPhase("Merge phase", 3, self.pb)
798
590
        if do_merge:
799
591
            self.do_merge()
800
 
        if pp is not None:
801
 
            warnings.warn("pp argument to Merge3Merger is deprecated")
802
 
        if pb is not None:
803
 
            warnings.warn("pb argument to Merge3Merger is deprecated")
804
592
 
805
593
    def do_merge(self):
806
 
        operation = OperationWithCleanups(self._do_merge)
807
594
        self.this_tree.lock_tree_write()
808
 
        operation.add_cleanup(self.this_tree.unlock)
809
595
        self.base_tree.lock_read()
810
 
        operation.add_cleanup(self.base_tree.unlock)
811
596
        self.other_tree.lock_read()
812
 
        operation.add_cleanup(self.other_tree.unlock)
813
 
        operation.run()
814
 
 
815
 
    def _do_merge(self, operation):
816
 
        self.tt = transform.TreeTransform(self.this_tree, None)
817
 
        operation.add_cleanup(self.tt.finalize)
818
 
        self._compute_transform()
819
 
        results = self.tt.apply(no_conflicts=True)
820
 
        self.write_modified(results)
 
597
        self.tt = TreeTransform(self.this_tree, self.pb)
821
598
        try:
822
 
            self.this_tree.add_conflicts(self.cooked_conflicts)
823
 
        except errors.UnsupportedOperation:
824
 
            pass
 
599
            self.pp.next_phase()
 
600
            self._compute_transform()
 
601
            self.pp.next_phase()
 
602
            results = self.tt.apply(no_conflicts=True)
 
603
            self.write_modified(results)
 
604
            try:
 
605
                self.this_tree.add_conflicts(self.cooked_conflicts)
 
606
            except UnsupportedOperation:
 
607
                pass
 
608
        finally:
 
609
            self.tt.finalize()
 
610
            self.other_tree.unlock()
 
611
            self.base_tree.unlock()
 
612
            self.this_tree.unlock()
 
613
            self.pb.clear()
825
614
 
826
615
    def make_preview_transform(self):
827
 
        operation = OperationWithCleanups(self._make_preview_transform)
828
616
        self.base_tree.lock_read()
829
 
        operation.add_cleanup(self.base_tree.unlock)
830
617
        self.other_tree.lock_read()
831
 
        operation.add_cleanup(self.other_tree.unlock)
832
 
        return operation.run_simple()
833
 
 
834
 
    def _make_preview_transform(self):
835
 
        self.tt = transform.TransformPreview(self.this_tree)
836
 
        self._compute_transform()
 
618
        self.tt = TransformPreview(self.this_tree)
 
619
        try:
 
620
            self.pp.next_phase()
 
621
            self._compute_transform()
 
622
            self.pp.next_phase()
 
623
        finally:
 
624
            self.other_tree.unlock()
 
625
            self.base_tree.unlock()
 
626
            self.pb.clear()
837
627
        return self.tt
838
628
 
839
629
    def _compute_transform(self):
845
635
            resolver = self._lca_multi_way
846
636
        child_pb = ui.ui_factory.nested_progress_bar()
847
637
        try:
848
 
            factories = Merger.hooks['merge_file_content']
849
 
            hooks = [factory(self) for factory in factories] + [self]
850
 
            self.active_hooks = [hook for hook in hooks if hook is not None]
851
638
            for num, (file_id, changed, parents3, names3,
852
639
                      executable3) in enumerate(entries):
853
640
                child_pb.update('Preparing file merge', num, len(entries))
854
641
                self._merge_names(file_id, parents3, names3, resolver=resolver)
855
642
                if changed:
856
 
                    file_status = self._do_merge_contents(file_id)
 
643
                    file_status = self.merge_contents(file_id)
857
644
                else:
858
645
                    file_status = 'unmodified'
859
646
                self._merge_executable(file_id,
861
648
        finally:
862
649
            child_pb.finished()
863
650
        self.fix_root()
 
651
        self.pp.next_phase()
864
652
        child_pb = ui.ui_factory.nested_progress_bar()
865
653
        try:
866
 
            fs_conflicts = transform.resolve_conflicts(self.tt, child_pb,
867
 
                lambda t, c: transform.conflict_pass(t, c, self.other_tree))
 
654
            fs_conflicts = resolve_conflicts(self.tt, child_pb,
 
655
                lambda t, c: conflict_pass(t, c, self.other_tree))
868
656
        finally:
869
657
            child_pb.finished()
870
658
        if self.change_reporter is not None:
873
661
                self.tt.iter_changes(), self.change_reporter)
874
662
        self.cook_conflicts(fs_conflicts)
875
663
        for conflict in self.cooked_conflicts:
876
 
            trace.warning(conflict)
 
664
            warning(conflict)
877
665
 
878
666
    def _entries3(self):
879
667
        """Gather data about files modified between three trees.
1006
794
            content_changed = True
1007
795
            if kind_winner == 'this':
1008
796
                # No kind change in OTHER, see if there are *any* changes
1009
 
                if other_ie.kind == 'directory':
 
797
                if other_ie.kind == None:
 
798
                    # No content and 'this' wins the kind, so skip this?
 
799
                    # continue
 
800
                    pass
 
801
                elif other_ie.kind == 'directory':
1010
802
                    if parent_id_winner == 'this' and name_winner == 'this':
1011
803
                        # No change for this directory in OTHER, skip
1012
804
                        continue
1013
805
                    content_changed = False
1014
 
                elif other_ie.kind is None or other_ie.kind == 'file':
 
806
                elif other_ie.kind == 'file':
1015
807
                    def get_sha1(ie, tree):
1016
808
                        if ie.kind != 'file':
1017
809
                            return None
1064
856
                        continue
1065
857
                else:
1066
858
                    raise AssertionError('unhandled kind: %s' % other_ie.kind)
 
859
                # XXX: We need to handle kind == 'symlink'
1067
860
 
1068
861
            # If we have gotten this far, that means something has changed
1069
862
            result.append((file_id, content_changed,
1080
873
    def fix_root(self):
1081
874
        try:
1082
875
            self.tt.final_kind(self.tt.root)
1083
 
        except errors.NoSuchFile:
 
876
        except NoSuchFile:
1084
877
            self.tt.cancel_deletion(self.tt.root)
1085
878
        if self.tt.final_file_id(self.tt.root) is None:
1086
 
            self.tt.version_file(self.tt.tree_file_id(self.tt.root),
 
879
            self.tt.version_file(self.tt.tree_file_id(self.tt.root), 
1087
880
                                 self.tt.root)
1088
881
        other_root_file_id = self.other_tree.get_root_id()
1089
882
        if other_root_file_id is None:
1091
884
        other_root = self.tt.trans_id_file_id(other_root_file_id)
1092
885
        if other_root == self.tt.root:
1093
886
            return
 
887
        try:
 
888
            self.tt.final_kind(other_root)
 
889
        except NoSuchFile:
 
890
            return
1094
891
        if self.other_tree.inventory.root.file_id in self.this_tree.inventory:
1095
 
            # the other tree's root is a non-root in the current tree (as when
1096
 
            # a previously unrelated branch is merged into another)
 
892
            # the other tree's root is a non-root in the current tree
1097
893
            return
1098
 
        try:
1099
 
            self.tt.final_kind(other_root)
1100
 
            other_root_is_present = True
1101
 
        except errors.NoSuchFile:
1102
 
            # other_root doesn't have a physical representation. We still need
1103
 
            # to move any references to the actual root of the tree.
1104
 
            other_root_is_present = False
1105
 
        # 'other_tree.inventory.root' is not present in this tree. We are
1106
 
        # calling adjust_path for children which *want* to be present with a
1107
 
        # correct place to go.
1108
 
        for thing, child in self.other_tree.inventory.root.children.iteritems():
 
894
        self.reparent_children(self.other_tree.inventory.root, self.tt.root)
 
895
        self.tt.cancel_creation(other_root)
 
896
        self.tt.cancel_versioning(other_root)
 
897
 
 
898
    def reparent_children(self, ie, target):
 
899
        for thing, child in ie.children.iteritems():
1109
900
            trans_id = self.tt.trans_id_file_id(child.file_id)
1110
 
            if not other_root_is_present:
1111
 
                # FIXME: Make final_kind returns None instead of raising
1112
 
                # NoSuchFile to avoid the ugly construct below -- vila 20100402
1113
 
                try:
1114
 
                    self.tt.final_kind(trans_id)
1115
 
                    # The item exist in the final tree and has a defined place
1116
 
                    # to go already.
1117
 
                    continue
1118
 
                except errors.NoSuchFile, e:
1119
 
                    pass
1120
 
            # Move the item into the root
1121
 
            self.tt.adjust_path(self.tt.final_name(trans_id),
1122
 
                                self.tt.root, trans_id)
1123
 
        if other_root_is_present:
1124
 
            self.tt.cancel_creation(other_root)
1125
 
            self.tt.cancel_versioning(other_root)
 
901
            self.tt.adjust_path(self.tt.final_name(trans_id), target, trans_id)
1126
902
 
1127
903
    def write_modified(self, results):
1128
904
        modified_hashes = {}
1149
925
        if entry is None:
1150
926
            return None
1151
927
        return entry.name
1152
 
 
 
928
    
1153
929
    @staticmethod
1154
930
    def contents_sha1(tree, file_id):
1155
931
        """Determine the sha1 of the file contents (used as a key method)."""
1160
936
    @staticmethod
1161
937
    def executable(tree, file_id):
1162
938
        """Determine the executability of a file-id (used as a key method)."""
1163
 
        if not tree.has_id(file_id):
 
939
        if file_id not in tree:
1164
940
            return None
1165
941
        if tree.kind(file_id) != "file":
1166
942
            return False
1169
945
    @staticmethod
1170
946
    def kind(tree, file_id):
1171
947
        """Determine the kind of a file-id (used as a key method)."""
1172
 
        if not tree.has_id(file_id):
 
948
        if file_id not in tree:
1173
949
            return None
1174
950
        return tree.kind(file_id)
1175
951
 
1176
952
    @staticmethod
1177
953
    def _three_way(base, other, this):
 
954
        #if base == other, either they all agree, or only THIS has changed.
1178
955
        if base == other:
1179
 
            # if 'base == other', either they all agree, or only 'this' has
1180
 
            # changed.
1181
956
            return 'this'
1182
957
        elif this not in (base, other):
1183
 
            # 'this' is neither 'base' nor 'other', so both sides changed
1184
958
            return 'conflict'
 
959
        # "Ambiguous clean merge" -- both sides have made the same change.
1185
960
        elif this == other:
1186
 
            # "Ambiguous clean merge" -- both sides have made the same change.
1187
961
            return "this"
 
962
        # this == base: only other has changed.
1188
963
        else:
1189
 
            # this == base: only other has changed.
1190
964
            return "other"
1191
965
 
1192
966
    @staticmethod
1205
979
        :return: 'this', 'other', or 'conflict' depending on whether an entry
1206
980
            changed or not.
1207
981
        """
1208
 
        # See doc/developers/lca_tree_merging.txt for details about this
 
982
        # See doc/developers/lca_merge_resolution.txt for details about this
1209
983
        # algorithm.
1210
984
        if other == this:
1211
985
            # Either Ambiguously clean, or nothing was actually changed. We
1236
1010
                # only has an lca value
1237
1011
                return 'other'
1238
1012
 
1239
 
        # At this point, the lcas disagree, and the tip disagree
 
1013
        # At this point, the lcas disagree, and the tips disagree
1240
1014
        return 'conflict'
1241
1015
 
1242
1016
    @staticmethod
1243
 
    @deprecated_method(deprecated_in((2, 2, 0)))
1244
1017
    def scalar_three_way(this_tree, base_tree, other_tree, file_id, key):
1245
1018
        """Do a three-way test on a scalar.
1246
1019
        Return "this", "other" or "conflict", depending whether a value wins.
1261
1034
 
1262
1035
    def merge_names(self, file_id):
1263
1036
        def get_entry(tree):
1264
 
            if tree.has_id(file_id):
 
1037
            if file_id in tree.inventory:
1265
1038
                return tree.inventory[file_id]
1266
1039
            else:
1267
1040
                return None
1296
1069
                parent_id_winner = "other"
1297
1070
        if name_winner == "this" and parent_id_winner == "this":
1298
1071
            return
1299
 
        if name_winner == 'conflict' or parent_id_winner == 'conflict':
1300
 
            # Creating helpers (.OTHER or .THIS) here cause problems down the
1301
 
            # road if a ContentConflict needs to be created so we should not do
1302
 
            # that
1303
 
            trans_id = self.tt.trans_id_file_id(file_id)
1304
 
            self._raw_conflicts.append(('path conflict', trans_id, file_id,
1305
 
                                        this_parent, this_name,
1306
 
                                        other_parent, other_name))
 
1072
        if name_winner == "conflict":
 
1073
            trans_id = self.tt.trans_id_file_id(file_id)
 
1074
            self._raw_conflicts.append(('name conflict', trans_id, 
 
1075
                                        this_name, other_name))
 
1076
        if parent_id_winner == "conflict":
 
1077
            trans_id = self.tt.trans_id_file_id(file_id)
 
1078
            self._raw_conflicts.append(('parent conflict', trans_id, 
 
1079
                                        this_parent, other_parent))
1307
1080
        if other_name is None:
1308
 
            # it doesn't matter whether the result was 'other' or
 
1081
            # it doesn't matter whether the result was 'other' or 
1309
1082
            # 'conflict'-- if there's no 'other', we leave it alone.
1310
1083
            return
 
1084
        # if we get here, name_winner and parent_winner are set to safe values.
 
1085
        trans_id = self.tt.trans_id_file_id(file_id)
1311
1086
        parent_id = parents[self.winner_idx[parent_id_winner]]
1312
1087
        if parent_id is not None:
1313
 
            # if we get here, name_winner and parent_winner are set to safe
1314
 
            # values.
 
1088
            parent_trans_id = self.tt.trans_id_file_id(parent_id)
1315
1089
            self.tt.adjust_path(names[self.winner_idx[name_winner]],
1316
 
                                self.tt.trans_id_file_id(parent_id),
1317
 
                                self.tt.trans_id_file_id(file_id))
 
1090
                                parent_trans_id, trans_id)
1318
1091
 
1319
 
    def _do_merge_contents(self, file_id):
1320
 
        """Performs a merge on file_id contents."""
 
1092
    def merge_contents(self, file_id):
 
1093
        """Performa a merge on file_id contents."""
1321
1094
        def contents_pair(tree):
1322
1095
            if file_id not in tree:
1323
1096
                return (None, None)
1330
1103
                contents = None
1331
1104
            return kind, contents
1332
1105
 
 
1106
        def contents_conflict():
 
1107
            trans_id = self.tt.trans_id_file_id(file_id)
 
1108
            name = self.tt.final_name(trans_id)
 
1109
            parent_id = self.tt.final_parent(trans_id)
 
1110
            if file_id in self.this_tree.inventory:
 
1111
                self.tt.unversion_file(trans_id)
 
1112
                if file_id in self.this_tree:
 
1113
                    self.tt.delete_contents(trans_id)
 
1114
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1115
                                              set_version=True)
 
1116
            self._raw_conflicts.append(('contents conflict', file_group))
 
1117
 
1333
1118
        # See SPOT run.  run, SPOT, run.
1334
1119
        # So we're not QUITE repeating ourselves; we do tricky things with
1335
1120
        # file kind...
1336
1121
        base_pair = contents_pair(self.base_tree)
1337
1122
        other_pair = contents_pair(self.other_tree)
1338
 
        if self._lca_trees:
1339
 
            this_pair = contents_pair(self.this_tree)
1340
 
            lca_pairs = [contents_pair(tree) for tree in self._lca_trees]
1341
 
            winner = self._lca_multi_way((base_pair, lca_pairs), other_pair,
1342
 
                                         this_pair, allow_overriding_lca=False)
1343
 
        else:
1344
 
            if base_pair == other_pair:
1345
 
                winner = 'this'
1346
 
            else:
1347
 
                # We delayed evaluating this_pair as long as we can to avoid
1348
 
                # unnecessary sha1 calculation
1349
 
                this_pair = contents_pair(self.this_tree)
1350
 
                winner = self._three_way(base_pair, other_pair, this_pair)
1351
 
        if winner == 'this':
1352
 
            # No interesting changes introduced by OTHER
1353
 
            return "unmodified"
1354
 
        # We have a hypothetical conflict, but if we have files, then we
1355
 
        # can try to merge the content
1356
 
        trans_id = self.tt.trans_id_file_id(file_id)
1357
 
        params = MergeHookParams(self, file_id, trans_id, this_pair[0],
1358
 
            other_pair[0], winner)
1359
 
        hooks = self.active_hooks
1360
 
        hook_status = 'not_applicable'
1361
 
        for hook in hooks:
1362
 
            hook_status, lines = hook.merge_contents(params)
1363
 
            if hook_status != 'not_applicable':
1364
 
                # Don't try any more hooks, this one applies.
1365
 
                break
1366
 
        result = "modified"
1367
 
        if hook_status == 'not_applicable':
1368
 
            # This is a contents conflict, because none of the available
1369
 
            # functions could merge it.
1370
 
            result = None
1371
 
            name = self.tt.final_name(trans_id)
1372
 
            parent_id = self.tt.final_parent(trans_id)
1373
 
            if self.this_tree.has_id(file_id):
1374
 
                self.tt.unversion_file(trans_id)
1375
 
            file_group = self._dump_conflicts(name, parent_id, file_id,
1376
 
                                              set_version=True)
1377
 
            self._raw_conflicts.append(('contents conflict', file_group))
1378
 
        elif hook_status == 'success':
1379
 
            self.tt.create_file(lines, trans_id)
1380
 
        elif hook_status == 'conflicted':
1381
 
            # XXX: perhaps the hook should be able to provide
1382
 
            # the BASE/THIS/OTHER files?
1383
 
            self.tt.create_file(lines, trans_id)
1384
 
            self._raw_conflicts.append(('text conflict', trans_id))
1385
 
            name = self.tt.final_name(trans_id)
1386
 
            parent_id = self.tt.final_parent(trans_id)
1387
 
            self._dump_conflicts(name, parent_id, file_id)
1388
 
        elif hook_status == 'delete':
1389
 
            self.tt.unversion_file(trans_id)
1390
 
            result = "deleted"
1391
 
        elif hook_status == 'done':
1392
 
            # The hook function did whatever it needs to do directly, no
1393
 
            # further action needed here.
1394
 
            pass
1395
 
        else:
1396
 
            raise AssertionError('unknown hook_status: %r' % (hook_status,))
1397
 
        if not self.this_tree.has_id(file_id) and result == "modified":
1398
 
            self.tt.version_file(file_id, trans_id)
1399
 
        # The merge has been performed, so the old contents should not be
1400
 
        # retained.
1401
 
        try:
1402
 
            self.tt.delete_contents(trans_id)
1403
 
        except errors.NoSuchFile:
1404
 
            pass
1405
 
        return result
1406
 
 
1407
 
    def _default_other_winner_merge(self, merge_hook_params):
1408
 
        """Replace this contents with other."""
1409
 
        file_id = merge_hook_params.file_id
1410
 
        trans_id = merge_hook_params.trans_id
1411
 
        file_in_this = self.this_tree.has_id(file_id)
1412
 
        if self.other_tree.has_id(file_id):
1413
 
            # OTHER changed the file
1414
 
            wt = self.this_tree
1415
 
            if wt.supports_content_filtering():
1416
 
                # We get the path from the working tree if it exists.
1417
 
                # That fails though when OTHER is adding a file, so
1418
 
                # we fall back to the other tree to find the path if
1419
 
                # it doesn't exist locally.
1420
 
                try:
1421
 
                    filter_tree_path = wt.id2path(file_id)
1422
 
                except errors.NoSuchId:
1423
 
                    filter_tree_path = self.other_tree.id2path(file_id)
1424
 
            else:
1425
 
                # Skip the id2path lookup for older formats
1426
 
                filter_tree_path = None
1427
 
            transform.create_from_tree(self.tt, trans_id,
1428
 
                             self.other_tree, file_id,
1429
 
                             filter_tree_path=filter_tree_path)
1430
 
            return 'done', None
1431
 
        elif file_in_this:
1432
 
            # OTHER deleted the file
1433
 
            return 'delete', None
1434
 
        else:
1435
 
            raise AssertionError(
1436
 
                'winner is OTHER, but file_id %r not in THIS or OTHER tree'
1437
 
                % (file_id,))
1438
 
 
1439
 
    def merge_contents(self, merge_hook_params):
1440
 
        """Fallback merge logic after user installed hooks."""
1441
 
        # This function is used in merge hooks as the fallback instance.
1442
 
        # Perhaps making this function and the functions it calls be a 
1443
 
        # a separate class would be better.
1444
 
        if merge_hook_params.winner == 'other':
1445
 
            # OTHER is a straight winner, so replace this contents with other
1446
 
            return self._default_other_winner_merge(merge_hook_params)
1447
 
        elif merge_hook_params.is_file_merge():
1448
 
            # THIS and OTHER are both files, so text merge.  Either
1449
 
            # BASE is a file, or both converted to files, so at least we
1450
 
            # have agreement that output should be a file.
1451
 
            try:
1452
 
                self.text_merge(merge_hook_params.file_id,
1453
 
                    merge_hook_params.trans_id)
1454
 
            except errors.BinaryFile:
1455
 
                return 'not_applicable', None
1456
 
            return 'done', None
1457
 
        else:
1458
 
            return 'not_applicable', None
 
1123
        if base_pair == other_pair:
 
1124
            # OTHER introduced no changes
 
1125
            return "unmodified"
 
1126
        this_pair = contents_pair(self.this_tree)
 
1127
        if this_pair == other_pair:
 
1128
            # THIS and OTHER introduced the same changes
 
1129
            return "unmodified"
 
1130
        else:
 
1131
            trans_id = self.tt.trans_id_file_id(file_id)
 
1132
            if this_pair == base_pair:
 
1133
                # only OTHER introduced changes
 
1134
                if file_id in self.this_tree:
 
1135
                    # Remove any existing contents
 
1136
                    self.tt.delete_contents(trans_id)
 
1137
                if file_id in self.other_tree:
 
1138
                    # OTHER changed the file
 
1139
                    create_from_tree(self.tt, trans_id,
 
1140
                                     self.other_tree, file_id)
 
1141
                    if file_id not in self.this_tree:
 
1142
                        self.tt.version_file(file_id, trans_id)
 
1143
                    return "modified"
 
1144
                elif file_id in self.this_tree.inventory:
 
1145
                    # OTHER deleted the file
 
1146
                    self.tt.unversion_file(trans_id)
 
1147
                    return "deleted"
 
1148
            #BOTH THIS and OTHER introduced changes; scalar conflict
 
1149
            elif this_pair[0] == "file" and other_pair[0] == "file":
 
1150
                # THIS and OTHER are both files, so text merge.  Either
 
1151
                # BASE is a file, or both converted to files, so at least we
 
1152
                # have agreement that output should be a file.
 
1153
                try:
 
1154
                    self.text_merge(file_id, trans_id)
 
1155
                except BinaryFile:
 
1156
                    return contents_conflict()
 
1157
                if file_id not in self.this_tree:
 
1158
                    self.tt.version_file(file_id, trans_id)
 
1159
                try:
 
1160
                    self.tt.tree_kind(trans_id)
 
1161
                    self.tt.delete_contents(trans_id)
 
1162
                except NoSuchFile:
 
1163
                    pass
 
1164
                return "modified"
 
1165
            else:
 
1166
                # Scalar conflict, can't text merge.  Dump conflicts
 
1167
                return contents_conflict()
1459
1168
 
1460
1169
    def get_lines(self, tree, file_id):
1461
1170
        """Return the lines in a file, or an empty list."""
1462
 
        if tree.has_id(file_id):
1463
 
            return tree.get_file_lines(file_id)
 
1171
        if file_id in tree:
 
1172
            return tree.get_file(file_id).readlines()
1464
1173
        else:
1465
1174
            return []
1466
1175
 
1468
1177
        """Perform a three-way text merge on a file_id"""
1469
1178
        # it's possible that we got here with base as a different type.
1470
1179
        # if so, we just want two-way text conflicts.
1471
 
        if self.base_tree.has_id(file_id) and \
 
1180
        if file_id in self.base_tree and \
1472
1181
            self.base_tree.kind(file_id) == "file":
1473
1182
            base_lines = self.get_lines(self.base_tree, file_id)
1474
1183
        else:
1475
1184
            base_lines = []
1476
1185
        other_lines = self.get_lines(self.other_tree, file_id)
1477
1186
        this_lines = self.get_lines(self.this_tree, file_id)
1478
 
        m3 = merge3.Merge3(base_lines, this_lines, other_lines,
1479
 
                           is_cherrypick=self.cherrypick)
 
1187
        m3 = Merge3(base_lines, this_lines, other_lines,
 
1188
                    is_cherrypick=self.cherrypick)
1480
1189
        start_marker = "!START OF MERGE CONFLICT!" + "I HOPE THIS IS UNIQUE"
1481
1190
        if self.show_base is True:
1482
1191
            base_marker = '|' * 7
1485
1194
 
1486
1195
        def iter_merge3(retval):
1487
1196
            retval["text_conflicts"] = False
1488
 
            for line in m3.merge_lines(name_a = "TREE",
1489
 
                                       name_b = "MERGE-SOURCE",
 
1197
            for line in m3.merge_lines(name_a = "TREE", 
 
1198
                                       name_b = "MERGE-SOURCE", 
1490
1199
                                       name_base = "BASE-REVISION",
1491
 
                                       start_marker=start_marker,
 
1200
                                       start_marker=start_marker, 
1492
1201
                                       base_marker=base_marker,
1493
1202
                                       reprocess=self.reprocess):
1494
1203
                if line.startswith(start_marker):
1503
1212
            self._raw_conflicts.append(('text conflict', trans_id))
1504
1213
            name = self.tt.final_name(trans_id)
1505
1214
            parent_id = self.tt.final_parent(trans_id)
1506
 
            file_group = self._dump_conflicts(name, parent_id, file_id,
 
1215
            file_group = self._dump_conflicts(name, parent_id, file_id, 
1507
1216
                                              this_lines, base_lines,
1508
1217
                                              other_lines)
1509
1218
            file_group.append(trans_id)
1510
1219
 
1511
 
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None,
 
1220
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None, 
1512
1221
                        base_lines=None, other_lines=None, set_version=False,
1513
1222
                        no_base=False):
1514
1223
        """Emit conflict files.
1516
1225
        determined automatically.  If set_version is true, the .OTHER, .THIS
1517
1226
        or .BASE (in that order) will be created as versioned files.
1518
1227
        """
1519
 
        data = [('OTHER', self.other_tree, other_lines),
 
1228
        data = [('OTHER', self.other_tree, other_lines), 
1520
1229
                ('THIS', self.this_tree, this_lines)]
1521
1230
        if not no_base:
1522
1231
            data.append(('BASE', self.base_tree, base_lines))
1523
 
 
1524
 
        # We need to use the actual path in the working tree of the file here,
1525
 
        # ignoring the conflict suffixes
1526
 
        wt = self.this_tree
1527
 
        if wt.supports_content_filtering():
1528
 
            try:
1529
 
                filter_tree_path = wt.id2path(file_id)
1530
 
            except errors.NoSuchId:
1531
 
                # file has been deleted
1532
 
                filter_tree_path = None
1533
 
        else:
1534
 
            # Skip the id2path lookup for older formats
1535
 
            filter_tree_path = None
1536
 
 
1537
1232
        versioned = False
1538
1233
        file_group = []
1539
1234
        for suffix, tree, lines in data:
1540
 
            if tree.has_id(file_id):
 
1235
            if file_id in tree:
1541
1236
                trans_id = self._conflict_file(name, parent_id, tree, file_id,
1542
 
                                               suffix, lines, filter_tree_path)
 
1237
                                               suffix, lines)
1543
1238
                file_group.append(trans_id)
1544
1239
                if set_version and not versioned:
1545
1240
                    self.tt.version_file(file_id, trans_id)
1546
1241
                    versioned = True
1547
1242
        return file_group
1548
 
 
 
1243
           
1549
1244
    def _conflict_file(self, name, parent_id, tree, file_id, suffix,
1550
 
                       lines=None, filter_tree_path=None):
 
1245
                       lines=None):
1551
1246
        """Emit a single conflict file."""
1552
1247
        name = name + '.' + suffix
1553
1248
        trans_id = self.tt.create_path(name, parent_id)
1554
 
        transform.create_from_tree(self.tt, trans_id, tree, file_id, lines,
1555
 
            filter_tree_path)
 
1249
        create_from_tree(self.tt, trans_id, tree, file_id, lines)
1556
1250
        return trans_id
1557
1251
 
1558
1252
    def merge_executable(self, file_id, file_status):
1582
1276
        try:
1583
1277
            if self.tt.final_kind(trans_id) != "file":
1584
1278
                return
1585
 
        except errors.NoSuchFile:
 
1279
        except NoSuchFile:
1586
1280
            return
1587
1281
        if winner == "this":
1588
1282
            executability = this_executable
1589
1283
        else:
1590
 
            if self.other_tree.has_id(file_id):
 
1284
            if file_id in self.other_tree:
1591
1285
                executability = other_executable
1592
 
            elif self.this_tree.has_id(file_id):
 
1286
            elif file_id in self.this_tree:
1593
1287
                executability = this_executable
1594
 
            elif self.base_tree_has_id(file_id):
 
1288
            elif file_id in self.base_tree:
1595
1289
                executability = base_executable
1596
1290
        if executability is not None:
1597
1291
            trans_id = self.tt.trans_id_file_id(file_id)
1599
1293
 
1600
1294
    def cook_conflicts(self, fs_conflicts):
1601
1295
        """Convert all conflicts into a form that doesn't depend on trans_id"""
1602
 
        self.cooked_conflicts.extend(transform.cook_conflicts(
1603
 
                fs_conflicts, self.tt))
1604
 
        fp = transform.FinalPaths(self.tt)
 
1296
        from conflicts import Conflict
 
1297
        name_conflicts = {}
 
1298
        self.cooked_conflicts.extend(cook_conflicts(fs_conflicts, self.tt))
 
1299
        fp = FinalPaths(self.tt)
1605
1300
        for conflict in self._raw_conflicts:
1606
1301
            conflict_type = conflict[0]
1607
 
            if conflict_type == 'path conflict':
1608
 
                (trans_id, file_id,
1609
 
                this_parent, this_name,
1610
 
                other_parent, other_name) = conflict[1:]
1611
 
                if this_parent is None or this_name is None:
1612
 
                    this_path = '<deleted>'
1613
 
                else:
1614
 
                    parent_path =  fp.get_path(
1615
 
                        self.tt.trans_id_file_id(this_parent))
1616
 
                    this_path = osutils.pathjoin(parent_path, this_name)
1617
 
                if other_parent is None or other_name is None:
1618
 
                    other_path = '<deleted>'
1619
 
                else:
1620
 
                    parent_path =  fp.get_path(
1621
 
                        self.tt.trans_id_file_id(other_parent))
1622
 
                    other_path = osutils.pathjoin(parent_path, other_name)
1623
 
                c = _mod_conflicts.Conflict.factory(
1624
 
                    'path conflict', path=this_path,
1625
 
                    conflict_path=other_path,
1626
 
                    file_id=file_id)
1627
 
            elif conflict_type == 'contents conflict':
 
1302
            if conflict_type in ('name conflict', 'parent conflict'):
 
1303
                trans_id = conflict[1]
 
1304
                conflict_args = conflict[2:]
 
1305
                if trans_id not in name_conflicts:
 
1306
                    name_conflicts[trans_id] = {}
 
1307
                unique_add(name_conflicts[trans_id], conflict_type, 
 
1308
                           conflict_args)
 
1309
            if conflict_type == 'contents conflict':
1628
1310
                for trans_id in conflict[1]:
1629
1311
                    file_id = self.tt.final_file_id(trans_id)
1630
1312
                    if file_id is not None:
1634
1316
                    if path.endswith(suffix):
1635
1317
                        path = path[:-len(suffix)]
1636
1318
                        break
1637
 
                c = _mod_conflicts.Conflict.factory(conflict_type,
1638
 
                                                    path=path, file_id=file_id)
1639
 
            elif conflict_type == 'text conflict':
 
1319
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
 
1320
                self.cooked_conflicts.append(c)
 
1321
            if conflict_type == 'text conflict':
1640
1322
                trans_id = conflict[1]
1641
1323
                path = fp.get_path(trans_id)
1642
1324
                file_id = self.tt.final_file_id(trans_id)
1643
 
                c = _mod_conflicts.Conflict.factory(conflict_type,
1644
 
                                                    path=path, file_id=file_id)
 
1325
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
 
1326
                self.cooked_conflicts.append(c)
 
1327
 
 
1328
        for trans_id, conflicts in name_conflicts.iteritems():
 
1329
            try:
 
1330
                this_parent, other_parent = conflicts['parent conflict']
 
1331
                if this_parent == other_parent:
 
1332
                    raise AssertionError()
 
1333
            except KeyError:
 
1334
                this_parent = other_parent = \
 
1335
                    self.tt.final_file_id(self.tt.final_parent(trans_id))
 
1336
            try:
 
1337
                this_name, other_name = conflicts['name conflict']
 
1338
                if this_name == other_name:
 
1339
                    raise AssertionError()
 
1340
            except KeyError:
 
1341
                this_name = other_name = self.tt.final_name(trans_id)
 
1342
            other_path = fp.get_path(trans_id)
 
1343
            if this_parent is not None and this_name is not None:
 
1344
                this_parent_path = \
 
1345
                    fp.get_path(self.tt.trans_id_file_id(this_parent))
 
1346
                this_path = pathjoin(this_parent_path, this_name)
1645
1347
            else:
1646
 
                raise AssertionError('bad conflict type: %r' % (conflict,))
 
1348
                this_path = "<deleted>"
 
1349
            file_id = self.tt.final_file_id(trans_id)
 
1350
            c = Conflict.factory('path conflict', path=this_path,
 
1351
                                 conflict_path=other_path, file_id=file_id)
1647
1352
            self.cooked_conflicts.append(c)
1648
 
        self.cooked_conflicts.sort(key=_mod_conflicts.Conflict.sort_key)
 
1353
        self.cooked_conflicts.sort(key=Conflict.sort_key)
1649
1354
 
1650
1355
 
1651
1356
class WeaveMerger(Merge3Merger):
1655
1360
    supports_reverse_cherrypick = False
1656
1361
    history_based = True
1657
1362
 
1658
 
    def _generate_merge_plan(self, file_id, base):
1659
 
        return self.this_tree.plan_file_merge(file_id, self.other_tree,
 
1363
    def _merged_lines(self, file_id):
 
1364
        """Generate the merged lines.
 
1365
        There is no distinction between lines that are meant to contain <<<<<<<
 
1366
        and conflicts.
 
1367
        """
 
1368
        if self.cherrypick:
 
1369
            base = self.base_tree
 
1370
        else:
 
1371
            base = None
 
1372
        plan = self.this_tree.plan_file_merge(file_id, self.other_tree,
1660
1373
                                              base=base)
1661
 
 
1662
 
    def _merged_lines(self, file_id):
1663
 
        """Generate the merged lines.
1664
 
        There is no distinction between lines that are meant to contain <<<<<<<
1665
 
        and conflicts.
1666
 
        """
1667
 
        if self.cherrypick:
1668
 
            base = self.base_tree
1669
 
        else:
1670
 
            base = None
1671
 
        plan = self._generate_merge_plan(file_id, base)
1672
1374
        if 'merge' in debug.debug_flags:
1673
1375
            plan = list(plan)
1674
1376
            trans_id = self.tt.trans_id_file_id(file_id)
1675
1377
            name = self.tt.final_name(trans_id) + '.plan'
1676
 
            contents = ('%11s|%s' % l for l in plan)
 
1378
            contents = ('%10s|%s' % l for l in plan)
1677
1379
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
1678
 
        textmerge = versionedfile.PlanWeaveMerge(plan, '<<<<<<< TREE\n',
1679
 
                                                 '>>>>>>> MERGE-SOURCE\n')
1680
 
        lines, conflicts = textmerge.merge_lines(self.reprocess)
1681
 
        if conflicts:
1682
 
            base_lines = textmerge.base_from_plan()
1683
 
        else:
1684
 
            base_lines = None
1685
 
        return lines, base_lines
 
1380
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1381
            '>>>>>>> MERGE-SOURCE\n')
 
1382
        return textmerge.merge_lines(self.reprocess)
1686
1383
 
1687
1384
    def text_merge(self, file_id, trans_id):
1688
1385
        """Perform a (weave) text merge for a given file and file-id.
1689
1386
        If conflicts are encountered, .THIS and .OTHER files will be emitted,
1690
1387
        and a conflict will be noted.
1691
1388
        """
1692
 
        lines, base_lines = self._merged_lines(file_id)
 
1389
        lines, conflicts = self._merged_lines(file_id)
1693
1390
        lines = list(lines)
1694
 
        # Note we're checking whether the OUTPUT is binary in this case,
 
1391
        # Note we're checking whether the OUTPUT is binary in this case, 
1695
1392
        # because we don't want to get into weave merge guts.
1696
 
        textfile.check_text_lines(lines)
 
1393
        check_text_lines(lines)
1697
1394
        self.tt.create_file(lines, trans_id)
1698
 
        if base_lines is not None:
1699
 
            # Conflict
 
1395
        if conflicts:
1700
1396
            self._raw_conflicts.append(('text conflict', trans_id))
1701
1397
            name = self.tt.final_name(trans_id)
1702
1398
            parent_id = self.tt.final_parent(trans_id)
1703
 
            file_group = self._dump_conflicts(name, parent_id, file_id,
1704
 
                                              no_base=False,
1705
 
                                              base_lines=base_lines)
 
1399
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1400
                                              no_base=True)
1706
1401
            file_group.append(trans_id)
1707
1402
 
1708
1403
 
1709
1404
class LCAMerger(WeaveMerger):
1710
1405
 
1711
 
    def _generate_merge_plan(self, file_id, base):
1712
 
        return self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
 
1406
    def _merged_lines(self, file_id):
 
1407
        """Generate the merged lines.
 
1408
        There is no distinction between lines that are meant to contain <<<<<<<
 
1409
        and conflicts.
 
1410
        """
 
1411
        if self.cherrypick:
 
1412
            base = self.base_tree
 
1413
        else:
 
1414
            base = None
 
1415
        plan = self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
1713
1416
                                                  base=base)
 
1417
        if 'merge' in debug.debug_flags:
 
1418
            plan = list(plan)
 
1419
            trans_id = self.tt.trans_id_file_id(file_id)
 
1420
            name = self.tt.final_name(trans_id) + '.plan'
 
1421
            contents = ('%10s|%s' % l for l in plan)
 
1422
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
 
1423
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1424
            '>>>>>>> MERGE-SOURCE\n')
 
1425
        return textmerge.merge_lines(self.reprocess)
 
1426
 
1714
1427
 
1715
1428
class Diff3Merger(Merge3Merger):
1716
1429
    """Three-way merger using external diff3 for text merging"""
1717
1430
 
1718
1431
    def dump_file(self, temp_dir, name, tree, file_id):
1719
 
        out_path = osutils.pathjoin(temp_dir, name)
 
1432
        out_path = pathjoin(temp_dir, name)
1720
1433
        out_file = open(out_path, "wb")
1721
1434
        try:
1722
1435
            in_file = tree.get_file(file_id)
1734
1447
        import bzrlib.patch
1735
1448
        temp_dir = osutils.mkdtemp(prefix="bzr-")
1736
1449
        try:
1737
 
            new_file = osutils.pathjoin(temp_dir, "new")
 
1450
            new_file = pathjoin(temp_dir, "new")
1738
1451
            this = self.dump_file(temp_dir, "this", self.this_tree, file_id)
1739
1452
            base = self.dump_file(temp_dir, "base", self.base_tree, file_id)
1740
1453
            other = self.dump_file(temp_dir, "other", self.other_tree, file_id)
1741
1454
            status = bzrlib.patch.diff3(new_file, this, base, other)
1742
1455
            if status not in (0, 1):
1743
 
                raise errors.BzrError("Unhandled diff3 exit code")
 
1456
                raise BzrError("Unhandled diff3 exit code")
1744
1457
            f = open(new_file, 'rb')
1745
1458
            try:
1746
1459
                self.tt.create_file(f, trans_id)
1764
1477
                other_rev_id=None,
1765
1478
                interesting_files=None,
1766
1479
                this_tree=None,
1767
 
                pb=None,
 
1480
                pb=DummyProgress(),
1768
1481
                change_reporter=None):
1769
 
    """Primary interface for merging.
 
1482
    """Primary interface for merging. 
1770
1483
 
1771
 
        typical use is probably
 
1484
        typical use is probably 
1772
1485
        'merge_inner(branch, branch.get_revision_tree(other_revision),
1773
1486
                     branch.get_revision_tree(base_revision))'
1774
1487
        """
1775
1488
    if this_tree is None:
1776
 
        raise errors.BzrError("bzrlib.merge.merge_inner requires a this_tree "
1777
 
                              "parameter as of bzrlib version 0.8.")
 
1489
        raise BzrError("bzrlib.merge.merge_inner requires a this_tree "
 
1490
            "parameter as of bzrlib version 0.8.")
1778
1491
    merger = Merger(this_branch, other_tree, base_tree, this_tree=this_tree,
1779
1492
                    pb=pb, change_reporter=change_reporter)
1780
1493
    merger.backup_files = backup_files
1793
1506
    get_revision_id = getattr(base_tree, 'get_revision_id', None)
1794
1507
    if get_revision_id is None:
1795
1508
        get_revision_id = base_tree.last_revision
1796
 
    merger.cache_trees_with_revision_ids([other_tree, base_tree, this_tree])
1797
1509
    merger.set_base_revision(get_revision_id(), this_branch)
1798
1510
    return merger.do_merge()
1799
1511
 
1867
1579
 
1868
1580
    def get_lines(self, revisions):
1869
1581
        """Get lines for revisions from the backing VersionedFiles.
1870
 
 
 
1582
        
1871
1583
        :raises RevisionNotPresent: on absent texts.
1872
1584
        """
1873
1585
        keys = [(self._key_prefix + (rev,)) for rev in revisions]
1875
1587
        for record in self.vf.get_record_stream(keys, 'unordered', True):
1876
1588
            if record.storage_kind == 'absent':
1877
1589
                raise errors.RevisionNotPresent(record.key, self.vf)
1878
 
            result[record.key[-1]] = osutils.chunks_to_lines(
1879
 
                record.get_bytes_as('chunked'))
 
1590
            result[record.key[-1]] = osutils.split_lines(
 
1591
                record.get_bytes_as('fulltext'))
1880
1592
        return result
1881
1593
 
1882
1594
    def plan_merge(self):
1998
1710
        super(_PlanMerge, self).__init__(a_rev, b_rev, vf, key_prefix)
1999
1711
        self.a_key = self._key_prefix + (self.a_rev,)
2000
1712
        self.b_key = self._key_prefix + (self.b_rev,)
2001
 
        self.graph = _mod_graph.Graph(self.vf)
 
1713
        self.graph = Graph(self.vf)
2002
1714
        heads = self.graph.heads((self.a_key, self.b_key))
2003
1715
        if len(heads) == 1:
2004
1716
            # one side dominates, so we can just return its values, yay for
2009
1721
                other = b_rev
2010
1722
            else:
2011
1723
                other = a_rev
2012
 
            trace.mutter('found dominating revision for %s\n%s > %s', self.vf,
2013
 
                         self._head_key[-1], other)
 
1724
            mutter('found dominating revision for %s\n%s > %s', self.vf,
 
1725
                   self._head_key[-1], other)
2014
1726
            self._weave = None
2015
1727
        else:
2016
1728
            self._head_key = None
2030
1742
        while True:
2031
1743
            next_lcas = self.graph.find_lca(*cur_ancestors)
2032
1744
            # Map a plain NULL_REVISION to a simple no-ancestors
2033
 
            if next_lcas == set([_mod_revision.NULL_REVISION]):
 
1745
            if next_lcas == set([NULL_REVISION]):
2034
1746
                next_lcas = ()
2035
1747
            # Order the lca's based on when they were merged into the tip
2036
1748
            # While the actual merge portion of weave merge uses a set() of
2048
1760
            elif len(next_lcas) > 2:
2049
1761
                # More than 2 lca's, fall back to grabbing all nodes between
2050
1762
                # this and the unique lca.
2051
 
                trace.mutter('More than 2 LCAs, falling back to all nodes for:'
2052
 
                             ' %s, %s\n=> %s',
2053
 
                             self.a_key, self.b_key, cur_ancestors)
 
1763
                mutter('More than 2 LCAs, falling back to all nodes for:'
 
1764
                       ' %s, %s\n=> %s', self.a_key, self.b_key, cur_ancestors)
2054
1765
                cur_lcas = next_lcas
2055
1766
                while len(cur_lcas) > 1:
2056
1767
                    cur_lcas = self.graph.find_lca(*cur_lcas)
2059
1770
                    unique_lca = None
2060
1771
                else:
2061
1772
                    unique_lca = list(cur_lcas)[0]
2062
 
                    if unique_lca == _mod_revision.NULL_REVISION:
 
1773
                    if unique_lca == NULL_REVISION:
2063
1774
                        # find_lca will return a plain 'NULL_REVISION' rather
2064
1775
                        # than a key tuple when there is no common ancestor, we
2065
1776
                        # prefer to just use None, because it doesn't confuse
2073
1784
 
2074
1785
    def _find_unique_parents(self, tip_keys, base_key):
2075
1786
        """Find ancestors of tip that aren't ancestors of base.
2076
 
 
 
1787
        
2077
1788
        :param tip_keys: Nodes that are interesting
2078
1789
        :param base_key: Cull all ancestors of this node
2079
1790
        :return: The parent map for all revisions between tip_keys and
2088
1799
            # We remove NULL_REVISION because it isn't a proper tuple key, and
2089
1800
            # thus confuses things like _get_interesting_texts, and our logic
2090
1801
            # to add the texts into the memory weave.
2091
 
            if _mod_revision.NULL_REVISION in parent_map:
2092
 
                parent_map.pop(_mod_revision.NULL_REVISION)
 
1802
            if NULL_REVISION in parent_map:
 
1803
                parent_map.pop(NULL_REVISION)
2093
1804
        else:
2094
1805
            interesting = set()
2095
1806
            for tip in tip_keys:
2139
1850
    @staticmethod
2140
1851
    def _prune_tails(parent_map, child_map, tails_to_remove):
2141
1852
        """Remove tails from the parent map.
2142
 
 
 
1853
        
2143
1854
        This will remove the supplied revisions until no more children have 0
2144
1855
        parents.
2145
1856
 
2247
1958
        lcas = graph.find_lca(key_prefix + (a_rev,), key_prefix + (b_rev,))
2248
1959
        self.lcas = set()
2249
1960
        for lca in lcas:
2250
 
            if lca == _mod_revision.NULL_REVISION:
 
1961
            if lca == NULL_REVISION:
2251
1962
                self.lcas.add(lca)
2252
1963
            else:
2253
1964
                self.lcas.add(lca[-1])