20
20
from bzrlib.trace import mutter
22
# XXX: mbp: I'm not totally convinced that we should handle conflicts
23
# as part of changeset application, rather than only in the merge
26
"""Represent and apply a changeset
28
Conflicts in applying a changeset are represented as exceptions.
22
Represent and apply a changeset
31
24
__docformat__ = "restructuredtext"
39
class PatchApply(object):
40
"""Patch application as a kind of content change"""
41
def __init__(self, contents):
44
:param contents: The text of the patch to apply
45
:type contents: str"""
46
self.contents = contents
48
def __eq__(self, other):
49
if not isinstance(other, PatchApply):
51
elif self.contents != other.contents:
56
def __ne__(self, other):
57
return not (self == other)
59
def apply(self, filename, conflict_handler, reverse=False):
60
"""Applies the patch to the specified file.
62
:param filename: the file to apply the patch to
64
:param reverse: If true, apply the patch in reverse
67
input_name = filename+".orig"
69
os.rename(filename, input_name)
71
if e.errno != errno.ENOENT:
73
if conflict_handler.patch_target_missing(filename, self.contents)\
76
os.rename(filename, input_name)
79
status = patch.patch(self.contents, input_name, filename,
81
os.chmod(filename, os.stat(input_name).st_mode)
85
conflict_handler.failed_hunks(filename)
47
88
class ChangeUnixPermissions(object):
48
89
"""This is two-way change, suitable for file modification, creation,
328
369
class Diff3Merge(object):
329
def __init__(self, file_id, base, other):
330
self.file_id = file_id
370
def __init__(self, base_file, other_file):
371
self.base_file = base_file
372
self.other_file = other_file
334
374
def __eq__(self, other):
335
375
if not isinstance(other, Diff3Merge):
337
return (self.base == other.base and
338
self.other == other.other and self.file_id == other.file_id)
377
return (self.base_file == other.base_file and
378
self.other_file == other.other_file)
340
380
def __ne__(self, other):
341
381
return not (self == other)
343
383
def apply(self, filename, conflict_handler, reverse=False):
344
new_file = filename+".new"
345
base_file = self.base.readonly_path(self.file_id)
346
other_file = self.other.readonly_path(self.file_id)
384
new_file = filename+".new"
386
base = self.base_file
387
other = self.other_file
389
base = self.other_file
390
other = self.base_file
353
391
status = patch.diff3(new_file, filename, base, other)
355
393
os.chmod(new_file, os.stat(filename).st_mode)
359
397
assert(status == 1)
360
def get_lines(filename):
361
my_file = file(base, "rb")
362
lines = my_file.readlines()
364
base_lines = get_lines(base)
365
other_lines = get_lines(other)
366
conflict_handler.merge_conflict(new_file, filename, base_lines,
398
conflict_handler.merge_conflict(new_file, filename, base, other)
1022
1047
def rename_conflict(self, id, this_name, base_name, other_name):
1023
1048
raise RenameConflict(id, this_name, base_name, other_name)
1025
def move_conflict(self, id, this_dir, base_dir, other_dir):
1050
def move_conflict(self, id, inventory):
1051
this_dir = inventory.this.get_dir(id)
1052
base_dir = inventory.base.get_dir(id)
1053
other_dir = inventory.other.get_dir(id)
1026
1054
raise MoveConflict(id, this_dir, base_dir, other_dir)
1028
def merge_conflict(self, new_file, this_path, base_lines, other_lines):
1056
def merge_conflict(self, new_file, this_path, base_path, other_path):
1029
1057
os.unlink(new_file)
1030
1058
raise MergeConflict(this_path)
1059
1087
def missing_for_rename(self, filename):
1060
1088
raise MissingForRename(filename)
1062
def missing_for_merge(self, file_id, other_path):
1063
raise MissingForMerge(other_path)
1090
def missing_for_merge(self, file_id, inventory):
1091
raise MissingForMerge(inventory.other.get_path(file_id))
1065
1093
def new_contents_conflict(self, filename, other_contents):
1066
1094
raise NewContentsConflict(filename)
1287
1315
self.full_path = full_path
1288
1316
self.stat_result = stat_result
1290
def generate_changeset(tree_a, tree_b, interesting_ids=None):
1291
return ChangesetGenerator(tree_a, tree_b, interesting_ids)()
1318
def generate_changeset(tree_a, tree_b, inventory_a=None, inventory_b=None):
1319
return ChangesetGenerator(tree_a, tree_b, inventory_a, inventory_b)()
1293
1321
class ChangesetGenerator(object):
1294
def __init__(self, tree_a, tree_b, interesting_ids=None):
1322
def __init__(self, tree_a, tree_b, inventory_a=None, inventory_b=None):
1295
1323
object.__init__(self)
1296
1324
self.tree_a = tree_a
1297
1325
self.tree_b = tree_b
1298
self._interesting_ids = interesting_ids
1326
if inventory_a is not None:
1327
self.inventory_a = inventory_a
1329
self.inventory_a = tree_a.inventory()
1330
if inventory_b is not None:
1331
self.inventory_b = inventory_b
1333
self.inventory_b = tree_b.inventory()
1334
self.r_inventory_a = self.reverse_inventory(self.inventory_a)
1335
self.r_inventory_b = self.reverse_inventory(self.inventory_b)
1300
def iter_both_tree_ids(self):
1301
for file_id in self.tree_a:
1303
for file_id in self.tree_b:
1304
if file_id not in self.tree_a:
1337
def reverse_inventory(self, inventory):
1339
for entry in inventory.itervalues():
1340
if entry.id is None:
1342
r_inventory[entry.id] = entry
1307
1345
def __call__(self):
1308
1346
cset = Changeset()
1309
for file_id in self.iter_both_tree_ids():
1310
cs_entry = self.make_entry(file_id)
1347
for entry in self.inventory_a.itervalues():
1348
if entry.id is None:
1350
cs_entry = self.make_entry(entry.id)
1311
1351
if cs_entry is not None and not cs_entry.is_boring():
1312
1352
cset.add_entry(cs_entry)
1354
for entry in self.inventory_b.itervalues():
1355
if entry.id is None:
1357
if not self.r_inventory_a.has_key(entry.id):
1358
cs_entry = self.make_entry(entry.id)
1359
if cs_entry is not None and not cs_entry.is_boring():
1360
cset.add_entry(cs_entry)
1314
1361
for entry in list(cset.entries.itervalues()):
1315
1362
if entry.parent != entry.new_parent:
1316
1363
if not cset.entries.has_key(entry.parent) and\
1324
1371
cset.add_entry(parent_entry)
1327
def iter_inventory(self, tree):
1328
for file_id in tree:
1329
yield self.get_entry(file_id, tree)
1331
def get_entry(self, file_id, tree):
1332
if file_id not in tree:
1334
return tree.tree.inventory[file_id]
1336
def get_entry_parent(self, entry):
1339
return entry.parent_id
1341
def get_path(self, file_id, tree):
1342
if not tree.has_id(file_id):
1344
path = tree.id2path(file_id)
1350
def make_basic_entry(self, file_id, only_interesting):
1351
entry_a = self.get_entry(file_id, self.tree_a)
1352
entry_b = self.get_entry(file_id, self.tree_b)
1374
def get_entry_parent(self, entry, inventory):
1377
if entry.path == "./.":
1379
dirname = os.path.dirname(entry.path)
1382
parent = inventory[dirname]
1385
def get_path(self, entry, tree):
1388
if entry.path == ".":
1392
def make_basic_entry(self, id, only_interesting):
1393
entry_a = self.r_inventory_a.get(id)
1394
entry_b = self.r_inventory_b.get(id)
1353
1395
if only_interesting and not self.is_interesting(entry_a, entry_b):
1355
parent = self.get_entry_parent(entry_a)
1356
path = self.get_path(file_id, self.tree_a)
1357
cs_entry = ChangesetEntry(file_id, parent, path)
1358
new_parent = self.get_entry_parent(entry_b)
1360
new_path = self.get_path(file_id, self.tree_b)
1397
parent = self.get_entry_parent(entry_a, self.inventory_a)
1398
path = self.get_path(entry_a, self.tree_a)
1399
cs_entry = ChangesetEntry(id, parent, path)
1400
new_parent = self.get_entry_parent(entry_b, self.inventory_b)
1403
new_path = self.get_path(entry_b, self.tree_b)
1362
1405
cs_entry.new_path = new_path
1363
1406
cs_entry.new_parent = new_parent
1364
1407
return cs_entry
1366
1409
def is_interesting(self, entry_a, entry_b):
1367
if self._interesting_ids is None:
1369
1410
if entry_a is not None:
1370
file_id = entry_a.file_id
1371
elif entry_b is not None:
1372
file_id = entry_b.file_id
1375
return file_id in self._interesting_ids
1411
if entry_a.interesting:
1413
if entry_b is not None:
1414
if entry_b.interesting:
1377
1418
def make_boring_entry(self, id):
1378
1419
cs_entry = self.make_basic_entry(id, only_interesting=False)
1430
1471
if stat_a.st_ino == stat_b.st_ino and \
1431
1472
stat_a.st_dev == stat_b.st_dev:
1474
if file(full_path_a, "rb").read() == \
1475
file(full_path_b, "rb").read():
1478
patch_contents = patch.diff(full_path_a,
1479
file(full_path_b, "rb").read())
1480
if patch_contents is None:
1482
return PatchApply(patch_contents)
1434
1484
a_contents = self.get_contents(stat_a, full_path_a)
1435
1485
b_contents = self.get_contents(stat_b, full_path_b)