~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge.py

  • Committer: Martin Pool
  • Date: 2010-02-25 06:17:27 UTC
  • mfrom: (5055 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5057.
  • Revision ID: mbp@sourcefrog.net-20100225061727-4sd9lt0qmdc6087t
merge news

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
16
16
 
17
17
import warnings
18
18
 
19
 
from bzrlib.lazy_import import lazy_import
20
 
lazy_import(globals(), """
21
19
from bzrlib import (
22
20
    branch as _mod_branch,
23
 
    cleanup,
24
21
    conflicts as _mod_conflicts,
25
22
    debug,
26
 
    generate_ids,
 
23
    decorators,
 
24
    errors,
27
25
    graph as _mod_graph,
 
26
    hooks,
28
27
    merge3,
29
28
    osutils,
30
29
    patiencediff,
35
34
    tree as _mod_tree,
36
35
    tsort,
37
36
    ui,
38
 
    versionedfile,
39
 
    workingtree,
40
 
    )
41
 
from bzrlib.i18n import gettext
42
 
""")
43
 
from bzrlib import (
44
 
    decorators,
45
 
    errors,
46
 
    hooks,
47
 
    )
 
37
    versionedfile
 
38
    )
 
39
from bzrlib.cleanup import OperationWithCleanups
48
40
from bzrlib.symbol_versioning import (
49
41
    deprecated_in,
50
42
    deprecated_method,
54
46
 
55
47
def transform_tree(from_tree, to_tree, interesting_ids=None):
56
48
    from_tree.lock_tree_write()
57
 
    operation = cleanup.OperationWithCleanups(merge_inner)
 
49
    operation = OperationWithCleanups(merge_inner)
58
50
    operation.add_cleanup(from_tree.unlock)
59
51
    operation.run_simple(from_tree.branch, to_tree, from_tree,
60
52
        ignore_zero=True, interesting_ids=interesting_ids, this_tree=from_tree)
63
55
class MergeHooks(hooks.Hooks):
64
56
 
65
57
    def __init__(self):
66
 
        hooks.Hooks.__init__(self, "bzrlib.merge", "Merger.hooks")
67
 
        self.add_hook('merge_file_content',
 
58
        hooks.Hooks.__init__(self)
 
59
        self.create_hook(hooks.HookPoint('merge_file_content',
68
60
            "Called with a bzrlib.merge.Merger object to create a per file "
69
61
            "merge object when starting a merge. "
70
62
            "Should return either None or a subclass of "
74
66
            "side has deleted the file and the other has changed it). "
75
67
            "See the AbstractPerFileMerger API docs for details on how it is "
76
68
            "used by merge.",
77
 
            (2, 1))
 
69
            (2, 1), None))
78
70
 
79
71
 
80
72
class AbstractPerFileMerger(object):
93
85
        """Attempt to merge the contents of a single file.
94
86
        
95
87
        :param merge_params: A bzrlib.merge.MergeHookParams
96
 
        :return: A tuple of (status, chunks), where status is one of
 
88
        :return : A tuple of (status, chunks), where status is one of
97
89
            'not_applicable', 'success', 'conflicted', or 'delete'.  If status
98
90
            is 'success' or 'conflicted', then chunks should be an iterable of
99
91
            strings for the new file contents.
101
93
        return ('not applicable', None)
102
94
 
103
95
 
104
 
class PerFileMerger(AbstractPerFileMerger):
105
 
    """Merge individual files when self.file_matches returns True.
106
 
 
107
 
    This class is intended to be subclassed.  The file_matches and
108
 
    merge_matching methods should be overridden with concrete implementations.
109
 
    """
110
 
 
111
 
    def file_matches(self, params):
112
 
        """Return True if merge_matching should be called on this file.
113
 
 
114
 
        Only called with merges of plain files with no clear winner.
115
 
 
116
 
        Subclasses must override this.
117
 
        """
118
 
        raise NotImplementedError(self.file_matches)
119
 
 
120
 
    def get_filename(self, params, tree):
121
 
        """Lookup the filename (i.e. basename, not path), given a Tree (e.g.
122
 
        self.merger.this_tree) and a MergeHookParams.
123
 
        """
124
 
        return osutils.basename(tree.id2path(params.file_id))
125
 
 
126
 
    def get_filepath(self, params, tree):
127
 
        """Calculate the path to the file in a tree.
128
 
 
129
 
        :param params: A MergeHookParams describing the file to merge
130
 
        :param tree: a Tree, e.g. self.merger.this_tree.
131
 
        """
132
 
        return tree.id2path(params.file_id)
133
 
 
134
 
    def merge_contents(self, params):
135
 
        """Merge the contents of a single file."""
136
 
        # Check whether this custom merge logic should be used.
137
 
        if (
138
 
            # OTHER is a straight winner, rely on default merge.
139
 
            params.winner == 'other' or
140
 
            # THIS and OTHER aren't both files.
141
 
            not params.is_file_merge() or
142
 
            # The filename doesn't match *.xml
143
 
            not self.file_matches(params)):
144
 
            return 'not_applicable', None
145
 
        return self.merge_matching(params)
146
 
 
147
 
    def merge_matching(self, params):
148
 
        """Merge the contents of a single file that has matched the criteria
149
 
        in PerFileMerger.merge_contents (is a conflict, is a file,
150
 
        self.file_matches is True).
151
 
 
152
 
        Subclasses must override this.
153
 
        """
154
 
        raise NotImplementedError(self.merge_matching)
155
 
 
156
 
 
157
 
class ConfigurableFileMerger(PerFileMerger):
 
96
class ConfigurableFileMerger(AbstractPerFileMerger):
158
97
    """Merge individual files when configured via a .conf file.
159
98
 
160
99
    This is a base class for concrete custom file merging logic. Concrete
183
122
        if self.name_prefix is None:
184
123
            raise ValueError("name_prefix must be set.")
185
124
 
186
 
    def file_matches(self, params):
 
125
    def filename_matches_config(self, params):
187
126
        """Check whether the file should call the merge hook.
188
127
 
189
128
        <name_prefix>_merge_files configuration variable is a list of files
203
142
                affected_files = self.default_files
204
143
            self.affected_files = affected_files
205
144
        if affected_files:
206
 
            filepath = self.get_filepath(params, self.merger.this_tree)
207
 
            if filepath in affected_files:
 
145
            filename = self.merger.this_tree.id2path(params.file_id)
 
146
            if filename in affected_files:
208
147
                return True
209
148
        return False
210
149
 
211
 
    def merge_matching(self, params):
 
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
212
163
        return self.merge_text(params)
213
164
 
214
165
    def merge_text(self, params):
429
380
        return self._cached_trees[revision_id]
430
381
 
431
382
    def _get_tree(self, treespec, possible_transports=None):
 
383
        from bzrlib import workingtree
432
384
        location, revno = treespec
433
385
        if revno is None:
434
386
            tree = workingtree.WorkingTree.open_containing(location)[0]
459
411
    @deprecated_method(deprecated_in((2, 1, 0)))
460
412
    def file_revisions(self, file_id):
461
413
        self.ensure_revision_trees()
 
414
        def get_id(tree, file_id):
 
415
            revision_id = tree.inventory[file_id].revision
 
416
            return revision_id
462
417
        if self.this_rev_id is None:
463
418
            if self.this_basis_tree.get_file_sha1(file_id) != \
464
419
                self.this_tree.get_file_sha1(file_id):
465
420
                raise errors.WorkingTreeNotRevision(self.this_tree)
466
421
 
467
422
        trees = (self.this_basis_tree, self.other_tree)
468
 
        return [tree.get_file_revision(file_id) for tree in trees]
 
423
        return [get_id(tree, file_id) for tree in trees]
469
424
 
470
425
    @deprecated_method(deprecated_in((2, 1, 0)))
471
426
    def check_basis(self, check_clean, require_commits=True):
499
454
    def _add_parent(self):
500
455
        new_parents = self.this_tree.get_parent_ids() + [self.other_rev_id]
501
456
        new_parent_trees = []
502
 
        operation = cleanup.OperationWithCleanups(
503
 
            self.this_tree.set_parent_trees)
 
457
        operation = OperationWithCleanups(self.this_tree.set_parent_trees)
504
458
        for revision_id in new_parents:
505
459
            try:
506
460
                tree = self.revision_tree(revision_id)
581
535
            elif len(lcas) == 1:
582
536
                self.base_rev_id = list(lcas)[0]
583
537
            else: # len(lcas) > 1
584
 
                self._is_criss_cross = True
585
538
                if len(lcas) > 2:
586
539
                    # find_unique_lca can only handle 2 nodes, so we have to
587
540
                    # start back at the beginning. It is a shame to traverse
592
545
                else:
593
546
                    self.base_rev_id = self.revision_graph.find_unique_lca(
594
547
                                            *lcas)
595
 
                sorted_lca_keys = self.revision_graph.find_merge_order(                
596
 
                    revisions[0], lcas)
597
 
                if self.base_rev_id == _mod_revision.NULL_REVISION:
598
 
                    self.base_rev_id = sorted_lca_keys[0]
599
 
                
 
548
                self._is_criss_cross = True
600
549
            if self.base_rev_id == _mod_revision.NULL_REVISION:
601
550
                raise errors.UnrelatedBranches()
602
551
            if self._is_criss_cross:
603
552
                trace.warning('Warning: criss-cross merge encountered.  See bzr'
604
553
                              ' help criss-cross.')
605
554
                trace.mutter('Criss-cross lcas: %r' % lcas)
606
 
                if self.base_rev_id in lcas:
607
 
                    trace.mutter('Unable to find unique lca. '
608
 
                                 'Fallback %r as best option.' % self.base_rev_id)
609
 
                interesting_revision_ids = set(lcas)
610
 
                interesting_revision_ids.add(self.base_rev_id)
 
555
                interesting_revision_ids = [self.base_rev_id]
 
556
                interesting_revision_ids.extend(lcas)
611
557
                interesting_trees = dict((t.get_revision_id(), t)
612
558
                    for t in self.this_branch.repository.revision_trees(
613
559
                        interesting_revision_ids))
614
560
                self._cached_trees.update(interesting_trees)
615
 
                if self.base_rev_id in lcas:
616
 
                    self.base_tree = interesting_trees[self.base_rev_id]
617
 
                else:
618
 
                    self.base_tree = interesting_trees.pop(self.base_rev_id)
 
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)
619
564
                self._lca_trees = [interesting_trees[key]
620
565
                                   for key in sorted_lca_keys]
621
566
            else:
700
645
        return merge
701
646
 
702
647
    def do_merge(self):
703
 
        operation = cleanup.OperationWithCleanups(self._do_merge_to)
 
648
        operation = OperationWithCleanups(self._do_merge_to)
704
649
        self.this_tree.lock_tree_write()
705
650
        operation.add_cleanup(self.this_tree.unlock)
706
651
        if self.base_tree is not None:
712
657
        merge = operation.run_simple()
713
658
        if len(merge.cooked_conflicts) == 0:
714
659
            if not self.ignore_zero and not trace.is_quiet():
715
 
                trace.note(gettext("All changes applied successfully."))
 
660
                trace.note("All changes applied successfully.")
716
661
        else:
717
 
            trace.note(gettext("%d conflicts encountered.")
 
662
            trace.note("%d conflicts encountered."
718
663
                       % len(merge.cooked_conflicts))
719
664
 
720
665
        return len(merge.cooked_conflicts)
759
704
        :param this_tree: The local tree in the merge operation
760
705
        :param base_tree: The common tree in the merge operation
761
706
        :param other_tree: The other tree to merge changes from
762
 
        :param this_branch: The branch associated with this_tree.  Defaults to
763
 
            this_tree.branch if not supplied.
 
707
        :param this_branch: The branch associated with this_tree
764
708
        :param interesting_ids: The file_ids of files that should be
765
709
            participate in the merge.  May not be combined with
766
710
            interesting_files.
784
728
        if interesting_files is not None and interesting_ids is not None:
785
729
            raise ValueError(
786
730
                'specify either interesting_ids or interesting_files')
787
 
        if this_branch is None:
788
 
            this_branch = this_tree.branch
789
731
        self.interesting_ids = interesting_ids
790
732
        self.interesting_files = interesting_files
791
733
        self.this_tree = working_tree
812
754
            warnings.warn("pb argument to Merge3Merger is deprecated")
813
755
 
814
756
    def do_merge(self):
815
 
        operation = cleanup.OperationWithCleanups(self._do_merge)
 
757
        operation = OperationWithCleanups(self._do_merge)
816
758
        self.this_tree.lock_tree_write()
817
759
        operation.add_cleanup(self.this_tree.unlock)
818
760
        self.base_tree.lock_read()
833
775
            pass
834
776
 
835
777
    def make_preview_transform(self):
836
 
        operation = cleanup.OperationWithCleanups(self._make_preview_transform)
 
778
        operation = OperationWithCleanups(self._make_preview_transform)
837
779
        self.base_tree.lock_read()
838
780
        operation.add_cleanup(self.base_tree.unlock)
839
781
        self.other_tree.lock_read()
859
801
            self.active_hooks = [hook for hook in hooks if hook is not None]
860
802
            for num, (file_id, changed, parents3, names3,
861
803
                      executable3) in enumerate(entries):
862
 
                child_pb.update(gettext('Preparing file merge'), num, len(entries))
 
804
                child_pb.update('Preparing file merge', num, len(entries))
863
805
                self._merge_names(file_id, parents3, names3, resolver=resolver)
864
806
                if changed:
865
807
                    file_status = self._do_merge_contents(file_id)
869
811
                    executable3, file_status, resolver=resolver)
870
812
        finally:
871
813
            child_pb.finished()
872
 
        self.tt.fixup_new_roots()
873
 
        self._finish_computing_transform()
874
 
 
875
 
    def _finish_computing_transform(self):
876
 
        """Finalize the transform and report the changes.
877
 
 
878
 
        This is the second half of _compute_transform.
879
 
        """
 
814
        self.fix_root()
880
815
        child_pb = ui.ui_factory.nested_progress_bar()
881
816
        try:
882
817
            fs_conflicts = transform.resolve_conflicts(self.tt, child_pb,
889
824
                self.tt.iter_changes(), self.change_reporter)
890
825
        self.cook_conflicts(fs_conflicts)
891
826
        for conflict in self.cooked_conflicts:
892
 
            trace.warning(unicode(conflict))
 
827
            trace.warning(conflict)
893
828
 
894
829
    def _entries3(self):
895
830
        """Gather data about files modified between three trees.
902
837
        """
903
838
        result = []
904
839
        iterator = self.other_tree.iter_changes(self.base_tree,
905
 
                specific_files=self.interesting_files,
 
840
                include_unchanged=True, specific_files=self.interesting_files,
906
841
                extra_trees=[self.this_tree])
907
842
        this_entries = dict((e.file_id, e) for p, e in
908
843
                            self.this_tree.iter_entries_by_dir(
934
869
        it then compares with THIS and BASE.
935
870
 
936
871
        For the multi-valued entries, the format will be (BASE, [lca1, lca2])
937
 
 
938
 
        :return: [(file_id, changed, parents, names, executable)], where:
939
 
 
940
 
            * file_id: Simple file_id of the entry
941
 
            * changed: Boolean, True if the kind or contents changed else False
942
 
            * parents: ((base, [parent_id, in, lcas]), parent_id_other,
943
 
                        parent_id_this)
944
 
            * names:   ((base, [name, in, lcas]), name_in_other, name_in_this)
945
 
            * executable: ((base, [exec, in, lcas]), exec_in_other,
946
 
                        exec_in_this)
 
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)
947
880
        """
948
881
        if self.interesting_files is not None:
949
882
            lookup_trees = [self.this_tree, self.base_tree]
991
924
                else:
992
925
                    lca_entries.append(lca_ie)
993
926
 
994
 
            if base_inventory.has_id(file_id):
 
927
            if file_id in base_inventory:
995
928
                base_ie = base_inventory[file_id]
996
929
            else:
997
930
                base_ie = _none_entry
998
931
 
999
 
            if this_inventory.has_id(file_id):
 
932
            if file_id in this_inventory:
1000
933
                this_ie = this_inventory[file_id]
1001
934
            else:
1002
935
                this_ie = _none_entry
1082
1015
                        continue
1083
1016
                else:
1084
1017
                    raise AssertionError('unhandled kind: %s' % other_ie.kind)
 
1018
                # XXX: We need to handle kind == 'symlink'
1085
1019
 
1086
1020
            # If we have gotten this far, that means something has changed
1087
1021
            result.append((file_id, content_changed,
1094
1028
                          ))
1095
1029
        return result
1096
1030
 
1097
 
    @deprecated_method(deprecated_in((2, 4, 0)))
 
1031
 
1098
1032
    def fix_root(self):
1099
 
        if self.tt.final_kind(self.tt.root) is None:
 
1033
        try:
 
1034
            self.tt.final_kind(self.tt.root)
 
1035
        except errors.NoSuchFile:
1100
1036
            self.tt.cancel_deletion(self.tt.root)
1101
1037
        if self.tt.final_file_id(self.tt.root) is None:
1102
1038
            self.tt.version_file(self.tt.tree_file_id(self.tt.root),
1107
1043
        other_root = self.tt.trans_id_file_id(other_root_file_id)
1108
1044
        if other_root == self.tt.root:
1109
1045
            return
1110
 
        if self.this_tree.inventory.has_id(
1111
 
            self.other_tree.inventory.root.file_id):
1112
 
            # the other tree's root is a non-root in the current tree (as
1113
 
            # when a previously unrelated branch is merged into another)
1114
 
            return
1115
 
        if self.tt.final_kind(other_root) is not None:
1116
 
            other_root_is_present = True
1117
 
        else:
1118
 
            # other_root doesn't have a physical representation. We still need
1119
 
            # to move any references to the actual root of the tree.
1120
 
            other_root_is_present = False
1121
 
        # 'other_tree.inventory.root' is not present in this tree. We are
1122
 
        # calling adjust_path for children which *want* to be present with a
1123
 
        # correct place to go.
1124
 
        for _, child in self.other_tree.inventory.root.children.iteritems():
 
1046
        try:
 
1047
            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
 
1052
            return
 
1053
        self.reparent_children(self.other_tree.inventory.root, self.tt.root)
 
1054
        self.tt.cancel_creation(other_root)
 
1055
        self.tt.cancel_versioning(other_root)
 
1056
 
 
1057
    def reparent_children(self, ie, target):
 
1058
        for thing, child in ie.children.iteritems():
1125
1059
            trans_id = self.tt.trans_id_file_id(child.file_id)
1126
 
            if not other_root_is_present:
1127
 
                if self.tt.final_kind(trans_id) is not None:
1128
 
                    # The item exist in the final tree and has a defined place
1129
 
                    # to go already.
1130
 
                    continue
1131
 
            # Move the item into the root
1132
 
            try:
1133
 
                final_name = self.tt.final_name(trans_id)
1134
 
            except errors.NoFinalPath:
1135
 
                # This file is not present anymore, ignore it.
1136
 
                continue
1137
 
            self.tt.adjust_path(final_name, self.tt.root, trans_id)
1138
 
        if other_root_is_present:
1139
 
            self.tt.cancel_creation(other_root)
1140
 
            self.tt.cancel_versioning(other_root)
 
1060
            self.tt.adjust_path(self.tt.final_name(trans_id), target, trans_id)
1141
1061
 
1142
1062
    def write_modified(self, results):
1143
1063
        modified_hashes = {}
1168
1088
    @staticmethod
1169
1089
    def contents_sha1(tree, file_id):
1170
1090
        """Determine the sha1 of the file contents (used as a key method)."""
1171
 
        if not tree.has_id(file_id):
 
1091
        if file_id not in tree:
1172
1092
            return None
1173
1093
        return tree.get_file_sha1(file_id)
1174
1094
 
1190
1110
 
1191
1111
    @staticmethod
1192
1112
    def _three_way(base, other, this):
 
1113
        #if base == other, either they all agree, or only THIS has changed.
1193
1114
        if base == other:
1194
 
            # if 'base == other', either they all agree, or only 'this' has
1195
 
            # changed.
1196
1115
            return 'this'
1197
1116
        elif this not in (base, other):
1198
 
            # 'this' is neither 'base' nor 'other', so both sides changed
1199
1117
            return 'conflict'
 
1118
        # "Ambiguous clean merge" -- both sides have made the same change.
1200
1119
        elif this == other:
1201
 
            # "Ambiguous clean merge" -- both sides have made the same change.
1202
1120
            return "this"
 
1121
        # this == base: only other has changed.
1203
1122
        else:
1204
 
            # this == base: only other has changed.
1205
1123
            return "other"
1206
1124
 
1207
1125
    @staticmethod
1251
1169
                # only has an lca value
1252
1170
                return 'other'
1253
1171
 
1254
 
        # At this point, the lcas disagree, and the tip disagree
 
1172
        # At this point, the lcas disagree, and the tips disagree
1255
1173
        return 'conflict'
1256
1174
 
1257
1175
    @staticmethod
1258
 
    @deprecated_method(deprecated_in((2, 2, 0)))
1259
1176
    def scalar_three_way(this_tree, base_tree, other_tree, file_id, key):
1260
1177
        """Do a three-way test on a scalar.
1261
1178
        Return "this", "other" or "conflict", depending whether a value wins.
1311
1228
                parent_id_winner = "other"
1312
1229
        if name_winner == "this" and parent_id_winner == "this":
1313
1230
            return
1314
 
        if name_winner == 'conflict' or parent_id_winner == 'conflict':
1315
 
            # Creating helpers (.OTHER or .THIS) here cause problems down the
1316
 
            # road if a ContentConflict needs to be created so we should not do
1317
 
            # that
1318
 
            trans_id = self.tt.trans_id_file_id(file_id)
1319
 
            self._raw_conflicts.append(('path conflict', trans_id, file_id,
1320
 
                                        this_parent, this_name,
1321
 
                                        other_parent, other_name))
1322
 
        if not self.other_tree.has_id(file_id):
 
1231
        if name_winner == "conflict":
 
1232
            trans_id = self.tt.trans_id_file_id(file_id)
 
1233
            self._raw_conflicts.append(('name conflict', trans_id,
 
1234
                                        this_name, other_name))
 
1235
        if parent_id_winner == "conflict":
 
1236
            trans_id = self.tt.trans_id_file_id(file_id)
 
1237
            self._raw_conflicts.append(('parent conflict', trans_id,
 
1238
                                        this_parent, other_parent))
 
1239
        if other_name is None:
1323
1240
            # it doesn't matter whether the result was 'other' or
1324
 
            # 'conflict'-- if it has no file id, we leave it alone.
 
1241
            # 'conflict'-- if there's no 'other', we leave it alone.
1325
1242
            return
 
1243
        # if we get here, name_winner and parent_winner are set to safe values.
 
1244
        trans_id = self.tt.trans_id_file_id(file_id)
1326
1245
        parent_id = parents[self.winner_idx[parent_id_winner]]
1327
 
        name = names[self.winner_idx[name_winner]]
1328
 
        if parent_id is not None or name is not None:
1329
 
            # if we get here, name_winner and parent_winner are set to safe
1330
 
            # values.
1331
 
            if parent_id is None and name is not None:
1332
 
                # if parent_id is None and name is non-None, current file is
1333
 
                # the tree root.
1334
 
                if names[self.winner_idx[parent_id_winner]] != '':
1335
 
                    raise AssertionError(
1336
 
                        'File looks like a root, but named %s' %
1337
 
                        names[self.winner_idx[parent_id_winner]])
1338
 
                parent_trans_id = transform.ROOT_PARENT
1339
 
            else:
1340
 
                parent_trans_id = self.tt.trans_id_file_id(parent_id)
1341
 
            self.tt.adjust_path(name, parent_trans_id,
1342
 
                                self.tt.trans_id_file_id(file_id))
 
1246
        if parent_id is not None:
 
1247
            parent_trans_id = self.tt.trans_id_file_id(parent_id)
 
1248
            self.tt.adjust_path(names[self.winner_idx[name_winner]],
 
1249
                                parent_trans_id, trans_id)
1343
1250
 
1344
1251
    def _do_merge_contents(self, file_id):
1345
1252
        """Performs a merge on file_id contents."""
1346
1253
        def contents_pair(tree):
1347
 
            if not tree.has_id(file_id):
 
1254
            if file_id not in tree:
1348
1255
                return (None, None)
1349
1256
            kind = tree.kind(file_id)
1350
1257
            if kind == "file":
1423
1330
            self.tt.version_file(file_id, trans_id)
1424
1331
        # The merge has been performed, so the old contents should not be
1425
1332
        # retained.
1426
 
        self.tt.delete_contents(trans_id)
 
1333
        try:
 
1334
            self.tt.delete_contents(trans_id)
 
1335
        except errors.NoSuchFile:
 
1336
            pass
1427
1337
        return result
1428
1338
 
1429
1339
    def _default_other_winner_merge(self, merge_hook_params):
1482
1392
    def get_lines(self, tree, file_id):
1483
1393
        """Return the lines in a file, or an empty list."""
1484
1394
        if tree.has_id(file_id):
1485
 
            return tree.get_file_lines(file_id)
 
1395
            return tree.get_file(file_id).readlines()
1486
1396
        else:
1487
1397
            return []
1488
1398
 
1601
1511
        if winner == 'this' and file_status != "modified":
1602
1512
            return
1603
1513
        trans_id = self.tt.trans_id_file_id(file_id)
1604
 
        if self.tt.final_kind(trans_id) != "file":
 
1514
        try:
 
1515
            if self.tt.final_kind(trans_id) != "file":
 
1516
                return
 
1517
        except errors.NoSuchFile:
1605
1518
            return
1606
1519
        if winner == "this":
1607
1520
            executability = this_executable
1618
1531
 
1619
1532
    def cook_conflicts(self, fs_conflicts):
1620
1533
        """Convert all conflicts into a form that doesn't depend on trans_id"""
1621
 
        content_conflict_file_ids = set()
1622
 
        cooked_conflicts = transform.cook_conflicts(fs_conflicts, self.tt)
 
1534
        name_conflicts = {}
 
1535
        self.cooked_conflicts.extend(transform.cook_conflicts(
 
1536
                fs_conflicts, self.tt))
1623
1537
        fp = transform.FinalPaths(self.tt)
1624
1538
        for conflict in self._raw_conflicts:
1625
1539
            conflict_type = conflict[0]
1626
 
            if conflict_type == 'path conflict':
1627
 
                (trans_id, file_id,
1628
 
                this_parent, this_name,
1629
 
                other_parent, other_name) = conflict[1:]
1630
 
                if this_parent is None or this_name is None:
1631
 
                    this_path = '<deleted>'
1632
 
                else:
1633
 
                    parent_path =  fp.get_path(
1634
 
                        self.tt.trans_id_file_id(this_parent))
1635
 
                    this_path = osutils.pathjoin(parent_path, this_name)
1636
 
                if other_parent is None or other_name is None:
1637
 
                    other_path = '<deleted>'
1638
 
                else:
1639
 
                    if other_parent == self.other_tree.get_root_id():
1640
 
                        # The tree transform doesn't know about the other root,
1641
 
                        # so we special case here to avoid a NoFinalPath
1642
 
                        # exception
1643
 
                        parent_path = ''
1644
 
                    else:
1645
 
                        parent_path =  fp.get_path(
1646
 
                            self.tt.trans_id_file_id(other_parent))
1647
 
                    other_path = osutils.pathjoin(parent_path, other_name)
1648
 
                c = _mod_conflicts.Conflict.factory(
1649
 
                    'path conflict', path=this_path,
1650
 
                    conflict_path=other_path,
1651
 
                    file_id=file_id)
1652
 
            elif conflict_type == 'contents conflict':
 
1540
            if conflict_type in ('name conflict', 'parent conflict'):
 
1541
                trans_id = conflict[1]
 
1542
                conflict_args = conflict[2:]
 
1543
                if trans_id not in name_conflicts:
 
1544
                    name_conflicts[trans_id] = {}
 
1545
                transform.unique_add(name_conflicts[trans_id], conflict_type,
 
1546
                                     conflict_args)
 
1547
            if conflict_type == 'contents conflict':
1653
1548
                for trans_id in conflict[1]:
1654
1549
                    file_id = self.tt.final_file_id(trans_id)
1655
1550
                    if file_id is not None:
1661
1556
                        break
1662
1557
                c = _mod_conflicts.Conflict.factory(conflict_type,
1663
1558
                                                    path=path, file_id=file_id)
1664
 
                content_conflict_file_ids.add(file_id)
1665
 
            elif conflict_type == 'text conflict':
 
1559
                self.cooked_conflicts.append(c)
 
1560
            if conflict_type == 'text conflict':
1666
1561
                trans_id = conflict[1]
1667
1562
                path = fp.get_path(trans_id)
1668
1563
                file_id = self.tt.final_file_id(trans_id)
1669
1564
                c = _mod_conflicts.Conflict.factory(conflict_type,
1670
1565
                                                    path=path, file_id=file_id)
 
1566
                self.cooked_conflicts.append(c)
 
1567
 
 
1568
        for trans_id, conflicts in name_conflicts.iteritems():
 
1569
            try:
 
1570
                this_parent, other_parent = conflicts['parent conflict']
 
1571
                if this_parent == other_parent:
 
1572
                    raise AssertionError()
 
1573
            except KeyError:
 
1574
                this_parent = other_parent = \
 
1575
                    self.tt.final_file_id(self.tt.final_parent(trans_id))
 
1576
            try:
 
1577
                this_name, other_name = conflicts['name conflict']
 
1578
                if this_name == other_name:
 
1579
                    raise AssertionError()
 
1580
            except KeyError:
 
1581
                this_name = other_name = self.tt.final_name(trans_id)
 
1582
            other_path = fp.get_path(trans_id)
 
1583
            if this_parent is not None and this_name is not None:
 
1584
                this_parent_path = \
 
1585
                    fp.get_path(self.tt.trans_id_file_id(this_parent))
 
1586
                this_path = osutils.pathjoin(this_parent_path, this_name)
1671
1587
            else:
1672
 
                raise AssertionError('bad conflict type: %r' % (conflict,))
1673
 
            cooked_conflicts.append(c)
1674
 
 
1675
 
        self.cooked_conflicts = []
1676
 
        # We want to get rid of path conflicts when a corresponding contents
1677
 
        # conflict exists. This can occur when one branch deletes a file while
1678
 
        # the other renames *and* modifies it. In this case, the content
1679
 
        # conflict is enough.
1680
 
        for c in cooked_conflicts:
1681
 
            if (c.typestring == 'path conflict'
1682
 
                and c.file_id in content_conflict_file_ids):
1683
 
                continue
 
1588
                this_path = "<deleted>"
 
1589
            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)
1684
1593
            self.cooked_conflicts.append(c)
1685
1594
        self.cooked_conflicts.sort(key=_mod_conflicts.Conflict.sort_key)
1686
1595
 
1792
1701
            osutils.rmtree(temp_dir)
1793
1702
 
1794
1703
 
1795
 
class PathNotInTree(errors.BzrError):
1796
 
 
1797
 
    _fmt = """Merge-into failed because %(tree)s does not contain %(path)s."""
1798
 
 
1799
 
    def __init__(self, path, tree):
1800
 
        errors.BzrError.__init__(self, path=path, tree=tree)
1801
 
 
1802
 
 
1803
 
class MergeIntoMerger(Merger):
1804
 
    """Merger that understands other_tree will be merged into a subdir.
1805
 
 
1806
 
    This also changes the Merger api so that it uses real Branch, revision_id,
1807
 
    and RevisonTree objects, rather than using revision specs.
1808
 
    """
1809
 
 
1810
 
    def __init__(self, this_tree, other_branch, other_tree, target_subdir,
1811
 
            source_subpath, other_rev_id=None):
1812
 
        """Create a new MergeIntoMerger object.
1813
 
 
1814
 
        source_subpath in other_tree will be effectively copied to
1815
 
        target_subdir in this_tree.
1816
 
 
1817
 
        :param this_tree: The tree that we will be merging into.
1818
 
        :param other_branch: The Branch we will be merging from.
1819
 
        :param other_tree: The RevisionTree object we want to merge.
1820
 
        :param target_subdir: The relative path where we want to merge
1821
 
            other_tree into this_tree
1822
 
        :param source_subpath: The relative path specifying the subtree of
1823
 
            other_tree to merge into this_tree.
1824
 
        """
1825
 
        # It is assumed that we are merging a tree that is not in our current
1826
 
        # ancestry, which means we are using the "EmptyTree" as our basis.
1827
 
        null_ancestor_tree = this_tree.branch.repository.revision_tree(
1828
 
                                _mod_revision.NULL_REVISION)
1829
 
        super(MergeIntoMerger, self).__init__(
1830
 
            this_branch=this_tree.branch,
1831
 
            this_tree=this_tree,
1832
 
            other_tree=other_tree,
1833
 
            base_tree=null_ancestor_tree,
1834
 
            )
1835
 
        self._target_subdir = target_subdir
1836
 
        self._source_subpath = source_subpath
1837
 
        self.other_branch = other_branch
1838
 
        if other_rev_id is None:
1839
 
            other_rev_id = other_tree.get_revision_id()
1840
 
        self.other_rev_id = self.other_basis = other_rev_id
1841
 
        self.base_is_ancestor = True
1842
 
        self.backup_files = True
1843
 
        self.merge_type = Merge3Merger
1844
 
        self.show_base = False
1845
 
        self.reprocess = False
1846
 
        self.interesting_ids = None
1847
 
        self.merge_type = _MergeTypeParameterizer(MergeIntoMergeType,
1848
 
              target_subdir=self._target_subdir,
1849
 
              source_subpath=self._source_subpath)
1850
 
        if self._source_subpath != '':
1851
 
            # If this isn't a partial merge make sure the revisions will be
1852
 
            # present.
1853
 
            self._maybe_fetch(self.other_branch, self.this_branch,
1854
 
                self.other_basis)
1855
 
 
1856
 
    def set_pending(self):
1857
 
        if self._source_subpath != '':
1858
 
            return
1859
 
        Merger.set_pending(self)
1860
 
 
1861
 
 
1862
 
class _MergeTypeParameterizer(object):
1863
 
    """Wrap a merge-type class to provide extra parameters.
1864
 
    
1865
 
    This is hack used by MergeIntoMerger to pass some extra parameters to its
1866
 
    merge_type.  Merger.do_merge() sets up its own set of parameters to pass to
1867
 
    the 'merge_type' member.  It is difficult override do_merge without
1868
 
    re-writing the whole thing, so instead we create a wrapper which will pass
1869
 
    the extra parameters.
1870
 
    """
1871
 
 
1872
 
    def __init__(self, merge_type, **kwargs):
1873
 
        self._extra_kwargs = kwargs
1874
 
        self._merge_type = merge_type
1875
 
 
1876
 
    def __call__(self, *args, **kwargs):
1877
 
        kwargs.update(self._extra_kwargs)
1878
 
        return self._merge_type(*args, **kwargs)
1879
 
 
1880
 
    def __getattr__(self, name):
1881
 
        return getattr(self._merge_type, name)
1882
 
 
1883
 
 
1884
 
class MergeIntoMergeType(Merge3Merger):
1885
 
    """Merger that incorporates a tree (or part of a tree) into another."""
1886
 
 
1887
 
    def __init__(self, *args, **kwargs):
1888
 
        """Initialize the merger object.
1889
 
 
1890
 
        :param args: See Merge3Merger.__init__'s args.
1891
 
        :param kwargs: See Merge3Merger.__init__'s keyword args, except for
1892
 
            source_subpath and target_subdir.
1893
 
        :keyword source_subpath: The relative path specifying the subtree of
1894
 
            other_tree to merge into this_tree.
1895
 
        :keyword target_subdir: The relative path where we want to merge
1896
 
            other_tree into this_tree
1897
 
        """
1898
 
        # All of the interesting work happens during Merge3Merger.__init__(),
1899
 
        # so we have have to hack in to get our extra parameters set.
1900
 
        self._source_subpath = kwargs.pop('source_subpath')
1901
 
        self._target_subdir = kwargs.pop('target_subdir')
1902
 
        super(MergeIntoMergeType, self).__init__(*args, **kwargs)
1903
 
 
1904
 
    def _compute_transform(self):
1905
 
        child_pb = ui.ui_factory.nested_progress_bar()
1906
 
        try:
1907
 
            entries = self._entries_to_incorporate()
1908
 
            entries = list(entries)
1909
 
            for num, (entry, parent_id) in enumerate(entries):
1910
 
                child_pb.update(gettext('Preparing file merge'), num, len(entries))
1911
 
                parent_trans_id = self.tt.trans_id_file_id(parent_id)
1912
 
                trans_id = transform.new_by_entry(self.tt, entry,
1913
 
                    parent_trans_id, self.other_tree)
1914
 
        finally:
1915
 
            child_pb.finished()
1916
 
        self._finish_computing_transform()
1917
 
 
1918
 
    def _entries_to_incorporate(self):
1919
 
        """Yields pairs of (inventory_entry, new_parent)."""
1920
 
        other_inv = self.other_tree.inventory
1921
 
        subdir_id = other_inv.path2id(self._source_subpath)
1922
 
        if subdir_id is None:
1923
 
            # XXX: The error would be clearer if it gave the URL of the source
1924
 
            # branch, but we don't have a reference to that here.
1925
 
            raise PathNotInTree(self._source_subpath, "Source tree")
1926
 
        subdir = other_inv[subdir_id]
1927
 
        parent_in_target = osutils.dirname(self._target_subdir)
1928
 
        target_id = self.this_tree.inventory.path2id(parent_in_target)
1929
 
        if target_id is None:
1930
 
            raise PathNotInTree(self._target_subdir, "Target tree")
1931
 
        name_in_target = osutils.basename(self._target_subdir)
1932
 
        merge_into_root = subdir.copy()
1933
 
        merge_into_root.name = name_in_target
1934
 
        if self.this_tree.inventory.has_id(merge_into_root.file_id):
1935
 
            # Give the root a new file-id.
1936
 
            # This can happen fairly easily if the directory we are
1937
 
            # incorporating is the root, and both trees have 'TREE_ROOT' as
1938
 
            # their root_id.  Users will expect this to Just Work, so we
1939
 
            # change the file-id here.
1940
 
            # Non-root file-ids could potentially conflict too.  That's really
1941
 
            # an edge case, so we don't do anything special for those.  We let
1942
 
            # them cause conflicts.
1943
 
            merge_into_root.file_id = generate_ids.gen_file_id(name_in_target)
1944
 
        yield (merge_into_root, target_id)
1945
 
        if subdir.kind != 'directory':
1946
 
            # No children, so we are done.
1947
 
            return
1948
 
        for ignored_path, entry in other_inv.iter_entries_by_dir(subdir_id):
1949
 
            parent_id = entry.parent_id
1950
 
            if parent_id == subdir.file_id:
1951
 
                # The root's parent ID has changed, so make sure children of
1952
 
                # the root refer to the new ID.
1953
 
                parent_id = merge_into_root.file_id
1954
 
            yield (entry, parent_id)
1955
 
 
1956
 
 
1957
1704
def merge_inner(this_branch, other_tree, base_tree, ignore_zero=False,
1958
1705
                backup_files=False,
1959
1706
                merge_type=Merge3Merger,
1967
1714
                change_reporter=None):
1968
1715
    """Primary interface for merging.
1969
1716
 
1970
 
    Typical use is probably::
1971
 
 
1972
 
        merge_inner(branch, branch.get_revision_tree(other_revision),
1973
 
                    branch.get_revision_tree(base_revision))
1974
 
    """
 
1717
        typical use is probably
 
1718
        'merge_inner(branch, branch.get_revision_tree(other_revision),
 
1719
                     branch.get_revision_tree(base_revision))'
 
1720
        """
1975
1721
    if this_tree is None:
1976
1722
        raise errors.BzrError("bzrlib.merge.merge_inner requires a this_tree "
1977
 
                              "parameter")
 
1723
                              "parameter as of bzrlib version 0.8.")
1978
1724
    merger = Merger(this_branch, other_tree, base_tree, this_tree=this_tree,
1979
1725
                    pb=pb, change_reporter=change_reporter)
1980
1726
    merger.backup_files = backup_files
2434
2180
class _PlanLCAMerge(_PlanMergeBase):
2435
2181
    """
2436
2182
    This merge algorithm differs from _PlanMerge in that:
2437
 
 
2438
2183
    1. comparisons are done against LCAs only
2439
2184
    2. cases where a contested line is new versus one LCA but old versus
2440
2185
       another are marked as conflicts, by emitting the line as conflicted-a
2481
2226
 
2482
2227
        If a line is killed and new, this indicates that the two merge
2483
2228
        revisions contain differing conflict resolutions.
2484
 
 
2485
2229
        :param revision_id: The id of the revision in which the lines are
2486
2230
            unique
2487
2231
        :param unique_line_numbers: The line numbers of unique lines.
2488
 
        :return: a tuple of (new_this, killed_other)
 
2232
        :return a tuple of (new_this, killed_other):
2489
2233
        """
2490
2234
        new = set()
2491
2235
        killed = set()