~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-08-17 18:13:57 UTC
  • mfrom: (5268.7.29 transport-segments)
  • Revision ID: pqm@pqm.ubuntu.com-20110817181357-y5q5eth1hk8bl3om
(jelmer) Allow specifying the colocated branch to use in the branch URL,
 and retrieving the branch name using ControlDir._get_selected_branch.
 (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2008 Canonical Ltd
 
1
# Copyright (C) 2005-2011 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
18
 
import errno
19
 
from itertools import chain
20
 
import os
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
21
17
import warnings
22
18
 
 
19
from bzrlib.lazy_import import lazy_import
 
20
lazy_import(globals(), """
23
21
from bzrlib import (
 
22
    branch as _mod_branch,
 
23
    cleanup,
 
24
    conflicts as _mod_conflicts,
24
25
    debug,
25
 
    errors,
 
26
    generate_ids,
26
27
    graph as _mod_graph,
 
28
    merge3,
27
29
    osutils,
28
30
    patiencediff,
29
 
    registry,
30
31
    revision as _mod_revision,
 
32
    textfile,
 
33
    trace,
 
34
    transform,
 
35
    tree as _mod_tree,
31
36
    tsort,
32
 
    )
33
 
from bzrlib.branch import Branch
34
 
from bzrlib.conflicts import ConflictList, Conflict
35
 
from bzrlib.errors import (BzrCommandError,
36
 
                           BzrError,
37
 
                           NoCommonAncestor,
38
 
                           NoCommits,
39
 
                           NoSuchRevision,
40
 
                           NoSuchFile,
41
 
                           NotBranchError,
42
 
                           NotVersionedError,
43
 
                           UnrelatedBranches,
44
 
                           UnsupportedOperation,
45
 
                           WorkingTreeNotRevision,
46
 
                           BinaryFile,
47
 
                           )
48
 
from bzrlib.graph import Graph
49
 
from bzrlib.merge3 import Merge3
50
 
from bzrlib.osutils import rename, pathjoin
51
 
from progress import DummyProgress, ProgressPhase
52
 
from bzrlib.revision import (NULL_REVISION, ensure_null)
53
 
from bzrlib.textfile import check_text_lines
54
 
from bzrlib.trace import mutter, warning, note, is_quiet
55
 
from bzrlib.transform import (TransformPreview, TreeTransform,
56
 
                              resolve_conflicts, cook_conflicts,
57
 
                              conflict_pass, FinalPaths, create_by_entry,
58
 
                              unique_add, ROOT_PARENT)
59
 
from bzrlib.versionedfile import PlanWeaveMerge
60
 
from bzrlib import ui
61
 
 
 
37
    ui,
 
38
    versionedfile,
 
39
    workingtree,
 
40
    )
 
41
""")
 
42
from bzrlib import (
 
43
    decorators,
 
44
    errors,
 
45
    hooks,
 
46
    )
 
47
from bzrlib.symbol_versioning import (
 
48
    deprecated_in,
 
49
    deprecated_method,
 
50
    )
62
51
# TODO: Report back as changes are merged in
63
52
 
64
53
 
65
54
def transform_tree(from_tree, to_tree, interesting_ids=None):
66
 
    merge_inner(from_tree.branch, to_tree, from_tree, ignore_zero=True,
67
 
                interesting_ids=interesting_ids, this_tree=from_tree)
 
55
    from_tree.lock_tree_write()
 
56
    operation = cleanup.OperationWithCleanups(merge_inner)
 
57
    operation.add_cleanup(from_tree.unlock)
 
58
    operation.run_simple(from_tree.branch, to_tree, from_tree,
 
59
        ignore_zero=True, interesting_ids=interesting_ids, this_tree=from_tree)
 
60
 
 
61
 
 
62
class MergeHooks(hooks.Hooks):
 
63
 
 
64
    def __init__(self):
 
65
        hooks.Hooks.__init__(self, "bzrlib.merge", "Merger.hooks")
 
66
        self.add_hook('merge_file_content',
 
67
            "Called with a bzrlib.merge.Merger object to create a per file "
 
68
            "merge object when starting a merge. "
 
69
            "Should return either None or a subclass of "
 
70
            "``bzrlib.merge.AbstractPerFileMerger``. "
 
71
            "Such objects will then be called per file "
 
72
            "that needs to be merged (including when one "
 
73
            "side has deleted the file and the other has changed it). "
 
74
            "See the AbstractPerFileMerger API docs for details on how it is "
 
75
            "used by merge.",
 
76
            (2, 1))
 
77
 
 
78
 
 
79
class AbstractPerFileMerger(object):
 
80
    """PerFileMerger objects are used by plugins extending merge for bzrlib.
 
81
 
 
82
    See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
 
83
    
 
84
    :ivar merger: The Merge3Merger performing the merge.
 
85
    """
 
86
 
 
87
    def __init__(self, merger):
 
88
        """Create a PerFileMerger for use with merger."""
 
89
        self.merger = merger
 
90
 
 
91
    def merge_contents(self, merge_params):
 
92
        """Attempt to merge the contents of a single file.
 
93
        
 
94
        :param merge_params: A bzrlib.merge.MergeHookParams
 
95
        :return: A tuple of (status, chunks), where status is one of
 
96
            'not_applicable', 'success', 'conflicted', or 'delete'.  If status
 
97
            is 'success' or 'conflicted', then chunks should be an iterable of
 
98
            strings for the new file contents.
 
99
        """
 
100
        return ('not applicable', None)
 
101
 
 
102
 
 
103
class PerFileMerger(AbstractPerFileMerger):
 
104
    """Merge individual files when self.file_matches returns True.
 
105
 
 
106
    This class is intended to be subclassed.  The file_matches and
 
107
    merge_matching methods should be overridden with concrete implementations.
 
108
    """
 
109
 
 
110
    def file_matches(self, params):
 
111
        """Return True if merge_matching should be called on this file.
 
112
 
 
113
        Only called with merges of plain files with no clear winner.
 
114
 
 
115
        Subclasses must override this.
 
116
        """
 
117
        raise NotImplementedError(self.file_matches)
 
118
 
 
119
    def get_filename(self, params, tree):
 
120
        """Lookup the filename (i.e. basename, not path), given a Tree (e.g.
 
121
        self.merger.this_tree) and a MergeHookParams.
 
122
        """
 
123
        return osutils.basename(tree.id2path(params.file_id))
 
124
 
 
125
    def get_filepath(self, params, tree):
 
126
        """Calculate the path to the file in a tree.
 
127
 
 
128
        :param params: A MergeHookParams describing the file to merge
 
129
        :param tree: a Tree, e.g. self.merger.this_tree.
 
130
        """
 
131
        return tree.id2path(params.file_id)
 
132
 
 
133
    def merge_contents(self, params):
 
134
        """Merge the contents of a single file."""
 
135
        # Check whether this custom merge logic should be used.
 
136
        if (
 
137
            # OTHER is a straight winner, rely on default merge.
 
138
            params.winner == 'other' or
 
139
            # THIS and OTHER aren't both files.
 
140
            not params.is_file_merge() or
 
141
            # The filename doesn't match *.xml
 
142
            not self.file_matches(params)):
 
143
            return 'not_applicable', None
 
144
        return self.merge_matching(params)
 
145
 
 
146
    def merge_matching(self, params):
 
147
        """Merge the contents of a single file that has matched the criteria
 
148
        in PerFileMerger.merge_contents (is a conflict, is a file,
 
149
        self.file_matches is True).
 
150
 
 
151
        Subclasses must override this.
 
152
        """
 
153
        raise NotImplementedError(self.merge_matching)
 
154
 
 
155
 
 
156
class ConfigurableFileMerger(PerFileMerger):
 
157
    """Merge individual files when configured via a .conf file.
 
158
 
 
159
    This is a base class for concrete custom file merging logic. Concrete
 
160
    classes should implement ``merge_text``.
 
161
 
 
162
    See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
 
163
    
 
164
    :ivar affected_files: The configured file paths to merge.
 
165
 
 
166
    :cvar name_prefix: The prefix to use when looking up configuration
 
167
        details. <name_prefix>_merge_files describes the files targeted by the
 
168
        hook for example.
 
169
        
 
170
    :cvar default_files: The default file paths to merge when no configuration
 
171
        is present.
 
172
    """
 
173
 
 
174
    name_prefix = None
 
175
    default_files = None
 
176
 
 
177
    def __init__(self, merger):
 
178
        super(ConfigurableFileMerger, self).__init__(merger)
 
179
        self.affected_files = None
 
180
        self.default_files = self.__class__.default_files or []
 
181
        self.name_prefix = self.__class__.name_prefix
 
182
        if self.name_prefix is None:
 
183
            raise ValueError("name_prefix must be set.")
 
184
 
 
185
    def file_matches(self, params):
 
186
        """Check whether the file should call the merge hook.
 
187
 
 
188
        <name_prefix>_merge_files configuration variable is a list of files
 
189
        that should use the hook.
 
190
        """
 
191
        affected_files = self.affected_files
 
192
        if affected_files is None:
 
193
            config = self.merger.this_branch.get_config()
 
194
            # Until bzr provides a better policy for caching the config, we
 
195
            # just add the part we're interested in to the params to avoid
 
196
            # reading the config files repeatedly (bazaar.conf, location.conf,
 
197
            # branch.conf).
 
198
            config_key = self.name_prefix + '_merge_files'
 
199
            affected_files = config.get_user_option_as_list(config_key)
 
200
            if affected_files is None:
 
201
                # If nothing was specified in the config, use the default.
 
202
                affected_files = self.default_files
 
203
            self.affected_files = affected_files
 
204
        if affected_files:
 
205
            filepath = self.get_filepath(params, self.merger.this_tree)
 
206
            if filepath in affected_files:
 
207
                return True
 
208
        return False
 
209
 
 
210
    def merge_matching(self, params):
 
211
        return self.merge_text(params)
 
212
 
 
213
    def merge_text(self, params):
 
214
        """Merge the byte contents of a single file.
 
215
 
 
216
        This is called after checking that the merge should be performed in
 
217
        merge_contents, and it should behave as per
 
218
        ``bzrlib.merge.AbstractPerFileMerger.merge_contents``.
 
219
        """
 
220
        raise NotImplementedError(self.merge_text)
 
221
 
 
222
 
 
223
class MergeHookParams(object):
 
224
    """Object holding parameters passed to merge_file_content hooks.
 
225
 
 
226
    There are some fields hooks can access:
 
227
 
 
228
    :ivar file_id: the file ID of the file being merged
 
229
    :ivar trans_id: the transform ID for the merge of this file
 
230
    :ivar this_kind: kind of file_id in 'this' tree
 
231
    :ivar other_kind: kind of file_id in 'other' tree
 
232
    :ivar winner: one of 'this', 'other', 'conflict'
 
233
    """
 
234
 
 
235
    def __init__(self, merger, file_id, trans_id, this_kind, other_kind,
 
236
            winner):
 
237
        self._merger = merger
 
238
        self.file_id = file_id
 
239
        self.trans_id = trans_id
 
240
        self.this_kind = this_kind
 
241
        self.other_kind = other_kind
 
242
        self.winner = winner
 
243
 
 
244
    def is_file_merge(self):
 
245
        """True if this_kind and other_kind are both 'file'."""
 
246
        return self.this_kind == 'file' and self.other_kind == 'file'
 
247
 
 
248
    @decorators.cachedproperty
 
249
    def base_lines(self):
 
250
        """The lines of the 'base' version of the file."""
 
251
        return self._merger.get_lines(self._merger.base_tree, self.file_id)
 
252
 
 
253
    @decorators.cachedproperty
 
254
    def this_lines(self):
 
255
        """The lines of the 'this' version of the file."""
 
256
        return self._merger.get_lines(self._merger.this_tree, self.file_id)
 
257
 
 
258
    @decorators.cachedproperty
 
259
    def other_lines(self):
 
260
        """The lines of the 'other' version of the file."""
 
261
        return self._merger.get_lines(self._merger.other_tree, self.file_id)
68
262
 
69
263
 
70
264
class Merger(object):
 
265
 
 
266
    hooks = MergeHooks()
 
267
 
71
268
    def __init__(self, this_branch, other_tree=None, base_tree=None,
72
 
                 this_tree=None, pb=DummyProgress(), change_reporter=None,
 
269
                 this_tree=None, pb=None, change_reporter=None,
73
270
                 recurse='down', revision_graph=None):
74
271
        object.__init__(self)
75
272
        self.this_branch = this_branch
88
285
        self.interesting_files = None
89
286
        self.show_base = False
90
287
        self.reprocess = False
91
 
        self._pb = pb
 
288
        if pb is not None:
 
289
            warnings.warn("pb parameter to Merger() is deprecated and ignored")
92
290
        self.pp = None
93
291
        self.recurse = recurse
94
292
        self.change_reporter = change_reporter
96
294
        self._revision_graph = revision_graph
97
295
        self._base_is_ancestor = None
98
296
        self._base_is_other_ancestor = None
 
297
        self._is_criss_cross = None
 
298
        self._lca_trees = None
 
299
 
 
300
    def cache_trees_with_revision_ids(self, trees):
 
301
        """Cache any tree in trees if it has a revision_id."""
 
302
        for maybe_tree in trees:
 
303
            if maybe_tree is None:
 
304
                continue
 
305
            try:
 
306
                rev_id = maybe_tree.get_revision_id()
 
307
            except AttributeError:
 
308
                continue
 
309
            self._cached_trees[rev_id] = maybe_tree
99
310
 
100
311
    @property
101
312
    def revision_graph(self):
129
340
                                      _set_base_is_other_ancestor)
130
341
 
131
342
    @staticmethod
132
 
    def from_uncommitted(tree, other_tree, pb):
 
343
    def from_uncommitted(tree, other_tree, pb=None, base_tree=None):
133
344
        """Return a Merger for uncommitted changes in other_tree.
134
345
 
135
346
        :param tree: The tree to merge into
136
347
        :param other_tree: The tree to get uncommitted changes from
137
348
        :param pb: A progress indicator
 
349
        :param base_tree: The basis to use for the merge.  If unspecified,
 
350
            other_tree.basis_tree() will be used.
138
351
        """
139
 
        merger = Merger(tree.branch, other_tree, other_tree.basis_tree(), tree,
140
 
                        pb)
 
352
        if base_tree is None:
 
353
            base_tree = other_tree.basis_tree()
 
354
        merger = Merger(tree.branch, other_tree, base_tree, tree, pb)
141
355
        merger.base_rev_id = merger.base_tree.get_revision_id()
142
356
        merger.other_rev_id = None
143
357
        merger.other_basis = merger.base_rev_id
161
375
                base_revision_id, tree.branch.last_revision())):
162
376
                base_revision_id = None
163
377
            else:
164
 
                warning('Performing cherrypick')
 
378
                trace.warning('Performing cherrypick')
165
379
        merger = klass.from_revision_ids(pb, tree, other_revision_id,
166
380
                                         base_revision_id, revision_graph=
167
381
                                         revision_graph)
169
383
 
170
384
    @staticmethod
171
385
    def from_revision_ids(pb, tree, other, base=None, other_branch=None,
172
 
                          base_branch=None, revision_graph=None):
 
386
                          base_branch=None, revision_graph=None,
 
387
                          tree_branch=None):
173
388
        """Return a Merger for revision-ids.
174
389
 
 
390
        :param pb: A progress indicator
175
391
        :param tree: The tree to merge changes into
176
392
        :param other: The revision-id to use as OTHER
177
393
        :param base: The revision-id to use as BASE.  If not specified, will
182
398
            not supplied, other_branch or tree.branch will be used.
183
399
        :param revision_graph: If you have a revision_graph precomputed, pass
184
400
            it in, otherwise it will be created for you.
185
 
        :param pb: A progress indicator
 
401
        :param tree_branch: The branch associated with tree.  If not supplied,
 
402
            tree.branch will be used.
186
403
        """
187
 
        merger = Merger(tree.branch, this_tree=tree, pb=pb,
 
404
        if tree_branch is None:
 
405
            tree_branch = tree.branch
 
406
        merger = Merger(tree_branch, this_tree=tree, pb=pb,
188
407
                        revision_graph=revision_graph)
189
408
        if other_branch is None:
190
409
            other_branch = tree.branch
209
428
        return self._cached_trees[revision_id]
210
429
 
211
430
    def _get_tree(self, treespec, possible_transports=None):
212
 
        from bzrlib import workingtree
213
431
        location, revno = treespec
214
432
        if revno is None:
215
433
            tree = workingtree.WorkingTree.open_containing(location)[0]
216
434
            return tree.branch, tree
217
 
        branch = Branch.open_containing(location, possible_transports)[0]
 
435
        branch = _mod_branch.Branch.open_containing(
 
436
            location, possible_transports)[0]
218
437
        if revno == -1:
219
438
            revision_id = branch.last_revision()
220
439
        else:
221
440
            revision_id = branch.get_rev_id(revno)
222
 
        revision_id = ensure_null(revision_id)
 
441
        revision_id = _mod_revision.ensure_null(revision_id)
223
442
        return branch, self.revision_tree(revision_id, branch)
224
443
 
 
444
    @deprecated_method(deprecated_in((2, 1, 0)))
225
445
    def ensure_revision_trees(self):
226
446
        if self.this_revision_tree is None:
227
447
            self.this_basis_tree = self.revision_tree(self.this_basis)
230
450
 
231
451
        if self.other_rev_id is None:
232
452
            other_basis_tree = self.revision_tree(self.other_basis)
233
 
            changes = other_basis_tree.changes_from(self.other_tree)
234
 
            if changes.has_changed():
235
 
                raise WorkingTreeNotRevision(self.this_tree)
 
453
            if other_basis_tree.has_changes(self.other_tree):
 
454
                raise errors.WorkingTreeNotRevision(self.this_tree)
236
455
            other_rev_id = self.other_basis
237
456
            self.other_tree = other_basis_tree
238
457
 
 
458
    @deprecated_method(deprecated_in((2, 1, 0)))
239
459
    def file_revisions(self, file_id):
240
460
        self.ensure_revision_trees()
241
 
        def get_id(tree, file_id):
242
 
            revision_id = tree.inventory[file_id].revision
243
 
            return revision_id
244
461
        if self.this_rev_id is None:
245
462
            if self.this_basis_tree.get_file_sha1(file_id) != \
246
463
                self.this_tree.get_file_sha1(file_id):
247
 
                raise WorkingTreeNotRevision(self.this_tree)
 
464
                raise errors.WorkingTreeNotRevision(self.this_tree)
248
465
 
249
466
        trees = (self.this_basis_tree, self.other_tree)
250
 
        return [get_id(tree, file_id) for tree in trees]
 
467
        return [tree.get_file_revision(file_id) for tree in trees]
251
468
 
 
469
    @deprecated_method(deprecated_in((2, 1, 0)))
252
470
    def check_basis(self, check_clean, require_commits=True):
253
471
        if self.this_basis is None and require_commits is True:
254
 
            raise BzrCommandError("This branch has no commits."
255
 
                                  " (perhaps you would prefer 'bzr pull')")
 
472
            raise errors.BzrCommandError(
 
473
                "This branch has no commits."
 
474
                " (perhaps you would prefer 'bzr pull')")
256
475
        if check_clean:
257
476
            self.compare_basis()
258
477
            if self.this_basis != self.this_rev_id:
259
478
                raise errors.UncommittedChanges(self.this_tree)
260
479
 
 
480
    @deprecated_method(deprecated_in((2, 1, 0)))
261
481
    def compare_basis(self):
262
482
        try:
263
483
            basis_tree = self.revision_tree(self.this_tree.last_revision())
264
484
        except errors.NoSuchRevision:
265
485
            basis_tree = self.this_tree.basis_tree()
266
 
        changes = self.this_tree.changes_from(basis_tree)
267
 
        if not changes.has_changed():
 
486
        if not self.this_tree.has_changes(basis_tree):
268
487
            self.this_rev_id = self.this_basis
269
488
 
270
489
    def set_interesting_files(self, file_list):
271
490
        self.interesting_files = file_list
272
491
 
273
492
    def set_pending(self):
274
 
        if not self.base_is_ancestor or not self.base_is_other_ancestor or self.other_rev_id is None:
 
493
        if (not self.base_is_ancestor or not self.base_is_other_ancestor
 
494
            or self.other_rev_id is None):
275
495
            return
276
496
        self._add_parent()
277
497
 
278
498
    def _add_parent(self):
279
499
        new_parents = self.this_tree.get_parent_ids() + [self.other_rev_id]
280
500
        new_parent_trees = []
 
501
        operation = cleanup.OperationWithCleanups(
 
502
            self.this_tree.set_parent_trees)
281
503
        for revision_id in new_parents:
282
504
            try:
283
505
                tree = self.revision_tree(revision_id)
285
507
                tree = None
286
508
            else:
287
509
                tree.lock_read()
 
510
                operation.add_cleanup(tree.unlock)
288
511
            new_parent_trees.append((revision_id, tree))
289
 
        try:
290
 
            self.this_tree.set_parent_trees(new_parent_trees,
291
 
                                            allow_leftmost_as_ghost=True)
292
 
        finally:
293
 
            for _revision_id, tree in new_parent_trees:
294
 
                if tree is not None:
295
 
                    tree.unlock()
 
512
        operation.run_simple(new_parent_trees, allow_leftmost_as_ghost=True)
296
513
 
297
514
    def set_other(self, other_revision, possible_transports=None):
298
515
        """Set the revision and tree to merge from.
307
524
            self.other_rev_id = _mod_revision.ensure_null(
308
525
                self.other_branch.last_revision())
309
526
            if _mod_revision.is_null(self.other_rev_id):
310
 
                raise NoCommits(self.other_branch)
 
527
                raise errors.NoCommits(self.other_branch)
311
528
            self.other_basis = self.other_rev_id
312
529
        elif other_revision[1] is not None:
313
530
            self.other_rev_id = self.other_branch.get_rev_id(other_revision[1])
316
533
            self.other_rev_id = None
317
534
            self.other_basis = self.other_branch.last_revision()
318
535
            if self.other_basis is None:
319
 
                raise NoCommits(self.other_branch)
 
536
                raise errors.NoCommits(self.other_branch)
320
537
        if self.other_rev_id is not None:
321
538
            self._cached_trees[self.other_rev_id] = self.other_tree
322
539
        self._maybe_fetch(self.other_branch,self.this_branch, self.other_basis)
349
566
            target.fetch(source, revision_id)
350
567
 
351
568
    def find_base(self):
352
 
        revisions = [ensure_null(self.this_basis),
353
 
                     ensure_null(self.other_basis)]
354
 
        if NULL_REVISION in revisions:
355
 
            self.base_rev_id = NULL_REVISION
 
569
        revisions = [_mod_revision.ensure_null(self.this_basis),
 
570
                     _mod_revision.ensure_null(self.other_basis)]
 
571
        if _mod_revision.NULL_REVISION in revisions:
 
572
            self.base_rev_id = _mod_revision.NULL_REVISION
 
573
            self.base_tree = self.revision_tree(self.base_rev_id)
 
574
            self._is_criss_cross = False
356
575
        else:
357
 
            self.base_rev_id, steps = self.revision_graph.find_unique_lca(
358
 
                revisions[0], revisions[1], count_steps=True)
359
 
            if self.base_rev_id == NULL_REVISION:
360
 
                raise UnrelatedBranches()
361
 
            if steps > 1:
362
 
                warning('Warning: criss-cross merge encountered.  See bzr'
363
 
                        ' help criss-cross.')
364
 
        self.base_tree = self.revision_tree(self.base_rev_id)
 
576
            lcas = self.revision_graph.find_lca(revisions[0], revisions[1])
 
577
            self._is_criss_cross = False
 
578
            if len(lcas) == 0:
 
579
                self.base_rev_id = _mod_revision.NULL_REVISION
 
580
            elif len(lcas) == 1:
 
581
                self.base_rev_id = list(lcas)[0]
 
582
            else: # len(lcas) > 1
 
583
                self._is_criss_cross = True
 
584
                if len(lcas) > 2:
 
585
                    # find_unique_lca can only handle 2 nodes, so we have to
 
586
                    # start back at the beginning. It is a shame to traverse
 
587
                    # the graph again, but better than re-implementing
 
588
                    # find_unique_lca.
 
589
                    self.base_rev_id = self.revision_graph.find_unique_lca(
 
590
                                            revisions[0], revisions[1])
 
591
                else:
 
592
                    self.base_rev_id = self.revision_graph.find_unique_lca(
 
593
                                            *lcas)
 
594
                sorted_lca_keys = self.revision_graph.find_merge_order(                
 
595
                    revisions[0], lcas)
 
596
                if self.base_rev_id == _mod_revision.NULL_REVISION:
 
597
                    self.base_rev_id = sorted_lca_keys[0]
 
598
                
 
599
            if self.base_rev_id == _mod_revision.NULL_REVISION:
 
600
                raise errors.UnrelatedBranches()
 
601
            if self._is_criss_cross:
 
602
                trace.warning('Warning: criss-cross merge encountered.  See bzr'
 
603
                              ' help criss-cross.')
 
604
                trace.mutter('Criss-cross lcas: %r' % lcas)
 
605
                if self.base_rev_id in lcas:
 
606
                    trace.mutter('Unable to find unique lca. '
 
607
                                 'Fallback %r as best option.' % self.base_rev_id)
 
608
                interesting_revision_ids = set(lcas)
 
609
                interesting_revision_ids.add(self.base_rev_id)
 
610
                interesting_trees = dict((t.get_revision_id(), t)
 
611
                    for t in self.this_branch.repository.revision_trees(
 
612
                        interesting_revision_ids))
 
613
                self._cached_trees.update(interesting_trees)
 
614
                if self.base_rev_id in lcas:
 
615
                    self.base_tree = interesting_trees[self.base_rev_id]
 
616
                else:
 
617
                    self.base_tree = interesting_trees.pop(self.base_rev_id)
 
618
                self._lca_trees = [interesting_trees[key]
 
619
                                   for key in sorted_lca_keys]
 
620
            else:
 
621
                self.base_tree = self.revision_tree(self.base_rev_id)
365
622
        self.base_is_ancestor = True
366
623
        self.base_is_other_ancestor = True
 
624
        trace.mutter('Base revid: %r' % self.base_rev_id)
367
625
 
368
626
    def set_base(self, base_revision):
369
627
        """Set the base revision to use for the merge.
370
628
 
371
629
        :param base_revision: A 2-list containing a path and revision number.
372
630
        """
373
 
        mutter("doing merge() with no base_revision specified")
 
631
        trace.mutter("doing merge() with no base_revision specified")
374
632
        if base_revision == [None, None]:
375
633
            self.find_base()
376
634
        else:
389
647
                  'other_tree': self.other_tree,
390
648
                  'interesting_ids': self.interesting_ids,
391
649
                  'interesting_files': self.interesting_files,
392
 
                  'pp': self.pp,
 
650
                  'this_branch': self.this_branch,
393
651
                  'do_merge': False}
394
652
        if self.merge_type.requires_base:
395
653
            kwargs['base_tree'] = self.base_tree
396
654
        if self.merge_type.supports_reprocess:
397
655
            kwargs['reprocess'] = self.reprocess
398
656
        elif self.reprocess:
399
 
            raise BzrError("Conflict reduction is not supported for merge"
400
 
                                  " type %s." % self.merge_type)
 
657
            raise errors.BzrError(
 
658
                "Conflict reduction is not supported for merge"
 
659
                " type %s." % self.merge_type)
401
660
        if self.merge_type.supports_show_base:
402
661
            kwargs['show_base'] = self.show_base
403
662
        elif self.show_base:
404
 
            raise BzrError("Showing base is not supported for this"
405
 
                           " merge type. %s" % self.merge_type)
 
663
            raise errors.BzrError("Showing base is not supported for this"
 
664
                                  " merge type. %s" % self.merge_type)
406
665
        if (not getattr(self.merge_type, 'supports_reverse_cherrypick', True)
407
666
            and not self.base_is_other_ancestor):
408
667
            raise errors.CannotReverseCherrypick()
409
668
        if self.merge_type.supports_cherrypick:
410
669
            kwargs['cherrypick'] = (not self.base_is_ancestor or
411
670
                                    not self.base_is_other_ancestor)
412
 
        return self.merge_type(pb=self._pb,
 
671
        if self._is_criss_cross and getattr(self.merge_type,
 
672
                                            'supports_lca_trees', False):
 
673
            kwargs['lca_trees'] = self._lca_trees
 
674
        return self.merge_type(pb=None,
413
675
                               change_reporter=self.change_reporter,
414
676
                               **kwargs)
415
677
 
416
 
    def _do_merge_to(self, merge):
 
678
    def _do_merge_to(self):
 
679
        merge = self.make_merger()
 
680
        if self.other_branch is not None:
 
681
            self.other_branch.update_references(self.this_branch)
417
682
        merge.do_merge()
418
683
        if self.recurse == 'down':
419
684
            for relpath, file_id in self.this_tree.iter_references():
431
696
                    sub_tree.branch.repository.revision_tree(base_revision)
432
697
                sub_merge.base_rev_id = base_revision
433
698
                sub_merge.do_merge()
434
 
        
 
699
        return merge
 
700
 
435
701
    def do_merge(self):
 
702
        operation = cleanup.OperationWithCleanups(self._do_merge_to)
436
703
        self.this_tree.lock_tree_write()
437
 
        try:
438
 
            if self.base_tree is not None:
439
 
                self.base_tree.lock_read()
440
 
            try:
441
 
                if self.other_tree is not None:
442
 
                    self.other_tree.lock_read()
443
 
                try:
444
 
                    merge = self.make_merger()
445
 
                    self._do_merge_to(merge)
446
 
                finally:
447
 
                    if self.other_tree is not None:
448
 
                        self.other_tree.unlock()
449
 
            finally:
450
 
                if self.base_tree is not None:
451
 
                    self.base_tree.unlock()
452
 
        finally:
453
 
            self.this_tree.unlock()
 
704
        operation.add_cleanup(self.this_tree.unlock)
 
705
        if self.base_tree is not None:
 
706
            self.base_tree.lock_read()
 
707
            operation.add_cleanup(self.base_tree.unlock)
 
708
        if self.other_tree is not None:
 
709
            self.other_tree.lock_read()
 
710
            operation.add_cleanup(self.other_tree.unlock)
 
711
        merge = operation.run_simple()
454
712
        if len(merge.cooked_conflicts) == 0:
455
 
            if not self.ignore_zero and not is_quiet():
456
 
                note("All changes applied successfully.")
 
713
            if not self.ignore_zero and not trace.is_quiet():
 
714
                trace.note("All changes applied successfully.")
457
715
        else:
458
 
            note("%d conflicts encountered." % len(merge.cooked_conflicts))
 
716
            trace.note("%d conflicts encountered."
 
717
                       % len(merge.cooked_conflicts))
459
718
 
460
719
        return len(merge.cooked_conflicts)
461
720
 
462
721
 
 
722
class _InventoryNoneEntry(object):
 
723
    """This represents an inventory entry which *isn't there*.
 
724
 
 
725
    It simplifies the merging logic if we always have an InventoryEntry, even
 
726
    if it isn't actually present
 
727
    """
 
728
    executable = None
 
729
    kind = None
 
730
    name = None
 
731
    parent_id = None
 
732
    revision = None
 
733
    symlink_target = None
 
734
    text_sha1 = None
 
735
 
 
736
_none_entry = _InventoryNoneEntry()
 
737
 
 
738
 
463
739
class Merge3Merger(object):
464
740
    """Three-way merger that uses the merge3 text merger"""
465
741
    requires_base = True
469
745
    supports_cherrypick = True
470
746
    supports_reverse_cherrypick = True
471
747
    winner_idx = {"this": 2, "other": 1, "conflict": 1}
 
748
    supports_lca_trees = True
472
749
 
473
 
    def __init__(self, working_tree, this_tree, base_tree, other_tree, 
 
750
    def __init__(self, working_tree, this_tree, base_tree, other_tree,
474
751
                 interesting_ids=None, reprocess=False, show_base=False,
475
 
                 pb=DummyProgress(), pp=None, change_reporter=None,
 
752
                 pb=None, pp=None, change_reporter=None,
476
753
                 interesting_files=None, do_merge=True,
477
 
                 cherrypick=False):
 
754
                 cherrypick=False, lca_trees=None, this_branch=None):
478
755
        """Initialize the merger object and perform the merge.
479
756
 
480
757
        :param working_tree: The working tree to apply the merge to
481
758
        :param this_tree: The local tree in the merge operation
482
759
        :param base_tree: The common tree in the merge operation
483
 
        :param other_tree: The other other tree to merge changes from
 
760
        :param other_tree: The other tree to merge changes from
 
761
        :param this_branch: The branch associated with this_tree.  Defaults to
 
762
            this_tree.branch if not supplied.
484
763
        :param interesting_ids: The file_ids of files that should be
485
764
            participate in the merge.  May not be combined with
486
765
            interesting_files.
487
766
        :param: reprocess If True, perform conflict-reduction processing.
488
767
        :param show_base: If True, show the base revision in text conflicts.
489
768
            (incompatible with reprocess)
490
 
        :param pb: A Progress bar
 
769
        :param pb: ignored
491
770
        :param pp: A ProgressPhase object
492
771
        :param change_reporter: An object that should report changes made
493
772
        :param interesting_files: The tree-relative paths of files that should
496
775
            be combined with interesting_ids.  If neither interesting_files nor
497
776
            interesting_ids is specified, all files may participate in the
498
777
            merge.
 
778
        :param lca_trees: Can be set to a dictionary of {revision_id:rev_tree}
 
779
            if the ancestry was found to include a criss-cross merge.
 
780
            Otherwise should be None.
499
781
        """
500
782
        object.__init__(self)
501
783
        if interesting_files is not None and interesting_ids is not None:
502
784
            raise ValueError(
503
785
                'specify either interesting_ids or interesting_files')
 
786
        if this_branch is None:
 
787
            this_branch = this_tree.branch
504
788
        self.interesting_ids = interesting_ids
505
789
        self.interesting_files = interesting_files
506
790
        self.this_tree = working_tree
507
791
        self.base_tree = base_tree
508
792
        self.other_tree = other_tree
 
793
        self.this_branch = this_branch
509
794
        self._raw_conflicts = []
510
795
        self.cooked_conflicts = []
511
796
        self.reprocess = reprocess
512
797
        self.show_base = show_base
513
 
        self.pb = pb
514
 
        self.pp = pp
 
798
        self._lca_trees = lca_trees
 
799
        # Uncommenting this will change the default algorithm to always use
 
800
        # _entries_lca. This can be useful for running the test suite and
 
801
        # making sure we haven't missed any corner cases.
 
802
        # if lca_trees is None:
 
803
        #     self._lca_trees = [self.base_tree]
515
804
        self.change_reporter = change_reporter
516
805
        self.cherrypick = cherrypick
517
 
        if self.pp is None:
518
 
            self.pp = ProgressPhase("Merge phase", 3, self.pb)
519
806
        if do_merge:
520
807
            self.do_merge()
 
808
        if pp is not None:
 
809
            warnings.warn("pp argument to Merge3Merger is deprecated")
 
810
        if pb is not None:
 
811
            warnings.warn("pb argument to Merge3Merger is deprecated")
521
812
 
522
813
    def do_merge(self):
 
814
        operation = cleanup.OperationWithCleanups(self._do_merge)
523
815
        self.this_tree.lock_tree_write()
 
816
        operation.add_cleanup(self.this_tree.unlock)
524
817
        self.base_tree.lock_read()
 
818
        operation.add_cleanup(self.base_tree.unlock)
525
819
        self.other_tree.lock_read()
526
 
        self.tt = TreeTransform(self.this_tree, self.pb)
 
820
        operation.add_cleanup(self.other_tree.unlock)
 
821
        operation.run()
 
822
 
 
823
    def _do_merge(self, operation):
 
824
        self.tt = transform.TreeTransform(self.this_tree, None)
 
825
        operation.add_cleanup(self.tt.finalize)
 
826
        self._compute_transform()
 
827
        results = self.tt.apply(no_conflicts=True)
 
828
        self.write_modified(results)
527
829
        try:
528
 
            self.pp.next_phase()
529
 
            self._compute_transform()
530
 
            self.pp.next_phase()
531
 
            results = self.tt.apply(no_conflicts=True)
532
 
            self.write_modified(results)
533
 
            try:
534
 
                self.this_tree.add_conflicts(self.cooked_conflicts)
535
 
            except UnsupportedOperation:
536
 
                pass
537
 
        finally:
538
 
            self.tt.finalize()
539
 
            self.other_tree.unlock()
540
 
            self.base_tree.unlock()
541
 
            self.this_tree.unlock()
542
 
            self.pb.clear()
 
830
            self.this_tree.add_conflicts(self.cooked_conflicts)
 
831
        except errors.UnsupportedOperation:
 
832
            pass
543
833
 
544
834
    def make_preview_transform(self):
 
835
        operation = cleanup.OperationWithCleanups(self._make_preview_transform)
545
836
        self.base_tree.lock_read()
 
837
        operation.add_cleanup(self.base_tree.unlock)
546
838
        self.other_tree.lock_read()
547
 
        self.tt = TransformPreview(self.this_tree)
548
 
        try:
549
 
            self.pp.next_phase()
550
 
            self._compute_transform()
551
 
            self.pp.next_phase()
552
 
        finally:
553
 
            self.other_tree.unlock()
554
 
            self.base_tree.unlock()
555
 
            self.pb.clear()
 
839
        operation.add_cleanup(self.other_tree.unlock)
 
840
        return operation.run_simple()
 
841
 
 
842
    def _make_preview_transform(self):
 
843
        self.tt = transform.TransformPreview(self.this_tree)
 
844
        self._compute_transform()
556
845
        return self.tt
557
846
 
558
847
    def _compute_transform(self):
559
 
        entries = self._entries3()
 
848
        if self._lca_trees is None:
 
849
            entries = self._entries3()
 
850
            resolver = self._three_way
 
851
        else:
 
852
            entries = self._entries_lca()
 
853
            resolver = self._lca_multi_way
560
854
        child_pb = ui.ui_factory.nested_progress_bar()
561
855
        try:
 
856
            factories = Merger.hooks['merge_file_content']
 
857
            hooks = [factory(self) for factory in factories] + [self]
 
858
            self.active_hooks = [hook for hook in hooks if hook is not None]
562
859
            for num, (file_id, changed, parents3, names3,
563
860
                      executable3) in enumerate(entries):
564
861
                child_pb.update('Preparing file merge', num, len(entries))
565
 
                self._merge_names(file_id, parents3, names3)
 
862
                self._merge_names(file_id, parents3, names3, resolver=resolver)
566
863
                if changed:
567
 
                    file_status = self.merge_contents(file_id)
 
864
                    file_status = self._do_merge_contents(file_id)
568
865
                else:
569
866
                    file_status = 'unmodified'
570
867
                self._merge_executable(file_id,
571
 
                    executable3, file_status)
 
868
                    executable3, file_status, resolver=resolver)
572
869
        finally:
573
870
            child_pb.finished()
574
 
        self.fix_root()
575
 
        self.pp.next_phase()
 
871
        self.tt.fixup_new_roots()
 
872
        self._finish_computing_transform()
 
873
 
 
874
    def _finish_computing_transform(self):
 
875
        """Finalize the transform and report the changes.
 
876
 
 
877
        This is the second half of _compute_transform.
 
878
        """
576
879
        child_pb = ui.ui_factory.nested_progress_bar()
577
880
        try:
578
 
            fs_conflicts = resolve_conflicts(self.tt, child_pb,
579
 
                lambda t, c: conflict_pass(t, c, self.other_tree))
 
881
            fs_conflicts = transform.resolve_conflicts(self.tt, child_pb,
 
882
                lambda t, c: transform.conflict_pass(t, c, self.other_tree))
580
883
        finally:
581
884
            child_pb.finished()
582
885
        if self.change_reporter is not None:
585
888
                self.tt.iter_changes(), self.change_reporter)
586
889
        self.cook_conflicts(fs_conflicts)
587
890
        for conflict in self.cooked_conflicts:
588
 
            warning(conflict)
 
891
            trace.warning(unicode(conflict))
589
892
 
590
893
    def _entries3(self):
591
894
        """Gather data about files modified between three trees.
598
901
        """
599
902
        result = []
600
903
        iterator = self.other_tree.iter_changes(self.base_tree,
601
 
                include_unchanged=True, specific_files=self.interesting_files,
 
904
                specific_files=self.interesting_files,
602
905
                extra_trees=[self.this_tree])
 
906
        this_entries = dict((e.file_id, e) for p, e in
 
907
                            self.this_tree.iter_entries_by_dir(
 
908
                            self.interesting_ids))
603
909
        for (file_id, paths, changed, versioned, parents, names, kind,
604
910
             executable) in iterator:
605
911
            if (self.interesting_ids is not None and
606
912
                file_id not in self.interesting_ids):
607
913
                continue
608
 
            if file_id in self.this_tree.inventory:
609
 
                entry = self.this_tree.inventory[file_id]
 
914
            entry = this_entries.get(file_id)
 
915
            if entry is not None:
610
916
                this_name = entry.name
611
917
                this_parent = entry.parent_id
612
918
                this_executable = entry.executable
620
926
            result.append((file_id, changed, parents3, names3, executable3))
621
927
        return result
622
928
 
 
929
    def _entries_lca(self):
 
930
        """Gather data about files modified between multiple trees.
 
931
 
 
932
        This compares OTHER versus all LCA trees, and for interesting entries,
 
933
        it then compares with THIS and BASE.
 
934
 
 
935
        For the multi-valued entries, the format will be (BASE, [lca1, lca2])
 
936
 
 
937
        :return: [(file_id, changed, parents, names, executable)], where:
 
938
 
 
939
            * file_id: Simple file_id of the entry
 
940
            * changed: Boolean, True if the kind or contents changed else False
 
941
            * parents: ((base, [parent_id, in, lcas]), parent_id_other,
 
942
                        parent_id_this)
 
943
            * names:   ((base, [name, in, lcas]), name_in_other, name_in_this)
 
944
            * executable: ((base, [exec, in, lcas]), exec_in_other,
 
945
                        exec_in_this)
 
946
        """
 
947
        if self.interesting_files is not None:
 
948
            lookup_trees = [self.this_tree, self.base_tree]
 
949
            lookup_trees.extend(self._lca_trees)
 
950
            # I think we should include the lca trees as well
 
951
            interesting_ids = self.other_tree.paths2ids(self.interesting_files,
 
952
                                                        lookup_trees)
 
953
        else:
 
954
            interesting_ids = self.interesting_ids
 
955
        result = []
 
956
        walker = _mod_tree.MultiWalker(self.other_tree, self._lca_trees)
 
957
 
 
958
        base_inventory = self.base_tree.inventory
 
959
        this_inventory = self.this_tree.inventory
 
960
        for path, file_id, other_ie, lca_values in walker.iter_all():
 
961
            # Is this modified at all from any of the other trees?
 
962
            if other_ie is None:
 
963
                other_ie = _none_entry
 
964
            if interesting_ids is not None and file_id not in interesting_ids:
 
965
                continue
 
966
 
 
967
            # If other_revision is found in any of the lcas, that means this
 
968
            # node is uninteresting. This is because when merging, if there are
 
969
            # multiple heads(), we have to create a new node. So if we didn't,
 
970
            # we know that the ancestry is linear, and that OTHER did not
 
971
            # modify anything
 
972
            # See doc/developers/lca_merge_resolution.txt for details
 
973
            other_revision = other_ie.revision
 
974
            if other_revision is not None:
 
975
                # We can't use this shortcut when other_revision is None,
 
976
                # because it may be None because things are WorkingTrees, and
 
977
                # not because it is *actually* None.
 
978
                is_unmodified = False
 
979
                for lca_path, ie in lca_values:
 
980
                    if ie is not None and ie.revision == other_revision:
 
981
                        is_unmodified = True
 
982
                        break
 
983
                if is_unmodified:
 
984
                    continue
 
985
 
 
986
            lca_entries = []
 
987
            for lca_path, lca_ie in lca_values:
 
988
                if lca_ie is None:
 
989
                    lca_entries.append(_none_entry)
 
990
                else:
 
991
                    lca_entries.append(lca_ie)
 
992
 
 
993
            if base_inventory.has_id(file_id):
 
994
                base_ie = base_inventory[file_id]
 
995
            else:
 
996
                base_ie = _none_entry
 
997
 
 
998
            if this_inventory.has_id(file_id):
 
999
                this_ie = this_inventory[file_id]
 
1000
            else:
 
1001
                this_ie = _none_entry
 
1002
 
 
1003
            lca_kinds = []
 
1004
            lca_parent_ids = []
 
1005
            lca_names = []
 
1006
            lca_executable = []
 
1007
            for lca_ie in lca_entries:
 
1008
                lca_kinds.append(lca_ie.kind)
 
1009
                lca_parent_ids.append(lca_ie.parent_id)
 
1010
                lca_names.append(lca_ie.name)
 
1011
                lca_executable.append(lca_ie.executable)
 
1012
 
 
1013
            kind_winner = self._lca_multi_way(
 
1014
                (base_ie.kind, lca_kinds),
 
1015
                other_ie.kind, this_ie.kind)
 
1016
            parent_id_winner = self._lca_multi_way(
 
1017
                (base_ie.parent_id, lca_parent_ids),
 
1018
                other_ie.parent_id, this_ie.parent_id)
 
1019
            name_winner = self._lca_multi_way(
 
1020
                (base_ie.name, lca_names),
 
1021
                other_ie.name, this_ie.name)
 
1022
 
 
1023
            content_changed = True
 
1024
            if kind_winner == 'this':
 
1025
                # No kind change in OTHER, see if there are *any* changes
 
1026
                if other_ie.kind == 'directory':
 
1027
                    if parent_id_winner == 'this' and name_winner == 'this':
 
1028
                        # No change for this directory in OTHER, skip
 
1029
                        continue
 
1030
                    content_changed = False
 
1031
                elif other_ie.kind is None or other_ie.kind == 'file':
 
1032
                    def get_sha1(ie, tree):
 
1033
                        if ie.kind != 'file':
 
1034
                            return None
 
1035
                        return tree.get_file_sha1(file_id)
 
1036
                    base_sha1 = get_sha1(base_ie, self.base_tree)
 
1037
                    lca_sha1s = [get_sha1(ie, tree) for ie, tree
 
1038
                                 in zip(lca_entries, self._lca_trees)]
 
1039
                    this_sha1 = get_sha1(this_ie, self.this_tree)
 
1040
                    other_sha1 = get_sha1(other_ie, self.other_tree)
 
1041
                    sha1_winner = self._lca_multi_way(
 
1042
                        (base_sha1, lca_sha1s), other_sha1, this_sha1,
 
1043
                        allow_overriding_lca=False)
 
1044
                    exec_winner = self._lca_multi_way(
 
1045
                        (base_ie.executable, lca_executable),
 
1046
                        other_ie.executable, this_ie.executable)
 
1047
                    if (parent_id_winner == 'this' and name_winner == 'this'
 
1048
                        and sha1_winner == 'this' and exec_winner == 'this'):
 
1049
                        # No kind, parent, name, exec, or content change for
 
1050
                        # OTHER, so this node is not considered interesting
 
1051
                        continue
 
1052
                    if sha1_winner == 'this':
 
1053
                        content_changed = False
 
1054
                elif other_ie.kind == 'symlink':
 
1055
                    def get_target(ie, tree):
 
1056
                        if ie.kind != 'symlink':
 
1057
                            return None
 
1058
                        return tree.get_symlink_target(file_id)
 
1059
                    base_target = get_target(base_ie, self.base_tree)
 
1060
                    lca_targets = [get_target(ie, tree) for ie, tree
 
1061
                                   in zip(lca_entries, self._lca_trees)]
 
1062
                    this_target = get_target(this_ie, self.this_tree)
 
1063
                    other_target = get_target(other_ie, self.other_tree)
 
1064
                    target_winner = self._lca_multi_way(
 
1065
                        (base_target, lca_targets),
 
1066
                        other_target, this_target)
 
1067
                    if (parent_id_winner == 'this' and name_winner == 'this'
 
1068
                        and target_winner == 'this'):
 
1069
                        # No kind, parent, name, or symlink target change
 
1070
                        # not interesting
 
1071
                        continue
 
1072
                    if target_winner == 'this':
 
1073
                        content_changed = False
 
1074
                elif other_ie.kind == 'tree-reference':
 
1075
                    # The 'changed' information seems to be handled at a higher
 
1076
                    # level. At least, _entries3 returns False for content
 
1077
                    # changed, even when at a new revision_id.
 
1078
                    content_changed = False
 
1079
                    if (parent_id_winner == 'this' and name_winner == 'this'):
 
1080
                        # Nothing interesting
 
1081
                        continue
 
1082
                else:
 
1083
                    raise AssertionError('unhandled kind: %s' % other_ie.kind)
 
1084
 
 
1085
            # If we have gotten this far, that means something has changed
 
1086
            result.append((file_id, content_changed,
 
1087
                           ((base_ie.parent_id, lca_parent_ids),
 
1088
                            other_ie.parent_id, this_ie.parent_id),
 
1089
                           ((base_ie.name, lca_names),
 
1090
                            other_ie.name, this_ie.name),
 
1091
                           ((base_ie.executable, lca_executable),
 
1092
                            other_ie.executable, this_ie.executable)
 
1093
                          ))
 
1094
        return result
 
1095
 
 
1096
    @deprecated_method(deprecated_in((2, 4, 0)))
623
1097
    def fix_root(self):
624
 
        try:
625
 
            self.tt.final_kind(self.tt.root)
626
 
        except NoSuchFile:
 
1098
        if self.tt.final_kind(self.tt.root) is None:
627
1099
            self.tt.cancel_deletion(self.tt.root)
628
1100
        if self.tt.final_file_id(self.tt.root) is None:
629
 
            self.tt.version_file(self.tt.tree_file_id(self.tt.root), 
 
1101
            self.tt.version_file(self.tt.tree_file_id(self.tt.root),
630
1102
                                 self.tt.root)
631
 
        if self.other_tree.inventory.root is None:
632
 
            return
633
1103
        other_root_file_id = self.other_tree.get_root_id()
 
1104
        if other_root_file_id is None:
 
1105
            return
634
1106
        other_root = self.tt.trans_id_file_id(other_root_file_id)
635
1107
        if other_root == self.tt.root:
636
1108
            return
637
 
        try:
638
 
            self.tt.final_kind(other_root)
639
 
        except NoSuchFile:
640
 
            return
641
 
        if self.other_tree.inventory.root.file_id in self.this_tree.inventory:
642
 
            # the other tree's root is a non-root in the current tree
643
 
            return
644
 
        self.reparent_children(self.other_tree.inventory.root, self.tt.root)
645
 
        self.tt.cancel_creation(other_root)
646
 
        self.tt.cancel_versioning(other_root)
647
 
 
648
 
    def reparent_children(self, ie, target):
649
 
        for thing, child in ie.children.iteritems():
 
1109
        if self.this_tree.inventory.has_id(
 
1110
            self.other_tree.inventory.root.file_id):
 
1111
            # the other tree's root is a non-root in the current tree (as
 
1112
            # when a previously unrelated branch is merged into another)
 
1113
            return
 
1114
        if self.tt.final_kind(other_root) is not None:
 
1115
            other_root_is_present = True
 
1116
        else:
 
1117
            # other_root doesn't have a physical representation. We still need
 
1118
            # to move any references to the actual root of the tree.
 
1119
            other_root_is_present = False
 
1120
        # 'other_tree.inventory.root' is not present in this tree. We are
 
1121
        # calling adjust_path for children which *want* to be present with a
 
1122
        # correct place to go.
 
1123
        for _, child in self.other_tree.inventory.root.children.iteritems():
650
1124
            trans_id = self.tt.trans_id_file_id(child.file_id)
651
 
            self.tt.adjust_path(self.tt.final_name(trans_id), target, trans_id)
 
1125
            if not other_root_is_present:
 
1126
                if self.tt.final_kind(trans_id) is not None:
 
1127
                    # The item exist in the final tree and has a defined place
 
1128
                    # to go already.
 
1129
                    continue
 
1130
            # Move the item into the root
 
1131
            try:
 
1132
                final_name = self.tt.final_name(trans_id)
 
1133
            except errors.NoFinalPath:
 
1134
                # This file is not present anymore, ignore it.
 
1135
                continue
 
1136
            self.tt.adjust_path(final_name, self.tt.root, trans_id)
 
1137
        if other_root_is_present:
 
1138
            self.tt.cancel_creation(other_root)
 
1139
            self.tt.cancel_versioning(other_root)
652
1140
 
653
1141
    def write_modified(self, results):
654
1142
        modified_hashes = {}
675
1163
        if entry is None:
676
1164
            return None
677
1165
        return entry.name
678
 
    
 
1166
 
679
1167
    @staticmethod
680
1168
    def contents_sha1(tree, file_id):
681
1169
        """Determine the sha1 of the file contents (used as a key method)."""
682
 
        if file_id not in tree:
 
1170
        if not tree.has_id(file_id):
683
1171
            return None
684
1172
        return tree.get_file_sha1(file_id)
685
1173
 
686
1174
    @staticmethod
687
1175
    def executable(tree, file_id):
688
1176
        """Determine the executability of a file-id (used as a key method)."""
689
 
        if file_id not in tree:
 
1177
        if not tree.has_id(file_id):
690
1178
            return None
691
1179
        if tree.kind(file_id) != "file":
692
1180
            return False
695
1183
    @staticmethod
696
1184
    def kind(tree, file_id):
697
1185
        """Determine the kind of a file-id (used as a key method)."""
698
 
        if file_id not in tree:
 
1186
        if not tree.has_id(file_id):
699
1187
            return None
700
1188
        return tree.kind(file_id)
701
1189
 
702
1190
    @staticmethod
703
1191
    def _three_way(base, other, this):
704
 
        #if base == other, either they all agree, or only THIS has changed.
705
1192
        if base == other:
 
1193
            # if 'base == other', either they all agree, or only 'this' has
 
1194
            # changed.
706
1195
            return 'this'
707
1196
        elif this not in (base, other):
 
1197
            # 'this' is neither 'base' nor 'other', so both sides changed
708
1198
            return 'conflict'
709
 
        # "Ambiguous clean merge" -- both sides have made the same change.
710
1199
        elif this == other:
 
1200
            # "Ambiguous clean merge" -- both sides have made the same change.
711
1201
            return "this"
712
 
        # this == base: only other has changed.
713
1202
        else:
 
1203
            # this == base: only other has changed.
714
1204
            return "other"
715
1205
 
716
1206
    @staticmethod
 
1207
    def _lca_multi_way(bases, other, this, allow_overriding_lca=True):
 
1208
        """Consider LCAs when determining whether a change has occurred.
 
1209
 
 
1210
        If LCAS are all identical, this is the same as a _three_way comparison.
 
1211
 
 
1212
        :param bases: value in (BASE, [LCAS])
 
1213
        :param other: value in OTHER
 
1214
        :param this: value in THIS
 
1215
        :param allow_overriding_lca: If there is more than one unique lca
 
1216
            value, allow OTHER to override THIS if it has a new value, and
 
1217
            THIS only has an lca value, or vice versa. This is appropriate for
 
1218
            truly scalar values, not as much for non-scalars.
 
1219
        :return: 'this', 'other', or 'conflict' depending on whether an entry
 
1220
            changed or not.
 
1221
        """
 
1222
        # See doc/developers/lca_tree_merging.txt for details about this
 
1223
        # algorithm.
 
1224
        if other == this:
 
1225
            # Either Ambiguously clean, or nothing was actually changed. We
 
1226
            # don't really care
 
1227
            return 'this'
 
1228
        base_val, lca_vals = bases
 
1229
        # Remove 'base_val' from the lca_vals, because it is not interesting
 
1230
        filtered_lca_vals = [lca_val for lca_val in lca_vals
 
1231
                                      if lca_val != base_val]
 
1232
        if len(filtered_lca_vals) == 0:
 
1233
            return Merge3Merger._three_way(base_val, other, this)
 
1234
 
 
1235
        unique_lca_vals = set(filtered_lca_vals)
 
1236
        if len(unique_lca_vals) == 1:
 
1237
            return Merge3Merger._three_way(unique_lca_vals.pop(), other, this)
 
1238
 
 
1239
        if allow_overriding_lca:
 
1240
            if other in unique_lca_vals:
 
1241
                if this in unique_lca_vals:
 
1242
                    # Each side picked a different lca, conflict
 
1243
                    return 'conflict'
 
1244
                else:
 
1245
                    # This has a value which supersedes both lca values, and
 
1246
                    # other only has an lca value
 
1247
                    return 'this'
 
1248
            elif this in unique_lca_vals:
 
1249
                # OTHER has a value which supersedes both lca values, and this
 
1250
                # only has an lca value
 
1251
                return 'other'
 
1252
 
 
1253
        # At this point, the lcas disagree, and the tip disagree
 
1254
        return 'conflict'
 
1255
 
 
1256
    @staticmethod
 
1257
    @deprecated_method(deprecated_in((2, 2, 0)))
717
1258
    def scalar_three_way(this_tree, base_tree, other_tree, file_id, key):
718
1259
        """Do a three-way test on a scalar.
719
1260
        Return "this", "other" or "conflict", depending whether a value wins.
734
1275
 
735
1276
    def merge_names(self, file_id):
736
1277
        def get_entry(tree):
737
 
            if file_id in tree.inventory:
 
1278
            if tree.has_id(file_id):
738
1279
                return tree.inventory[file_id]
739
1280
            else:
740
1281
                return None
751
1292
            else:
752
1293
                names.append(entry.name)
753
1294
                parents.append(entry.parent_id)
754
 
        return self._merge_names(file_id, parents, names)
 
1295
        return self._merge_names(file_id, parents, names,
 
1296
                                 resolver=self._three_way)
755
1297
 
756
 
    def _merge_names(self, file_id, parents, names):
 
1298
    def _merge_names(self, file_id, parents, names, resolver):
757
1299
        """Perform a merge on file_id names and parents"""
758
1300
        base_name, other_name, this_name = names
759
1301
        base_parent, other_parent, this_parent = parents
760
1302
 
761
 
        name_winner = self._three_way(*names)
 
1303
        name_winner = resolver(*names)
762
1304
 
763
 
        parent_id_winner = self._three_way(*parents)
 
1305
        parent_id_winner = resolver(*parents)
764
1306
        if this_name is None:
765
1307
            if name_winner == "this":
766
1308
                name_winner = "other"
768
1310
                parent_id_winner = "other"
769
1311
        if name_winner == "this" and parent_id_winner == "this":
770
1312
            return
771
 
        if name_winner == "conflict":
772
 
            trans_id = self.tt.trans_id_file_id(file_id)
773
 
            self._raw_conflicts.append(('name conflict', trans_id, 
774
 
                                        this_name, other_name))
775
 
        if parent_id_winner == "conflict":
776
 
            trans_id = self.tt.trans_id_file_id(file_id)
777
 
            self._raw_conflicts.append(('parent conflict', trans_id, 
778
 
                                        this_parent, other_parent))
779
 
        if other_name is None:
780
 
            # it doesn't matter whether the result was 'other' or 
781
 
            # 'conflict'-- if there's no 'other', we leave it alone.
 
1313
        if name_winner == 'conflict' or parent_id_winner == 'conflict':
 
1314
            # Creating helpers (.OTHER or .THIS) here cause problems down the
 
1315
            # road if a ContentConflict needs to be created so we should not do
 
1316
            # that
 
1317
            trans_id = self.tt.trans_id_file_id(file_id)
 
1318
            self._raw_conflicts.append(('path conflict', trans_id, file_id,
 
1319
                                        this_parent, this_name,
 
1320
                                        other_parent, other_name))
 
1321
        if not self.other_tree.has_id(file_id):
 
1322
            # it doesn't matter whether the result was 'other' or
 
1323
            # 'conflict'-- if it has no file id, we leave it alone.
782
1324
            return
783
 
        # if we get here, name_winner and parent_winner are set to safe values.
784
 
        trans_id = self.tt.trans_id_file_id(file_id)
785
1325
        parent_id = parents[self.winner_idx[parent_id_winner]]
786
 
        if parent_id is not None:
787
 
            parent_trans_id = self.tt.trans_id_file_id(parent_id)
788
 
            self.tt.adjust_path(names[self.winner_idx[name_winner]],
789
 
                                parent_trans_id, trans_id)
 
1326
        name = names[self.winner_idx[name_winner]]
 
1327
        if parent_id is not None or name is not None:
 
1328
            # if we get here, name_winner and parent_winner are set to safe
 
1329
            # values.
 
1330
            if parent_id is None and name is not None:
 
1331
                # if parent_id is None and name is non-None, current file is
 
1332
                # the tree root.
 
1333
                if names[self.winner_idx[parent_id_winner]] != '':
 
1334
                    raise AssertionError(
 
1335
                        'File looks like a root, but named %s' %
 
1336
                        names[self.winner_idx[parent_id_winner]])
 
1337
                parent_trans_id = transform.ROOT_PARENT
 
1338
            else:
 
1339
                parent_trans_id = self.tt.trans_id_file_id(parent_id)
 
1340
            self.tt.adjust_path(name, parent_trans_id,
 
1341
                                self.tt.trans_id_file_id(file_id))
790
1342
 
791
 
    def merge_contents(self, file_id):
792
 
        """Performa a merge on file_id contents."""
 
1343
    def _do_merge_contents(self, file_id):
 
1344
        """Performs a merge on file_id contents."""
793
1345
        def contents_pair(tree):
794
 
            if file_id not in tree:
 
1346
            if not tree.has_id(file_id):
795
1347
                return (None, None)
796
1348
            kind = tree.kind(file_id)
797
1349
            if kind == "file":
802
1354
                contents = None
803
1355
            return kind, contents
804
1356
 
805
 
        def contents_conflict():
806
 
            trans_id = self.tt.trans_id_file_id(file_id)
807
 
            name = self.tt.final_name(trans_id)
808
 
            parent_id = self.tt.final_parent(trans_id)
809
 
            if file_id in self.this_tree.inventory:
810
 
                self.tt.unversion_file(trans_id)
811
 
                if file_id in self.this_tree:
812
 
                    self.tt.delete_contents(trans_id)
813
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
814
 
                                              set_version=True)
815
 
            self._raw_conflicts.append(('contents conflict', file_group))
816
 
 
817
1357
        # See SPOT run.  run, SPOT, run.
818
1358
        # So we're not QUITE repeating ourselves; we do tricky things with
819
1359
        # file kind...
820
1360
        base_pair = contents_pair(self.base_tree)
821
1361
        other_pair = contents_pair(self.other_tree)
822
 
        if base_pair == other_pair:
823
 
            # OTHER introduced no changes
824
 
            return "unmodified"
825
 
        this_pair = contents_pair(self.this_tree)
826
 
        if this_pair == other_pair:
827
 
            # THIS and OTHER introduced the same changes
828
 
            return "unmodified"
829
 
        else:
830
 
            trans_id = self.tt.trans_id_file_id(file_id)
831
 
            if this_pair == base_pair:
832
 
                # only OTHER introduced changes
833
 
                if file_id in self.this_tree:
834
 
                    # Remove any existing contents
835
 
                    self.tt.delete_contents(trans_id)
836
 
                if file_id in self.other_tree:
837
 
                    # OTHER changed the file
838
 
                    create_by_entry(self.tt, 
839
 
                                    self.other_tree.inventory[file_id], 
840
 
                                    self.other_tree, trans_id)
841
 
                    if file_id not in self.this_tree.inventory:
842
 
                        self.tt.version_file(file_id, trans_id)
843
 
                    return "modified"
844
 
                elif file_id in self.this_tree.inventory:
845
 
                    # OTHER deleted the file
846
 
                    self.tt.unversion_file(trans_id)
847
 
                    return "deleted"
848
 
            #BOTH THIS and OTHER introduced changes; scalar conflict
849
 
            elif this_pair[0] == "file" and other_pair[0] == "file":
850
 
                # THIS and OTHER are both files, so text merge.  Either
851
 
                # BASE is a file, or both converted to files, so at least we
852
 
                # have agreement that output should be a file.
853
 
                try:
854
 
                    self.text_merge(file_id, trans_id)
855
 
                except BinaryFile:
856
 
                    return contents_conflict()
857
 
                if file_id not in self.this_tree.inventory:
858
 
                    self.tt.version_file(file_id, trans_id)
859
 
                try:
860
 
                    self.tt.tree_kind(trans_id)
861
 
                    self.tt.delete_contents(trans_id)
862
 
                except NoSuchFile:
863
 
                    pass
864
 
                return "modified"
865
 
            else:
866
 
                # Scalar conflict, can't text merge.  Dump conflicts
867
 
                return contents_conflict()
 
1362
        if self._lca_trees:
 
1363
            this_pair = contents_pair(self.this_tree)
 
1364
            lca_pairs = [contents_pair(tree) for tree in self._lca_trees]
 
1365
            winner = self._lca_multi_way((base_pair, lca_pairs), other_pair,
 
1366
                                         this_pair, allow_overriding_lca=False)
 
1367
        else:
 
1368
            if base_pair == other_pair:
 
1369
                winner = 'this'
 
1370
            else:
 
1371
                # We delayed evaluating this_pair as long as we can to avoid
 
1372
                # unnecessary sha1 calculation
 
1373
                this_pair = contents_pair(self.this_tree)
 
1374
                winner = self._three_way(base_pair, other_pair, this_pair)
 
1375
        if winner == 'this':
 
1376
            # No interesting changes introduced by OTHER
 
1377
            return "unmodified"
 
1378
        # We have a hypothetical conflict, but if we have files, then we
 
1379
        # can try to merge the content
 
1380
        trans_id = self.tt.trans_id_file_id(file_id)
 
1381
        params = MergeHookParams(self, file_id, trans_id, this_pair[0],
 
1382
            other_pair[0], winner)
 
1383
        hooks = self.active_hooks
 
1384
        hook_status = 'not_applicable'
 
1385
        for hook in hooks:
 
1386
            hook_status, lines = hook.merge_contents(params)
 
1387
            if hook_status != 'not_applicable':
 
1388
                # Don't try any more hooks, this one applies.
 
1389
                break
 
1390
        result = "modified"
 
1391
        if hook_status == 'not_applicable':
 
1392
            # This is a contents conflict, because none of the available
 
1393
            # functions could merge it.
 
1394
            result = None
 
1395
            name = self.tt.final_name(trans_id)
 
1396
            parent_id = self.tt.final_parent(trans_id)
 
1397
            if self.this_tree.has_id(file_id):
 
1398
                self.tt.unversion_file(trans_id)
 
1399
            file_group = self._dump_conflicts(name, parent_id, file_id,
 
1400
                                              set_version=True)
 
1401
            self._raw_conflicts.append(('contents conflict', file_group))
 
1402
        elif hook_status == 'success':
 
1403
            self.tt.create_file(lines, trans_id)
 
1404
        elif hook_status == 'conflicted':
 
1405
            # XXX: perhaps the hook should be able to provide
 
1406
            # the BASE/THIS/OTHER files?
 
1407
            self.tt.create_file(lines, trans_id)
 
1408
            self._raw_conflicts.append(('text conflict', trans_id))
 
1409
            name = self.tt.final_name(trans_id)
 
1410
            parent_id = self.tt.final_parent(trans_id)
 
1411
            self._dump_conflicts(name, parent_id, file_id)
 
1412
        elif hook_status == 'delete':
 
1413
            self.tt.unversion_file(trans_id)
 
1414
            result = "deleted"
 
1415
        elif hook_status == 'done':
 
1416
            # The hook function did whatever it needs to do directly, no
 
1417
            # further action needed here.
 
1418
            pass
 
1419
        else:
 
1420
            raise AssertionError('unknown hook_status: %r' % (hook_status,))
 
1421
        if not self.this_tree.has_id(file_id) and result == "modified":
 
1422
            self.tt.version_file(file_id, trans_id)
 
1423
        # The merge has been performed, so the old contents should not be
 
1424
        # retained.
 
1425
        self.tt.delete_contents(trans_id)
 
1426
        return result
 
1427
 
 
1428
    def _default_other_winner_merge(self, merge_hook_params):
 
1429
        """Replace this contents with other."""
 
1430
        file_id = merge_hook_params.file_id
 
1431
        trans_id = merge_hook_params.trans_id
 
1432
        file_in_this = self.this_tree.has_id(file_id)
 
1433
        if self.other_tree.has_id(file_id):
 
1434
            # OTHER changed the file
 
1435
            wt = self.this_tree
 
1436
            if wt.supports_content_filtering():
 
1437
                # We get the path from the working tree if it exists.
 
1438
                # That fails though when OTHER is adding a file, so
 
1439
                # we fall back to the other tree to find the path if
 
1440
                # it doesn't exist locally.
 
1441
                try:
 
1442
                    filter_tree_path = wt.id2path(file_id)
 
1443
                except errors.NoSuchId:
 
1444
                    filter_tree_path = self.other_tree.id2path(file_id)
 
1445
            else:
 
1446
                # Skip the id2path lookup for older formats
 
1447
                filter_tree_path = None
 
1448
            transform.create_from_tree(self.tt, trans_id,
 
1449
                             self.other_tree, file_id,
 
1450
                             filter_tree_path=filter_tree_path)
 
1451
            return 'done', None
 
1452
        elif file_in_this:
 
1453
            # OTHER deleted the file
 
1454
            return 'delete', None
 
1455
        else:
 
1456
            raise AssertionError(
 
1457
                'winner is OTHER, but file_id %r not in THIS or OTHER tree'
 
1458
                % (file_id,))
 
1459
 
 
1460
    def merge_contents(self, merge_hook_params):
 
1461
        """Fallback merge logic after user installed hooks."""
 
1462
        # This function is used in merge hooks as the fallback instance.
 
1463
        # Perhaps making this function and the functions it calls be a 
 
1464
        # a separate class would be better.
 
1465
        if merge_hook_params.winner == 'other':
 
1466
            # OTHER is a straight winner, so replace this contents with other
 
1467
            return self._default_other_winner_merge(merge_hook_params)
 
1468
        elif merge_hook_params.is_file_merge():
 
1469
            # THIS and OTHER are both files, so text merge.  Either
 
1470
            # BASE is a file, or both converted to files, so at least we
 
1471
            # have agreement that output should be a file.
 
1472
            try:
 
1473
                self.text_merge(merge_hook_params.file_id,
 
1474
                    merge_hook_params.trans_id)
 
1475
            except errors.BinaryFile:
 
1476
                return 'not_applicable', None
 
1477
            return 'done', None
 
1478
        else:
 
1479
            return 'not_applicable', None
868
1480
 
869
1481
    def get_lines(self, tree, file_id):
870
1482
        """Return the lines in a file, or an empty list."""
871
 
        if file_id in tree:
872
 
            return tree.get_file(file_id).readlines()
 
1483
        if tree.has_id(file_id):
 
1484
            return tree.get_file_lines(file_id)
873
1485
        else:
874
1486
            return []
875
1487
 
877
1489
        """Perform a three-way text merge on a file_id"""
878
1490
        # it's possible that we got here with base as a different type.
879
1491
        # if so, we just want two-way text conflicts.
880
 
        if file_id in self.base_tree and \
 
1492
        if self.base_tree.has_id(file_id) and \
881
1493
            self.base_tree.kind(file_id) == "file":
882
1494
            base_lines = self.get_lines(self.base_tree, file_id)
883
1495
        else:
884
1496
            base_lines = []
885
1497
        other_lines = self.get_lines(self.other_tree, file_id)
886
1498
        this_lines = self.get_lines(self.this_tree, file_id)
887
 
        m3 = Merge3(base_lines, this_lines, other_lines,
888
 
                    is_cherrypick=self.cherrypick)
 
1499
        m3 = merge3.Merge3(base_lines, this_lines, other_lines,
 
1500
                           is_cherrypick=self.cherrypick)
889
1501
        start_marker = "!START OF MERGE CONFLICT!" + "I HOPE THIS IS UNIQUE"
890
1502
        if self.show_base is True:
891
1503
            base_marker = '|' * 7
894
1506
 
895
1507
        def iter_merge3(retval):
896
1508
            retval["text_conflicts"] = False
897
 
            for line in m3.merge_lines(name_a = "TREE", 
898
 
                                       name_b = "MERGE-SOURCE", 
 
1509
            for line in m3.merge_lines(name_a = "TREE",
 
1510
                                       name_b = "MERGE-SOURCE",
899
1511
                                       name_base = "BASE-REVISION",
900
 
                                       start_marker=start_marker, 
 
1512
                                       start_marker=start_marker,
901
1513
                                       base_marker=base_marker,
902
1514
                                       reprocess=self.reprocess):
903
1515
                if line.startswith(start_marker):
912
1524
            self._raw_conflicts.append(('text conflict', trans_id))
913
1525
            name = self.tt.final_name(trans_id)
914
1526
            parent_id = self.tt.final_parent(trans_id)
915
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1527
            file_group = self._dump_conflicts(name, parent_id, file_id,
916
1528
                                              this_lines, base_lines,
917
1529
                                              other_lines)
918
1530
            file_group.append(trans_id)
919
1531
 
920
 
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None, 
 
1532
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None,
921
1533
                        base_lines=None, other_lines=None, set_version=False,
922
1534
                        no_base=False):
923
1535
        """Emit conflict files.
925
1537
        determined automatically.  If set_version is true, the .OTHER, .THIS
926
1538
        or .BASE (in that order) will be created as versioned files.
927
1539
        """
928
 
        data = [('OTHER', self.other_tree, other_lines), 
 
1540
        data = [('OTHER', self.other_tree, other_lines),
929
1541
                ('THIS', self.this_tree, this_lines)]
930
1542
        if not no_base:
931
1543
            data.append(('BASE', self.base_tree, base_lines))
 
1544
 
 
1545
        # We need to use the actual path in the working tree of the file here,
 
1546
        # ignoring the conflict suffixes
 
1547
        wt = self.this_tree
 
1548
        if wt.supports_content_filtering():
 
1549
            try:
 
1550
                filter_tree_path = wt.id2path(file_id)
 
1551
            except errors.NoSuchId:
 
1552
                # file has been deleted
 
1553
                filter_tree_path = None
 
1554
        else:
 
1555
            # Skip the id2path lookup for older formats
 
1556
            filter_tree_path = None
 
1557
 
932
1558
        versioned = False
933
1559
        file_group = []
934
1560
        for suffix, tree, lines in data:
935
 
            if file_id in tree:
 
1561
            if tree.has_id(file_id):
936
1562
                trans_id = self._conflict_file(name, parent_id, tree, file_id,
937
 
                                               suffix, lines)
 
1563
                                               suffix, lines, filter_tree_path)
938
1564
                file_group.append(trans_id)
939
1565
                if set_version and not versioned:
940
1566
                    self.tt.version_file(file_id, trans_id)
941
1567
                    versioned = True
942
1568
        return file_group
943
 
           
944
 
    def _conflict_file(self, name, parent_id, tree, file_id, suffix, 
945
 
                       lines=None):
 
1569
 
 
1570
    def _conflict_file(self, name, parent_id, tree, file_id, suffix,
 
1571
                       lines=None, filter_tree_path=None):
946
1572
        """Emit a single conflict file."""
947
1573
        name = name + '.' + suffix
948
1574
        trans_id = self.tt.create_path(name, parent_id)
949
 
        entry = tree.inventory[file_id]
950
 
        create_by_entry(self.tt, entry, tree, trans_id, lines)
 
1575
        transform.create_from_tree(self.tt, trans_id, tree, file_id, lines,
 
1576
            filter_tree_path)
951
1577
        return trans_id
952
1578
 
953
1579
    def merge_executable(self, file_id, file_status):
954
1580
        """Perform a merge on the execute bit."""
955
1581
        executable = [self.executable(t, file_id) for t in (self.base_tree,
956
1582
                      self.other_tree, self.this_tree)]
957
 
        self._merge_executable(file_id, executable, file_status)
 
1583
        self._merge_executable(file_id, executable, file_status,
 
1584
                               resolver=self._three_way)
958
1585
 
959
 
    def _merge_executable(self, file_id, executable, file_status):
 
1586
    def _merge_executable(self, file_id, executable, file_status,
 
1587
                          resolver):
960
1588
        """Perform a merge on the execute bit."""
961
1589
        base_executable, other_executable, this_executable = executable
962
1590
        if file_status == "deleted":
963
1591
            return
964
 
        winner = self._three_way(*executable)
 
1592
        winner = resolver(*executable)
965
1593
        if winner == "conflict":
966
1594
        # There must be a None in here, if we have a conflict, but we
967
1595
        # need executability since file status was not deleted.
972
1600
        if winner == 'this' and file_status != "modified":
973
1601
            return
974
1602
        trans_id = self.tt.trans_id_file_id(file_id)
975
 
        try:
976
 
            if self.tt.final_kind(trans_id) != "file":
977
 
                return
978
 
        except NoSuchFile:
 
1603
        if self.tt.final_kind(trans_id) != "file":
979
1604
            return
980
1605
        if winner == "this":
981
1606
            executability = this_executable
982
1607
        else:
983
 
            if file_id in self.other_tree:
 
1608
            if self.other_tree.has_id(file_id):
984
1609
                executability = other_executable
985
 
            elif file_id in self.this_tree:
 
1610
            elif self.this_tree.has_id(file_id):
986
1611
                executability = this_executable
987
 
            elif file_id in self.base_tree:
 
1612
            elif self.base_tree_has_id(file_id):
988
1613
                executability = base_executable
989
1614
        if executability is not None:
990
1615
            trans_id = self.tt.trans_id_file_id(file_id)
992
1617
 
993
1618
    def cook_conflicts(self, fs_conflicts):
994
1619
        """Convert all conflicts into a form that doesn't depend on trans_id"""
995
 
        from conflicts import Conflict
996
 
        name_conflicts = {}
997
 
        self.cooked_conflicts.extend(cook_conflicts(fs_conflicts, self.tt))
998
 
        fp = FinalPaths(self.tt)
 
1620
        content_conflict_file_ids = set()
 
1621
        cooked_conflicts = transform.cook_conflicts(fs_conflicts, self.tt)
 
1622
        fp = transform.FinalPaths(self.tt)
999
1623
        for conflict in self._raw_conflicts:
1000
1624
            conflict_type = conflict[0]
1001
 
            if conflict_type in ('name conflict', 'parent conflict'):
1002
 
                trans_id = conflict[1]
1003
 
                conflict_args = conflict[2:]
1004
 
                if trans_id not in name_conflicts:
1005
 
                    name_conflicts[trans_id] = {}
1006
 
                unique_add(name_conflicts[trans_id], conflict_type, 
1007
 
                           conflict_args)
1008
 
            if conflict_type == 'contents conflict':
 
1625
            if conflict_type == 'path conflict':
 
1626
                (trans_id, file_id,
 
1627
                this_parent, this_name,
 
1628
                other_parent, other_name) = conflict[1:]
 
1629
                if this_parent is None or this_name is None:
 
1630
                    this_path = '<deleted>'
 
1631
                else:
 
1632
                    parent_path =  fp.get_path(
 
1633
                        self.tt.trans_id_file_id(this_parent))
 
1634
                    this_path = osutils.pathjoin(parent_path, this_name)
 
1635
                if other_parent is None or other_name is None:
 
1636
                    other_path = '<deleted>'
 
1637
                else:
 
1638
                    if other_parent == self.other_tree.get_root_id():
 
1639
                        # The tree transform doesn't know about the other root,
 
1640
                        # so we special case here to avoid a NoFinalPath
 
1641
                        # exception
 
1642
                        parent_path = ''
 
1643
                    else:
 
1644
                        parent_path =  fp.get_path(
 
1645
                            self.tt.trans_id_file_id(other_parent))
 
1646
                    other_path = osutils.pathjoin(parent_path, other_name)
 
1647
                c = _mod_conflicts.Conflict.factory(
 
1648
                    'path conflict', path=this_path,
 
1649
                    conflict_path=other_path,
 
1650
                    file_id=file_id)
 
1651
            elif conflict_type == 'contents conflict':
1009
1652
                for trans_id in conflict[1]:
1010
1653
                    file_id = self.tt.final_file_id(trans_id)
1011
1654
                    if file_id is not None:
1015
1658
                    if path.endswith(suffix):
1016
1659
                        path = path[:-len(suffix)]
1017
1660
                        break
1018
 
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
1019
 
                self.cooked_conflicts.append(c)
1020
 
            if conflict_type == 'text conflict':
 
1661
                c = _mod_conflicts.Conflict.factory(conflict_type,
 
1662
                                                    path=path, file_id=file_id)
 
1663
                content_conflict_file_ids.add(file_id)
 
1664
            elif conflict_type == 'text conflict':
1021
1665
                trans_id = conflict[1]
1022
1666
                path = fp.get_path(trans_id)
1023
1667
                file_id = self.tt.final_file_id(trans_id)
1024
 
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
1025
 
                self.cooked_conflicts.append(c)
 
1668
                c = _mod_conflicts.Conflict.factory(conflict_type,
 
1669
                                                    path=path, file_id=file_id)
 
1670
            else:
 
1671
                raise AssertionError('bad conflict type: %r' % (conflict,))
 
1672
            cooked_conflicts.append(c)
1026
1673
 
1027
 
        for trans_id, conflicts in name_conflicts.iteritems():
1028
 
            try:
1029
 
                this_parent, other_parent = conflicts['parent conflict']
1030
 
                if this_parent == other_parent:
1031
 
                    raise AssertionError()
1032
 
            except KeyError:
1033
 
                this_parent = other_parent = \
1034
 
                    self.tt.final_file_id(self.tt.final_parent(trans_id))
1035
 
            try:
1036
 
                this_name, other_name = conflicts['name conflict']
1037
 
                if this_name == other_name:
1038
 
                    raise AssertionError()
1039
 
            except KeyError:
1040
 
                this_name = other_name = self.tt.final_name(trans_id)
1041
 
            other_path = fp.get_path(trans_id)
1042
 
            if this_parent is not None and this_name is not None:
1043
 
                this_parent_path = \
1044
 
                    fp.get_path(self.tt.trans_id_file_id(this_parent))
1045
 
                this_path = pathjoin(this_parent_path, this_name)
1046
 
            else:
1047
 
                this_path = "<deleted>"
1048
 
            file_id = self.tt.final_file_id(trans_id)
1049
 
            c = Conflict.factory('path conflict', path=this_path,
1050
 
                                 conflict_path=other_path, file_id=file_id)
 
1674
        self.cooked_conflicts = []
 
1675
        # We want to get rid of path conflicts when a corresponding contents
 
1676
        # conflict exists. This can occur when one branch deletes a file while
 
1677
        # the other renames *and* modifies it. In this case, the content
 
1678
        # conflict is enough.
 
1679
        for c in cooked_conflicts:
 
1680
            if (c.typestring == 'path conflict'
 
1681
                and c.file_id in content_conflict_file_ids):
 
1682
                continue
1051
1683
            self.cooked_conflicts.append(c)
1052
 
        self.cooked_conflicts.sort(key=Conflict.sort_key)
 
1684
        self.cooked_conflicts.sort(key=_mod_conflicts.Conflict.sort_key)
1053
1685
 
1054
1686
 
1055
1687
class WeaveMerger(Merge3Merger):
1059
1691
    supports_reverse_cherrypick = False
1060
1692
    history_based = True
1061
1693
 
1062
 
    def _merged_lines(self, file_id):
1063
 
        """Generate the merged lines.
1064
 
        There is no distinction between lines that are meant to contain <<<<<<<
1065
 
        and conflicts.
1066
 
        """
1067
 
        if self.cherrypick:
1068
 
            base = self.base_tree
1069
 
        else:
1070
 
            base = None
1071
 
        plan = self.this_tree.plan_file_merge(file_id, self.other_tree,
 
1694
    def _generate_merge_plan(self, file_id, base):
 
1695
        return self.this_tree.plan_file_merge(file_id, self.other_tree,
1072
1696
                                              base=base)
 
1697
 
 
1698
    def _merged_lines(self, file_id):
 
1699
        """Generate the merged lines.
 
1700
        There is no distinction between lines that are meant to contain <<<<<<<
 
1701
        and conflicts.
 
1702
        """
 
1703
        if self.cherrypick:
 
1704
            base = self.base_tree
 
1705
        else:
 
1706
            base = None
 
1707
        plan = self._generate_merge_plan(file_id, base)
1073
1708
        if 'merge' in debug.debug_flags:
1074
1709
            plan = list(plan)
1075
1710
            trans_id = self.tt.trans_id_file_id(file_id)
1076
1711
            name = self.tt.final_name(trans_id) + '.plan'
1077
 
            contents = ('%10s|%s' % l for l in plan)
 
1712
            contents = ('%11s|%s' % l for l in plan)
1078
1713
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
1079
 
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
1080
 
            '>>>>>>> MERGE-SOURCE\n')
1081
 
        return textmerge.merge_lines(self.reprocess)
 
1714
        textmerge = versionedfile.PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1715
                                                 '>>>>>>> MERGE-SOURCE\n')
 
1716
        lines, conflicts = textmerge.merge_lines(self.reprocess)
 
1717
        if conflicts:
 
1718
            base_lines = textmerge.base_from_plan()
 
1719
        else:
 
1720
            base_lines = None
 
1721
        return lines, base_lines
1082
1722
 
1083
1723
    def text_merge(self, file_id, trans_id):
1084
1724
        """Perform a (weave) text merge for a given file and file-id.
1085
1725
        If conflicts are encountered, .THIS and .OTHER files will be emitted,
1086
1726
        and a conflict will be noted.
1087
1727
        """
1088
 
        lines, conflicts = self._merged_lines(file_id)
 
1728
        lines, base_lines = self._merged_lines(file_id)
1089
1729
        lines = list(lines)
1090
 
        # Note we're checking whether the OUTPUT is binary in this case, 
 
1730
        # Note we're checking whether the OUTPUT is binary in this case,
1091
1731
        # because we don't want to get into weave merge guts.
1092
 
        check_text_lines(lines)
 
1732
        textfile.check_text_lines(lines)
1093
1733
        self.tt.create_file(lines, trans_id)
1094
 
        if conflicts:
 
1734
        if base_lines is not None:
 
1735
            # Conflict
1095
1736
            self._raw_conflicts.append(('text conflict', trans_id))
1096
1737
            name = self.tt.final_name(trans_id)
1097
1738
            parent_id = self.tt.final_parent(trans_id)
1098
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
1099
 
                                              no_base=True)
 
1739
            file_group = self._dump_conflicts(name, parent_id, file_id,
 
1740
                                              no_base=False,
 
1741
                                              base_lines=base_lines)
1100
1742
            file_group.append(trans_id)
1101
1743
 
1102
1744
 
1103
1745
class LCAMerger(WeaveMerger):
1104
1746
 
1105
 
    def _merged_lines(self, file_id):
1106
 
        """Generate the merged lines.
1107
 
        There is no distinction between lines that are meant to contain <<<<<<<
1108
 
        and conflicts.
1109
 
        """
1110
 
        if self.cherrypick:
1111
 
            base = self.base_tree
1112
 
        else:
1113
 
            base = None
1114
 
        plan = self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
 
1747
    def _generate_merge_plan(self, file_id, base):
 
1748
        return self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
1115
1749
                                                  base=base)
1116
 
        if 'merge' in debug.debug_flags:
1117
 
            plan = list(plan)
1118
 
            trans_id = self.tt.trans_id_file_id(file_id)
1119
 
            name = self.tt.final_name(trans_id) + '.plan'
1120
 
            contents = ('%10s|%s' % l for l in plan)
1121
 
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
1122
 
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
1123
 
            '>>>>>>> MERGE-SOURCE\n')
1124
 
        return textmerge.merge_lines(self.reprocess)
1125
 
 
1126
1750
 
1127
1751
class Diff3Merger(Merge3Merger):
1128
1752
    """Three-way merger using external diff3 for text merging"""
1129
1753
 
1130
1754
    def dump_file(self, temp_dir, name, tree, file_id):
1131
 
        out_path = pathjoin(temp_dir, name)
 
1755
        out_path = osutils.pathjoin(temp_dir, name)
1132
1756
        out_file = open(out_path, "wb")
1133
1757
        try:
1134
1758
            in_file = tree.get_file(file_id)
1146
1770
        import bzrlib.patch
1147
1771
        temp_dir = osutils.mkdtemp(prefix="bzr-")
1148
1772
        try:
1149
 
            new_file = pathjoin(temp_dir, "new")
 
1773
            new_file = osutils.pathjoin(temp_dir, "new")
1150
1774
            this = self.dump_file(temp_dir, "this", self.this_tree, file_id)
1151
1775
            base = self.dump_file(temp_dir, "base", self.base_tree, file_id)
1152
1776
            other = self.dump_file(temp_dir, "other", self.other_tree, file_id)
1153
1777
            status = bzrlib.patch.diff3(new_file, this, base, other)
1154
1778
            if status not in (0, 1):
1155
 
                raise BzrError("Unhandled diff3 exit code")
 
1779
                raise errors.BzrError("Unhandled diff3 exit code")
1156
1780
            f = open(new_file, 'rb')
1157
1781
            try:
1158
1782
                self.tt.create_file(f, trans_id)
1167
1791
            osutils.rmtree(temp_dir)
1168
1792
 
1169
1793
 
 
1794
class PathNotInTree(errors.BzrError):
 
1795
 
 
1796
    _fmt = """Merge-into failed because %(tree)s does not contain %(path)s."""
 
1797
 
 
1798
    def __init__(self, path, tree):
 
1799
        errors.BzrError.__init__(self, path=path, tree=tree)
 
1800
 
 
1801
 
 
1802
class MergeIntoMerger(Merger):
 
1803
    """Merger that understands other_tree will be merged into a subdir.
 
1804
 
 
1805
    This also changes the Merger api so that it uses real Branch, revision_id,
 
1806
    and RevisonTree objects, rather than using revision specs.
 
1807
    """
 
1808
 
 
1809
    def __init__(self, this_tree, other_branch, other_tree, target_subdir,
 
1810
            source_subpath, other_rev_id=None):
 
1811
        """Create a new MergeIntoMerger object.
 
1812
 
 
1813
        source_subpath in other_tree will be effectively copied to
 
1814
        target_subdir in this_tree.
 
1815
 
 
1816
        :param this_tree: The tree that we will be merging into.
 
1817
        :param other_branch: The Branch we will be merging from.
 
1818
        :param other_tree: The RevisionTree object we want to merge.
 
1819
        :param target_subdir: The relative path where we want to merge
 
1820
            other_tree into this_tree
 
1821
        :param source_subpath: The relative path specifying the subtree of
 
1822
            other_tree to merge into this_tree.
 
1823
        """
 
1824
        # It is assumed that we are merging a tree that is not in our current
 
1825
        # ancestry, which means we are using the "EmptyTree" as our basis.
 
1826
        null_ancestor_tree = this_tree.branch.repository.revision_tree(
 
1827
                                _mod_revision.NULL_REVISION)
 
1828
        super(MergeIntoMerger, self).__init__(
 
1829
            this_branch=this_tree.branch,
 
1830
            this_tree=this_tree,
 
1831
            other_tree=other_tree,
 
1832
            base_tree=null_ancestor_tree,
 
1833
            )
 
1834
        self._target_subdir = target_subdir
 
1835
        self._source_subpath = source_subpath
 
1836
        self.other_branch = other_branch
 
1837
        if other_rev_id is None:
 
1838
            other_rev_id = other_tree.get_revision_id()
 
1839
        self.other_rev_id = self.other_basis = other_rev_id
 
1840
        self.base_is_ancestor = True
 
1841
        self.backup_files = True
 
1842
        self.merge_type = Merge3Merger
 
1843
        self.show_base = False
 
1844
        self.reprocess = False
 
1845
        self.interesting_ids = None
 
1846
        self.merge_type = _MergeTypeParameterizer(MergeIntoMergeType,
 
1847
              target_subdir=self._target_subdir,
 
1848
              source_subpath=self._source_subpath)
 
1849
        if self._source_subpath != '':
 
1850
            # If this isn't a partial merge make sure the revisions will be
 
1851
            # present.
 
1852
            self._maybe_fetch(self.other_branch, self.this_branch,
 
1853
                self.other_basis)
 
1854
 
 
1855
    def set_pending(self):
 
1856
        if self._source_subpath != '':
 
1857
            return
 
1858
        Merger.set_pending(self)
 
1859
 
 
1860
 
 
1861
class _MergeTypeParameterizer(object):
 
1862
    """Wrap a merge-type class to provide extra parameters.
 
1863
    
 
1864
    This is hack used by MergeIntoMerger to pass some extra parameters to its
 
1865
    merge_type.  Merger.do_merge() sets up its own set of parameters to pass to
 
1866
    the 'merge_type' member.  It is difficult override do_merge without
 
1867
    re-writing the whole thing, so instead we create a wrapper which will pass
 
1868
    the extra parameters.
 
1869
    """
 
1870
 
 
1871
    def __init__(self, merge_type, **kwargs):
 
1872
        self._extra_kwargs = kwargs
 
1873
        self._merge_type = merge_type
 
1874
 
 
1875
    def __call__(self, *args, **kwargs):
 
1876
        kwargs.update(self._extra_kwargs)
 
1877
        return self._merge_type(*args, **kwargs)
 
1878
 
 
1879
    def __getattr__(self, name):
 
1880
        return getattr(self._merge_type, name)
 
1881
 
 
1882
 
 
1883
class MergeIntoMergeType(Merge3Merger):
 
1884
    """Merger that incorporates a tree (or part of a tree) into another."""
 
1885
 
 
1886
    def __init__(self, *args, **kwargs):
 
1887
        """Initialize the merger object.
 
1888
 
 
1889
        :param args: See Merge3Merger.__init__'s args.
 
1890
        :param kwargs: See Merge3Merger.__init__'s keyword args, except for
 
1891
            source_subpath and target_subdir.
 
1892
        :keyword source_subpath: The relative path specifying the subtree of
 
1893
            other_tree to merge into this_tree.
 
1894
        :keyword target_subdir: The relative path where we want to merge
 
1895
            other_tree into this_tree
 
1896
        """
 
1897
        # All of the interesting work happens during Merge3Merger.__init__(),
 
1898
        # so we have have to hack in to get our extra parameters set.
 
1899
        self._source_subpath = kwargs.pop('source_subpath')
 
1900
        self._target_subdir = kwargs.pop('target_subdir')
 
1901
        super(MergeIntoMergeType, self).__init__(*args, **kwargs)
 
1902
 
 
1903
    def _compute_transform(self):
 
1904
        child_pb = ui.ui_factory.nested_progress_bar()
 
1905
        try:
 
1906
            entries = self._entries_to_incorporate()
 
1907
            entries = list(entries)
 
1908
            for num, (entry, parent_id) in enumerate(entries):
 
1909
                child_pb.update('Preparing file merge', num, len(entries))
 
1910
                parent_trans_id = self.tt.trans_id_file_id(parent_id)
 
1911
                trans_id = transform.new_by_entry(self.tt, entry,
 
1912
                    parent_trans_id, self.other_tree)
 
1913
        finally:
 
1914
            child_pb.finished()
 
1915
        self._finish_computing_transform()
 
1916
 
 
1917
    def _entries_to_incorporate(self):
 
1918
        """Yields pairs of (inventory_entry, new_parent)."""
 
1919
        other_inv = self.other_tree.inventory
 
1920
        subdir_id = other_inv.path2id(self._source_subpath)
 
1921
        if subdir_id is None:
 
1922
            # XXX: The error would be clearer if it gave the URL of the source
 
1923
            # branch, but we don't have a reference to that here.
 
1924
            raise PathNotInTree(self._source_subpath, "Source tree")
 
1925
        subdir = other_inv[subdir_id]
 
1926
        parent_in_target = osutils.dirname(self._target_subdir)
 
1927
        target_id = self.this_tree.inventory.path2id(parent_in_target)
 
1928
        if target_id is None:
 
1929
            raise PathNotInTree(self._target_subdir, "Target tree")
 
1930
        name_in_target = osutils.basename(self._target_subdir)
 
1931
        merge_into_root = subdir.copy()
 
1932
        merge_into_root.name = name_in_target
 
1933
        if self.this_tree.inventory.has_id(merge_into_root.file_id):
 
1934
            # Give the root a new file-id.
 
1935
            # This can happen fairly easily if the directory we are
 
1936
            # incorporating is the root, and both trees have 'TREE_ROOT' as
 
1937
            # their root_id.  Users will expect this to Just Work, so we
 
1938
            # change the file-id here.
 
1939
            # Non-root file-ids could potentially conflict too.  That's really
 
1940
            # an edge case, so we don't do anything special for those.  We let
 
1941
            # them cause conflicts.
 
1942
            merge_into_root.file_id = generate_ids.gen_file_id(name_in_target)
 
1943
        yield (merge_into_root, target_id)
 
1944
        if subdir.kind != 'directory':
 
1945
            # No children, so we are done.
 
1946
            return
 
1947
        for ignored_path, entry in other_inv.iter_entries_by_dir(subdir_id):
 
1948
            parent_id = entry.parent_id
 
1949
            if parent_id == subdir.file_id:
 
1950
                # The root's parent ID has changed, so make sure children of
 
1951
                # the root refer to the new ID.
 
1952
                parent_id = merge_into_root.file_id
 
1953
            yield (entry, parent_id)
 
1954
 
 
1955
 
1170
1956
def merge_inner(this_branch, other_tree, base_tree, ignore_zero=False,
1171
1957
                backup_files=False,
1172
1958
                merge_type=Merge3Merger,
1176
1962
                other_rev_id=None,
1177
1963
                interesting_files=None,
1178
1964
                this_tree=None,
1179
 
                pb=DummyProgress(),
 
1965
                pb=None,
1180
1966
                change_reporter=None):
1181
 
    """Primary interface for merging. 
1182
 
 
1183
 
        typical use is probably 
1184
 
        'merge_inner(branch, branch.get_revision_tree(other_revision),
1185
 
                     branch.get_revision_tree(base_revision))'
1186
 
        """
 
1967
    """Primary interface for merging.
 
1968
 
 
1969
    Typical use is probably::
 
1970
 
 
1971
        merge_inner(branch, branch.get_revision_tree(other_revision),
 
1972
                    branch.get_revision_tree(base_revision))
 
1973
    """
1187
1974
    if this_tree is None:
1188
 
        raise BzrError("bzrlib.merge.merge_inner requires a this_tree "
1189
 
            "parameter as of bzrlib version 0.8.")
 
1975
        raise errors.BzrError("bzrlib.merge.merge_inner requires a this_tree "
 
1976
                              "parameter")
1190
1977
    merger = Merger(this_branch, other_tree, base_tree, this_tree=this_tree,
1191
1978
                    pb=pb, change_reporter=change_reporter)
1192
1979
    merger.backup_files = backup_files
1205
1992
    get_revision_id = getattr(base_tree, 'get_revision_id', None)
1206
1993
    if get_revision_id is None:
1207
1994
        get_revision_id = base_tree.last_revision
 
1995
    merger.cache_trees_with_revision_ids([other_tree, base_tree, this_tree])
1208
1996
    merger.set_base_revision(get_revision_id(), this_branch)
1209
1997
    return merger.do_merge()
1210
1998
 
1278
2066
 
1279
2067
    def get_lines(self, revisions):
1280
2068
        """Get lines for revisions from the backing VersionedFiles.
1281
 
        
 
2069
 
1282
2070
        :raises RevisionNotPresent: on absent texts.
1283
2071
        """
1284
2072
        keys = [(self._key_prefix + (rev,)) for rev in revisions]
1286
2074
        for record in self.vf.get_record_stream(keys, 'unordered', True):
1287
2075
            if record.storage_kind == 'absent':
1288
2076
                raise errors.RevisionNotPresent(record.key, self.vf)
1289
 
            result[record.key[-1]] = osutils.split_lines(
1290
 
                record.get_bytes_as('fulltext'))
 
2077
            result[record.key[-1]] = osutils.chunks_to_lines(
 
2078
                record.get_bytes_as('chunked'))
1291
2079
        return result
1292
2080
 
1293
2081
    def plan_merge(self):
1409
2197
        super(_PlanMerge, self).__init__(a_rev, b_rev, vf, key_prefix)
1410
2198
        self.a_key = self._key_prefix + (self.a_rev,)
1411
2199
        self.b_key = self._key_prefix + (self.b_rev,)
1412
 
        self.graph = Graph(self.vf)
 
2200
        self.graph = _mod_graph.Graph(self.vf)
1413
2201
        heads = self.graph.heads((self.a_key, self.b_key))
1414
2202
        if len(heads) == 1:
1415
2203
            # one side dominates, so we can just return its values, yay for
1420
2208
                other = b_rev
1421
2209
            else:
1422
2210
                other = a_rev
1423
 
            mutter('found dominating revision for %s\n%s > %s', self.vf,
1424
 
                   self._head_key[-1], other)
 
2211
            trace.mutter('found dominating revision for %s\n%s > %s', self.vf,
 
2212
                         self._head_key[-1], other)
1425
2213
            self._weave = None
1426
2214
        else:
1427
2215
            self._head_key = None
1441
2229
        while True:
1442
2230
            next_lcas = self.graph.find_lca(*cur_ancestors)
1443
2231
            # Map a plain NULL_REVISION to a simple no-ancestors
1444
 
            if next_lcas == set([NULL_REVISION]):
 
2232
            if next_lcas == set([_mod_revision.NULL_REVISION]):
1445
2233
                next_lcas = ()
1446
2234
            # Order the lca's based on when they were merged into the tip
1447
2235
            # While the actual merge portion of weave merge uses a set() of
1459
2247
            elif len(next_lcas) > 2:
1460
2248
                # More than 2 lca's, fall back to grabbing all nodes between
1461
2249
                # this and the unique lca.
1462
 
                mutter('More than 2 LCAs, falling back to all nodes for:'
1463
 
                       ' %s, %s\n=> %s', self.a_key, self.b_key, cur_ancestors)
 
2250
                trace.mutter('More than 2 LCAs, falling back to all nodes for:'
 
2251
                             ' %s, %s\n=> %s',
 
2252
                             self.a_key, self.b_key, cur_ancestors)
1464
2253
                cur_lcas = next_lcas
1465
2254
                while len(cur_lcas) > 1:
1466
2255
                    cur_lcas = self.graph.find_lca(*cur_lcas)
1469
2258
                    unique_lca = None
1470
2259
                else:
1471
2260
                    unique_lca = list(cur_lcas)[0]
1472
 
                    if unique_lca == NULL_REVISION:
 
2261
                    if unique_lca == _mod_revision.NULL_REVISION:
1473
2262
                        # find_lca will return a plain 'NULL_REVISION' rather
1474
2263
                        # than a key tuple when there is no common ancestor, we
1475
2264
                        # prefer to just use None, because it doesn't confuse
1483
2272
 
1484
2273
    def _find_unique_parents(self, tip_keys, base_key):
1485
2274
        """Find ancestors of tip that aren't ancestors of base.
1486
 
        
 
2275
 
1487
2276
        :param tip_keys: Nodes that are interesting
1488
2277
        :param base_key: Cull all ancestors of this node
1489
2278
        :return: The parent map for all revisions between tip_keys and
1498
2287
            # We remove NULL_REVISION because it isn't a proper tuple key, and
1499
2288
            # thus confuses things like _get_interesting_texts, and our logic
1500
2289
            # to add the texts into the memory weave.
1501
 
            if NULL_REVISION in parent_map:
1502
 
                parent_map.pop(NULL_REVISION)
 
2290
            if _mod_revision.NULL_REVISION in parent_map:
 
2291
                parent_map.pop(_mod_revision.NULL_REVISION)
1503
2292
        else:
1504
2293
            interesting = set()
1505
2294
            for tip in tip_keys:
1549
2338
    @staticmethod
1550
2339
    def _prune_tails(parent_map, child_map, tails_to_remove):
1551
2340
        """Remove tails from the parent map.
1552
 
        
 
2341
 
1553
2342
        This will remove the supplied revisions until no more children have 0
1554
2343
        parents.
1555
2344
 
1644
2433
class _PlanLCAMerge(_PlanMergeBase):
1645
2434
    """
1646
2435
    This merge algorithm differs from _PlanMerge in that:
 
2436
 
1647
2437
    1. comparisons are done against LCAs only
1648
2438
    2. cases where a contested line is new versus one LCA but old versus
1649
2439
       another are marked as conflicts, by emitting the line as conflicted-a
1657
2447
        lcas = graph.find_lca(key_prefix + (a_rev,), key_prefix + (b_rev,))
1658
2448
        self.lcas = set()
1659
2449
        for lca in lcas:
1660
 
            if lca == NULL_REVISION:
 
2450
            if lca == _mod_revision.NULL_REVISION:
1661
2451
                self.lcas.add(lca)
1662
2452
            else:
1663
2453
                self.lcas.add(lca[-1])
1690
2480
 
1691
2481
        If a line is killed and new, this indicates that the two merge
1692
2482
        revisions contain differing conflict resolutions.
 
2483
 
1693
2484
        :param revision_id: The id of the revision in which the lines are
1694
2485
            unique
1695
2486
        :param unique_line_numbers: The line numbers of unique lines.
1696
 
        :return a tuple of (new_this, killed_other):
 
2487
        :return: a tuple of (new_this, killed_other)
1697
2488
        """
1698
2489
        new = set()
1699
2490
        killed = set()