~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: 2010-03-11 13:47:06 UTC
  • mfrom: (5051.3.16 use-branch-open)
  • Revision ID: pqm@pqm.ubuntu.com-20100311134706-kaerqhx3lf7xn6rh
(Jelmer) Pass colocated branch names further down the call stack.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2008 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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
 
23
19
from bzrlib import (
 
20
    branch as _mod_branch,
 
21
    conflicts as _mod_conflicts,
24
22
    debug,
 
23
    decorators,
25
24
    errors,
26
25
    graph as _mod_graph,
 
26
    hooks,
 
27
    merge3,
27
28
    osutils,
28
29
    patiencediff,
29
 
    registry,
30
30
    revision as _mod_revision,
 
31
    textfile,
 
32
    trace,
 
33
    transform,
 
34
    tree as _mod_tree,
31
35
    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
 
 
 
36
    ui,
 
37
    versionedfile
 
38
    )
 
39
from bzrlib.cleanup import OperationWithCleanups
 
40
from bzrlib.symbol_versioning import (
 
41
    deprecated_in,
 
42
    deprecated_method,
 
43
    )
62
44
# TODO: Report back as changes are merged in
63
45
 
64
46
 
65
47
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)
 
48
    from_tree.lock_tree_write()
 
49
    operation = OperationWithCleanups(merge_inner)
 
50
    operation.add_cleanup(from_tree.unlock)
 
51
    operation.run_simple(from_tree.branch, to_tree, from_tree,
 
52
        ignore_zero=True, interesting_ids=interesting_ids, this_tree=from_tree)
 
53
 
 
54
 
 
55
class MergeHooks(hooks.Hooks):
 
56
 
 
57
    def __init__(self):
 
58
        hooks.Hooks.__init__(self)
 
59
        self.create_hook(hooks.HookPoint('merge_file_content',
 
60
            "Called with a bzrlib.merge.Merger object to create a per file "
 
61
            "merge object when starting a merge. "
 
62
            "Should return either None or a subclass of "
 
63
            "``bzrlib.merge.AbstractPerFileMerger``. "
 
64
            "Such objects will then be called per file "
 
65
            "that needs to be merged (including when one "
 
66
            "side has deleted the file and the other has changed it). "
 
67
            "See the AbstractPerFileMerger API docs for details on how it is "
 
68
            "used by merge.",
 
69
            (2, 1), None))
 
70
 
 
71
 
 
72
class AbstractPerFileMerger(object):
 
73
    """PerFileMerger objects are used by plugins extending merge for bzrlib.
 
74
 
 
75
    See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
 
76
    
 
77
    :ivar merger: The Merge3Merger performing the merge.
 
78
    """
 
79
 
 
80
    def __init__(self, merger):
 
81
        """Create a PerFileMerger for use with merger."""
 
82
        self.merger = merger
 
83
 
 
84
    def merge_contents(self, merge_params):
 
85
        """Attempt to merge the contents of a single file.
 
86
        
 
87
        :param merge_params: A bzrlib.merge.MergeHookParams
 
88
        :return : A tuple of (status, chunks), where status is one of
 
89
            'not_applicable', 'success', 'conflicted', or 'delete'.  If status
 
90
            is 'success' or 'conflicted', then chunks should be an iterable of
 
91
            strings for the new file contents.
 
92
        """
 
93
        return ('not applicable', None)
 
94
 
 
95
 
 
96
class ConfigurableFileMerger(AbstractPerFileMerger):
 
97
    """Merge individual files when configured via a .conf file.
 
98
 
 
99
    This is a base class for concrete custom file merging logic. Concrete
 
100
    classes should implement ``merge_text``.
 
101
 
 
102
    See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
 
103
    
 
104
    :ivar affected_files: The configured file paths to merge.
 
105
 
 
106
    :cvar name_prefix: The prefix to use when looking up configuration
 
107
        details. <name_prefix>_merge_files describes the files targeted by the
 
108
        hook for example.
 
109
        
 
110
    :cvar default_files: The default file paths to merge when no configuration
 
111
        is present.
 
112
    """
 
113
 
 
114
    name_prefix = None
 
115
    default_files = None
 
116
 
 
117
    def __init__(self, merger):
 
118
        super(ConfigurableFileMerger, self).__init__(merger)
 
119
        self.affected_files = None
 
120
        self.default_files = self.__class__.default_files or []
 
121
        self.name_prefix = self.__class__.name_prefix
 
122
        if self.name_prefix is None:
 
123
            raise ValueError("name_prefix must be set.")
 
124
 
 
125
    def filename_matches_config(self, params):
 
126
        """Check whether the file should call the merge hook.
 
127
 
 
128
        <name_prefix>_merge_files configuration variable is a list of files
 
129
        that should use the hook.
 
130
        """
 
131
        affected_files = self.affected_files
 
132
        if affected_files is None:
 
133
            config = self.merger.this_branch.get_config()
 
134
            # Until bzr provides a better policy for caching the config, we
 
135
            # just add the part we're interested in to the params to avoid
 
136
            # reading the config files repeatedly (bazaar.conf, location.conf,
 
137
            # branch.conf).
 
138
            config_key = self.name_prefix + '_merge_files'
 
139
            affected_files = config.get_user_option_as_list(config_key)
 
140
            if affected_files is None:
 
141
                # If nothing was specified in the config, use the default.
 
142
                affected_files = self.default_files
 
143
            self.affected_files = affected_files
 
144
        if affected_files:
 
145
            filename = self.merger.this_tree.id2path(params.file_id)
 
146
            if filename in affected_files:
 
147
                return True
 
148
        return False
 
149
 
 
150
    def merge_contents(self, params):
 
151
        """Merge the contents of a single file."""
 
152
        # First, check whether this custom merge logic should be used.  We
 
153
        # expect most files should not be merged by this handler.
 
154
        if (
 
155
            # OTHER is a straight winner, rely on default merge.
 
156
            params.winner == 'other' or
 
157
            # THIS and OTHER aren't both files.
 
158
            not params.is_file_merge() or
 
159
            # The filename isn't listed in the 'NAME_merge_files' config
 
160
            # option.
 
161
            not self.filename_matches_config(params)):
 
162
            return 'not_applicable', None
 
163
        return self.merge_text(params)
 
164
 
 
165
    def merge_text(self, params):
 
166
        """Merge the byte contents of a single file.
 
167
 
 
168
        This is called after checking that the merge should be performed in
 
169
        merge_contents, and it should behave as per
 
170
        ``bzrlib.merge.AbstractPerFileMerger.merge_contents``.
 
171
        """
 
172
        raise NotImplementedError(self.merge_text)
 
173
 
 
174
 
 
175
class MergeHookParams(object):
 
176
    """Object holding parameters passed to merge_file_content hooks.
 
177
 
 
178
    There are some fields hooks can access:
 
179
 
 
180
    :ivar file_id: the file ID of the file being merged
 
181
    :ivar trans_id: the transform ID for the merge of this file
 
182
    :ivar this_kind: kind of file_id in 'this' tree
 
183
    :ivar other_kind: kind of file_id in 'other' tree
 
184
    :ivar winner: one of 'this', 'other', 'conflict'
 
185
    """
 
186
 
 
187
    def __init__(self, merger, file_id, trans_id, this_kind, other_kind,
 
188
            winner):
 
189
        self._merger = merger
 
190
        self.file_id = file_id
 
191
        self.trans_id = trans_id
 
192
        self.this_kind = this_kind
 
193
        self.other_kind = other_kind
 
194
        self.winner = winner
 
195
 
 
196
    def is_file_merge(self):
 
197
        """True if this_kind and other_kind are both 'file'."""
 
198
        return self.this_kind == 'file' and self.other_kind == 'file'
 
199
 
 
200
    @decorators.cachedproperty
 
201
    def base_lines(self):
 
202
        """The lines of the 'base' version of the file."""
 
203
        return self._merger.get_lines(self._merger.base_tree, self.file_id)
 
204
 
 
205
    @decorators.cachedproperty
 
206
    def this_lines(self):
 
207
        """The lines of the 'this' version of the file."""
 
208
        return self._merger.get_lines(self._merger.this_tree, self.file_id)
 
209
 
 
210
    @decorators.cachedproperty
 
211
    def other_lines(self):
 
212
        """The lines of the 'other' version of the file."""
 
213
        return self._merger.get_lines(self._merger.other_tree, self.file_id)
68
214
 
69
215
 
70
216
class Merger(object):
 
217
 
 
218
    hooks = MergeHooks()
 
219
 
71
220
    def __init__(self, this_branch, other_tree=None, base_tree=None,
72
 
                 this_tree=None, pb=DummyProgress(), change_reporter=None,
 
221
                 this_tree=None, pb=None, change_reporter=None,
73
222
                 recurse='down', revision_graph=None):
74
223
        object.__init__(self)
75
224
        self.this_branch = this_branch
88
237
        self.interesting_files = None
89
238
        self.show_base = False
90
239
        self.reprocess = False
91
 
        self._pb = pb
 
240
        if pb is not None:
 
241
            warnings.warn("pb parameter to Merger() is deprecated and ignored")
92
242
        self.pp = None
93
243
        self.recurse = recurse
94
244
        self.change_reporter = change_reporter
96
246
        self._revision_graph = revision_graph
97
247
        self._base_is_ancestor = None
98
248
        self._base_is_other_ancestor = None
 
249
        self._is_criss_cross = None
 
250
        self._lca_trees = None
 
251
 
 
252
    def cache_trees_with_revision_ids(self, trees):
 
253
        """Cache any tree in trees if it has a revision_id."""
 
254
        for maybe_tree in trees:
 
255
            if maybe_tree is None:
 
256
                continue
 
257
            try:
 
258
                rev_id = maybe_tree.get_revision_id()
 
259
            except AttributeError:
 
260
                continue
 
261
            self._cached_trees[rev_id] = maybe_tree
99
262
 
100
263
    @property
101
264
    def revision_graph(self):
129
292
                                      _set_base_is_other_ancestor)
130
293
 
131
294
    @staticmethod
132
 
    def from_uncommitted(tree, other_tree, pb):
 
295
    def from_uncommitted(tree, other_tree, pb=None, base_tree=None):
133
296
        """Return a Merger for uncommitted changes in other_tree.
134
297
 
135
298
        :param tree: The tree to merge into
136
299
        :param other_tree: The tree to get uncommitted changes from
137
300
        :param pb: A progress indicator
 
301
        :param base_tree: The basis to use for the merge.  If unspecified,
 
302
            other_tree.basis_tree() will be used.
138
303
        """
139
 
        merger = Merger(tree.branch, other_tree, other_tree.basis_tree(), tree,
140
 
                        pb)
 
304
        if base_tree is None:
 
305
            base_tree = other_tree.basis_tree()
 
306
        merger = Merger(tree.branch, other_tree, base_tree, tree, pb)
141
307
        merger.base_rev_id = merger.base_tree.get_revision_id()
142
308
        merger.other_rev_id = None
143
309
        merger.other_basis = merger.base_rev_id
161
327
                base_revision_id, tree.branch.last_revision())):
162
328
                base_revision_id = None
163
329
            else:
164
 
                warning('Performing cherrypick')
 
330
                trace.warning('Performing cherrypick')
165
331
        merger = klass.from_revision_ids(pb, tree, other_revision_id,
166
332
                                         base_revision_id, revision_graph=
167
333
                                         revision_graph)
169
335
 
170
336
    @staticmethod
171
337
    def from_revision_ids(pb, tree, other, base=None, other_branch=None,
172
 
                          base_branch=None, revision_graph=None):
 
338
                          base_branch=None, revision_graph=None,
 
339
                          tree_branch=None):
173
340
        """Return a Merger for revision-ids.
174
341
 
 
342
        :param pb: A progress indicator
175
343
        :param tree: The tree to merge changes into
176
344
        :param other: The revision-id to use as OTHER
177
345
        :param base: The revision-id to use as BASE.  If not specified, will
182
350
            not supplied, other_branch or tree.branch will be used.
183
351
        :param revision_graph: If you have a revision_graph precomputed, pass
184
352
            it in, otherwise it will be created for you.
185
 
        :param pb: A progress indicator
 
353
        :param tree_branch: The branch associated with tree.  If not supplied,
 
354
            tree.branch will be used.
186
355
        """
187
 
        merger = Merger(tree.branch, this_tree=tree, pb=pb,
 
356
        if tree_branch is None:
 
357
            tree_branch = tree.branch
 
358
        merger = Merger(tree_branch, this_tree=tree, pb=pb,
188
359
                        revision_graph=revision_graph)
189
360
        if other_branch is None:
190
361
            other_branch = tree.branch
214
385
        if revno is None:
215
386
            tree = workingtree.WorkingTree.open_containing(location)[0]
216
387
            return tree.branch, tree
217
 
        branch = Branch.open_containing(location, possible_transports)[0]
 
388
        branch = _mod_branch.Branch.open_containing(
 
389
            location, possible_transports)[0]
218
390
        if revno == -1:
219
391
            revision_id = branch.last_revision()
220
392
        else:
221
393
            revision_id = branch.get_rev_id(revno)
222
 
        revision_id = ensure_null(revision_id)
 
394
        revision_id = _mod_revision.ensure_null(revision_id)
223
395
        return branch, self.revision_tree(revision_id, branch)
224
396
 
 
397
    @deprecated_method(deprecated_in((2, 1, 0)))
225
398
    def ensure_revision_trees(self):
226
399
        if self.this_revision_tree is None:
227
400
            self.this_basis_tree = self.revision_tree(self.this_basis)
230
403
 
231
404
        if self.other_rev_id is None:
232
405
            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)
 
406
            if other_basis_tree.has_changes(self.other_tree):
 
407
                raise errors.WorkingTreeNotRevision(self.this_tree)
236
408
            other_rev_id = self.other_basis
237
409
            self.other_tree = other_basis_tree
238
410
 
 
411
    @deprecated_method(deprecated_in((2, 1, 0)))
239
412
    def file_revisions(self, file_id):
240
413
        self.ensure_revision_trees()
241
414
        def get_id(tree, file_id):
244
417
        if self.this_rev_id is None:
245
418
            if self.this_basis_tree.get_file_sha1(file_id) != \
246
419
                self.this_tree.get_file_sha1(file_id):
247
 
                raise WorkingTreeNotRevision(self.this_tree)
 
420
                raise errors.WorkingTreeNotRevision(self.this_tree)
248
421
 
249
422
        trees = (self.this_basis_tree, self.other_tree)
250
423
        return [get_id(tree, file_id) for tree in trees]
251
424
 
 
425
    @deprecated_method(deprecated_in((2, 1, 0)))
252
426
    def check_basis(self, check_clean, require_commits=True):
253
427
        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')")
 
428
            raise errors.BzrCommandError(
 
429
                "This branch has no commits."
 
430
                " (perhaps you would prefer 'bzr pull')")
256
431
        if check_clean:
257
432
            self.compare_basis()
258
433
            if self.this_basis != self.this_rev_id:
259
434
                raise errors.UncommittedChanges(self.this_tree)
260
435
 
 
436
    @deprecated_method(deprecated_in((2, 1, 0)))
261
437
    def compare_basis(self):
262
438
        try:
263
439
            basis_tree = self.revision_tree(self.this_tree.last_revision())
264
440
        except errors.NoSuchRevision:
265
441
            basis_tree = self.this_tree.basis_tree()
266
 
        changes = self.this_tree.changes_from(basis_tree)
267
 
        if not changes.has_changed():
 
442
        if not self.this_tree.has_changes(basis_tree):
268
443
            self.this_rev_id = self.this_basis
269
444
 
270
445
    def set_interesting_files(self, file_list):
271
446
        self.interesting_files = file_list
272
447
 
273
448
    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:
 
449
        if (not self.base_is_ancestor or not self.base_is_other_ancestor
 
450
            or self.other_rev_id is None):
275
451
            return
276
452
        self._add_parent()
277
453
 
278
454
    def _add_parent(self):
279
455
        new_parents = self.this_tree.get_parent_ids() + [self.other_rev_id]
280
456
        new_parent_trees = []
 
457
        operation = OperationWithCleanups(self.this_tree.set_parent_trees)
281
458
        for revision_id in new_parents:
282
459
            try:
283
460
                tree = self.revision_tree(revision_id)
285
462
                tree = None
286
463
            else:
287
464
                tree.lock_read()
 
465
                operation.add_cleanup(tree.unlock)
288
466
            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()
 
467
        operation.run_simple(new_parent_trees, allow_leftmost_as_ghost=True)
296
468
 
297
469
    def set_other(self, other_revision, possible_transports=None):
298
470
        """Set the revision and tree to merge from.
307
479
            self.other_rev_id = _mod_revision.ensure_null(
308
480
                self.other_branch.last_revision())
309
481
            if _mod_revision.is_null(self.other_rev_id):
310
 
                raise NoCommits(self.other_branch)
 
482
                raise errors.NoCommits(self.other_branch)
311
483
            self.other_basis = self.other_rev_id
312
484
        elif other_revision[1] is not None:
313
485
            self.other_rev_id = self.other_branch.get_rev_id(other_revision[1])
316
488
            self.other_rev_id = None
317
489
            self.other_basis = self.other_branch.last_revision()
318
490
            if self.other_basis is None:
319
 
                raise NoCommits(self.other_branch)
 
491
                raise errors.NoCommits(self.other_branch)
320
492
        if self.other_rev_id is not None:
321
493
            self._cached_trees[self.other_rev_id] = self.other_tree
322
494
        self._maybe_fetch(self.other_branch,self.this_branch, self.other_basis)
349
521
            target.fetch(source, revision_id)
350
522
 
351
523
    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
 
524
        revisions = [_mod_revision.ensure_null(self.this_basis),
 
525
                     _mod_revision.ensure_null(self.other_basis)]
 
526
        if _mod_revision.NULL_REVISION in revisions:
 
527
            self.base_rev_id = _mod_revision.NULL_REVISION
 
528
            self.base_tree = self.revision_tree(self.base_rev_id)
 
529
            self._is_criss_cross = False
356
530
        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)
 
531
            lcas = self.revision_graph.find_lca(revisions[0], revisions[1])
 
532
            self._is_criss_cross = False
 
533
            if len(lcas) == 0:
 
534
                self.base_rev_id = _mod_revision.NULL_REVISION
 
535
            elif len(lcas) == 1:
 
536
                self.base_rev_id = list(lcas)[0]
 
537
            else: # len(lcas) > 1
 
538
                if len(lcas) > 2:
 
539
                    # find_unique_lca can only handle 2 nodes, so we have to
 
540
                    # start back at the beginning. It is a shame to traverse
 
541
                    # the graph again, but better than re-implementing
 
542
                    # find_unique_lca.
 
543
                    self.base_rev_id = self.revision_graph.find_unique_lca(
 
544
                                            revisions[0], revisions[1])
 
545
                else:
 
546
                    self.base_rev_id = self.revision_graph.find_unique_lca(
 
547
                                            *lcas)
 
548
                self._is_criss_cross = True
 
549
            if self.base_rev_id == _mod_revision.NULL_REVISION:
 
550
                raise errors.UnrelatedBranches()
 
551
            if self._is_criss_cross:
 
552
                trace.warning('Warning: criss-cross merge encountered.  See bzr'
 
553
                              ' help criss-cross.')
 
554
                trace.mutter('Criss-cross lcas: %r' % lcas)
 
555
                interesting_revision_ids = [self.base_rev_id]
 
556
                interesting_revision_ids.extend(lcas)
 
557
                interesting_trees = dict((t.get_revision_id(), t)
 
558
                    for t in self.this_branch.repository.revision_trees(
 
559
                        interesting_revision_ids))
 
560
                self._cached_trees.update(interesting_trees)
 
561
                self.base_tree = interesting_trees.pop(self.base_rev_id)
 
562
                sorted_lca_keys = self.revision_graph.find_merge_order(
 
563
                    revisions[0], lcas)
 
564
                self._lca_trees = [interesting_trees[key]
 
565
                                   for key in sorted_lca_keys]
 
566
            else:
 
567
                self.base_tree = self.revision_tree(self.base_rev_id)
365
568
        self.base_is_ancestor = True
366
569
        self.base_is_other_ancestor = True
 
570
        trace.mutter('Base revid: %r' % self.base_rev_id)
367
571
 
368
572
    def set_base(self, base_revision):
369
573
        """Set the base revision to use for the merge.
370
574
 
371
575
        :param base_revision: A 2-list containing a path and revision number.
372
576
        """
373
 
        mutter("doing merge() with no base_revision specified")
 
577
        trace.mutter("doing merge() with no base_revision specified")
374
578
        if base_revision == [None, None]:
375
579
            self.find_base()
376
580
        else:
389
593
                  'other_tree': self.other_tree,
390
594
                  'interesting_ids': self.interesting_ids,
391
595
                  'interesting_files': self.interesting_files,
392
 
                  'pp': self.pp,
 
596
                  'this_branch': self.this_branch,
393
597
                  'do_merge': False}
394
598
        if self.merge_type.requires_base:
395
599
            kwargs['base_tree'] = self.base_tree
396
600
        if self.merge_type.supports_reprocess:
397
601
            kwargs['reprocess'] = self.reprocess
398
602
        elif self.reprocess:
399
 
            raise BzrError("Conflict reduction is not supported for merge"
400
 
                                  " type %s." % self.merge_type)
 
603
            raise errors.BzrError(
 
604
                "Conflict reduction is not supported for merge"
 
605
                " type %s." % self.merge_type)
401
606
        if self.merge_type.supports_show_base:
402
607
            kwargs['show_base'] = self.show_base
403
608
        elif self.show_base:
404
 
            raise BzrError("Showing base is not supported for this"
405
 
                           " merge type. %s" % self.merge_type)
 
609
            raise errors.BzrError("Showing base is not supported for this"
 
610
                                  " merge type. %s" % self.merge_type)
406
611
        if (not getattr(self.merge_type, 'supports_reverse_cherrypick', True)
407
612
            and not self.base_is_other_ancestor):
408
613
            raise errors.CannotReverseCherrypick()
409
614
        if self.merge_type.supports_cherrypick:
410
615
            kwargs['cherrypick'] = (not self.base_is_ancestor or
411
616
                                    not self.base_is_other_ancestor)
412
 
        return self.merge_type(pb=self._pb,
 
617
        if self._is_criss_cross and getattr(self.merge_type,
 
618
                                            'supports_lca_trees', False):
 
619
            kwargs['lca_trees'] = self._lca_trees
 
620
        return self.merge_type(pb=None,
413
621
                               change_reporter=self.change_reporter,
414
622
                               **kwargs)
415
623
 
 
624
    def _do_merge_to(self):
 
625
        merge = self.make_merger()
 
626
        if self.other_branch is not None:
 
627
            self.other_branch.update_references(self.this_branch)
 
628
        merge.do_merge()
 
629
        if self.recurse == 'down':
 
630
            for relpath, file_id in self.this_tree.iter_references():
 
631
                sub_tree = self.this_tree.get_nested_tree(file_id, relpath)
 
632
                other_revision = self.other_tree.get_reference_revision(
 
633
                    file_id, relpath)
 
634
                if  other_revision == sub_tree.last_revision():
 
635
                    continue
 
636
                sub_merge = Merger(sub_tree.branch, this_tree=sub_tree)
 
637
                sub_merge.merge_type = self.merge_type
 
638
                other_branch = self.other_branch.reference_parent(file_id, relpath)
 
639
                sub_merge.set_other_revision(other_revision, other_branch)
 
640
                base_revision = self.base_tree.get_reference_revision(file_id)
 
641
                sub_merge.base_tree = \
 
642
                    sub_tree.branch.repository.revision_tree(base_revision)
 
643
                sub_merge.base_rev_id = base_revision
 
644
                sub_merge.do_merge()
 
645
        return merge
 
646
 
416
647
    def do_merge(self):
 
648
        operation = OperationWithCleanups(self._do_merge_to)
417
649
        self.this_tree.lock_tree_write()
 
650
        operation.add_cleanup(self.this_tree.unlock)
418
651
        if self.base_tree is not None:
419
652
            self.base_tree.lock_read()
 
653
            operation.add_cleanup(self.base_tree.unlock)
420
654
        if self.other_tree is not None:
421
655
            self.other_tree.lock_read()
422
 
        try:
423
 
            merge = self.make_merger()
424
 
            merge.do_merge()
425
 
            if self.recurse == 'down':
426
 
                for relpath, file_id in self.this_tree.iter_references():
427
 
                    sub_tree = self.this_tree.get_nested_tree(file_id, relpath)
428
 
                    other_revision = self.other_tree.get_reference_revision(
429
 
                        file_id, relpath)
430
 
                    if  other_revision == sub_tree.last_revision():
431
 
                        continue
432
 
                    sub_merge = Merger(sub_tree.branch, this_tree=sub_tree)
433
 
                    sub_merge.merge_type = self.merge_type
434
 
                    other_branch = self.other_branch.reference_parent(file_id, relpath)
435
 
                    sub_merge.set_other_revision(other_revision, other_branch)
436
 
                    base_revision = self.base_tree.get_reference_revision(file_id)
437
 
                    sub_merge.base_tree = \
438
 
                        sub_tree.branch.repository.revision_tree(base_revision)
439
 
                    sub_merge.base_rev_id = base_revision
440
 
                    sub_merge.do_merge()
441
 
 
442
 
        finally:
443
 
            if self.other_tree is not None:
444
 
                self.other_tree.unlock()
445
 
            if self.base_tree is not None:
446
 
                self.base_tree.unlock()
447
 
            self.this_tree.unlock()
 
656
            operation.add_cleanup(self.other_tree.unlock)
 
657
        merge = operation.run_simple()
448
658
        if len(merge.cooked_conflicts) == 0:
449
 
            if not self.ignore_zero and not is_quiet():
450
 
                note("All changes applied successfully.")
 
659
            if not self.ignore_zero and not trace.is_quiet():
 
660
                trace.note("All changes applied successfully.")
451
661
        else:
452
 
            note("%d conflicts encountered." % len(merge.cooked_conflicts))
 
662
            trace.note("%d conflicts encountered."
 
663
                       % len(merge.cooked_conflicts))
453
664
 
454
665
        return len(merge.cooked_conflicts)
455
666
 
456
667
 
 
668
class _InventoryNoneEntry(object):
 
669
    """This represents an inventory entry which *isn't there*.
 
670
 
 
671
    It simplifies the merging logic if we always have an InventoryEntry, even
 
672
    if it isn't actually present
 
673
    """
 
674
    executable = None
 
675
    kind = None
 
676
    name = None
 
677
    parent_id = None
 
678
    revision = None
 
679
    symlink_target = None
 
680
    text_sha1 = None
 
681
 
 
682
_none_entry = _InventoryNoneEntry()
 
683
 
 
684
 
457
685
class Merge3Merger(object):
458
686
    """Three-way merger that uses the merge3 text merger"""
459
687
    requires_base = True
463
691
    supports_cherrypick = True
464
692
    supports_reverse_cherrypick = True
465
693
    winner_idx = {"this": 2, "other": 1, "conflict": 1}
 
694
    supports_lca_trees = True
466
695
 
467
 
    def __init__(self, working_tree, this_tree, base_tree, other_tree, 
 
696
    def __init__(self, working_tree, this_tree, base_tree, other_tree,
468
697
                 interesting_ids=None, reprocess=False, show_base=False,
469
 
                 pb=DummyProgress(), pp=None, change_reporter=None,
 
698
                 pb=None, pp=None, change_reporter=None,
470
699
                 interesting_files=None, do_merge=True,
471
 
                 cherrypick=False):
 
700
                 cherrypick=False, lca_trees=None, this_branch=None):
472
701
        """Initialize the merger object and perform the merge.
473
702
 
474
703
        :param working_tree: The working tree to apply the merge to
475
704
        :param this_tree: The local tree in the merge operation
476
705
        :param base_tree: The common tree in the merge operation
477
 
        :param other_tree: The other other tree to merge changes from
 
706
        :param other_tree: The other tree to merge changes from
 
707
        :param this_branch: The branch associated with this_tree
478
708
        :param interesting_ids: The file_ids of files that should be
479
709
            participate in the merge.  May not be combined with
480
710
            interesting_files.
481
711
        :param: reprocess If True, perform conflict-reduction processing.
482
712
        :param show_base: If True, show the base revision in text conflicts.
483
713
            (incompatible with reprocess)
484
 
        :param pb: A Progress bar
 
714
        :param pb: ignored
485
715
        :param pp: A ProgressPhase object
486
716
        :param change_reporter: An object that should report changes made
487
717
        :param interesting_files: The tree-relative paths of files that should
490
720
            be combined with interesting_ids.  If neither interesting_files nor
491
721
            interesting_ids is specified, all files may participate in the
492
722
            merge.
 
723
        :param lca_trees: Can be set to a dictionary of {revision_id:rev_tree}
 
724
            if the ancestry was found to include a criss-cross merge.
 
725
            Otherwise should be None.
493
726
        """
494
727
        object.__init__(self)
495
728
        if interesting_files is not None and interesting_ids is not None:
500
733
        self.this_tree = working_tree
501
734
        self.base_tree = base_tree
502
735
        self.other_tree = other_tree
 
736
        self.this_branch = this_branch
503
737
        self._raw_conflicts = []
504
738
        self.cooked_conflicts = []
505
739
        self.reprocess = reprocess
506
740
        self.show_base = show_base
507
 
        self.pb = pb
508
 
        self.pp = pp
 
741
        self._lca_trees = lca_trees
 
742
        # Uncommenting this will change the default algorithm to always use
 
743
        # _entries_lca. This can be useful for running the test suite and
 
744
        # making sure we haven't missed any corner cases.
 
745
        # if lca_trees is None:
 
746
        #     self._lca_trees = [self.base_tree]
509
747
        self.change_reporter = change_reporter
510
748
        self.cherrypick = cherrypick
511
 
        if self.pp is None:
512
 
            self.pp = ProgressPhase("Merge phase", 3, self.pb)
513
749
        if do_merge:
514
750
            self.do_merge()
 
751
        if pp is not None:
 
752
            warnings.warn("pp argument to Merge3Merger is deprecated")
 
753
        if pb is not None:
 
754
            warnings.warn("pb argument to Merge3Merger is deprecated")
515
755
 
516
756
    def do_merge(self):
 
757
        operation = OperationWithCleanups(self._do_merge)
517
758
        self.this_tree.lock_tree_write()
 
759
        operation.add_cleanup(self.this_tree.unlock)
518
760
        self.base_tree.lock_read()
 
761
        operation.add_cleanup(self.base_tree.unlock)
519
762
        self.other_tree.lock_read()
520
 
        self.tt = TreeTransform(self.this_tree, self.pb)
 
763
        operation.add_cleanup(self.other_tree.unlock)
 
764
        operation.run()
 
765
 
 
766
    def _do_merge(self, operation):
 
767
        self.tt = transform.TreeTransform(self.this_tree, None)
 
768
        operation.add_cleanup(self.tt.finalize)
 
769
        self._compute_transform()
 
770
        results = self.tt.apply(no_conflicts=True)
 
771
        self.write_modified(results)
521
772
        try:
522
 
            self.pp.next_phase()
523
 
            self._compute_transform()
524
 
            self.pp.next_phase()
525
 
            results = self.tt.apply(no_conflicts=True)
526
 
            self.write_modified(results)
527
 
            try:
528
 
                self.this_tree.add_conflicts(self.cooked_conflicts)
529
 
            except UnsupportedOperation:
530
 
                pass
531
 
        finally:
532
 
            self.tt.finalize()
533
 
            self.other_tree.unlock()
534
 
            self.base_tree.unlock()
535
 
            self.this_tree.unlock()
536
 
            self.pb.clear()
 
773
            self.this_tree.add_conflicts(self.cooked_conflicts)
 
774
        except errors.UnsupportedOperation:
 
775
            pass
537
776
 
538
777
    def make_preview_transform(self):
 
778
        operation = OperationWithCleanups(self._make_preview_transform)
539
779
        self.base_tree.lock_read()
 
780
        operation.add_cleanup(self.base_tree.unlock)
540
781
        self.other_tree.lock_read()
541
 
        self.tt = TransformPreview(self.this_tree)
542
 
        try:
543
 
            self.pp.next_phase()
544
 
            self._compute_transform()
545
 
            self.pp.next_phase()
546
 
        finally:
547
 
            self.other_tree.unlock()
548
 
            self.base_tree.unlock()
549
 
            self.pb.clear()
 
782
        operation.add_cleanup(self.other_tree.unlock)
 
783
        return operation.run_simple()
 
784
 
 
785
    def _make_preview_transform(self):
 
786
        self.tt = transform.TransformPreview(self.this_tree)
 
787
        self._compute_transform()
550
788
        return self.tt
551
789
 
552
790
    def _compute_transform(self):
553
 
        entries = self._entries3()
 
791
        if self._lca_trees is None:
 
792
            entries = self._entries3()
 
793
            resolver = self._three_way
 
794
        else:
 
795
            entries = self._entries_lca()
 
796
            resolver = self._lca_multi_way
554
797
        child_pb = ui.ui_factory.nested_progress_bar()
555
798
        try:
 
799
            factories = Merger.hooks['merge_file_content']
 
800
            hooks = [factory(self) for factory in factories] + [self]
 
801
            self.active_hooks = [hook for hook in hooks if hook is not None]
556
802
            for num, (file_id, changed, parents3, names3,
557
803
                      executable3) in enumerate(entries):
558
804
                child_pb.update('Preparing file merge', num, len(entries))
559
 
                self._merge_names(file_id, parents3, names3)
 
805
                self._merge_names(file_id, parents3, names3, resolver=resolver)
560
806
                if changed:
561
 
                    file_status = self.merge_contents(file_id)
 
807
                    file_status = self._do_merge_contents(file_id)
562
808
                else:
563
809
                    file_status = 'unmodified'
564
810
                self._merge_executable(file_id,
565
 
                    executable3, file_status)
 
811
                    executable3, file_status, resolver=resolver)
566
812
        finally:
567
813
            child_pb.finished()
568
814
        self.fix_root()
569
 
        self.pp.next_phase()
570
815
        child_pb = ui.ui_factory.nested_progress_bar()
571
816
        try:
572
 
            fs_conflicts = resolve_conflicts(self.tt, child_pb,
573
 
                lambda t, c: conflict_pass(t, c, self.other_tree))
 
817
            fs_conflicts = transform.resolve_conflicts(self.tt, child_pb,
 
818
                lambda t, c: transform.conflict_pass(t, c, self.other_tree))
574
819
        finally:
575
820
            child_pb.finished()
576
821
        if self.change_reporter is not None:
579
824
                self.tt.iter_changes(), self.change_reporter)
580
825
        self.cook_conflicts(fs_conflicts)
581
826
        for conflict in self.cooked_conflicts:
582
 
            warning(conflict)
 
827
            trace.warning(conflict)
583
828
 
584
829
    def _entries3(self):
585
830
        """Gather data about files modified between three trees.
594
839
        iterator = self.other_tree.iter_changes(self.base_tree,
595
840
                include_unchanged=True, specific_files=self.interesting_files,
596
841
                extra_trees=[self.this_tree])
 
842
        this_entries = dict((e.file_id, e) for p, e in
 
843
                            self.this_tree.iter_entries_by_dir(
 
844
                            self.interesting_ids))
597
845
        for (file_id, paths, changed, versioned, parents, names, kind,
598
846
             executable) in iterator:
599
847
            if (self.interesting_ids is not None and
600
848
                file_id not in self.interesting_ids):
601
849
                continue
602
 
            if file_id in self.this_tree.inventory:
603
 
                entry = self.this_tree.inventory[file_id]
 
850
            entry = this_entries.get(file_id)
 
851
            if entry is not None:
604
852
                this_name = entry.name
605
853
                this_parent = entry.parent_id
606
854
                this_executable = entry.executable
614
862
            result.append((file_id, changed, parents3, names3, executable3))
615
863
        return result
616
864
 
 
865
    def _entries_lca(self):
 
866
        """Gather data about files modified between multiple trees.
 
867
 
 
868
        This compares OTHER versus all LCA trees, and for interesting entries,
 
869
        it then compares with THIS and BASE.
 
870
 
 
871
        For the multi-valued entries, the format will be (BASE, [lca1, lca2])
 
872
        :return: [(file_id, changed, parents, names, executable)]
 
873
            file_id     Simple file_id of the entry
 
874
            changed     Boolean, True if the kind or contents changed
 
875
                        else False
 
876
            parents     ((base, [parent_id, in, lcas]), parent_id_other,
 
877
                         parent_id_this)
 
878
            names       ((base, [name, in, lcas]), name_in_other, name_in_this)
 
879
            executable  ((base, [exec, in, lcas]), exec_in_other, exec_in_this)
 
880
        """
 
881
        if self.interesting_files is not None:
 
882
            lookup_trees = [self.this_tree, self.base_tree]
 
883
            lookup_trees.extend(self._lca_trees)
 
884
            # I think we should include the lca trees as well
 
885
            interesting_ids = self.other_tree.paths2ids(self.interesting_files,
 
886
                                                        lookup_trees)
 
887
        else:
 
888
            interesting_ids = self.interesting_ids
 
889
        result = []
 
890
        walker = _mod_tree.MultiWalker(self.other_tree, self._lca_trees)
 
891
 
 
892
        base_inventory = self.base_tree.inventory
 
893
        this_inventory = self.this_tree.inventory
 
894
        for path, file_id, other_ie, lca_values in walker.iter_all():
 
895
            # Is this modified at all from any of the other trees?
 
896
            if other_ie is None:
 
897
                other_ie = _none_entry
 
898
            if interesting_ids is not None and file_id not in interesting_ids:
 
899
                continue
 
900
 
 
901
            # If other_revision is found in any of the lcas, that means this
 
902
            # node is uninteresting. This is because when merging, if there are
 
903
            # multiple heads(), we have to create a new node. So if we didn't,
 
904
            # we know that the ancestry is linear, and that OTHER did not
 
905
            # modify anything
 
906
            # See doc/developers/lca_merge_resolution.txt for details
 
907
            other_revision = other_ie.revision
 
908
            if other_revision is not None:
 
909
                # We can't use this shortcut when other_revision is None,
 
910
                # because it may be None because things are WorkingTrees, and
 
911
                # not because it is *actually* None.
 
912
                is_unmodified = False
 
913
                for lca_path, ie in lca_values:
 
914
                    if ie is not None and ie.revision == other_revision:
 
915
                        is_unmodified = True
 
916
                        break
 
917
                if is_unmodified:
 
918
                    continue
 
919
 
 
920
            lca_entries = []
 
921
            for lca_path, lca_ie in lca_values:
 
922
                if lca_ie is None:
 
923
                    lca_entries.append(_none_entry)
 
924
                else:
 
925
                    lca_entries.append(lca_ie)
 
926
 
 
927
            if file_id in base_inventory:
 
928
                base_ie = base_inventory[file_id]
 
929
            else:
 
930
                base_ie = _none_entry
 
931
 
 
932
            if file_id in this_inventory:
 
933
                this_ie = this_inventory[file_id]
 
934
            else:
 
935
                this_ie = _none_entry
 
936
 
 
937
            lca_kinds = []
 
938
            lca_parent_ids = []
 
939
            lca_names = []
 
940
            lca_executable = []
 
941
            for lca_ie in lca_entries:
 
942
                lca_kinds.append(lca_ie.kind)
 
943
                lca_parent_ids.append(lca_ie.parent_id)
 
944
                lca_names.append(lca_ie.name)
 
945
                lca_executable.append(lca_ie.executable)
 
946
 
 
947
            kind_winner = self._lca_multi_way(
 
948
                (base_ie.kind, lca_kinds),
 
949
                other_ie.kind, this_ie.kind)
 
950
            parent_id_winner = self._lca_multi_way(
 
951
                (base_ie.parent_id, lca_parent_ids),
 
952
                other_ie.parent_id, this_ie.parent_id)
 
953
            name_winner = self._lca_multi_way(
 
954
                (base_ie.name, lca_names),
 
955
                other_ie.name, this_ie.name)
 
956
 
 
957
            content_changed = True
 
958
            if kind_winner == 'this':
 
959
                # No kind change in OTHER, see if there are *any* changes
 
960
                if other_ie.kind == 'directory':
 
961
                    if parent_id_winner == 'this' and name_winner == 'this':
 
962
                        # No change for this directory in OTHER, skip
 
963
                        continue
 
964
                    content_changed = False
 
965
                elif other_ie.kind is None or other_ie.kind == 'file':
 
966
                    def get_sha1(ie, tree):
 
967
                        if ie.kind != 'file':
 
968
                            return None
 
969
                        return tree.get_file_sha1(file_id)
 
970
                    base_sha1 = get_sha1(base_ie, self.base_tree)
 
971
                    lca_sha1s = [get_sha1(ie, tree) for ie, tree
 
972
                                 in zip(lca_entries, self._lca_trees)]
 
973
                    this_sha1 = get_sha1(this_ie, self.this_tree)
 
974
                    other_sha1 = get_sha1(other_ie, self.other_tree)
 
975
                    sha1_winner = self._lca_multi_way(
 
976
                        (base_sha1, lca_sha1s), other_sha1, this_sha1,
 
977
                        allow_overriding_lca=False)
 
978
                    exec_winner = self._lca_multi_way(
 
979
                        (base_ie.executable, lca_executable),
 
980
                        other_ie.executable, this_ie.executable)
 
981
                    if (parent_id_winner == 'this' and name_winner == 'this'
 
982
                        and sha1_winner == 'this' and exec_winner == 'this'):
 
983
                        # No kind, parent, name, exec, or content change for
 
984
                        # OTHER, so this node is not considered interesting
 
985
                        continue
 
986
                    if sha1_winner == 'this':
 
987
                        content_changed = False
 
988
                elif other_ie.kind == 'symlink':
 
989
                    def get_target(ie, tree):
 
990
                        if ie.kind != 'symlink':
 
991
                            return None
 
992
                        return tree.get_symlink_target(file_id)
 
993
                    base_target = get_target(base_ie, self.base_tree)
 
994
                    lca_targets = [get_target(ie, tree) for ie, tree
 
995
                                   in zip(lca_entries, self._lca_trees)]
 
996
                    this_target = get_target(this_ie, self.this_tree)
 
997
                    other_target = get_target(other_ie, self.other_tree)
 
998
                    target_winner = self._lca_multi_way(
 
999
                        (base_target, lca_targets),
 
1000
                        other_target, this_target)
 
1001
                    if (parent_id_winner == 'this' and name_winner == 'this'
 
1002
                        and target_winner == 'this'):
 
1003
                        # No kind, parent, name, or symlink target change
 
1004
                        # not interesting
 
1005
                        continue
 
1006
                    if target_winner == 'this':
 
1007
                        content_changed = False
 
1008
                elif other_ie.kind == 'tree-reference':
 
1009
                    # The 'changed' information seems to be handled at a higher
 
1010
                    # level. At least, _entries3 returns False for content
 
1011
                    # changed, even when at a new revision_id.
 
1012
                    content_changed = False
 
1013
                    if (parent_id_winner == 'this' and name_winner == 'this'):
 
1014
                        # Nothing interesting
 
1015
                        continue
 
1016
                else:
 
1017
                    raise AssertionError('unhandled kind: %s' % other_ie.kind)
 
1018
                # XXX: We need to handle kind == 'symlink'
 
1019
 
 
1020
            # If we have gotten this far, that means something has changed
 
1021
            result.append((file_id, content_changed,
 
1022
                           ((base_ie.parent_id, lca_parent_ids),
 
1023
                            other_ie.parent_id, this_ie.parent_id),
 
1024
                           ((base_ie.name, lca_names),
 
1025
                            other_ie.name, this_ie.name),
 
1026
                           ((base_ie.executable, lca_executable),
 
1027
                            other_ie.executable, this_ie.executable)
 
1028
                          ))
 
1029
        return result
 
1030
 
 
1031
 
617
1032
    def fix_root(self):
618
1033
        try:
619
1034
            self.tt.final_kind(self.tt.root)
620
 
        except NoSuchFile:
 
1035
        except errors.NoSuchFile:
621
1036
            self.tt.cancel_deletion(self.tt.root)
622
1037
        if self.tt.final_file_id(self.tt.root) is None:
623
 
            self.tt.version_file(self.tt.tree_file_id(self.tt.root), 
 
1038
            self.tt.version_file(self.tt.tree_file_id(self.tt.root),
624
1039
                                 self.tt.root)
625
 
        if self.other_tree.inventory.root is None:
626
 
            return
627
1040
        other_root_file_id = self.other_tree.get_root_id()
 
1041
        if other_root_file_id is None:
 
1042
            return
628
1043
        other_root = self.tt.trans_id_file_id(other_root_file_id)
629
1044
        if other_root == self.tt.root:
630
1045
            return
631
1046
        try:
632
1047
            self.tt.final_kind(other_root)
633
 
        except NoSuchFile:
 
1048
        except errors.NoSuchFile:
634
1049
            return
635
 
        if self.other_tree.inventory.root.file_id in self.this_tree.inventory:
 
1050
        if self.this_tree.has_id(self.other_tree.inventory.root.file_id):
636
1051
            # the other tree's root is a non-root in the current tree
637
1052
            return
638
1053
        self.reparent_children(self.other_tree.inventory.root, self.tt.root)
669
1084
        if entry is None:
670
1085
            return None
671
1086
        return entry.name
672
 
    
 
1087
 
673
1088
    @staticmethod
674
1089
    def contents_sha1(tree, file_id):
675
1090
        """Determine the sha1 of the file contents (used as a key method)."""
680
1095
    @staticmethod
681
1096
    def executable(tree, file_id):
682
1097
        """Determine the executability of a file-id (used as a key method)."""
683
 
        if file_id not in tree:
 
1098
        if not tree.has_id(file_id):
684
1099
            return None
685
1100
        if tree.kind(file_id) != "file":
686
1101
            return False
689
1104
    @staticmethod
690
1105
    def kind(tree, file_id):
691
1106
        """Determine the kind of a file-id (used as a key method)."""
692
 
        if file_id not in tree:
 
1107
        if not tree.has_id(file_id):
693
1108
            return None
694
1109
        return tree.kind(file_id)
695
1110
 
708
1123
            return "other"
709
1124
 
710
1125
    @staticmethod
 
1126
    def _lca_multi_way(bases, other, this, allow_overriding_lca=True):
 
1127
        """Consider LCAs when determining whether a change has occurred.
 
1128
 
 
1129
        If LCAS are all identical, this is the same as a _three_way comparison.
 
1130
 
 
1131
        :param bases: value in (BASE, [LCAS])
 
1132
        :param other: value in OTHER
 
1133
        :param this: value in THIS
 
1134
        :param allow_overriding_lca: If there is more than one unique lca
 
1135
            value, allow OTHER to override THIS if it has a new value, and
 
1136
            THIS only has an lca value, or vice versa. This is appropriate for
 
1137
            truly scalar values, not as much for non-scalars.
 
1138
        :return: 'this', 'other', or 'conflict' depending on whether an entry
 
1139
            changed or not.
 
1140
        """
 
1141
        # See doc/developers/lca_tree_merging.txt for details about this
 
1142
        # algorithm.
 
1143
        if other == this:
 
1144
            # Either Ambiguously clean, or nothing was actually changed. We
 
1145
            # don't really care
 
1146
            return 'this'
 
1147
        base_val, lca_vals = bases
 
1148
        # Remove 'base_val' from the lca_vals, because it is not interesting
 
1149
        filtered_lca_vals = [lca_val for lca_val in lca_vals
 
1150
                                      if lca_val != base_val]
 
1151
        if len(filtered_lca_vals) == 0:
 
1152
            return Merge3Merger._three_way(base_val, other, this)
 
1153
 
 
1154
        unique_lca_vals = set(filtered_lca_vals)
 
1155
        if len(unique_lca_vals) == 1:
 
1156
            return Merge3Merger._three_way(unique_lca_vals.pop(), other, this)
 
1157
 
 
1158
        if allow_overriding_lca:
 
1159
            if other in unique_lca_vals:
 
1160
                if this in unique_lca_vals:
 
1161
                    # Each side picked a different lca, conflict
 
1162
                    return 'conflict'
 
1163
                else:
 
1164
                    # This has a value which supersedes both lca values, and
 
1165
                    # other only has an lca value
 
1166
                    return 'this'
 
1167
            elif this in unique_lca_vals:
 
1168
                # OTHER has a value which supersedes both lca values, and this
 
1169
                # only has an lca value
 
1170
                return 'other'
 
1171
 
 
1172
        # At this point, the lcas disagree, and the tips disagree
 
1173
        return 'conflict'
 
1174
 
 
1175
    @staticmethod
 
1176
    @deprecated_method(deprecated_in((2, 2, 0)))
711
1177
    def scalar_three_way(this_tree, base_tree, other_tree, file_id, key):
712
1178
        """Do a three-way test on a scalar.
713
1179
        Return "this", "other" or "conflict", depending whether a value wins.
728
1194
 
729
1195
    def merge_names(self, file_id):
730
1196
        def get_entry(tree):
731
 
            if file_id in tree.inventory:
 
1197
            if tree.has_id(file_id):
732
1198
                return tree.inventory[file_id]
733
1199
            else:
734
1200
                return None
745
1211
            else:
746
1212
                names.append(entry.name)
747
1213
                parents.append(entry.parent_id)
748
 
        return self._merge_names(file_id, parents, names)
 
1214
        return self._merge_names(file_id, parents, names,
 
1215
                                 resolver=self._three_way)
749
1216
 
750
 
    def _merge_names(self, file_id, parents, names):
 
1217
    def _merge_names(self, file_id, parents, names, resolver):
751
1218
        """Perform a merge on file_id names and parents"""
752
1219
        base_name, other_name, this_name = names
753
1220
        base_parent, other_parent, this_parent = parents
754
1221
 
755
 
        name_winner = self._three_way(*names)
 
1222
        name_winner = resolver(*names)
756
1223
 
757
 
        parent_id_winner = self._three_way(*parents)
 
1224
        parent_id_winner = resolver(*parents)
758
1225
        if this_name is None:
759
1226
            if name_winner == "this":
760
1227
                name_winner = "other"
764
1231
            return
765
1232
        if name_winner == "conflict":
766
1233
            trans_id = self.tt.trans_id_file_id(file_id)
767
 
            self._raw_conflicts.append(('name conflict', trans_id, 
 
1234
            self._raw_conflicts.append(('name conflict', trans_id,
768
1235
                                        this_name, other_name))
769
1236
        if parent_id_winner == "conflict":
770
1237
            trans_id = self.tt.trans_id_file_id(file_id)
771
 
            self._raw_conflicts.append(('parent conflict', trans_id, 
 
1238
            self._raw_conflicts.append(('parent conflict', trans_id,
772
1239
                                        this_parent, other_parent))
773
1240
        if other_name is None:
774
 
            # it doesn't matter whether the result was 'other' or 
 
1241
            # it doesn't matter whether the result was 'other' or
775
1242
            # 'conflict'-- if there's no 'other', we leave it alone.
776
1243
            return
777
1244
        # if we get here, name_winner and parent_winner are set to safe values.
782
1249
            self.tt.adjust_path(names[self.winner_idx[name_winner]],
783
1250
                                parent_trans_id, trans_id)
784
1251
 
785
 
    def merge_contents(self, file_id):
786
 
        """Performa a merge on file_id contents."""
 
1252
    def _do_merge_contents(self, file_id):
 
1253
        """Performs a merge on file_id contents."""
787
1254
        def contents_pair(tree):
788
1255
            if file_id not in tree:
789
1256
                return (None, None)
796
1263
                contents = None
797
1264
            return kind, contents
798
1265
 
799
 
        def contents_conflict():
800
 
            trans_id = self.tt.trans_id_file_id(file_id)
801
 
            name = self.tt.final_name(trans_id)
802
 
            parent_id = self.tt.final_parent(trans_id)
803
 
            if file_id in self.this_tree.inventory:
804
 
                self.tt.unversion_file(trans_id)
805
 
                if file_id in self.this_tree:
806
 
                    self.tt.delete_contents(trans_id)
807
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
808
 
                                              set_version=True)
809
 
            self._raw_conflicts.append(('contents conflict', file_group))
810
 
 
811
1266
        # See SPOT run.  run, SPOT, run.
812
1267
        # So we're not QUITE repeating ourselves; we do tricky things with
813
1268
        # file kind...
814
1269
        base_pair = contents_pair(self.base_tree)
815
1270
        other_pair = contents_pair(self.other_tree)
816
 
        if base_pair == other_pair:
817
 
            # OTHER introduced no changes
818
 
            return "unmodified"
819
 
        this_pair = contents_pair(self.this_tree)
820
 
        if this_pair == other_pair:
821
 
            # THIS and OTHER introduced the same changes
822
 
            return "unmodified"
823
 
        else:
824
 
            trans_id = self.tt.trans_id_file_id(file_id)
825
 
            if this_pair == base_pair:
826
 
                # only OTHER introduced changes
827
 
                if file_id in self.this_tree:
828
 
                    # Remove any existing contents
829
 
                    self.tt.delete_contents(trans_id)
830
 
                if file_id in self.other_tree:
831
 
                    # OTHER changed the file
832
 
                    create_by_entry(self.tt, 
833
 
                                    self.other_tree.inventory[file_id], 
834
 
                                    self.other_tree, trans_id)
835
 
                    if file_id not in self.this_tree.inventory:
836
 
                        self.tt.version_file(file_id, trans_id)
837
 
                    return "modified"
838
 
                elif file_id in self.this_tree.inventory:
839
 
                    # OTHER deleted the file
840
 
                    self.tt.unversion_file(trans_id)
841
 
                    return "deleted"
842
 
            #BOTH THIS and OTHER introduced changes; scalar conflict
843
 
            elif this_pair[0] == "file" and other_pair[0] == "file":
844
 
                # THIS and OTHER are both files, so text merge.  Either
845
 
                # BASE is a file, or both converted to files, so at least we
846
 
                # have agreement that output should be a file.
847
 
                try:
848
 
                    self.text_merge(file_id, trans_id)
849
 
                except BinaryFile:
850
 
                    return contents_conflict()
851
 
                if file_id not in self.this_tree.inventory:
852
 
                    self.tt.version_file(file_id, trans_id)
853
 
                try:
854
 
                    self.tt.tree_kind(trans_id)
855
 
                    self.tt.delete_contents(trans_id)
856
 
                except NoSuchFile:
857
 
                    pass
858
 
                return "modified"
859
 
            else:
860
 
                # Scalar conflict, can't text merge.  Dump conflicts
861
 
                return contents_conflict()
 
1271
        if self._lca_trees:
 
1272
            this_pair = contents_pair(self.this_tree)
 
1273
            lca_pairs = [contents_pair(tree) for tree in self._lca_trees]
 
1274
            winner = self._lca_multi_way((base_pair, lca_pairs), other_pair,
 
1275
                                         this_pair, allow_overriding_lca=False)
 
1276
        else:
 
1277
            if base_pair == other_pair:
 
1278
                winner = 'this'
 
1279
            else:
 
1280
                # We delayed evaluating this_pair as long as we can to avoid
 
1281
                # unnecessary sha1 calculation
 
1282
                this_pair = contents_pair(self.this_tree)
 
1283
                winner = self._three_way(base_pair, other_pair, this_pair)
 
1284
        if winner == 'this':
 
1285
            # No interesting changes introduced by OTHER
 
1286
            return "unmodified"
 
1287
        # We have a hypothetical conflict, but if we have files, then we
 
1288
        # can try to merge the content
 
1289
        trans_id = self.tt.trans_id_file_id(file_id)
 
1290
        params = MergeHookParams(self, file_id, trans_id, this_pair[0],
 
1291
            other_pair[0], winner)
 
1292
        hooks = self.active_hooks
 
1293
        hook_status = 'not_applicable'
 
1294
        for hook in hooks:
 
1295
            hook_status, lines = hook.merge_contents(params)
 
1296
            if hook_status != 'not_applicable':
 
1297
                # Don't try any more hooks, this one applies.
 
1298
                break
 
1299
        result = "modified"
 
1300
        if hook_status == 'not_applicable':
 
1301
            # This is a contents conflict, because none of the available
 
1302
            # functions could merge it.
 
1303
            result = None
 
1304
            name = self.tt.final_name(trans_id)
 
1305
            parent_id = self.tt.final_parent(trans_id)
 
1306
            if self.this_tree.has_id(file_id):
 
1307
                self.tt.unversion_file(trans_id)
 
1308
            file_group = self._dump_conflicts(name, parent_id, file_id,
 
1309
                                              set_version=True)
 
1310
            self._raw_conflicts.append(('contents conflict', file_group))
 
1311
        elif hook_status == 'success':
 
1312
            self.tt.create_file(lines, trans_id)
 
1313
        elif hook_status == 'conflicted':
 
1314
            # XXX: perhaps the hook should be able to provide
 
1315
            # the BASE/THIS/OTHER files?
 
1316
            self.tt.create_file(lines, trans_id)
 
1317
            self._raw_conflicts.append(('text conflict', trans_id))
 
1318
            name = self.tt.final_name(trans_id)
 
1319
            parent_id = self.tt.final_parent(trans_id)
 
1320
            self._dump_conflicts(name, parent_id, file_id)
 
1321
        elif hook_status == 'delete':
 
1322
            self.tt.unversion_file(trans_id)
 
1323
            result = "deleted"
 
1324
        elif hook_status == 'done':
 
1325
            # The hook function did whatever it needs to do directly, no
 
1326
            # further action needed here.
 
1327
            pass
 
1328
        else:
 
1329
            raise AssertionError('unknown hook_status: %r' % (hook_status,))
 
1330
        if not self.this_tree.has_id(file_id) and result == "modified":
 
1331
            self.tt.version_file(file_id, trans_id)
 
1332
        # The merge has been performed, so the old contents should not be
 
1333
        # retained.
 
1334
        try:
 
1335
            self.tt.delete_contents(trans_id)
 
1336
        except errors.NoSuchFile:
 
1337
            pass
 
1338
        return result
 
1339
 
 
1340
    def _default_other_winner_merge(self, merge_hook_params):
 
1341
        """Replace this contents with other."""
 
1342
        file_id = merge_hook_params.file_id
 
1343
        trans_id = merge_hook_params.trans_id
 
1344
        file_in_this = self.this_tree.has_id(file_id)
 
1345
        if self.other_tree.has_id(file_id):
 
1346
            # OTHER changed the file
 
1347
            wt = self.this_tree
 
1348
            if wt.supports_content_filtering():
 
1349
                # We get the path from the working tree if it exists.
 
1350
                # That fails though when OTHER is adding a file, so
 
1351
                # we fall back to the other tree to find the path if
 
1352
                # it doesn't exist locally.
 
1353
                try:
 
1354
                    filter_tree_path = wt.id2path(file_id)
 
1355
                except errors.NoSuchId:
 
1356
                    filter_tree_path = self.other_tree.id2path(file_id)
 
1357
            else:
 
1358
                # Skip the id2path lookup for older formats
 
1359
                filter_tree_path = None
 
1360
            transform.create_from_tree(self.tt, trans_id,
 
1361
                             self.other_tree, file_id,
 
1362
                             filter_tree_path=filter_tree_path)
 
1363
            return 'done', None
 
1364
        elif file_in_this:
 
1365
            # OTHER deleted the file
 
1366
            return 'delete', None
 
1367
        else:
 
1368
            raise AssertionError(
 
1369
                'winner is OTHER, but file_id %r not in THIS or OTHER tree'
 
1370
                % (file_id,))
 
1371
 
 
1372
    def merge_contents(self, merge_hook_params):
 
1373
        """Fallback merge logic after user installed hooks."""
 
1374
        # This function is used in merge hooks as the fallback instance.
 
1375
        # Perhaps making this function and the functions it calls be a 
 
1376
        # a separate class would be better.
 
1377
        if merge_hook_params.winner == 'other':
 
1378
            # OTHER is a straight winner, so replace this contents with other
 
1379
            return self._default_other_winner_merge(merge_hook_params)
 
1380
        elif merge_hook_params.is_file_merge():
 
1381
            # THIS and OTHER are both files, so text merge.  Either
 
1382
            # BASE is a file, or both converted to files, so at least we
 
1383
            # have agreement that output should be a file.
 
1384
            try:
 
1385
                self.text_merge(merge_hook_params.file_id,
 
1386
                    merge_hook_params.trans_id)
 
1387
            except errors.BinaryFile:
 
1388
                return 'not_applicable', None
 
1389
            return 'done', None
 
1390
        else:
 
1391
            return 'not_applicable', None
862
1392
 
863
1393
    def get_lines(self, tree, file_id):
864
1394
        """Return the lines in a file, or an empty list."""
865
 
        if file_id in tree:
 
1395
        if tree.has_id(file_id):
866
1396
            return tree.get_file(file_id).readlines()
867
1397
        else:
868
1398
            return []
871
1401
        """Perform a three-way text merge on a file_id"""
872
1402
        # it's possible that we got here with base as a different type.
873
1403
        # if so, we just want two-way text conflicts.
874
 
        if file_id in self.base_tree and \
 
1404
        if self.base_tree.has_id(file_id) and \
875
1405
            self.base_tree.kind(file_id) == "file":
876
1406
            base_lines = self.get_lines(self.base_tree, file_id)
877
1407
        else:
878
1408
            base_lines = []
879
1409
        other_lines = self.get_lines(self.other_tree, file_id)
880
1410
        this_lines = self.get_lines(self.this_tree, file_id)
881
 
        m3 = Merge3(base_lines, this_lines, other_lines,
882
 
                    is_cherrypick=self.cherrypick)
 
1411
        m3 = merge3.Merge3(base_lines, this_lines, other_lines,
 
1412
                           is_cherrypick=self.cherrypick)
883
1413
        start_marker = "!START OF MERGE CONFLICT!" + "I HOPE THIS IS UNIQUE"
884
1414
        if self.show_base is True:
885
1415
            base_marker = '|' * 7
888
1418
 
889
1419
        def iter_merge3(retval):
890
1420
            retval["text_conflicts"] = False
891
 
            for line in m3.merge_lines(name_a = "TREE", 
892
 
                                       name_b = "MERGE-SOURCE", 
 
1421
            for line in m3.merge_lines(name_a = "TREE",
 
1422
                                       name_b = "MERGE-SOURCE",
893
1423
                                       name_base = "BASE-REVISION",
894
 
                                       start_marker=start_marker, 
 
1424
                                       start_marker=start_marker,
895
1425
                                       base_marker=base_marker,
896
1426
                                       reprocess=self.reprocess):
897
1427
                if line.startswith(start_marker):
906
1436
            self._raw_conflicts.append(('text conflict', trans_id))
907
1437
            name = self.tt.final_name(trans_id)
908
1438
            parent_id = self.tt.final_parent(trans_id)
909
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1439
            file_group = self._dump_conflicts(name, parent_id, file_id,
910
1440
                                              this_lines, base_lines,
911
1441
                                              other_lines)
912
1442
            file_group.append(trans_id)
913
1443
 
914
 
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None, 
 
1444
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None,
915
1445
                        base_lines=None, other_lines=None, set_version=False,
916
1446
                        no_base=False):
917
1447
        """Emit conflict files.
919
1449
        determined automatically.  If set_version is true, the .OTHER, .THIS
920
1450
        or .BASE (in that order) will be created as versioned files.
921
1451
        """
922
 
        data = [('OTHER', self.other_tree, other_lines), 
 
1452
        data = [('OTHER', self.other_tree, other_lines),
923
1453
                ('THIS', self.this_tree, this_lines)]
924
1454
        if not no_base:
925
1455
            data.append(('BASE', self.base_tree, base_lines))
 
1456
 
 
1457
        # We need to use the actual path in the working tree of the file here,
 
1458
        # ignoring the conflict suffixes
 
1459
        wt = self.this_tree
 
1460
        if wt.supports_content_filtering():
 
1461
            try:
 
1462
                filter_tree_path = wt.id2path(file_id)
 
1463
            except errors.NoSuchId:
 
1464
                # file has been deleted
 
1465
                filter_tree_path = None
 
1466
        else:
 
1467
            # Skip the id2path lookup for older formats
 
1468
            filter_tree_path = None
 
1469
 
926
1470
        versioned = False
927
1471
        file_group = []
928
1472
        for suffix, tree, lines in data:
929
 
            if file_id in tree:
 
1473
            if tree.has_id(file_id):
930
1474
                trans_id = self._conflict_file(name, parent_id, tree, file_id,
931
 
                                               suffix, lines)
 
1475
                                               suffix, lines, filter_tree_path)
932
1476
                file_group.append(trans_id)
933
1477
                if set_version and not versioned:
934
1478
                    self.tt.version_file(file_id, trans_id)
935
1479
                    versioned = True
936
1480
        return file_group
937
 
           
938
 
    def _conflict_file(self, name, parent_id, tree, file_id, suffix, 
939
 
                       lines=None):
 
1481
 
 
1482
    def _conflict_file(self, name, parent_id, tree, file_id, suffix,
 
1483
                       lines=None, filter_tree_path=None):
940
1484
        """Emit a single conflict file."""
941
1485
        name = name + '.' + suffix
942
1486
        trans_id = self.tt.create_path(name, parent_id)
943
 
        entry = tree.inventory[file_id]
944
 
        create_by_entry(self.tt, entry, tree, trans_id, lines)
 
1487
        transform.create_from_tree(self.tt, trans_id, tree, file_id, lines,
 
1488
            filter_tree_path)
945
1489
        return trans_id
946
1490
 
947
1491
    def merge_executable(self, file_id, file_status):
948
1492
        """Perform a merge on the execute bit."""
949
1493
        executable = [self.executable(t, file_id) for t in (self.base_tree,
950
1494
                      self.other_tree, self.this_tree)]
951
 
        self._merge_executable(file_id, executable, file_status)
 
1495
        self._merge_executable(file_id, executable, file_status,
 
1496
                               resolver=self._three_way)
952
1497
 
953
 
    def _merge_executable(self, file_id, executable, file_status):
 
1498
    def _merge_executable(self, file_id, executable, file_status,
 
1499
                          resolver):
954
1500
        """Perform a merge on the execute bit."""
955
1501
        base_executable, other_executable, this_executable = executable
956
1502
        if file_status == "deleted":
957
1503
            return
958
 
        winner = self._three_way(*executable)
 
1504
        winner = resolver(*executable)
959
1505
        if winner == "conflict":
960
1506
        # There must be a None in here, if we have a conflict, but we
961
1507
        # need executability since file status was not deleted.
969
1515
        try:
970
1516
            if self.tt.final_kind(trans_id) != "file":
971
1517
                return
972
 
        except NoSuchFile:
 
1518
        except errors.NoSuchFile:
973
1519
            return
974
1520
        if winner == "this":
975
1521
            executability = this_executable
976
1522
        else:
977
 
            if file_id in self.other_tree:
 
1523
            if self.other_tree.has_id(file_id):
978
1524
                executability = other_executable
979
 
            elif file_id in self.this_tree:
 
1525
            elif self.this_tree.has_id(file_id):
980
1526
                executability = this_executable
981
 
            elif file_id in self.base_tree:
 
1527
            elif self.base_tree_has_id(file_id):
982
1528
                executability = base_executable
983
1529
        if executability is not None:
984
1530
            trans_id = self.tt.trans_id_file_id(file_id)
986
1532
 
987
1533
    def cook_conflicts(self, fs_conflicts):
988
1534
        """Convert all conflicts into a form that doesn't depend on trans_id"""
989
 
        from conflicts import Conflict
990
1535
        name_conflicts = {}
991
 
        self.cooked_conflicts.extend(cook_conflicts(fs_conflicts, self.tt))
992
 
        fp = FinalPaths(self.tt)
 
1536
        self.cooked_conflicts.extend(transform.cook_conflicts(
 
1537
                fs_conflicts, self.tt))
 
1538
        fp = transform.FinalPaths(self.tt)
993
1539
        for conflict in self._raw_conflicts:
994
1540
            conflict_type = conflict[0]
995
1541
            if conflict_type in ('name conflict', 'parent conflict'):
997
1543
                conflict_args = conflict[2:]
998
1544
                if trans_id not in name_conflicts:
999
1545
                    name_conflicts[trans_id] = {}
1000
 
                unique_add(name_conflicts[trans_id], conflict_type, 
1001
 
                           conflict_args)
 
1546
                transform.unique_add(name_conflicts[trans_id], conflict_type,
 
1547
                                     conflict_args)
1002
1548
            if conflict_type == 'contents conflict':
1003
1549
                for trans_id in conflict[1]:
1004
1550
                    file_id = self.tt.final_file_id(trans_id)
1009
1555
                    if path.endswith(suffix):
1010
1556
                        path = path[:-len(suffix)]
1011
1557
                        break
1012
 
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
 
1558
                c = _mod_conflicts.Conflict.factory(conflict_type,
 
1559
                                                    path=path, file_id=file_id)
1013
1560
                self.cooked_conflicts.append(c)
1014
1561
            if conflict_type == 'text conflict':
1015
1562
                trans_id = conflict[1]
1016
1563
                path = fp.get_path(trans_id)
1017
1564
                file_id = self.tt.final_file_id(trans_id)
1018
 
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
 
1565
                c = _mod_conflicts.Conflict.factory(conflict_type,
 
1566
                                                    path=path, file_id=file_id)
1019
1567
                self.cooked_conflicts.append(c)
1020
1568
 
1021
1569
        for trans_id, conflicts in name_conflicts.iteritems():
1036
1584
            if this_parent is not None and this_name is not None:
1037
1585
                this_parent_path = \
1038
1586
                    fp.get_path(self.tt.trans_id_file_id(this_parent))
1039
 
                this_path = pathjoin(this_parent_path, this_name)
 
1587
                this_path = osutils.pathjoin(this_parent_path, this_name)
1040
1588
            else:
1041
1589
                this_path = "<deleted>"
1042
1590
            file_id = self.tt.final_file_id(trans_id)
1043
 
            c = Conflict.factory('path conflict', path=this_path,
1044
 
                                 conflict_path=other_path, file_id=file_id)
 
1591
            c = _mod_conflicts.Conflict.factory('path conflict', path=this_path,
 
1592
                                                conflict_path=other_path,
 
1593
                                                file_id=file_id)
1045
1594
            self.cooked_conflicts.append(c)
1046
 
        self.cooked_conflicts.sort(key=Conflict.sort_key)
 
1595
        self.cooked_conflicts.sort(key=_mod_conflicts.Conflict.sort_key)
1047
1596
 
1048
1597
 
1049
1598
class WeaveMerger(Merge3Merger):
1053
1602
    supports_reverse_cherrypick = False
1054
1603
    history_based = True
1055
1604
 
1056
 
    def _merged_lines(self, file_id):
1057
 
        """Generate the merged lines.
1058
 
        There is no distinction between lines that are meant to contain <<<<<<<
1059
 
        and conflicts.
1060
 
        """
1061
 
        if self.cherrypick:
1062
 
            base = self.base_tree
1063
 
        else:
1064
 
            base = None
1065
 
        plan = self.this_tree.plan_file_merge(file_id, self.other_tree,
 
1605
    def _generate_merge_plan(self, file_id, base):
 
1606
        return self.this_tree.plan_file_merge(file_id, self.other_tree,
1066
1607
                                              base=base)
 
1608
 
 
1609
    def _merged_lines(self, file_id):
 
1610
        """Generate the merged lines.
 
1611
        There is no distinction between lines that are meant to contain <<<<<<<
 
1612
        and conflicts.
 
1613
        """
 
1614
        if self.cherrypick:
 
1615
            base = self.base_tree
 
1616
        else:
 
1617
            base = None
 
1618
        plan = self._generate_merge_plan(file_id, base)
1067
1619
        if 'merge' in debug.debug_flags:
1068
1620
            plan = list(plan)
1069
1621
            trans_id = self.tt.trans_id_file_id(file_id)
1070
1622
            name = self.tt.final_name(trans_id) + '.plan'
1071
 
            contents = ('%10s|%s' % l for l in plan)
 
1623
            contents = ('%11s|%s' % l for l in plan)
1072
1624
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
1073
 
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
1074
 
            '>>>>>>> MERGE-SOURCE\n')
1075
 
        return textmerge.merge_lines(self.reprocess)
 
1625
        textmerge = versionedfile.PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1626
                                                 '>>>>>>> MERGE-SOURCE\n')
 
1627
        lines, conflicts = textmerge.merge_lines(self.reprocess)
 
1628
        if conflicts:
 
1629
            base_lines = textmerge.base_from_plan()
 
1630
        else:
 
1631
            base_lines = None
 
1632
        return lines, base_lines
1076
1633
 
1077
1634
    def text_merge(self, file_id, trans_id):
1078
1635
        """Perform a (weave) text merge for a given file and file-id.
1079
1636
        If conflicts are encountered, .THIS and .OTHER files will be emitted,
1080
1637
        and a conflict will be noted.
1081
1638
        """
1082
 
        lines, conflicts = self._merged_lines(file_id)
 
1639
        lines, base_lines = self._merged_lines(file_id)
1083
1640
        lines = list(lines)
1084
 
        # Note we're checking whether the OUTPUT is binary in this case, 
 
1641
        # Note we're checking whether the OUTPUT is binary in this case,
1085
1642
        # because we don't want to get into weave merge guts.
1086
 
        check_text_lines(lines)
 
1643
        textfile.check_text_lines(lines)
1087
1644
        self.tt.create_file(lines, trans_id)
1088
 
        if conflicts:
 
1645
        if base_lines is not None:
 
1646
            # Conflict
1089
1647
            self._raw_conflicts.append(('text conflict', trans_id))
1090
1648
            name = self.tt.final_name(trans_id)
1091
1649
            parent_id = self.tt.final_parent(trans_id)
1092
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
1093
 
                                              no_base=True)
 
1650
            file_group = self._dump_conflicts(name, parent_id, file_id,
 
1651
                                              no_base=False,
 
1652
                                              base_lines=base_lines)
1094
1653
            file_group.append(trans_id)
1095
1654
 
1096
1655
 
1097
1656
class LCAMerger(WeaveMerger):
1098
1657
 
1099
 
    def _merged_lines(self, file_id):
1100
 
        """Generate the merged lines.
1101
 
        There is no distinction between lines that are meant to contain <<<<<<<
1102
 
        and conflicts.
1103
 
        """
1104
 
        if self.cherrypick:
1105
 
            base = self.base_tree
1106
 
        else:
1107
 
            base = None
1108
 
        plan = self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
 
1658
    def _generate_merge_plan(self, file_id, base):
 
1659
        return self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
1109
1660
                                                  base=base)
1110
 
        if 'merge' in debug.debug_flags:
1111
 
            plan = list(plan)
1112
 
            trans_id = self.tt.trans_id_file_id(file_id)
1113
 
            name = self.tt.final_name(trans_id) + '.plan'
1114
 
            contents = ('%10s|%s' % l for l in plan)
1115
 
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
1116
 
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
1117
 
            '>>>>>>> MERGE-SOURCE\n')
1118
 
        return textmerge.merge_lines(self.reprocess)
1119
 
 
1120
1661
 
1121
1662
class Diff3Merger(Merge3Merger):
1122
1663
    """Three-way merger using external diff3 for text merging"""
1123
1664
 
1124
1665
    def dump_file(self, temp_dir, name, tree, file_id):
1125
 
        out_path = pathjoin(temp_dir, name)
 
1666
        out_path = osutils.pathjoin(temp_dir, name)
1126
1667
        out_file = open(out_path, "wb")
1127
1668
        try:
1128
1669
            in_file = tree.get_file(file_id)
1140
1681
        import bzrlib.patch
1141
1682
        temp_dir = osutils.mkdtemp(prefix="bzr-")
1142
1683
        try:
1143
 
            new_file = pathjoin(temp_dir, "new")
 
1684
            new_file = osutils.pathjoin(temp_dir, "new")
1144
1685
            this = self.dump_file(temp_dir, "this", self.this_tree, file_id)
1145
1686
            base = self.dump_file(temp_dir, "base", self.base_tree, file_id)
1146
1687
            other = self.dump_file(temp_dir, "other", self.other_tree, file_id)
1147
1688
            status = bzrlib.patch.diff3(new_file, this, base, other)
1148
1689
            if status not in (0, 1):
1149
 
                raise BzrError("Unhandled diff3 exit code")
 
1690
                raise errors.BzrError("Unhandled diff3 exit code")
1150
1691
            f = open(new_file, 'rb')
1151
1692
            try:
1152
1693
                self.tt.create_file(f, trans_id)
1170
1711
                other_rev_id=None,
1171
1712
                interesting_files=None,
1172
1713
                this_tree=None,
1173
 
                pb=DummyProgress(),
 
1714
                pb=None,
1174
1715
                change_reporter=None):
1175
 
    """Primary interface for merging. 
 
1716
    """Primary interface for merging.
1176
1717
 
1177
 
        typical use is probably 
 
1718
        typical use is probably
1178
1719
        'merge_inner(branch, branch.get_revision_tree(other_revision),
1179
1720
                     branch.get_revision_tree(base_revision))'
1180
1721
        """
1181
1722
    if this_tree is None:
1182
 
        raise BzrError("bzrlib.merge.merge_inner requires a this_tree "
1183
 
            "parameter as of bzrlib version 0.8.")
 
1723
        raise errors.BzrError("bzrlib.merge.merge_inner requires a this_tree "
 
1724
                              "parameter as of bzrlib version 0.8.")
1184
1725
    merger = Merger(this_branch, other_tree, base_tree, this_tree=this_tree,
1185
1726
                    pb=pb, change_reporter=change_reporter)
1186
1727
    merger.backup_files = backup_files
1199
1740
    get_revision_id = getattr(base_tree, 'get_revision_id', None)
1200
1741
    if get_revision_id is None:
1201
1742
        get_revision_id = base_tree.last_revision
 
1743
    merger.cache_trees_with_revision_ids([other_tree, base_tree, this_tree])
1202
1744
    merger.set_base_revision(get_revision_id(), this_branch)
1203
1745
    return merger.do_merge()
1204
1746
 
1272
1814
 
1273
1815
    def get_lines(self, revisions):
1274
1816
        """Get lines for revisions from the backing VersionedFiles.
1275
 
        
 
1817
 
1276
1818
        :raises RevisionNotPresent: on absent texts.
1277
1819
        """
1278
1820
        keys = [(self._key_prefix + (rev,)) for rev in revisions]
1280
1822
        for record in self.vf.get_record_stream(keys, 'unordered', True):
1281
1823
            if record.storage_kind == 'absent':
1282
1824
                raise errors.RevisionNotPresent(record.key, self.vf)
1283
 
            result[record.key[-1]] = osutils.split_lines(
1284
 
                record.get_bytes_as('fulltext'))
 
1825
            result[record.key[-1]] = osutils.chunks_to_lines(
 
1826
                record.get_bytes_as('chunked'))
1285
1827
        return result
1286
1828
 
1287
1829
    def plan_merge(self):
1403
1945
        super(_PlanMerge, self).__init__(a_rev, b_rev, vf, key_prefix)
1404
1946
        self.a_key = self._key_prefix + (self.a_rev,)
1405
1947
        self.b_key = self._key_prefix + (self.b_rev,)
1406
 
        self.graph = Graph(self.vf)
 
1948
        self.graph = _mod_graph.Graph(self.vf)
1407
1949
        heads = self.graph.heads((self.a_key, self.b_key))
1408
1950
        if len(heads) == 1:
1409
1951
            # one side dominates, so we can just return its values, yay for
1414
1956
                other = b_rev
1415
1957
            else:
1416
1958
                other = a_rev
1417
 
            mutter('found dominating revision for %s\n%s > %s', self.vf,
1418
 
                   self._head_key[-1], other)
 
1959
            trace.mutter('found dominating revision for %s\n%s > %s', self.vf,
 
1960
                         self._head_key[-1], other)
1419
1961
            self._weave = None
1420
1962
        else:
1421
1963
            self._head_key = None
1435
1977
        while True:
1436
1978
            next_lcas = self.graph.find_lca(*cur_ancestors)
1437
1979
            # Map a plain NULL_REVISION to a simple no-ancestors
1438
 
            if next_lcas == set([NULL_REVISION]):
 
1980
            if next_lcas == set([_mod_revision.NULL_REVISION]):
1439
1981
                next_lcas = ()
1440
1982
            # Order the lca's based on when they were merged into the tip
1441
1983
            # While the actual merge portion of weave merge uses a set() of
1453
1995
            elif len(next_lcas) > 2:
1454
1996
                # More than 2 lca's, fall back to grabbing all nodes between
1455
1997
                # this and the unique lca.
1456
 
                mutter('More than 2 LCAs, falling back to all nodes for:'
1457
 
                       ' %s, %s\n=> %s', self.a_key, self.b_key, cur_ancestors)
 
1998
                trace.mutter('More than 2 LCAs, falling back to all nodes for:'
 
1999
                             ' %s, %s\n=> %s',
 
2000
                             self.a_key, self.b_key, cur_ancestors)
1458
2001
                cur_lcas = next_lcas
1459
2002
                while len(cur_lcas) > 1:
1460
2003
                    cur_lcas = self.graph.find_lca(*cur_lcas)
1463
2006
                    unique_lca = None
1464
2007
                else:
1465
2008
                    unique_lca = list(cur_lcas)[0]
1466
 
                    if unique_lca == NULL_REVISION:
 
2009
                    if unique_lca == _mod_revision.NULL_REVISION:
1467
2010
                        # find_lca will return a plain 'NULL_REVISION' rather
1468
2011
                        # than a key tuple when there is no common ancestor, we
1469
2012
                        # prefer to just use None, because it doesn't confuse
1477
2020
 
1478
2021
    def _find_unique_parents(self, tip_keys, base_key):
1479
2022
        """Find ancestors of tip that aren't ancestors of base.
1480
 
        
 
2023
 
1481
2024
        :param tip_keys: Nodes that are interesting
1482
2025
        :param base_key: Cull all ancestors of this node
1483
2026
        :return: The parent map for all revisions between tip_keys and
1492
2035
            # We remove NULL_REVISION because it isn't a proper tuple key, and
1493
2036
            # thus confuses things like _get_interesting_texts, and our logic
1494
2037
            # to add the texts into the memory weave.
1495
 
            if NULL_REVISION in parent_map:
1496
 
                parent_map.pop(NULL_REVISION)
 
2038
            if _mod_revision.NULL_REVISION in parent_map:
 
2039
                parent_map.pop(_mod_revision.NULL_REVISION)
1497
2040
        else:
1498
2041
            interesting = set()
1499
2042
            for tip in tip_keys:
1543
2086
    @staticmethod
1544
2087
    def _prune_tails(parent_map, child_map, tails_to_remove):
1545
2088
        """Remove tails from the parent map.
1546
 
        
 
2089
 
1547
2090
        This will remove the supplied revisions until no more children have 0
1548
2091
        parents.
1549
2092
 
1651
2194
        lcas = graph.find_lca(key_prefix + (a_rev,), key_prefix + (b_rev,))
1652
2195
        self.lcas = set()
1653
2196
        for lca in lcas:
1654
 
            if lca == NULL_REVISION:
 
2197
            if lca == _mod_revision.NULL_REVISION:
1655
2198
                self.lcas.add(lca)
1656
2199
            else:
1657
2200
                self.lcas.add(lca[-1])