24
24
import bzrlib.revision
25
25
from bzrlib.merge_core import merge_flex, ApplyMerge3, BackupBeforeChange
26
26
from bzrlib.changeset import generate_changeset, ExceptionConflictHandler
27
from bzrlib.changeset import Inventory, Diff3Merge
27
from bzrlib.changeset import Inventory, Diff3Merge, ReplaceContents
28
28
from bzrlib.branch import Branch
29
29
from bzrlib.errors import BzrCommandError, UnrelatedBranches, NoCommonAncestor
30
30
from bzrlib.errors import NoCommits
36
36
from bzrlib.revision import common_ancestor, MultipleRevisionSources
37
37
from bzrlib.errors import NoSuchRevision
39
# TODO: build_working_dir can be built on something simpler than merge()
41
# FIXME: merge() parameters seem oriented towards the command line
40
43
# comments from abentley on irc: merge happens in two stages, each
41
44
# of which generates a changeset object
50
53
conflict that are not explicitly handled cause an exception and
51
54
terminate the merge.
53
def __init__(self, ignore_zero=False):
56
def __init__(self, this_tree, base_tree, other_tree, ignore_zero=False):
54
57
ExceptionConflictHandler.__init__(self)
56
59
self.ignore_zero = ignore_zero
60
self.this_tree = this_tree
61
self.base_tree = base_tree
62
self.other_tree = other_tree
58
64
def copy(self, source, dest):
59
65
"""Copy the text and mode of a file
140
def rem_contents_conflict(self, filename, this_contents, base_contents):
141
base_contents(filename+".BASE", self, False)
142
this_contents(filename+".THIS", self, False)
143
self.conflict("Other branch deleted locally modified file %s" %
145
return ReplaceContents(this_contents, None)
147
def abs_this_path(self, file_id):
148
"""Return the absolute path for a file_id in the this tree."""
149
relpath = self.this_tree.id2path(file_id)
150
return self.this_tree.tree.abspath(relpath)
152
def add_missing_parents(self, file_id, tree):
153
"""If some of the parents for file_id are missing, add them."""
154
entry = tree.tree.inventory[file_id]
155
if entry.parent_id not in self.this_tree:
156
return self.create_all_missing(entry.parent_id, tree)
158
return self.abs_this_path(entry.parent_id)
160
def create_all_missing(self, file_id, tree):
161
"""Add contents for a file_id and all its parents to a tree."""
162
entry = tree.tree.inventory[file_id]
163
if entry.parent_id is not None and entry.parent_id not in self.this_tree:
164
abspath = self.create_all_missing(entry.parent_id, tree)
166
abspath = self.abs_this_path(entry.parent_id)
167
entry_path = os.path.join(abspath, entry.name)
168
if not os.path.isdir(entry_path):
169
self.create(file_id, entry_path, tree)
172
def create(self, file_id, path, tree, reverse=False):
173
"""Uses tree data to create a filesystem object for the file_id"""
174
from merge_core import get_id_contents
175
get_id_contents(file_id, tree)(path, self, reverse)
177
def missing_for_merge(self, file_id, other_path):
178
"""The file_id doesn't exist in THIS, but does in OTHER and BASE"""
179
self.conflict("Other branch modified locally deleted file %s" %
181
parent_dir = self.add_missing_parents(file_id, self.other_tree)
182
stem = os.path.join(parent_dir, os.path.basename(other_path))
183
self.create(file_id, stem+".OTHER", self.other_tree)
184
self.create(file_id, stem+".BASE", self.base_tree)
134
186
def finalize(self):
135
187
if not self.ignore_zero:
136
188
print "%d conflicts encountered.\n" % self.conflicts
222
274
return self.cached[id]
277
def build_working_dir(to_dir):
278
"""Build a working directory in an empty directory.
280
to_dir is a directory containing branch metadata but no working files,
281
typically constructed by cloning an existing branch.
283
This is split out as a special idiomatic case of merge. It could
284
eventually be done by just building the tree directly calling into
285
lower-level code (e.g. constructing a changeset).
287
merge((to_dir, -1), (to_dir, 0), this_dir=to_dir,
288
check_clean=False, ignore_zero=True)
226
291
def merge(other_revision, base_revision,
227
292
check_clean=True, ignore_zero=False,
352
420
inv_changes = merge_flex(this_tree, base_tree, other_tree,
353
421
generate_cset_optimized, get_inventory,
354
MergeConflictHandler(ignore_zero=ignore_zero),
422
MergeConflictHandler(this_tree, base_tree,
423
other_tree, ignore_zero=ignore_zero),
355
424
merge_factory=merge_factory,
356
425
interesting_ids=interesting_ids)