57
57
def __init__(self):
58
58
hooks.Hooks.__init__(self)
59
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 "
60
"Called when file content needs to be merged (including when one "
61
"side has deleted the file and the other has changed it)."
62
"merge_file_content is called with a "
63
"bzrlib.merge.MergeHookParams. The function should return a tuple "
64
"of (status, lines), where status is one of 'not_applicable', "
65
"'success', 'conflicted', or 'delete'. If status is success or "
66
"conflicted, then lines should be an iterable of strings of the "
72
class AbstractPerFileMerger(object):
73
"""PerFileMerger objects are used by plugins extending merge for bzrlib.
75
See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
77
:ivar merger: The Merge3Merger performing the merge.
80
def __init__(self, merger):
81
"""Create a PerFileMerger for use with merger."""
84
def merge_contents(self, merge_params):
85
"""Attempt to merge the contents of a single file.
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.
93
return ('not applicable', None)
96
class PerFileMerger(AbstractPerFileMerger):
97
"""Merge individual files when self.file_matches returns True.
99
This class is intended to be subclassed. The file_matches and
100
merge_matching methods should be overridden with concrete implementations.
103
def file_matches(self, params):
104
"""Return True if merge_matching should be called on this file.
106
Only called with merges of plain files with no clear winner.
108
Subclasses must override this.
110
raise NotImplementedError(self.file_matches)
112
def get_filename(self, params, tree):
113
"""Lookup the filename (i.e. basename, not path), given a Tree (e.g.
114
self.merger.this_tree) and a MergeHookParams.
116
return osutils.basename(tree.id2path(params.file_id))
118
def get_filepath(self, params, tree):
119
"""Calculate the path to the file in a tree.
121
:param params: A MergeHookParams describing the file to merge
122
:param tree: a Tree, e.g. self.merger.this_tree.
124
return tree.id2path(params.file_id)
126
def merge_contents(self, params):
127
"""Merge the contents of a single file."""
128
# Check whether this custom merge logic should be used.
130
# OTHER is a straight winner, rely on default merge.
131
params.winner == 'other' or
132
# THIS and OTHER aren't both files.
133
not params.is_file_merge() or
134
# The filename doesn't match *.xml
135
not self.file_matches(params)):
136
return 'not_applicable', None
137
return self.merge_matching(params)
139
def merge_matching(self, params):
140
"""Merge the contents of a single file that has matched the criteria
141
in PerFileMerger.merge_contents (is a conflict, is a file,
142
self.file_matches is True).
144
Subclasses must override this.
146
raise NotImplementedError(self.merge_matching)
149
class ConfigurableFileMerger(PerFileMerger):
150
"""Merge individual files when configured via a .conf file.
152
This is a base class for concrete custom file merging logic. Concrete
153
classes should implement ``merge_text``.
155
See ``bzrlib.plugins.news_merge.news_merge`` for an example concrete class.
157
:ivar affected_files: The configured file paths to merge.
159
:cvar name_prefix: The prefix to use when looking up configuration
160
details. <name_prefix>_merge_files describes the files targeted by the
163
:cvar default_files: The default file paths to merge when no configuration
170
def __init__(self, merger):
171
super(ConfigurableFileMerger, self).__init__(merger)
172
self.affected_files = None
173
self.default_files = self.__class__.default_files or []
174
self.name_prefix = self.__class__.name_prefix
175
if self.name_prefix is None:
176
raise ValueError("name_prefix must be set.")
178
def file_matches(self, params):
179
"""Check whether the file should call the merge hook.
181
<name_prefix>_merge_files configuration variable is a list of files
182
that should use the hook.
184
affected_files = self.affected_files
185
if affected_files is None:
186
config = self.merger.this_branch.get_config()
187
# Until bzr provides a better policy for caching the config, we
188
# just add the part we're interested in to the params to avoid
189
# reading the config files repeatedly (bazaar.conf, location.conf,
191
config_key = self.name_prefix + '_merge_files'
192
affected_files = config.get_user_option_as_list(config_key)
193
if affected_files is None:
194
# If nothing was specified in the config, use the default.
195
affected_files = self.default_files
196
self.affected_files = affected_files
198
filepath = self.get_filepath(params, self.merger.this_tree)
199
if filepath in affected_files:
203
def merge_matching(self, params):
204
return self.merge_text(params)
206
def merge_text(self, params):
207
"""Merge the byte contents of a single file.
209
This is called after checking that the merge should be performed in
210
merge_contents, and it should behave as per
211
``bzrlib.merge.AbstractPerFileMerger.merge_contents``.
213
raise NotImplementedError(self.merge_text)
216
71
class MergeHookParams(object):
217
72
"""Object holding parameters passed to merge_file_content hooks.
219
There are some fields hooks can access:
74
There are 3 fields hooks can access:
76
:ivar merger: the Merger object
221
77
:ivar file_id: the file ID of the file being merged
222
78
:ivar trans_id: the transform ID for the merge of this file
223
79
:ivar this_kind: kind of file_id in 'this' tree
788
652
# making sure we haven't missed any corner cases.
789
653
# if lca_trees is None:
790
654
# self._lca_trees = [self.base_tree]
791
657
self.change_reporter = change_reporter
792
658
self.cherrypick = cherrypick
660
self.pp = progress.ProgressPhase("Merge phase", 3, self.pb)
796
warnings.warn("pp argument to Merge3Merger is deprecated")
798
warnings.warn("pb argument to Merge3Merger is deprecated")
800
664
def do_merge(self):
801
operation = OperationWithCleanups(self._do_merge)
802
665
self.this_tree.lock_tree_write()
803
operation.add_cleanup(self.this_tree.unlock)
804
666
self.base_tree.lock_read()
805
operation.add_cleanup(self.base_tree.unlock)
806
667
self.other_tree.lock_read()
807
operation.add_cleanup(self.other_tree.unlock)
810
def _do_merge(self, operation):
811
self.tt = transform.TreeTransform(self.this_tree, None)
812
operation.add_cleanup(self.tt.finalize)
813
self._compute_transform()
814
results = self.tt.apply(no_conflicts=True)
815
self.write_modified(results)
817
self.this_tree.add_conflicts(self.cooked_conflicts)
818
except errors.UnsupportedOperation:
669
self.tt = transform.TreeTransform(self.this_tree, self.pb)
672
self._compute_transform()
674
results = self.tt.apply(no_conflicts=True)
675
self.write_modified(results)
677
self.this_tree.add_conflicts(self.cooked_conflicts)
678
except errors.UnsupportedOperation:
683
self.other_tree.unlock()
684
self.base_tree.unlock()
685
self.this_tree.unlock()
821
688
def make_preview_transform(self):
822
operation = OperationWithCleanups(self._make_preview_transform)
823
689
self.base_tree.lock_read()
824
operation.add_cleanup(self.base_tree.unlock)
825
690
self.other_tree.lock_read()
826
operation.add_cleanup(self.other_tree.unlock)
827
return operation.run_simple()
829
def _make_preview_transform(self):
830
691
self.tt = transform.TransformPreview(self.this_tree)
831
self._compute_transform()
694
self._compute_transform()
697
self.other_tree.unlock()
698
self.base_tree.unlock()
834
702
def _compute_transform(self):
1086
953
other_root = self.tt.trans_id_file_id(other_root_file_id)
1087
954
if other_root == self.tt.root:
1089
if self.other_tree.inventory.root.file_id in self.this_tree.inventory:
1090
# the other tree's root is a non-root in the current tree (as when
1091
# a previously unrelated branch is merged into another)
1094
957
self.tt.final_kind(other_root)
1095
other_root_is_present = True
1096
958
except errors.NoSuchFile:
1097
# other_root doesn't have a physical representation. We still need
1098
# to move any references to the actual root of the tree.
1099
other_root_is_present = False
1100
# 'other_tree.inventory.root' is not present in this tree. We are
1101
# calling adjust_path for children which *want* to be present with a
1102
# correct place to go.
1103
for thing, child in self.other_tree.inventory.root.children.iteritems():
960
if self.this_tree.has_id(self.other_tree.inventory.root.file_id):
961
# the other tree's root is a non-root in the current tree
963
self.reparent_children(self.other_tree.inventory.root, self.tt.root)
964
self.tt.cancel_creation(other_root)
965
self.tt.cancel_versioning(other_root)
967
def reparent_children(self, ie, target):
968
for thing, child in ie.children.iteritems():
1104
969
trans_id = self.tt.trans_id_file_id(child.file_id)
1105
if not other_root_is_present:
1106
# FIXME: Make final_kind returns None instead of raising
1107
# NoSuchFile to avoid the ugly construct below -- vila 20100402
1109
self.tt.final_kind(trans_id)
1110
# The item exist in the final tree and has a defined place
1113
except errors.NoSuchFile, e:
1115
# Move the item into the root
1116
self.tt.adjust_path(self.tt.final_name(trans_id),
1117
self.tt.root, trans_id)
1118
if other_root_is_present:
1119
self.tt.cancel_creation(other_root)
1120
self.tt.cancel_versioning(other_root)
970
self.tt.adjust_path(self.tt.final_name(trans_id), target, trans_id)
1122
972
def write_modified(self, results):
1123
973
modified_hashes = {}
1291
1138
parent_id_winner = "other"
1292
1139
if name_winner == "this" and parent_id_winner == "this":
1294
if name_winner == 'conflict' or parent_id_winner == 'conflict':
1295
# Creating helpers (.OTHER or .THIS) here cause problems down the
1296
# road if a ContentConflict needs to be created so we should not do
1298
trans_id = self.tt.trans_id_file_id(file_id)
1299
self._raw_conflicts.append(('path conflict', trans_id, file_id,
1300
this_parent, this_name,
1301
other_parent, other_name))
1141
if name_winner == "conflict":
1142
trans_id = self.tt.trans_id_file_id(file_id)
1143
self._raw_conflicts.append(('name conflict', trans_id,
1144
this_name, other_name))
1145
if parent_id_winner == "conflict":
1146
trans_id = self.tt.trans_id_file_id(file_id)
1147
self._raw_conflicts.append(('parent conflict', trans_id,
1148
this_parent, other_parent))
1302
1149
if other_name is None:
1303
1150
# it doesn't matter whether the result was 'other' or
1304
1151
# 'conflict'-- if there's no 'other', we leave it alone.
1153
# if we get here, name_winner and parent_winner are set to safe values.
1154
trans_id = self.tt.trans_id_file_id(file_id)
1306
1155
parent_id = parents[self.winner_idx[parent_id_winner]]
1307
1156
if parent_id is not None:
1308
# if we get here, name_winner and parent_winner are set to safe
1157
parent_trans_id = self.tt.trans_id_file_id(parent_id)
1310
1158
self.tt.adjust_path(names[self.winner_idx[name_winner]],
1311
self.tt.trans_id_file_id(parent_id),
1312
self.tt.trans_id_file_id(file_id))
1159
parent_trans_id, trans_id)
1314
def _do_merge_contents(self, file_id):
1161
def merge_contents(self, file_id):
1315
1162
"""Performs a merge on file_id contents."""
1316
1163
def contents_pair(tree):
1317
1164
if file_id not in tree:
1595
1439
def cook_conflicts(self, fs_conflicts):
1596
1440
"""Convert all conflicts into a form that doesn't depend on trans_id"""
1597
1442
self.cooked_conflicts.extend(transform.cook_conflicts(
1598
1443
fs_conflicts, self.tt))
1599
1444
fp = transform.FinalPaths(self.tt)
1600
1445
for conflict in self._raw_conflicts:
1601
1446
conflict_type = conflict[0]
1602
if conflict_type == 'path conflict':
1604
this_parent, this_name,
1605
other_parent, other_name) = conflict[1:]
1606
if this_parent is None or this_name is None:
1607
this_path = '<deleted>'
1609
parent_path = fp.get_path(
1610
self.tt.trans_id_file_id(this_parent))
1611
this_path = osutils.pathjoin(parent_path, this_name)
1612
if other_parent is None or other_name is None:
1613
other_path = '<deleted>'
1615
parent_path = fp.get_path(
1616
self.tt.trans_id_file_id(other_parent))
1617
other_path = osutils.pathjoin(parent_path, other_name)
1618
c = _mod_conflicts.Conflict.factory(
1619
'path conflict', path=this_path,
1620
conflict_path=other_path,
1622
elif conflict_type == 'contents conflict':
1447
if conflict_type in ('name conflict', 'parent conflict'):
1448
trans_id = conflict[1]
1449
conflict_args = conflict[2:]
1450
if trans_id not in name_conflicts:
1451
name_conflicts[trans_id] = {}
1452
transform.unique_add(name_conflicts[trans_id], conflict_type,
1454
if conflict_type == 'contents conflict':
1623
1455
for trans_id in conflict[1]:
1624
1456
file_id = self.tt.final_file_id(trans_id)
1625
1457
if file_id is not None:
1632
1464
c = _mod_conflicts.Conflict.factory(conflict_type,
1633
1465
path=path, file_id=file_id)
1634
elif conflict_type == 'text conflict':
1466
self.cooked_conflicts.append(c)
1467
if conflict_type == 'text conflict':
1635
1468
trans_id = conflict[1]
1636
1469
path = fp.get_path(trans_id)
1637
1470
file_id = self.tt.final_file_id(trans_id)
1638
1471
c = _mod_conflicts.Conflict.factory(conflict_type,
1639
1472
path=path, file_id=file_id)
1473
self.cooked_conflicts.append(c)
1475
for trans_id, conflicts in name_conflicts.iteritems():
1477
this_parent, other_parent = conflicts['parent conflict']
1478
if this_parent == other_parent:
1479
raise AssertionError()
1481
this_parent = other_parent = \
1482
self.tt.final_file_id(self.tt.final_parent(trans_id))
1484
this_name, other_name = conflicts['name conflict']
1485
if this_name == other_name:
1486
raise AssertionError()
1488
this_name = other_name = self.tt.final_name(trans_id)
1489
other_path = fp.get_path(trans_id)
1490
if this_parent is not None and this_name is not None:
1491
this_parent_path = \
1492
fp.get_path(self.tt.trans_id_file_id(this_parent))
1493
this_path = osutils.pathjoin(this_parent_path, this_name)
1641
raise AssertionError('bad conflict type: %r' % (conflict,))
1495
this_path = "<deleted>"
1496
file_id = self.tt.final_file_id(trans_id)
1497
c = _mod_conflicts.Conflict.factory('path conflict', path=this_path,
1498
conflict_path=other_path,
1642
1500
self.cooked_conflicts.append(c)
1643
1501
self.cooked_conflicts.sort(key=_mod_conflicts.Conflict.sort_key)