587
592
self.base_rev_id = self.revision_graph.find_unique_lca(
589
self._is_criss_cross = True
594
sorted_lca_keys = self.revision_graph.find_merge_order(
596
if self.base_rev_id == _mod_revision.NULL_REVISION:
597
self.base_rev_id = sorted_lca_keys[0]
590
599
if self.base_rev_id == _mod_revision.NULL_REVISION:
591
600
raise errors.UnrelatedBranches()
592
601
if self._is_criss_cross:
593
602
trace.warning('Warning: criss-cross merge encountered. See bzr'
594
603
' help criss-cross.')
595
604
trace.mutter('Criss-cross lcas: %r' % lcas)
596
interesting_revision_ids = [self.base_rev_id]
597
interesting_revision_ids.extend(lcas)
605
if self.base_rev_id in lcas:
606
trace.mutter('Unable to find unique lca. '
607
'Fallback %r as best option.' % self.base_rev_id)
608
interesting_revision_ids = set(lcas)
609
interesting_revision_ids.add(self.base_rev_id)
598
610
interesting_trees = dict((t.get_revision_id(), t)
599
611
for t in self.this_branch.repository.revision_trees(
600
612
interesting_revision_ids))
601
613
self._cached_trees.update(interesting_trees)
602
self.base_tree = interesting_trees.pop(self.base_rev_id)
603
sorted_lca_keys = self.revision_graph.find_merge_order(
614
if self.base_rev_id in lcas:
615
self.base_tree = interesting_trees[self.base_rev_id]
617
self.base_tree = interesting_trees.pop(self.base_rev_id)
605
618
self._lca_trees = [interesting_trees[key]
606
619
for key in sorted_lca_keys]
913
933
it then compares with THIS and BASE.
915
935
For the multi-valued entries, the format will be (BASE, [lca1, lca2])
916
:return: [(file_id, changed, parents, names, executable)]
917
file_id Simple file_id of the entry
918
changed Boolean, True if the kind or contents changed
920
parents ((base, [parent_id, in, lcas]), parent_id_other,
922
names ((base, [name, in, lcas]), name_in_other, name_in_this)
923
executable ((base, [exec, in, lcas]), exec_in_other, exec_in_this)
937
:return: [(file_id, changed, parents, names, executable)], where:
939
* file_id: Simple file_id of the entry
940
* changed: Boolean, True if the kind or contents changed else False
941
* parents: ((base, [parent_id, in, lcas]), parent_id_other,
943
* names: ((base, [name, in, lcas]), name_in_other, name_in_this)
944
* executable: ((base, [exec, in, lcas]), exec_in_other,
925
947
if self.interesting_files is not None:
926
948
lookup_trees = [self.this_tree, self.base_tree]
1090
1109
# the other tree's root is a non-root in the current tree (as when
1091
1110
# a previously unrelated branch is merged into another)
1094
self.tt.final_kind(other_root)
1112
if self.tt.final_kind(other_root) is not None:
1095
1113
other_root_is_present = True
1096
except errors.NoSuchFile:
1097
1115
# other_root doesn't have a physical representation. We still need
1098
1116
# to move any references to the actual root of the tree.
1099
1117
other_root_is_present = False
1100
1118
# 'other_tree.inventory.root' is not present in this tree. We are
1101
1119
# calling adjust_path for children which *want* to be present with a
1102
1120
# correct place to go.
1103
for thing, child in self.other_tree.inventory.root.children.iteritems():
1121
for _, child in self.other_tree.inventory.root.children.iteritems():
1104
1122
trans_id = self.tt.trans_id_file_id(child.file_id)
1105
1123
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)
1124
if self.tt.final_kind(trans_id) is not None:
1110
1125
# The item exist in the final tree and has a defined place
1111
1126
# to go already.
1113
except errors.NoSuchFile, e:
1115
1128
# Move the item into the root
1116
self.tt.adjust_path(self.tt.final_name(trans_id),
1117
self.tt.root, trans_id)
1130
final_name = self.tt.final_name(trans_id)
1131
except errors.NoFinalPath:
1132
# This file is not present anymore, ignore it.
1134
self.tt.adjust_path(final_name, self.tt.root, trans_id)
1118
1135
if other_root_is_present:
1119
1136
self.tt.cancel_creation(other_root)
1120
1137
self.tt.cancel_versioning(other_root)
1750
1761
osutils.rmtree(temp_dir)
1764
class PathNotInTree(errors.BzrError):
1766
_fmt = """Merge-into failed because %(tree)s does not contain %(path)s."""
1768
def __init__(self, path, tree):
1769
errors.BzrError.__init__(self, path=path, tree=tree)
1772
class MergeIntoMerger(Merger):
1773
"""Merger that understands other_tree will be merged into a subdir.
1775
This also changes the Merger api so that it uses real Branch, revision_id,
1776
and RevisonTree objects, rather than using revision specs.
1779
def __init__(self, this_tree, other_branch, other_tree, target_subdir,
1780
source_subpath, other_rev_id=None):
1781
"""Create a new MergeIntoMerger object.
1783
source_subpath in other_tree will be effectively copied to
1784
target_subdir in this_tree.
1786
:param this_tree: The tree that we will be merging into.
1787
:param other_branch: The Branch we will be merging from.
1788
:param other_tree: The RevisionTree object we want to merge.
1789
:param target_subdir: The relative path where we want to merge
1790
other_tree into this_tree
1791
:param source_subpath: The relative path specifying the subtree of
1792
other_tree to merge into this_tree.
1794
# It is assumed that we are merging a tree that is not in our current
1795
# ancestry, which means we are using the "EmptyTree" as our basis.
1796
null_ancestor_tree = this_tree.branch.repository.revision_tree(
1797
_mod_revision.NULL_REVISION)
1798
super(MergeIntoMerger, self).__init__(
1799
this_branch=this_tree.branch,
1800
this_tree=this_tree,
1801
other_tree=other_tree,
1802
base_tree=null_ancestor_tree,
1804
self._target_subdir = target_subdir
1805
self._source_subpath = source_subpath
1806
self.other_branch = other_branch
1807
if other_rev_id is None:
1808
other_rev_id = other_tree.get_revision_id()
1809
self.other_rev_id = self.other_basis = other_rev_id
1810
self.base_is_ancestor = True
1811
self.backup_files = True
1812
self.merge_type = Merge3Merger
1813
self.show_base = False
1814
self.reprocess = False
1815
self.interesting_ids = None
1816
self.merge_type = _MergeTypeParameterizer(MergeIntoMergeType,
1817
target_subdir=self._target_subdir,
1818
source_subpath=self._source_subpath)
1819
if self._source_subpath != '':
1820
# If this isn't a partial merge make sure the revisions will be
1822
self._maybe_fetch(self.other_branch, self.this_branch,
1825
def set_pending(self):
1826
if self._source_subpath != '':
1828
Merger.set_pending(self)
1831
class _MergeTypeParameterizer(object):
1832
"""Wrap a merge-type class to provide extra parameters.
1834
This is hack used by MergeIntoMerger to pass some extra parameters to its
1835
merge_type. Merger.do_merge() sets up its own set of parameters to pass to
1836
the 'merge_type' member. It is difficult override do_merge without
1837
re-writing the whole thing, so instead we create a wrapper which will pass
1838
the extra parameters.
1841
def __init__(self, merge_type, **kwargs):
1842
self._extra_kwargs = kwargs
1843
self._merge_type = merge_type
1845
def __call__(self, *args, **kwargs):
1846
kwargs.update(self._extra_kwargs)
1847
return self._merge_type(*args, **kwargs)
1849
def __getattr__(self, name):
1850
return getattr(self._merge_type, name)
1853
class MergeIntoMergeType(Merge3Merger):
1854
"""Merger that incorporates a tree (or part of a tree) into another."""
1856
def __init__(self, *args, **kwargs):
1857
"""Initialize the merger object.
1859
:param args: See Merge3Merger.__init__'s args.
1860
:param kwargs: See Merge3Merger.__init__'s keyword args, except for
1861
source_subpath and target_subdir.
1862
:keyword source_subpath: The relative path specifying the subtree of
1863
other_tree to merge into this_tree.
1864
:keyword target_subdir: The relative path where we want to merge
1865
other_tree into this_tree
1867
# All of the interesting work happens during Merge3Merger.__init__(),
1868
# so we have have to hack in to get our extra parameters set.
1869
self._source_subpath = kwargs.pop('source_subpath')
1870
self._target_subdir = kwargs.pop('target_subdir')
1871
super(MergeIntoMergeType, self).__init__(*args, **kwargs)
1873
def _compute_transform(self):
1874
child_pb = ui.ui_factory.nested_progress_bar()
1876
entries = self._entries_to_incorporate()
1877
entries = list(entries)
1878
for num, (entry, parent_id) in enumerate(entries):
1879
child_pb.update('Preparing file merge', num, len(entries))
1880
parent_trans_id = self.tt.trans_id_file_id(parent_id)
1881
trans_id = transform.new_by_entry(self.tt, entry,
1882
parent_trans_id, self.other_tree)
1885
self._finish_computing_transform()
1887
def _entries_to_incorporate(self):
1888
"""Yields pairs of (inventory_entry, new_parent)."""
1889
other_inv = self.other_tree.inventory
1890
subdir_id = other_inv.path2id(self._source_subpath)
1891
if subdir_id is None:
1892
# XXX: The error would be clearer if it gave the URL of the source
1893
# branch, but we don't have a reference to that here.
1894
raise PathNotInTree(self._source_subpath, "Source tree")
1895
subdir = other_inv[subdir_id]
1896
parent_in_target = osutils.dirname(self._target_subdir)
1897
target_id = self.this_tree.inventory.path2id(parent_in_target)
1898
if target_id is None:
1899
raise PathNotInTree(self._target_subdir, "Target tree")
1900
name_in_target = osutils.basename(self._target_subdir)
1901
merge_into_root = subdir.copy()
1902
merge_into_root.name = name_in_target
1903
if merge_into_root.file_id in self.this_tree.inventory:
1904
# Give the root a new file-id.
1905
# This can happen fairly easily if the directory we are
1906
# incorporating is the root, and both trees have 'TREE_ROOT' as
1907
# their root_id. Users will expect this to Just Work, so we
1908
# change the file-id here.
1909
# Non-root file-ids could potentially conflict too. That's really
1910
# an edge case, so we don't do anything special for those. We let
1911
# them cause conflicts.
1912
merge_into_root.file_id = generate_ids.gen_file_id(name_in_target)
1913
yield (merge_into_root, target_id)
1914
if subdir.kind != 'directory':
1915
# No children, so we are done.
1917
for ignored_path, entry in other_inv.iter_entries_by_dir(subdir_id):
1918
parent_id = entry.parent_id
1919
if parent_id == subdir.file_id:
1920
# The root's parent ID has changed, so make sure children of
1921
# the root refer to the new ID.
1922
parent_id = merge_into_root.file_id
1923
yield (entry, parent_id)
1753
1926
def merge_inner(this_branch, other_tree, base_tree, ignore_zero=False,
1754
1927
backup_files=False,
1755
1928
merge_type=Merge3Merger,