~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge.py

  • Committer: John Arbash Meinel
  • Date: 2008-08-25 21:50:11 UTC
  • mfrom: (0.11.3 tools)
  • mto: This revision was merged to the branch mainline in revision 3659.
  • Revision ID: john@arbash-meinel.com-20080825215011-de9esmzgkue3e522
Merge in Lukáš's helper scripts.
Update the packaging documents to describe how to do the releases
using bzr-builddeb to package all distro platforms
simultaneously.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2008 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
import errno
 
19
from itertools import chain
 
20
import os
17
21
import warnings
18
22
 
19
23
from bzrlib import (
20
 
    branch as _mod_branch,
21
 
    conflicts as _mod_conflicts,
22
24
    debug,
23
 
    decorators,
24
25
    errors,
25
26
    graph as _mod_graph,
26
 
    hooks,
27
 
    merge3,
28
27
    osutils,
29
28
    patiencediff,
 
29
    registry,
30
30
    revision as _mod_revision,
31
 
    textfile,
32
 
    trace,
33
 
    transform,
34
 
    tree as _mod_tree,
35
31
    tsort,
36
 
    ui,
37
 
    versionedfile
38
 
    )
39
 
from bzrlib.cleanup import OperationWithCleanups
40
 
from bzrlib.symbol_versioning import (
41
 
    deprecated_in,
42
 
    deprecated_method,
43
 
    )
 
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
 
44
62
# TODO: Report back as changes are merged in
45
63
 
46
64
 
47
65
def transform_tree(from_tree, to_tree, interesting_ids=None):
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_tree.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)
 
66
    merge_inner(from_tree.branch, to_tree, from_tree, ignore_zero=True,
 
67
                interesting_ids=interesting_ids, this_tree=from_tree)
214
68
 
215
69
 
216
70
class Merger(object):
217
 
 
218
 
    hooks = MergeHooks()
219
 
 
220
71
    def __init__(self, this_branch, other_tree=None, base_tree=None,
221
 
                 this_tree=None, pb=None, change_reporter=None,
 
72
                 this_tree=None, pb=DummyProgress(), change_reporter=None,
222
73
                 recurse='down', revision_graph=None):
223
74
        object.__init__(self)
224
75
        self.this_branch = this_branch
237
88
        self.interesting_files = None
238
89
        self.show_base = False
239
90
        self.reprocess = False
240
 
        if pb is not None:
241
 
            warnings.warn("pb parameter to Merger() is deprecated and ignored")
 
91
        self._pb = pb
242
92
        self.pp = None
243
93
        self.recurse = recurse
244
94
        self.change_reporter = change_reporter
246
96
        self._revision_graph = revision_graph
247
97
        self._base_is_ancestor = None
248
98
        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
262
99
 
263
100
    @property
264
101
    def revision_graph(self):
292
129
                                      _set_base_is_other_ancestor)
293
130
 
294
131
    @staticmethod
295
 
    def from_uncommitted(tree, other_tree, pb=None, base_tree=None):
 
132
    def from_uncommitted(tree, other_tree, pb):
296
133
        """Return a Merger for uncommitted changes in other_tree.
297
134
 
298
135
        :param tree: The tree to merge into
299
136
        :param other_tree: The tree to get uncommitted changes from
300
137
        :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.
303
138
        """
304
 
        if base_tree is None:
305
 
            base_tree = other_tree.basis_tree()
306
 
        merger = Merger(tree.branch, other_tree, base_tree, tree, pb)
 
139
        merger = Merger(tree.branch, other_tree, other_tree.basis_tree(), tree,
 
140
                        pb)
307
141
        merger.base_rev_id = merger.base_tree.get_revision_id()
308
142
        merger.other_rev_id = None
309
143
        merger.other_basis = merger.base_rev_id
327
161
                base_revision_id, tree.branch.last_revision())):
328
162
                base_revision_id = None
329
163
            else:
330
 
                trace.warning('Performing cherrypick')
 
164
                warning('Performing cherrypick')
331
165
        merger = klass.from_revision_ids(pb, tree, other_revision_id,
332
166
                                         base_revision_id, revision_graph=
333
167
                                         revision_graph)
335
169
 
336
170
    @staticmethod
337
171
    def from_revision_ids(pb, tree, other, base=None, other_branch=None,
338
 
                          base_branch=None, revision_graph=None,
339
 
                          tree_branch=None):
 
172
                          base_branch=None, revision_graph=None):
340
173
        """Return a Merger for revision-ids.
341
174
 
342
 
        :param pb: A progress indicator
343
175
        :param tree: The tree to merge changes into
344
176
        :param other: The revision-id to use as OTHER
345
177
        :param base: The revision-id to use as BASE.  If not specified, will
350
182
            not supplied, other_branch or tree.branch will be used.
351
183
        :param revision_graph: If you have a revision_graph precomputed, pass
352
184
            it in, otherwise it will be created for you.
353
 
        :param tree_branch: The branch associated with tree.  If not supplied,
354
 
            tree.branch will be used.
 
185
        :param pb: A progress indicator
355
186
        """
356
 
        if tree_branch is None:
357
 
            tree_branch = tree.branch
358
 
        merger = Merger(tree_branch, this_tree=tree, pb=pb,
 
187
        merger = Merger(tree.branch, this_tree=tree, pb=pb,
359
188
                        revision_graph=revision_graph)
360
189
        if other_branch is None:
361
190
            other_branch = tree.branch
385
214
        if revno is None:
386
215
            tree = workingtree.WorkingTree.open_containing(location)[0]
387
216
            return tree.branch, tree
388
 
        branch = _mod_branch.Branch.open_containing(
389
 
            location, possible_transports)[0]
 
217
        branch = Branch.open_containing(location, possible_transports)[0]
390
218
        if revno == -1:
391
219
            revision_id = branch.last_revision()
392
220
        else:
393
221
            revision_id = branch.get_rev_id(revno)
394
 
        revision_id = _mod_revision.ensure_null(revision_id)
 
222
        revision_id = ensure_null(revision_id)
395
223
        return branch, self.revision_tree(revision_id, branch)
396
224
 
397
 
    @deprecated_method(deprecated_in((2, 1, 0)))
398
225
    def ensure_revision_trees(self):
399
226
        if self.this_revision_tree is None:
400
227
            self.this_basis_tree = self.revision_tree(self.this_basis)
403
230
 
404
231
        if self.other_rev_id is None:
405
232
            other_basis_tree = self.revision_tree(self.other_basis)
406
 
            if other_basis_tree.has_changes(self.other_tree):
407
 
                raise errors.WorkingTreeNotRevision(self.this_tree)
 
233
            changes = other_basis_tree.changes_from(self.other_tree)
 
234
            if changes.has_changed():
 
235
                raise WorkingTreeNotRevision(self.this_tree)
408
236
            other_rev_id = self.other_basis
409
237
            self.other_tree = other_basis_tree
410
238
 
411
 
    @deprecated_method(deprecated_in((2, 1, 0)))
412
239
    def file_revisions(self, file_id):
413
240
        self.ensure_revision_trees()
414
241
        def get_id(tree, file_id):
417
244
        if self.this_rev_id is None:
418
245
            if self.this_basis_tree.get_file_sha1(file_id) != \
419
246
                self.this_tree.get_file_sha1(file_id):
420
 
                raise errors.WorkingTreeNotRevision(self.this_tree)
 
247
                raise WorkingTreeNotRevision(self.this_tree)
421
248
 
422
249
        trees = (self.this_basis_tree, self.other_tree)
423
250
        return [get_id(tree, file_id) for tree in trees]
424
251
 
425
 
    @deprecated_method(deprecated_in((2, 1, 0)))
426
252
    def check_basis(self, check_clean, require_commits=True):
427
253
        if self.this_basis is None and require_commits is True:
428
 
            raise errors.BzrCommandError(
429
 
                "This branch has no commits."
430
 
                " (perhaps you would prefer 'bzr pull')")
 
254
            raise BzrCommandError("This branch has no commits."
 
255
                                  " (perhaps you would prefer 'bzr pull')")
431
256
        if check_clean:
432
257
            self.compare_basis()
433
258
            if self.this_basis != self.this_rev_id:
434
259
                raise errors.UncommittedChanges(self.this_tree)
435
260
 
436
 
    @deprecated_method(deprecated_in((2, 1, 0)))
437
261
    def compare_basis(self):
438
262
        try:
439
263
            basis_tree = self.revision_tree(self.this_tree.last_revision())
440
264
        except errors.NoSuchRevision:
441
265
            basis_tree = self.this_tree.basis_tree()
442
 
        if not self.this_tree.has_changes(basis_tree):
 
266
        changes = self.this_tree.changes_from(basis_tree)
 
267
        if not changes.has_changed():
443
268
            self.this_rev_id = self.this_basis
444
269
 
445
270
    def set_interesting_files(self, file_list):
446
271
        self.interesting_files = file_list
447
272
 
448
273
    def set_pending(self):
449
 
        if (not self.base_is_ancestor or not self.base_is_other_ancestor
450
 
            or self.other_rev_id is None):
 
274
        if not self.base_is_ancestor or not self.base_is_other_ancestor or self.other_rev_id is None:
451
275
            return
452
276
        self._add_parent()
453
277
 
454
278
    def _add_parent(self):
455
279
        new_parents = self.this_tree.get_parent_ids() + [self.other_rev_id]
456
280
        new_parent_trees = []
457
 
        operation = OperationWithCleanups(self.this_tree.set_parent_trees)
458
281
        for revision_id in new_parents:
459
282
            try:
460
283
                tree = self.revision_tree(revision_id)
462
285
                tree = None
463
286
            else:
464
287
                tree.lock_read()
465
 
                operation.add_cleanup(tree.unlock)
466
288
            new_parent_trees.append((revision_id, tree))
467
 
        operation.run_simple(new_parent_trees, allow_leftmost_as_ghost=True)
 
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()
468
296
 
469
297
    def set_other(self, other_revision, possible_transports=None):
470
298
        """Set the revision and tree to merge from.
479
307
            self.other_rev_id = _mod_revision.ensure_null(
480
308
                self.other_branch.last_revision())
481
309
            if _mod_revision.is_null(self.other_rev_id):
482
 
                raise errors.NoCommits(self.other_branch)
 
310
                raise NoCommits(self.other_branch)
483
311
            self.other_basis = self.other_rev_id
484
312
        elif other_revision[1] is not None:
485
313
            self.other_rev_id = self.other_branch.get_rev_id(other_revision[1])
488
316
            self.other_rev_id = None
489
317
            self.other_basis = self.other_branch.last_revision()
490
318
            if self.other_basis is None:
491
 
                raise errors.NoCommits(self.other_branch)
 
319
                raise NoCommits(self.other_branch)
492
320
        if self.other_rev_id is not None:
493
321
            self._cached_trees[self.other_rev_id] = self.other_tree
494
322
        self._maybe_fetch(self.other_branch,self.this_branch, self.other_basis)
521
349
            target.fetch(source, revision_id)
522
350
 
523
351
    def find_base(self):
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
 
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
530
356
        else:
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)
 
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)
568
365
        self.base_is_ancestor = True
569
366
        self.base_is_other_ancestor = True
570
 
        trace.mutter('Base revid: %r' % self.base_rev_id)
571
367
 
572
368
    def set_base(self, base_revision):
573
369
        """Set the base revision to use for the merge.
574
370
 
575
371
        :param base_revision: A 2-list containing a path and revision number.
576
372
        """
577
 
        trace.mutter("doing merge() with no base_revision specified")
 
373
        mutter("doing merge() with no base_revision specified")
578
374
        if base_revision == [None, None]:
579
375
            self.find_base()
580
376
        else:
593
389
                  'other_tree': self.other_tree,
594
390
                  'interesting_ids': self.interesting_ids,
595
391
                  'interesting_files': self.interesting_files,
596
 
                  'this_branch': self.this_branch,
 
392
                  'pp': self.pp,
597
393
                  'do_merge': False}
598
394
        if self.merge_type.requires_base:
599
395
            kwargs['base_tree'] = self.base_tree
600
396
        if self.merge_type.supports_reprocess:
601
397
            kwargs['reprocess'] = self.reprocess
602
398
        elif self.reprocess:
603
 
            raise errors.BzrError(
604
 
                "Conflict reduction is not supported for merge"
605
 
                " type %s." % self.merge_type)
 
399
            raise BzrError("Conflict reduction is not supported for merge"
 
400
                                  " type %s." % self.merge_type)
606
401
        if self.merge_type.supports_show_base:
607
402
            kwargs['show_base'] = self.show_base
608
403
        elif self.show_base:
609
 
            raise errors.BzrError("Showing base is not supported for this"
610
 
                                  " merge type. %s" % self.merge_type)
 
404
            raise BzrError("Showing base is not supported for this"
 
405
                           " merge type. %s" % self.merge_type)
611
406
        if (not getattr(self.merge_type, 'supports_reverse_cherrypick', True)
612
407
            and not self.base_is_other_ancestor):
613
408
            raise errors.CannotReverseCherrypick()
614
409
        if self.merge_type.supports_cherrypick:
615
410
            kwargs['cherrypick'] = (not self.base_is_ancestor or
616
411
                                    not self.base_is_other_ancestor)
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,
 
412
        return self.merge_type(pb=self._pb,
621
413
                               change_reporter=self.change_reporter,
622
414
                               **kwargs)
623
415
 
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
 
 
647
416
    def do_merge(self):
648
 
        operation = OperationWithCleanups(self._do_merge_to)
649
417
        self.this_tree.lock_tree_write()
650
 
        operation.add_cleanup(self.this_tree.unlock)
651
418
        if self.base_tree is not None:
652
419
            self.base_tree.lock_read()
653
 
            operation.add_cleanup(self.base_tree.unlock)
654
420
        if self.other_tree is not None:
655
421
            self.other_tree.lock_read()
656
 
            operation.add_cleanup(self.other_tree.unlock)
657
 
        merge = operation.run_simple()
 
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()
658
448
        if len(merge.cooked_conflicts) == 0:
659
 
            if not self.ignore_zero and not trace.is_quiet():
660
 
                trace.note("All changes applied successfully.")
 
449
            if not self.ignore_zero and not is_quiet():
 
450
                note("All changes applied successfully.")
661
451
        else:
662
 
            trace.note("%d conflicts encountered."
663
 
                       % len(merge.cooked_conflicts))
 
452
            note("%d conflicts encountered." % len(merge.cooked_conflicts))
664
453
 
665
454
        return len(merge.cooked_conflicts)
666
455
 
667
456
 
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
 
 
685
457
class Merge3Merger(object):
686
458
    """Three-way merger that uses the merge3 text merger"""
687
459
    requires_base = True
691
463
    supports_cherrypick = True
692
464
    supports_reverse_cherrypick = True
693
465
    winner_idx = {"this": 2, "other": 1, "conflict": 1}
694
 
    supports_lca_trees = True
695
466
 
696
 
    def __init__(self, working_tree, this_tree, base_tree, other_tree,
 
467
    def __init__(self, working_tree, this_tree, base_tree, other_tree, 
697
468
                 interesting_ids=None, reprocess=False, show_base=False,
698
 
                 pb=None, pp=None, change_reporter=None,
 
469
                 pb=DummyProgress(), pp=None, change_reporter=None,
699
470
                 interesting_files=None, do_merge=True,
700
 
                 cherrypick=False, lca_trees=None, this_branch=None):
 
471
                 cherrypick=False):
701
472
        """Initialize the merger object and perform the merge.
702
473
 
703
474
        :param working_tree: The working tree to apply the merge to
704
475
        :param this_tree: The local tree in the merge operation
705
476
        :param base_tree: The common tree in the merge operation
706
 
        :param other_tree: The other tree to merge changes from
707
 
        :param this_branch: The branch associated with this_tree
 
477
        :param other_tree: The other other tree to merge changes from
708
478
        :param interesting_ids: The file_ids of files that should be
709
479
            participate in the merge.  May not be combined with
710
480
            interesting_files.
711
481
        :param: reprocess If True, perform conflict-reduction processing.
712
482
        :param show_base: If True, show the base revision in text conflicts.
713
483
            (incompatible with reprocess)
714
 
        :param pb: ignored
 
484
        :param pb: A Progress bar
715
485
        :param pp: A ProgressPhase object
716
486
        :param change_reporter: An object that should report changes made
717
487
        :param interesting_files: The tree-relative paths of files that should
720
490
            be combined with interesting_ids.  If neither interesting_files nor
721
491
            interesting_ids is specified, all files may participate in the
722
492
            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.
726
493
        """
727
494
        object.__init__(self)
728
495
        if interesting_files is not None and interesting_ids is not None:
733
500
        self.this_tree = working_tree
734
501
        self.base_tree = base_tree
735
502
        self.other_tree = other_tree
736
 
        self.this_branch = this_branch
737
503
        self._raw_conflicts = []
738
504
        self.cooked_conflicts = []
739
505
        self.reprocess = reprocess
740
506
        self.show_base = show_base
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]
 
507
        self.pb = pb
 
508
        self.pp = pp
747
509
        self.change_reporter = change_reporter
748
510
        self.cherrypick = cherrypick
 
511
        if self.pp is None:
 
512
            self.pp = ProgressPhase("Merge phase", 3, self.pb)
749
513
        if do_merge:
750
514
            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")
755
515
 
756
516
    def do_merge(self):
757
 
        operation = OperationWithCleanups(self._do_merge)
758
517
        self.this_tree.lock_tree_write()
759
 
        operation.add_cleanup(self.this_tree.unlock)
760
518
        self.base_tree.lock_read()
761
 
        operation.add_cleanup(self.base_tree.unlock)
762
519
        self.other_tree.lock_read()
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)
 
520
        self.tt = TreeTransform(self.this_tree, self.pb)
772
521
        try:
773
 
            self.this_tree.add_conflicts(self.cooked_conflicts)
774
 
        except errors.UnsupportedOperation:
775
 
            pass
 
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()
776
537
 
777
538
    def make_preview_transform(self):
778
 
        operation = OperationWithCleanups(self._make_preview_transform)
779
539
        self.base_tree.lock_read()
780
 
        operation.add_cleanup(self.base_tree.unlock)
781
540
        self.other_tree.lock_read()
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()
 
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()
788
550
        return self.tt
789
551
 
790
552
    def _compute_transform(self):
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
 
553
        entries = self._entries3()
797
554
        child_pb = ui.ui_factory.nested_progress_bar()
798
555
        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]
802
556
            for num, (file_id, changed, parents3, names3,
803
557
                      executable3) in enumerate(entries):
804
558
                child_pb.update('Preparing file merge', num, len(entries))
805
 
                self._merge_names(file_id, parents3, names3, resolver=resolver)
 
559
                self._merge_names(file_id, parents3, names3)
806
560
                if changed:
807
 
                    file_status = self._do_merge_contents(file_id)
 
561
                    file_status = self.merge_contents(file_id)
808
562
                else:
809
563
                    file_status = 'unmodified'
810
564
                self._merge_executable(file_id,
811
 
                    executable3, file_status, resolver=resolver)
 
565
                    executable3, file_status)
812
566
        finally:
813
567
            child_pb.finished()
814
568
        self.fix_root()
 
569
        self.pp.next_phase()
815
570
        child_pb = ui.ui_factory.nested_progress_bar()
816
571
        try:
817
 
            fs_conflicts = transform.resolve_conflicts(self.tt, child_pb,
818
 
                lambda t, c: transform.conflict_pass(t, c, self.other_tree))
 
572
            fs_conflicts = resolve_conflicts(self.tt, child_pb,
 
573
                lambda t, c: conflict_pass(t, c, self.other_tree))
819
574
        finally:
820
575
            child_pb.finished()
821
576
        if self.change_reporter is not None:
824
579
                self.tt.iter_changes(), self.change_reporter)
825
580
        self.cook_conflicts(fs_conflicts)
826
581
        for conflict in self.cooked_conflicts:
827
 
            trace.warning(conflict)
 
582
            warning(conflict)
828
583
 
829
584
    def _entries3(self):
830
585
        """Gather data about files modified between three trees.
839
594
        iterator = self.other_tree.iter_changes(self.base_tree,
840
595
                include_unchanged=True, specific_files=self.interesting_files,
841
596
                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))
845
597
        for (file_id, paths, changed, versioned, parents, names, kind,
846
598
             executable) in iterator:
847
599
            if (self.interesting_ids is not None and
848
600
                file_id not in self.interesting_ids):
849
601
                continue
850
 
            entry = this_entries.get(file_id)
851
 
            if entry is not None:
 
602
            if file_id in self.this_tree.inventory:
 
603
                entry = self.this_tree.inventory[file_id]
852
604
                this_name = entry.name
853
605
                this_parent = entry.parent_id
854
606
                this_executable = entry.executable
862
614
            result.append((file_id, changed, parents3, names3, executable3))
863
615
        return result
864
616
 
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
 
 
1032
617
    def fix_root(self):
1033
618
        try:
1034
619
            self.tt.final_kind(self.tt.root)
1035
 
        except errors.NoSuchFile:
 
620
        except NoSuchFile:
1036
621
            self.tt.cancel_deletion(self.tt.root)
1037
622
        if self.tt.final_file_id(self.tt.root) is None:
1038
 
            self.tt.version_file(self.tt.tree_file_id(self.tt.root),
 
623
            self.tt.version_file(self.tt.tree_file_id(self.tt.root), 
1039
624
                                 self.tt.root)
 
625
        if self.other_tree.inventory.root is None:
 
626
            return
1040
627
        other_root_file_id = self.other_tree.get_root_id()
1041
 
        if other_root_file_id is None:
1042
 
            return
1043
628
        other_root = self.tt.trans_id_file_id(other_root_file_id)
1044
629
        if other_root == self.tt.root:
1045
630
            return
1046
631
        try:
1047
632
            self.tt.final_kind(other_root)
1048
 
        except errors.NoSuchFile:
1049
 
            return
1050
 
        if self.this_tree.has_id(self.other_tree.inventory.root.file_id):
1051
 
            # the other tree's root is a non-root in the current tree
 
633
        except NoSuchFile:
1052
634
            return
1053
635
        self.reparent_children(self.other_tree.inventory.root, self.tt.root)
1054
636
        self.tt.cancel_creation(other_root)
1084
666
        if entry is None:
1085
667
            return None
1086
668
        return entry.name
1087
 
 
 
669
    
1088
670
    @staticmethod
1089
671
    def contents_sha1(tree, file_id):
1090
672
        """Determine the sha1 of the file contents (used as a key method)."""
1095
677
    @staticmethod
1096
678
    def executable(tree, file_id):
1097
679
        """Determine the executability of a file-id (used as a key method)."""
1098
 
        if not tree.has_id(file_id):
 
680
        if file_id not in tree:
1099
681
            return None
1100
682
        if tree.kind(file_id) != "file":
1101
683
            return False
1104
686
    @staticmethod
1105
687
    def kind(tree, file_id):
1106
688
        """Determine the kind of a file-id (used as a key method)."""
1107
 
        if not tree.has_id(file_id):
 
689
        if file_id not in tree:
1108
690
            return None
1109
691
        return tree.kind(file_id)
1110
692
 
1123
705
            return "other"
1124
706
 
1125
707
    @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
708
    def scalar_three_way(this_tree, base_tree, other_tree, file_id, key):
1177
709
        """Do a three-way test on a scalar.
1178
710
        Return "this", "other" or "conflict", depending whether a value wins.
1193
725
 
1194
726
    def merge_names(self, file_id):
1195
727
        def get_entry(tree):
1196
 
            if tree.has_id(file_id):
 
728
            if file_id in tree.inventory:
1197
729
                return tree.inventory[file_id]
1198
730
            else:
1199
731
                return None
1210
742
            else:
1211
743
                names.append(entry.name)
1212
744
                parents.append(entry.parent_id)
1213
 
        return self._merge_names(file_id, parents, names,
1214
 
                                 resolver=self._three_way)
 
745
        return self._merge_names(file_id, parents, names)
1215
746
 
1216
 
    def _merge_names(self, file_id, parents, names, resolver):
 
747
    def _merge_names(self, file_id, parents, names):
1217
748
        """Perform a merge on file_id names and parents"""
1218
749
        base_name, other_name, this_name = names
1219
750
        base_parent, other_parent, this_parent = parents
1220
751
 
1221
 
        name_winner = resolver(*names)
 
752
        name_winner = self._three_way(*names)
1222
753
 
1223
 
        parent_id_winner = resolver(*parents)
 
754
        parent_id_winner = self._three_way(*parents)
1224
755
        if this_name is None:
1225
756
            if name_winner == "this":
1226
757
                name_winner = "other"
1230
761
            return
1231
762
        if name_winner == "conflict":
1232
763
            trans_id = self.tt.trans_id_file_id(file_id)
1233
 
            self._raw_conflicts.append(('name conflict', trans_id,
 
764
            self._raw_conflicts.append(('name conflict', trans_id, 
1234
765
                                        this_name, other_name))
1235
766
        if parent_id_winner == "conflict":
1236
767
            trans_id = self.tt.trans_id_file_id(file_id)
1237
 
            self._raw_conflicts.append(('parent conflict', trans_id,
 
768
            self._raw_conflicts.append(('parent conflict', trans_id, 
1238
769
                                        this_parent, other_parent))
1239
770
        if other_name is None:
1240
 
            # it doesn't matter whether the result was 'other' or
 
771
            # it doesn't matter whether the result was 'other' or 
1241
772
            # 'conflict'-- if there's no 'other', we leave it alone.
1242
773
            return
1243
774
        # if we get here, name_winner and parent_winner are set to safe values.
1248
779
            self.tt.adjust_path(names[self.winner_idx[name_winner]],
1249
780
                                parent_trans_id, trans_id)
1250
781
 
1251
 
    def _do_merge_contents(self, file_id):
1252
 
        """Performs a merge on file_id contents."""
 
782
    def merge_contents(self, file_id):
 
783
        """Performa a merge on file_id contents."""
1253
784
        def contents_pair(tree):
1254
785
            if file_id not in tree:
1255
786
                return (None, None)
1262
793
                contents = None
1263
794
            return kind, contents
1264
795
 
 
796
        def contents_conflict():
 
797
            trans_id = self.tt.trans_id_file_id(file_id)
 
798
            name = self.tt.final_name(trans_id)
 
799
            parent_id = self.tt.final_parent(trans_id)
 
800
            if file_id in self.this_tree.inventory:
 
801
                self.tt.unversion_file(trans_id)
 
802
                if file_id in self.this_tree:
 
803
                    self.tt.delete_contents(trans_id)
 
804
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
805
                                              set_version=True)
 
806
            self._raw_conflicts.append(('contents conflict', file_group))
 
807
 
1265
808
        # See SPOT run.  run, SPOT, run.
1266
809
        # So we're not QUITE repeating ourselves; we do tricky things with
1267
810
        # file kind...
1268
811
        base_pair = contents_pair(self.base_tree)
1269
812
        other_pair = contents_pair(self.other_tree)
1270
 
        if self._lca_trees:
1271
 
            this_pair = contents_pair(self.this_tree)
1272
 
            lca_pairs = [contents_pair(tree) for tree in self._lca_trees]
1273
 
            winner = self._lca_multi_way((base_pair, lca_pairs), other_pair,
1274
 
                                         this_pair, allow_overriding_lca=False)
1275
 
        else:
1276
 
            if base_pair == other_pair:
1277
 
                winner = 'this'
1278
 
            else:
1279
 
                # We delayed evaluating this_pair as long as we can to avoid
1280
 
                # unnecessary sha1 calculation
1281
 
                this_pair = contents_pair(self.this_tree)
1282
 
                winner = self._three_way(base_pair, other_pair, this_pair)
1283
 
        if winner == 'this':
1284
 
            # No interesting changes introduced by OTHER
1285
 
            return "unmodified"
1286
 
        # We have a hypothetical conflict, but if we have files, then we
1287
 
        # can try to merge the content
1288
 
        trans_id = self.tt.trans_id_file_id(file_id)
1289
 
        params = MergeHookParams(self, file_id, trans_id, this_pair[0],
1290
 
            other_pair[0], winner)
1291
 
        hooks = self.active_hooks
1292
 
        hook_status = 'not_applicable'
1293
 
        for hook in hooks:
1294
 
            hook_status, lines = hook.merge_contents(params)
1295
 
            if hook_status != 'not_applicable':
1296
 
                # Don't try any more hooks, this one applies.
1297
 
                break
1298
 
        result = "modified"
1299
 
        if hook_status == 'not_applicable':
1300
 
            # This is a contents conflict, because none of the available
1301
 
            # functions could merge it.
1302
 
            result = None
1303
 
            name = self.tt.final_name(trans_id)
1304
 
            parent_id = self.tt.final_parent(trans_id)
1305
 
            if self.this_tree.has_id(file_id):
1306
 
                self.tt.unversion_file(trans_id)
1307
 
            file_group = self._dump_conflicts(name, parent_id, file_id,
1308
 
                                              set_version=True)
1309
 
            self._raw_conflicts.append(('contents conflict', file_group))
1310
 
        elif hook_status == 'success':
1311
 
            self.tt.create_file(lines, trans_id)
1312
 
        elif hook_status == 'conflicted':
1313
 
            # XXX: perhaps the hook should be able to provide
1314
 
            # the BASE/THIS/OTHER files?
1315
 
            self.tt.create_file(lines, trans_id)
1316
 
            self._raw_conflicts.append(('text conflict', trans_id))
1317
 
            name = self.tt.final_name(trans_id)
1318
 
            parent_id = self.tt.final_parent(trans_id)
1319
 
            self._dump_conflicts(name, parent_id, file_id)
1320
 
        elif hook_status == 'delete':
1321
 
            self.tt.unversion_file(trans_id)
1322
 
            result = "deleted"
1323
 
        elif hook_status == 'done':
1324
 
            # The hook function did whatever it needs to do directly, no
1325
 
            # further action needed here.
1326
 
            pass
1327
 
        else:
1328
 
            raise AssertionError('unknown hook_status: %r' % (hook_status,))
1329
 
        if not self.this_tree.has_id(file_id) and result == "modified":
1330
 
            self.tt.version_file(file_id, trans_id)
1331
 
        # The merge has been performed, so the old contents should not be
1332
 
        # retained.
1333
 
        try:
1334
 
            self.tt.delete_contents(trans_id)
1335
 
        except errors.NoSuchFile:
1336
 
            pass
1337
 
        return result
1338
 
 
1339
 
    def _default_other_winner_merge(self, merge_hook_params):
1340
 
        """Replace this contents with other."""
1341
 
        file_id = merge_hook_params.file_id
1342
 
        trans_id = merge_hook_params.trans_id
1343
 
        file_in_this = self.this_tree.has_id(file_id)
1344
 
        if self.other_tree.has_id(file_id):
1345
 
            # OTHER changed the file
1346
 
            wt = self.this_tree
1347
 
            if wt.supports_content_filtering():
1348
 
                # We get the path from the working tree if it exists.
1349
 
                # That fails though when OTHER is adding a file, so
1350
 
                # we fall back to the other tree to find the path if
1351
 
                # it doesn't exist locally.
1352
 
                try:
1353
 
                    filter_tree_path = wt.id2path(file_id)
1354
 
                except errors.NoSuchId:
1355
 
                    filter_tree_path = self.other_tree.id2path(file_id)
1356
 
            else:
1357
 
                # Skip the id2path lookup for older formats
1358
 
                filter_tree_path = None
1359
 
            transform.create_from_tree(self.tt, trans_id,
1360
 
                             self.other_tree, file_id,
1361
 
                             filter_tree_path=filter_tree_path)
1362
 
            return 'done', None
1363
 
        elif file_in_this:
1364
 
            # OTHER deleted the file
1365
 
            return 'delete', None
1366
 
        else:
1367
 
            raise AssertionError(
1368
 
                'winner is OTHER, but file_id %r not in THIS or OTHER tree'
1369
 
                % (file_id,))
1370
 
 
1371
 
    def merge_contents(self, merge_hook_params):
1372
 
        """Fallback merge logic after user installed hooks."""
1373
 
        # This function is used in merge hooks as the fallback instance.
1374
 
        # Perhaps making this function and the functions it calls be a 
1375
 
        # a separate class would be better.
1376
 
        if merge_hook_params.winner == 'other':
1377
 
            # OTHER is a straight winner, so replace this contents with other
1378
 
            return self._default_other_winner_merge(merge_hook_params)
1379
 
        elif merge_hook_params.is_file_merge():
1380
 
            # THIS and OTHER are both files, so text merge.  Either
1381
 
            # BASE is a file, or both converted to files, so at least we
1382
 
            # have agreement that output should be a file.
1383
 
            try:
1384
 
                self.text_merge(merge_hook_params.file_id,
1385
 
                    merge_hook_params.trans_id)
1386
 
            except errors.BinaryFile:
1387
 
                return 'not_applicable', None
1388
 
            return 'done', None
1389
 
        else:
1390
 
            return 'not_applicable', None
 
813
        if base_pair == other_pair:
 
814
            # OTHER introduced no changes
 
815
            return "unmodified"
 
816
        this_pair = contents_pair(self.this_tree)
 
817
        if this_pair == other_pair:
 
818
            # THIS and OTHER introduced the same changes
 
819
            return "unmodified"
 
820
        else:
 
821
            trans_id = self.tt.trans_id_file_id(file_id)
 
822
            if this_pair == base_pair:
 
823
                # only OTHER introduced changes
 
824
                if file_id in self.this_tree:
 
825
                    # Remove any existing contents
 
826
                    self.tt.delete_contents(trans_id)
 
827
                if file_id in self.other_tree:
 
828
                    # OTHER changed the file
 
829
                    create_by_entry(self.tt, 
 
830
                                    self.other_tree.inventory[file_id], 
 
831
                                    self.other_tree, trans_id)
 
832
                    if file_id not in self.this_tree.inventory:
 
833
                        self.tt.version_file(file_id, trans_id)
 
834
                    return "modified"
 
835
                elif file_id in self.this_tree.inventory:
 
836
                    # OTHER deleted the file
 
837
                    self.tt.unversion_file(trans_id)
 
838
                    return "deleted"
 
839
            #BOTH THIS and OTHER introduced changes; scalar conflict
 
840
            elif this_pair[0] == "file" and other_pair[0] == "file":
 
841
                # THIS and OTHER are both files, so text merge.  Either
 
842
                # BASE is a file, or both converted to files, so at least we
 
843
                # have agreement that output should be a file.
 
844
                try:
 
845
                    self.text_merge(file_id, trans_id)
 
846
                except BinaryFile:
 
847
                    return contents_conflict()
 
848
                if file_id not in self.this_tree.inventory:
 
849
                    self.tt.version_file(file_id, trans_id)
 
850
                try:
 
851
                    self.tt.tree_kind(trans_id)
 
852
                    self.tt.delete_contents(trans_id)
 
853
                except NoSuchFile:
 
854
                    pass
 
855
                return "modified"
 
856
            else:
 
857
                # Scalar conflict, can't text merge.  Dump conflicts
 
858
                return contents_conflict()
1391
859
 
1392
860
    def get_lines(self, tree, file_id):
1393
861
        """Return the lines in a file, or an empty list."""
1394
 
        if tree.has_id(file_id):
 
862
        if file_id in tree:
1395
863
            return tree.get_file(file_id).readlines()
1396
864
        else:
1397
865
            return []
1400
868
        """Perform a three-way text merge on a file_id"""
1401
869
        # it's possible that we got here with base as a different type.
1402
870
        # if so, we just want two-way text conflicts.
1403
 
        if self.base_tree.has_id(file_id) and \
 
871
        if file_id in self.base_tree and \
1404
872
            self.base_tree.kind(file_id) == "file":
1405
873
            base_lines = self.get_lines(self.base_tree, file_id)
1406
874
        else:
1407
875
            base_lines = []
1408
876
        other_lines = self.get_lines(self.other_tree, file_id)
1409
877
        this_lines = self.get_lines(self.this_tree, file_id)
1410
 
        m3 = merge3.Merge3(base_lines, this_lines, other_lines,
1411
 
                           is_cherrypick=self.cherrypick)
 
878
        m3 = Merge3(base_lines, this_lines, other_lines,
 
879
                    is_cherrypick=self.cherrypick)
1412
880
        start_marker = "!START OF MERGE CONFLICT!" + "I HOPE THIS IS UNIQUE"
1413
881
        if self.show_base is True:
1414
882
            base_marker = '|' * 7
1417
885
 
1418
886
        def iter_merge3(retval):
1419
887
            retval["text_conflicts"] = False
1420
 
            for line in m3.merge_lines(name_a = "TREE",
1421
 
                                       name_b = "MERGE-SOURCE",
 
888
            for line in m3.merge_lines(name_a = "TREE", 
 
889
                                       name_b = "MERGE-SOURCE", 
1422
890
                                       name_base = "BASE-REVISION",
1423
 
                                       start_marker=start_marker,
 
891
                                       start_marker=start_marker, 
1424
892
                                       base_marker=base_marker,
1425
893
                                       reprocess=self.reprocess):
1426
894
                if line.startswith(start_marker):
1435
903
            self._raw_conflicts.append(('text conflict', trans_id))
1436
904
            name = self.tt.final_name(trans_id)
1437
905
            parent_id = self.tt.final_parent(trans_id)
1438
 
            file_group = self._dump_conflicts(name, parent_id, file_id,
 
906
            file_group = self._dump_conflicts(name, parent_id, file_id, 
1439
907
                                              this_lines, base_lines,
1440
908
                                              other_lines)
1441
909
            file_group.append(trans_id)
1442
910
 
1443
 
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None,
 
911
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None, 
1444
912
                        base_lines=None, other_lines=None, set_version=False,
1445
913
                        no_base=False):
1446
914
        """Emit conflict files.
1448
916
        determined automatically.  If set_version is true, the .OTHER, .THIS
1449
917
        or .BASE (in that order) will be created as versioned files.
1450
918
        """
1451
 
        data = [('OTHER', self.other_tree, other_lines),
 
919
        data = [('OTHER', self.other_tree, other_lines), 
1452
920
                ('THIS', self.this_tree, this_lines)]
1453
921
        if not no_base:
1454
922
            data.append(('BASE', self.base_tree, base_lines))
1455
 
 
1456
 
        # We need to use the actual path in the working tree of the file here,
1457
 
        # ignoring the conflict suffixes
1458
 
        wt = self.this_tree
1459
 
        if wt.supports_content_filtering():
1460
 
            try:
1461
 
                filter_tree_path = wt.id2path(file_id)
1462
 
            except errors.NoSuchId:
1463
 
                # file has been deleted
1464
 
                filter_tree_path = None
1465
 
        else:
1466
 
            # Skip the id2path lookup for older formats
1467
 
            filter_tree_path = None
1468
 
 
1469
923
        versioned = False
1470
924
        file_group = []
1471
925
        for suffix, tree, lines in data:
1472
 
            if tree.has_id(file_id):
 
926
            if file_id in tree:
1473
927
                trans_id = self._conflict_file(name, parent_id, tree, file_id,
1474
 
                                               suffix, lines, filter_tree_path)
 
928
                                               suffix, lines)
1475
929
                file_group.append(trans_id)
1476
930
                if set_version and not versioned:
1477
931
                    self.tt.version_file(file_id, trans_id)
1478
932
                    versioned = True
1479
933
        return file_group
1480
 
 
1481
 
    def _conflict_file(self, name, parent_id, tree, file_id, suffix,
1482
 
                       lines=None, filter_tree_path=None):
 
934
           
 
935
    def _conflict_file(self, name, parent_id, tree, file_id, suffix, 
 
936
                       lines=None):
1483
937
        """Emit a single conflict file."""
1484
938
        name = name + '.' + suffix
1485
939
        trans_id = self.tt.create_path(name, parent_id)
1486
 
        transform.create_from_tree(self.tt, trans_id, tree, file_id, lines,
1487
 
            filter_tree_path)
 
940
        entry = tree.inventory[file_id]
 
941
        create_by_entry(self.tt, entry, tree, trans_id, lines)
1488
942
        return trans_id
1489
943
 
1490
944
    def merge_executable(self, file_id, file_status):
1491
945
        """Perform a merge on the execute bit."""
1492
946
        executable = [self.executable(t, file_id) for t in (self.base_tree,
1493
947
                      self.other_tree, self.this_tree)]
1494
 
        self._merge_executable(file_id, executable, file_status,
1495
 
                               resolver=self._three_way)
 
948
        self._merge_executable(file_id, executable, file_status)
1496
949
 
1497
 
    def _merge_executable(self, file_id, executable, file_status,
1498
 
                          resolver):
 
950
    def _merge_executable(self, file_id, executable, file_status):
1499
951
        """Perform a merge on the execute bit."""
1500
952
        base_executable, other_executable, this_executable = executable
1501
953
        if file_status == "deleted":
1502
954
            return
1503
 
        winner = resolver(*executable)
 
955
        winner = self._three_way(*executable)
1504
956
        if winner == "conflict":
1505
957
        # There must be a None in here, if we have a conflict, but we
1506
958
        # need executability since file status was not deleted.
1514
966
        try:
1515
967
            if self.tt.final_kind(trans_id) != "file":
1516
968
                return
1517
 
        except errors.NoSuchFile:
 
969
        except NoSuchFile:
1518
970
            return
1519
971
        if winner == "this":
1520
972
            executability = this_executable
1521
973
        else:
1522
 
            if self.other_tree.has_id(file_id):
 
974
            if file_id in self.other_tree:
1523
975
                executability = other_executable
1524
 
            elif self.this_tree.has_id(file_id):
 
976
            elif file_id in self.this_tree:
1525
977
                executability = this_executable
1526
 
            elif self.base_tree_has_id(file_id):
 
978
            elif file_id in self.base_tree:
1527
979
                executability = base_executable
1528
980
        if executability is not None:
1529
981
            trans_id = self.tt.trans_id_file_id(file_id)
1531
983
 
1532
984
    def cook_conflicts(self, fs_conflicts):
1533
985
        """Convert all conflicts into a form that doesn't depend on trans_id"""
 
986
        from conflicts import Conflict
1534
987
        name_conflicts = {}
1535
 
        self.cooked_conflicts.extend(transform.cook_conflicts(
1536
 
                fs_conflicts, self.tt))
1537
 
        fp = transform.FinalPaths(self.tt)
 
988
        self.cooked_conflicts.extend(cook_conflicts(fs_conflicts, self.tt))
 
989
        fp = FinalPaths(self.tt)
1538
990
        for conflict in self._raw_conflicts:
1539
991
            conflict_type = conflict[0]
1540
992
            if conflict_type in ('name conflict', 'parent conflict'):
1542
994
                conflict_args = conflict[2:]
1543
995
                if trans_id not in name_conflicts:
1544
996
                    name_conflicts[trans_id] = {}
1545
 
                transform.unique_add(name_conflicts[trans_id], conflict_type,
1546
 
                                     conflict_args)
 
997
                unique_add(name_conflicts[trans_id], conflict_type, 
 
998
                           conflict_args)
1547
999
            if conflict_type == 'contents conflict':
1548
1000
                for trans_id in conflict[1]:
1549
1001
                    file_id = self.tt.final_file_id(trans_id)
1554
1006
                    if path.endswith(suffix):
1555
1007
                        path = path[:-len(suffix)]
1556
1008
                        break
1557
 
                c = _mod_conflicts.Conflict.factory(conflict_type,
1558
 
                                                    path=path, file_id=file_id)
 
1009
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
1559
1010
                self.cooked_conflicts.append(c)
1560
1011
            if conflict_type == 'text conflict':
1561
1012
                trans_id = conflict[1]
1562
1013
                path = fp.get_path(trans_id)
1563
1014
                file_id = self.tt.final_file_id(trans_id)
1564
 
                c = _mod_conflicts.Conflict.factory(conflict_type,
1565
 
                                                    path=path, file_id=file_id)
 
1015
                c = Conflict.factory(conflict_type, path=path, file_id=file_id)
1566
1016
                self.cooked_conflicts.append(c)
1567
1017
 
1568
1018
        for trans_id, conflicts in name_conflicts.iteritems():
1583
1033
            if this_parent is not None and this_name is not None:
1584
1034
                this_parent_path = \
1585
1035
                    fp.get_path(self.tt.trans_id_file_id(this_parent))
1586
 
                this_path = osutils.pathjoin(this_parent_path, this_name)
 
1036
                this_path = pathjoin(this_parent_path, this_name)
1587
1037
            else:
1588
1038
                this_path = "<deleted>"
1589
1039
            file_id = self.tt.final_file_id(trans_id)
1590
 
            c = _mod_conflicts.Conflict.factory('path conflict', path=this_path,
1591
 
                                                conflict_path=other_path,
1592
 
                                                file_id=file_id)
 
1040
            c = Conflict.factory('path conflict', path=this_path,
 
1041
                                 conflict_path=other_path, file_id=file_id)
1593
1042
            self.cooked_conflicts.append(c)
1594
 
        self.cooked_conflicts.sort(key=_mod_conflicts.Conflict.sort_key)
 
1043
        self.cooked_conflicts.sort(key=Conflict.sort_key)
1595
1044
 
1596
1045
 
1597
1046
class WeaveMerger(Merge3Merger):
1601
1050
    supports_reverse_cherrypick = False
1602
1051
    history_based = True
1603
1052
 
1604
 
    def _generate_merge_plan(self, file_id, base):
1605
 
        return self.this_tree.plan_file_merge(file_id, self.other_tree,
 
1053
    def _merged_lines(self, file_id):
 
1054
        """Generate the merged lines.
 
1055
        There is no distinction between lines that are meant to contain <<<<<<<
 
1056
        and conflicts.
 
1057
        """
 
1058
        if self.cherrypick:
 
1059
            base = self.base_tree
 
1060
        else:
 
1061
            base = None
 
1062
        plan = self.this_tree.plan_file_merge(file_id, self.other_tree,
1606
1063
                                              base=base)
1607
 
 
1608
 
    def _merged_lines(self, file_id):
1609
 
        """Generate the merged lines.
1610
 
        There is no distinction between lines that are meant to contain <<<<<<<
1611
 
        and conflicts.
1612
 
        """
1613
 
        if self.cherrypick:
1614
 
            base = self.base_tree
1615
 
        else:
1616
 
            base = None
1617
 
        plan = self._generate_merge_plan(file_id, base)
1618
1064
        if 'merge' in debug.debug_flags:
1619
1065
            plan = list(plan)
1620
1066
            trans_id = self.tt.trans_id_file_id(file_id)
1621
1067
            name = self.tt.final_name(trans_id) + '.plan'
1622
 
            contents = ('%11s|%s' % l for l in plan)
 
1068
            contents = ('%10s|%s' % l for l in plan)
1623
1069
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
1624
 
        textmerge = versionedfile.PlanWeaveMerge(plan, '<<<<<<< TREE\n',
1625
 
                                                 '>>>>>>> MERGE-SOURCE\n')
1626
 
        lines, conflicts = textmerge.merge_lines(self.reprocess)
1627
 
        if conflicts:
1628
 
            base_lines = textmerge.base_from_plan()
1629
 
        else:
1630
 
            base_lines = None
1631
 
        return lines, base_lines
 
1070
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1071
            '>>>>>>> MERGE-SOURCE\n')
 
1072
        return textmerge.merge_lines(self.reprocess)
1632
1073
 
1633
1074
    def text_merge(self, file_id, trans_id):
1634
1075
        """Perform a (weave) text merge for a given file and file-id.
1635
1076
        If conflicts are encountered, .THIS and .OTHER files will be emitted,
1636
1077
        and a conflict will be noted.
1637
1078
        """
1638
 
        lines, base_lines = self._merged_lines(file_id)
 
1079
        lines, conflicts = self._merged_lines(file_id)
1639
1080
        lines = list(lines)
1640
 
        # Note we're checking whether the OUTPUT is binary in this case,
 
1081
        # Note we're checking whether the OUTPUT is binary in this case, 
1641
1082
        # because we don't want to get into weave merge guts.
1642
 
        textfile.check_text_lines(lines)
 
1083
        check_text_lines(lines)
1643
1084
        self.tt.create_file(lines, trans_id)
1644
 
        if base_lines is not None:
1645
 
            # Conflict
 
1085
        if conflicts:
1646
1086
            self._raw_conflicts.append(('text conflict', trans_id))
1647
1087
            name = self.tt.final_name(trans_id)
1648
1088
            parent_id = self.tt.final_parent(trans_id)
1649
 
            file_group = self._dump_conflicts(name, parent_id, file_id,
1650
 
                                              no_base=False,
1651
 
                                              base_lines=base_lines)
 
1089
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1090
                                              no_base=True)
1652
1091
            file_group.append(trans_id)
1653
1092
 
1654
1093
 
1655
1094
class LCAMerger(WeaveMerger):
1656
1095
 
1657
 
    def _generate_merge_plan(self, file_id, base):
1658
 
        return self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
 
1096
    def _merged_lines(self, file_id):
 
1097
        """Generate the merged lines.
 
1098
        There is no distinction between lines that are meant to contain <<<<<<<
 
1099
        and conflicts.
 
1100
        """
 
1101
        if self.cherrypick:
 
1102
            base = self.base_tree
 
1103
        else:
 
1104
            base = None
 
1105
        plan = self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
1659
1106
                                                  base=base)
 
1107
        if 'merge' in debug.debug_flags:
 
1108
            plan = list(plan)
 
1109
            trans_id = self.tt.trans_id_file_id(file_id)
 
1110
            name = self.tt.final_name(trans_id) + '.plan'
 
1111
            contents = ('%10s|%s' % l for l in plan)
 
1112
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
 
1113
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1114
            '>>>>>>> MERGE-SOURCE\n')
 
1115
        return textmerge.merge_lines(self.reprocess)
 
1116
 
1660
1117
 
1661
1118
class Diff3Merger(Merge3Merger):
1662
1119
    """Three-way merger using external diff3 for text merging"""
1663
1120
 
1664
1121
    def dump_file(self, temp_dir, name, tree, file_id):
1665
 
        out_path = osutils.pathjoin(temp_dir, name)
 
1122
        out_path = pathjoin(temp_dir, name)
1666
1123
        out_file = open(out_path, "wb")
1667
1124
        try:
1668
1125
            in_file = tree.get_file(file_id)
1680
1137
        import bzrlib.patch
1681
1138
        temp_dir = osutils.mkdtemp(prefix="bzr-")
1682
1139
        try:
1683
 
            new_file = osutils.pathjoin(temp_dir, "new")
 
1140
            new_file = pathjoin(temp_dir, "new")
1684
1141
            this = self.dump_file(temp_dir, "this", self.this_tree, file_id)
1685
1142
            base = self.dump_file(temp_dir, "base", self.base_tree, file_id)
1686
1143
            other = self.dump_file(temp_dir, "other", self.other_tree, file_id)
1687
1144
            status = bzrlib.patch.diff3(new_file, this, base, other)
1688
1145
            if status not in (0, 1):
1689
 
                raise errors.BzrError("Unhandled diff3 exit code")
 
1146
                raise BzrError("Unhandled diff3 exit code")
1690
1147
            f = open(new_file, 'rb')
1691
1148
            try:
1692
1149
                self.tt.create_file(f, trans_id)
1710
1167
                other_rev_id=None,
1711
1168
                interesting_files=None,
1712
1169
                this_tree=None,
1713
 
                pb=None,
 
1170
                pb=DummyProgress(),
1714
1171
                change_reporter=None):
1715
 
    """Primary interface for merging.
 
1172
    """Primary interface for merging. 
1716
1173
 
1717
 
        typical use is probably
 
1174
        typical use is probably 
1718
1175
        'merge_inner(branch, branch.get_revision_tree(other_revision),
1719
1176
                     branch.get_revision_tree(base_revision))'
1720
1177
        """
1721
1178
    if this_tree is None:
1722
 
        raise errors.BzrError("bzrlib.merge.merge_inner requires a this_tree "
1723
 
                              "parameter as of bzrlib version 0.8.")
 
1179
        raise BzrError("bzrlib.merge.merge_inner requires a this_tree "
 
1180
            "parameter as of bzrlib version 0.8.")
1724
1181
    merger = Merger(this_branch, other_tree, base_tree, this_tree=this_tree,
1725
1182
                    pb=pb, change_reporter=change_reporter)
1726
1183
    merger.backup_files = backup_files
1739
1196
    get_revision_id = getattr(base_tree, 'get_revision_id', None)
1740
1197
    if get_revision_id is None:
1741
1198
        get_revision_id = base_tree.last_revision
1742
 
    merger.cache_trees_with_revision_ids([other_tree, base_tree, this_tree])
1743
1199
    merger.set_base_revision(get_revision_id(), this_branch)
1744
1200
    return merger.do_merge()
1745
1201
 
1813
1269
 
1814
1270
    def get_lines(self, revisions):
1815
1271
        """Get lines for revisions from the backing VersionedFiles.
1816
 
 
 
1272
        
1817
1273
        :raises RevisionNotPresent: on absent texts.
1818
1274
        """
1819
1275
        keys = [(self._key_prefix + (rev,)) for rev in revisions]
1821
1277
        for record in self.vf.get_record_stream(keys, 'unordered', True):
1822
1278
            if record.storage_kind == 'absent':
1823
1279
                raise errors.RevisionNotPresent(record.key, self.vf)
1824
 
            result[record.key[-1]] = osutils.chunks_to_lines(
1825
 
                record.get_bytes_as('chunked'))
 
1280
            result[record.key[-1]] = osutils.split_lines(
 
1281
                record.get_bytes_as('fulltext'))
1826
1282
        return result
1827
1283
 
1828
1284
    def plan_merge(self):
1944
1400
        super(_PlanMerge, self).__init__(a_rev, b_rev, vf, key_prefix)
1945
1401
        self.a_key = self._key_prefix + (self.a_rev,)
1946
1402
        self.b_key = self._key_prefix + (self.b_rev,)
1947
 
        self.graph = _mod_graph.Graph(self.vf)
 
1403
        self.graph = Graph(self.vf)
1948
1404
        heads = self.graph.heads((self.a_key, self.b_key))
1949
1405
        if len(heads) == 1:
1950
1406
            # one side dominates, so we can just return its values, yay for
1955
1411
                other = b_rev
1956
1412
            else:
1957
1413
                other = a_rev
1958
 
            trace.mutter('found dominating revision for %s\n%s > %s', self.vf,
1959
 
                         self._head_key[-1], other)
 
1414
            mutter('found dominating revision for %s\n%s > %s', self.vf,
 
1415
                   self._head_key[-1], other)
1960
1416
            self._weave = None
1961
1417
        else:
1962
1418
            self._head_key = None
1976
1432
        while True:
1977
1433
            next_lcas = self.graph.find_lca(*cur_ancestors)
1978
1434
            # Map a plain NULL_REVISION to a simple no-ancestors
1979
 
            if next_lcas == set([_mod_revision.NULL_REVISION]):
 
1435
            if next_lcas == set([NULL_REVISION]):
1980
1436
                next_lcas = ()
1981
1437
            # Order the lca's based on when they were merged into the tip
1982
1438
            # While the actual merge portion of weave merge uses a set() of
1994
1450
            elif len(next_lcas) > 2:
1995
1451
                # More than 2 lca's, fall back to grabbing all nodes between
1996
1452
                # this and the unique lca.
1997
 
                trace.mutter('More than 2 LCAs, falling back to all nodes for:'
1998
 
                             ' %s, %s\n=> %s',
1999
 
                             self.a_key, self.b_key, cur_ancestors)
 
1453
                mutter('More than 2 LCAs, falling back to all nodes for:'
 
1454
                       ' %s, %s\n=> %s', self.a_key, self.b_key, cur_ancestors)
2000
1455
                cur_lcas = next_lcas
2001
1456
                while len(cur_lcas) > 1:
2002
1457
                    cur_lcas = self.graph.find_lca(*cur_lcas)
2005
1460
                    unique_lca = None
2006
1461
                else:
2007
1462
                    unique_lca = list(cur_lcas)[0]
2008
 
                    if unique_lca == _mod_revision.NULL_REVISION:
 
1463
                    if unique_lca == NULL_REVISION:
2009
1464
                        # find_lca will return a plain 'NULL_REVISION' rather
2010
1465
                        # than a key tuple when there is no common ancestor, we
2011
1466
                        # prefer to just use None, because it doesn't confuse
2019
1474
 
2020
1475
    def _find_unique_parents(self, tip_keys, base_key):
2021
1476
        """Find ancestors of tip that aren't ancestors of base.
2022
 
 
 
1477
        
2023
1478
        :param tip_keys: Nodes that are interesting
2024
1479
        :param base_key: Cull all ancestors of this node
2025
1480
        :return: The parent map for all revisions between tip_keys and
2034
1489
            # We remove NULL_REVISION because it isn't a proper tuple key, and
2035
1490
            # thus confuses things like _get_interesting_texts, and our logic
2036
1491
            # to add the texts into the memory weave.
2037
 
            if _mod_revision.NULL_REVISION in parent_map:
2038
 
                parent_map.pop(_mod_revision.NULL_REVISION)
 
1492
            if NULL_REVISION in parent_map:
 
1493
                parent_map.pop(NULL_REVISION)
2039
1494
        else:
2040
1495
            interesting = set()
2041
1496
            for tip in tip_keys:
2085
1540
    @staticmethod
2086
1541
    def _prune_tails(parent_map, child_map, tails_to_remove):
2087
1542
        """Remove tails from the parent map.
2088
 
 
 
1543
        
2089
1544
        This will remove the supplied revisions until no more children have 0
2090
1545
        parents.
2091
1546
 
2193
1648
        lcas = graph.find_lca(key_prefix + (a_rev,), key_prefix + (b_rev,))
2194
1649
        self.lcas = set()
2195
1650
        for lca in lcas:
2196
 
            if lca == _mod_revision.NULL_REVISION:
 
1651
            if lca == NULL_REVISION:
2197
1652
                self.lcas.add(lca)
2198
1653
            else:
2199
1654
                self.lcas.add(lca[-1])