13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""Represent and apply a changeset.
19
Conflicts in applying a changeset are represented as exceptions.
21
This only handles the in-memory objects representing changesets, which are
22
primarily used by the merge code.
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.
28
from shutil import rmtree
29
from itertools import izip
31
from bzrlib.trace import mutter, warning
32
from bzrlib.osutils import rename, sha_file, pathjoin, mkdtemp
34
from bzrlib.errors import BzrCheckError
31
36
__docformat__ = "restructuredtext"
35
42
class OldFailedTreeOp(Exception):
36
43
def __init__(self):
37
44
Exception.__init__(self, "bzr-tree-change contains files from a"
38
45
" previous failed merge operation.")
39
48
def invert_dict(dict):
41
50
for (key,value) in dict.iteritems():
42
51
newdict[value] = key
47
class ChangeUnixPermissions(object):
55
class ChangeExecFlag(object):
48
56
"""This is two-way change, suitable for file modification, creation,
50
def __init__(self, old_mode, new_mode):
51
self.old_mode = old_mode
52
self.new_mode = new_mode
58
def __init__(self, old_exec_flag, new_exec_flag):
59
self.old_exec_flag = old_exec_flag
60
self.new_exec_flag = new_exec_flag
54
def apply(self, filename, conflict_handler, reverse=False):
56
from_mode = self.old_mode
57
to_mode = self.new_mode
59
from_mode = self.new_mode
60
to_mode = self.old_mode
62
def apply(self, filename, conflict_handler):
63
from_exec_flag = self.old_exec_flag
64
to_exec_flag = self.new_exec_flag
62
current_mode = os.stat(filename).st_mode &0777
66
current_exec_flag = bool(os.stat(filename).st_mode & 0111)
64
68
if e.errno == errno.ENOENT:
65
if conflict_handler.missing_for_chmod(filename) == "skip":
69
if conflict_handler.missing_for_exec_flag(filename) == "skip":
68
current_mode = from_mode
72
current_exec_flag = from_exec_flag
70
if from_mode is not None and current_mode != from_mode:
71
if conflict_handler.wrong_old_perms(filename, from_mode,
72
current_mode) != "continue":
74
if from_exec_flag is not None and current_exec_flag != from_exec_flag:
75
if conflict_handler.wrong_old_exec_flag(filename,
76
from_exec_flag, current_exec_flag) != "continue":
75
if to_mode is not None:
79
if to_exec_flag is not None:
80
current_mode = os.stat(filename).st_mode
84
to_mode = current_mode | (0100 & ~umask)
85
# Enable x-bit for others only if they can read it.
86
if current_mode & 0004:
87
to_mode |= 0001 & ~umask
88
if current_mode & 0040:
89
to_mode |= 0010 & ~umask
91
to_mode = current_mode & ~0111
77
93
os.chmod(filename, to_mode)
79
95
if e.errno == errno.ENOENT:
80
conflict_handler.missing_for_chmod(filename)
96
conflict_handler.missing_for_exec_flag(filename)
82
98
def __eq__(self, other):
83
if not isinstance(other, ChangeUnixPermissions):
85
elif self.old_mode != other.old_mode:
87
elif self.new_mode != other.new_mode:
99
return (isinstance(other, ChangeExecFlag) and
100
self.old_exec_flag == other.old_exec_flag and
101
self.new_exec_flag == other.new_exec_flag)
92
103
def __ne__(self, other):
93
104
return not (self == other)
95
def dir_create(filename, conflict_handler, reverse):
107
def dir_create(filename, conflict_handler, reverse=False):
96
108
"""Creates the directory, or deletes it if reverse is true. Intended to be
97
109
used with ReplaceContents.
221
235
if conflict_handler.missing_for_rm(filename, undo) == "skip":
226
def reversed(sequence):
227
max = len(sequence) - 1
228
for i in range(len(sequence)):
229
yield sequence[max - i]
239
class TreeFileCreate(object):
240
"""Create or delete a file (for use with ReplaceContents)"""
241
def __init__(self, tree, file_id):
244
:param contents: The contents of the file to write
248
self.file_id = file_id
251
return "TreeFileCreate(%s)" % self.file_id
253
def __eq__(self, other):
254
if not isinstance(other, TreeFileCreate):
256
return self.tree.get_file_sha1(self.file_id) == \
257
other.tree.get_file_sha1(other.file_id)
259
def __ne__(self, other):
260
return not (self == other)
262
def write_file(self, filename):
263
outfile = file(filename, "wb")
264
for line in self.tree.get_file(self.file_id):
267
def same_text(self, filename):
268
in_file = file(filename, "rb")
269
return sha_file(in_file) == self.tree.get_file_sha1(self.file_id)
271
def __call__(self, filename, conflict_handler, reverse=False):
272
"""Create or delete a file
274
:param filename: The name of the file to create
276
:param reverse: Delete the file instead of creating it
281
self.write_file(filename)
283
if e.errno == errno.ENOENT:
284
if conflict_handler.missing_parent(filename)=="continue":
285
self.write_file(filename)
291
if not self.same_text(filename):
292
direction = conflict_handler.wrong_old_contents(filename,
293
self.tree.get_file(self.file_id).read())
294
if direction != "continue":
298
if e.errno != errno.ENOENT:
300
if conflict_handler.missing_for_rm(filename, undo) == "skip":
231
304
class ReplaceContents(object):
232
305
"""A contents-replacement framework. It allows a file/directory/symlink to
292
359
undo(filename, conflict_handler, reverse=True)
293
360
if perform is not None:
294
perform(filename, conflict_handler, reverse=False)
361
perform(filename, conflict_handler)
295
362
if mode is not None:
296
363
os.chmod(filename, mode)
298
class ApplySequence(object):
299
def __init__(self, changes=None):
301
if changes is not None:
302
self.changes.extend(changes)
304
def __eq__(self, other):
305
if not isinstance(other, ApplySequence):
307
elif len(other.changes) != len(self.changes):
310
for i in range(len(self.changes)):
311
if self.changes[i] != other.changes[i]:
315
def __ne__(self, other):
316
return not (self == other)
319
def apply(self, filename, conflict_handler, reverse=False):
323
iter = reversed(self.changes)
325
change.apply(filename, conflict_handler, reverse)
365
def is_creation(self):
366
return self.new_contents is not None and self.old_contents is None
368
def is_deletion(self):
369
return self.old_contents is not None and self.new_contents is None
328
372
class Diff3Merge(object):
373
history_based = False
329
374
def __init__(self, file_id, base, other):
330
375
self.file_id = file_id
332
377
self.other = other
379
def is_creation(self):
382
def is_deletion(self):
334
385
def __eq__(self, other):
335
386
if not isinstance(other, Diff3Merge):
340
391
def __ne__(self, other):
341
392
return not (self == other)
343
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)
394
def dump_file(self, temp_dir, name, tree):
395
out_path = pathjoin(temp_dir, name)
396
out_file = file(out_path, "wb")
397
in_file = tree.get_file(self.file_id)
402
def apply(self, filename, conflict_handler):
404
temp_dir = mkdtemp(prefix="bzr-")
406
new_file = filename+".new"
407
base_file = self.dump_file(temp_dir, "base", self.base)
408
other_file = self.dump_file(temp_dir, "other", self.other)
349
410
other = other_file
353
status = patch.diff3(new_file, filename, base, other)
355
os.chmod(new_file, os.stat(filename).st_mode)
356
os.rename(new_file, filename)
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,
411
status = bzrlib.patch.diff3(new_file, filename, base, other)
413
os.chmod(new_file, os.stat(filename).st_mode)
414
rename(new_file, filename)
418
def get_lines(filename):
419
my_file = file(filename, "rb")
420
lines = my_file.readlines()
423
base_lines = get_lines(base)
424
other_lines = get_lines(other)
425
conflict_handler.merge_conflict(new_file, filename, base_lines,
577
651
return (self.parent != self.new_parent or self.name != self.new_name)
579
def is_deletion(self, reverse):
653
def is_deletion(self, reverse=False):
580
654
"""Return true if applying the entry would delete a file/directory.
582
656
:param reverse: if true, the changeset is being applied in reverse
585
return ((self.new_parent is None and not reverse) or
586
(self.parent is None and reverse))
659
return self.is_creation(not reverse)
588
def is_creation(self, reverse):
661
def is_creation(self, reverse=False):
589
662
"""Return true if applying the entry would create a file/directory.
591
664
:param reverse: if true, the changeset is being applied in reverse
594
return ((self.parent is None and not reverse) or
595
(self.new_parent is None and reverse))
667
if self.contents_change is None:
670
return self.contents_change.is_deletion()
672
return self.contents_change.is_creation()
597
674
def is_creation_or_deletion(self):
598
675
"""Return true if applying the entry would create or delete a
629
def summarize_name(self, reverse=False):
706
def summarize_name(self):
630
707
"""Produce a one-line summary of the filename. Indicates renames as
631
708
old => new, indicates creation as None => new, indicates deletion as
634
:param changeset: The changeset to get paths from
635
:type changeset: `Changeset`
636
:param reverse: If true, reverse the names in the output
640
713
orig_path = self.get_cset_path(False)
641
714
mod_path = self.get_cset_path(True)
642
if orig_path is not None:
715
if orig_path and orig_path.startswith('./'):
643
716
orig_path = orig_path[2:]
644
if mod_path is not None:
717
if mod_path and mod_path.startswith('./'):
645
718
mod_path = mod_path[2:]
646
719
if orig_path == mod_path:
650
return "%s => %s" % (orig_path, mod_path)
652
return "%s => %s" % (mod_path, orig_path)
655
def get_new_path(self, id_map, changeset, reverse=False):
722
return "%s => %s" % (orig_path, mod_path)
724
def get_new_path(self, id_map, changeset):
656
725
"""Determine the full pathname to rename to
658
727
:param id_map: The map of ids to filenames for the tree
659
728
:type id_map: Dictionary
660
729
:param changeset: The changeset to get data from
661
730
:type changeset: `Changeset`
662
:param reverse: If true, we're applying the changeset in reverse
666
mutter("Finding new path for %s" % self.summarize_name())
670
from_dir = self.new_dir
672
from_name = self.new_name
674
parent = self.new_parent
675
to_dir = self.new_dir
677
to_name = self.new_name
678
from_name = self.name
733
mutter("Finding new path for %s", self.summarize_name())
734
parent = self.new_parent
735
to_dir = self.new_dir
737
to_name = self.new_name
738
from_name = self.name
680
740
if to_name is None:
683
743
if parent == NULL_ID or parent is None:
685
745
raise SourceRootHasName(self, to_name)
688
if from_dir == to_dir:
748
parent_entry = changeset.entries.get(parent)
749
if parent_entry is None:
689
750
dir = os.path.dirname(id_map[self.id])
691
mutter("path, new_path: %r %r" % (self.path, self.new_path))
692
parent_entry = changeset.entries[parent]
693
dir = parent_entry.get_new_path(id_map, changeset, reverse)
752
mutter("path, new_path: %r %r", self.path, self.new_path)
753
dir = parent_entry.get_new_path(id_map, changeset)
694
754
if from_name == to_name:
695
755
name = os.path.basename(id_map[self.id])
698
758
assert(from_name is None or from_name == os.path.basename(id_map[self.id]))
699
return os.path.join(dir, name)
759
return pathjoin(dir, name)
701
761
def is_boring(self):
702
762
"""Determines whether the entry does nothing
718
def apply(self, filename, conflict_handler, reverse=False):
778
def apply(self, filename, conflict_handler):
719
779
"""Applies the file content and/or metadata changes.
721
781
:param filename: the filename of the entry
722
782
:type filename: str
723
:param reverse: If true, apply the changes in reverse
726
if self.is_deletion(reverse) and self.metadata_change is not None:
727
self.metadata_change.apply(filename, conflict_handler, reverse)
784
if self.is_deletion() and self.metadata_change is not None:
785
self.metadata_change.apply(filename, conflict_handler)
728
786
if self.contents_change is not None:
729
self.contents_change.apply(filename, conflict_handler, reverse)
730
if not self.is_deletion(reverse) and self.metadata_change is not None:
731
self.metadata_change.apply(filename, conflict_handler, reverse)
787
self.contents_change.apply(filename, conflict_handler)
788
if not self.is_deletion() and self.metadata_change is not None:
789
self.metadata_change.apply(filename, conflict_handler)
733
792
class IDPresent(Exception):
734
793
def __init__(self, id):
790
my_sort(source_entries, longest_to_shortest, reverse=True)
833
source_entries.sort(None, longest_to_shortest, True)
792
835
target_entries = source_entries[:]
793
836
# These are done from shortest to longest path, to avoid creating a
794
837
# child before its parent has been created/renamed
795
838
def shortest_to_longest(entry):
796
path = entry.get_new_path(inventory, changeset, reverse)
839
path = entry.get_new_path(inventory, changeset)
801
my_sort(target_entries, shortest_to_longest)
844
target_entries.sort(None, shortest_to_longest)
802
845
return (source_entries, target_entries)
804
848
def rename_to_temp_delete(source_entries, inventory, dir, temp_dir,
805
conflict_handler, reverse):
806
850
"""Delete and rename entries as appropriate. Entries are renamed to temp
807
851
names. A map of id -> temp name (or None, for deletions) is returned.
812
856
:type inventory: Dictionary
813
857
:param dir: The directory to apply changes to
815
:param reverse: Apply changes in reverse
817
859
:return: a mapping of id to temporary name
818
860
:rtype: Dictionary
821
863
for i in range(len(source_entries)):
822
864
entry = source_entries[i]
823
if entry.is_deletion(reverse):
824
path = os.path.join(dir, inventory[entry.id])
825
entry.apply(path, conflict_handler, reverse)
865
if entry.is_deletion():
866
path = pathjoin(dir, inventory[entry.id])
867
entry.apply(path, conflict_handler)
826
868
temp_name[entry.id] = None
829
to_name = os.path.join(temp_dir, str(i))
870
elif entry.needs_rename():
871
if entry.is_creation():
873
to_name = pathjoin(temp_dir, str(i))
830
874
src_path = inventory.get(entry.id)
831
875
if src_path is not None:
832
src_path = os.path.join(dir, src_path)
876
src_path = pathjoin(dir, src_path)
834
os.rename(src_path, to_name)
878
rename(src_path, to_name)
835
879
temp_name[entry.id] = to_name
836
880
except OSError, e:
837
881
if e.errno != errno.ENOENT:
839
if conflict_handler.missing_for_rename(src_path) == "skip":
883
if conflict_handler.missing_for_rename(src_path, to_name) \
845
890
def rename_to_new_create(changed_inventory, target_entries, inventory,
846
changeset, dir, conflict_handler, reverse):
891
changeset, dir, conflict_handler):
847
892
"""Rename entries with temp names to their final names, create new files.
849
894
:param changed_inventory: A mapping of id to temporary name
854
899
:type changeset: `Changeset`
855
900
:param dir: The directory to apply changes to
857
:param reverse: If true, apply changes in reverse
860
903
for entry in target_entries:
861
new_tree_path = entry.get_new_path(inventory, changeset, reverse)
904
new_tree_path = entry.get_new_path(inventory, changeset)
862
905
if new_tree_path is None:
864
new_path = os.path.join(dir, new_tree_path)
907
new_path = pathjoin(dir, new_tree_path)
865
908
old_path = changed_inventory.get(entry.id)
866
if os.path.exists(new_path):
909
if bzrlib.osutils.lexists(new_path):
867
910
if conflict_handler.target_exists(entry, new_path, old_path) == \
870
if entry.is_creation(reverse):
871
entry.apply(new_path, conflict_handler, reverse)
913
if entry.is_creation():
914
entry.apply(new_path, conflict_handler)
872
915
changed_inventory[entry.id] = new_tree_path
916
elif entry.needs_rename():
917
if entry.is_deletion():
874
919
if old_path is None:
877
os.rename(old_path, new_path)
922
mutter('rename %s to final name %s', old_path, new_path)
923
rename(old_path, new_path)
878
924
changed_inventory[entry.id] = new_tree_path
879
925
except OSError, e:
880
raise Exception ("%s is missing" % new_path)
926
raise BzrCheckError('failed to rename %s to %s for changeset entry %s: %s'
927
% (old_path, new_path, entry, e))
882
930
class TargetExists(Exception):
883
931
def __init__(self, entry, target):
910
960
self.base_parent = base_parent
911
961
self_other_parent = other_parent
913
964
class MergeConflict(Exception):
914
965
def __init__(self, this_path):
915
966
Exception.__init__(self, "Conflict applying changes to %s" % this_path)
916
967
self.this_path = this_path
918
class MergePermissionConflict(Exception):
919
def __init__(self, this_path, base_path, other_path):
920
this_perms = os.stat(this_path).st_mode & 0755
921
base_perms = os.stat(base_path).st_mode & 0755
922
other_perms = os.stat(other_path).st_mode & 0755
923
msg = """Conflicting permission for %s
927
""" % (this_path, this_perms, base_perms, other_perms)
928
self.this_path = this_path
929
self.base_path = base_path
930
self.other_path = other_path
931
Exception.__init__(self, msg)
933
970
class WrongOldContents(Exception):
934
971
def __init__(self, filename):
980
1022
class MissingForRename(Exception):
981
def __init__(self, filename):
982
msg = "Attempt to move missing path %s" % (filename)
1023
def __init__(self, filename, to_path):
1024
msg = "Attempt to move missing path %s to %s" % (filename, to_path)
983
1025
Exception.__init__(self, msg)
984
1026
self.filename = filename
986
1029
class NewContentsConflict(Exception):
987
1030
def __init__(self, filename):
988
1031
msg = "Conflicting contents for new file %s" % (filename)
989
1032
Exception.__init__(self, msg)
1035
class WeaveMergeConflict(Exception):
1036
def __init__(self, filename):
1037
msg = "Conflicting contents for file %s" % (filename)
1038
Exception.__init__(self, msg)
1041
class ThreewayContentsConflict(Exception):
1042
def __init__(self, filename):
1043
msg = "Conflicting contents for file %s" % (filename)
1044
Exception.__init__(self, msg)
992
1047
class MissingForMerge(Exception):
993
1048
def __init__(self, filename):
994
1049
msg = "The file %s was modified, but does not exist in this tree"\
1029
1081
os.unlink(new_file)
1030
1082
raise MergeConflict(this_path)
1032
def permission_conflict(self, this_path, base_path, other_path):
1033
raise MergePermissionConflict(this_path, base_path, other_path)
1035
1084
def wrong_old_contents(self, filename, expected_contents):
1036
1085
raise WrongOldContents(filename)
1038
1087
def rem_contents_conflict(self, filename, this_contents, base_contents):
1039
1088
raise RemoveContentsConflict(filename)
1041
def wrong_old_perms(self, filename, old_perms, new_perms):
1042
raise WrongOldPermissions(filename, old_perms, new_perms)
1090
def wrong_old_exec_flag(self, filename, old_exec_flag, new_exec_flag):
1091
raise WrongOldExecFlag(filename, old_exec_flag, new_exec_flag)
1044
1093
def rmdir_non_empty(self, filename):
1045
1094
raise DeletingNonEmptyDirectory(filename)
1102
1156
#apply changes that don't affect filenames
1103
1157
for entry in changeset.entries.itervalues():
1104
if not entry.is_creation_or_deletion():
1105
path = os.path.join(dir, inventory[entry.id])
1106
entry.apply(path, conflict_handler, reverse)
1158
if not entry.is_creation_or_deletion() and not entry.is_boring():
1159
if entry.id not in inventory:
1160
warning("entry {%s} no longer present, can't be updated",
1163
path = pathjoin(dir, inventory[entry.id])
1164
entry.apply(path, conflict_handler)
1108
1166
# Apply renames in stages, to minimize conflicts:
1109
1167
# Only files whose name or parent change are interesting, because their
1110
1168
# target name may exist in the source tree. If a directory's name changes,
1111
1169
# that doesn't make its children interesting.
1112
(source_entries, target_entries) = get_rename_entries(changeset, inventory,
1170
(source_entries, target_entries) = get_rename_entries(changeset, inventory)
1115
1172
changed_inventory = rename_to_temp_delete(source_entries, inventory, dir,
1116
temp_dir, conflict_handler,
1173
temp_dir, conflict_handler)
1119
1175
rename_to_new_create(changed_inventory, target_entries, inventory,
1120
changeset, dir, conflict_handler, reverse)
1176
changeset, dir, conflict_handler)
1121
1177
os.rmdir(temp_dir)
1122
1178
return changed_inventory
1125
def apply_changeset_tree(cset, tree, reverse=False):
1181
def apply_changeset_tree(cset, tree):
1126
1182
r_inventory = {}
1127
1183
for entry in tree.source_inventory().itervalues():
1128
1184
inventory[entry.id] = entry.path
1129
new_inventory = apply_changeset(cset, r_inventory, tree.root,
1185
new_inventory = apply_changeset(cset, r_inventory, tree.basedir)
1131
1186
new_entries, remove_entries = \
1132
get_inventory_change(inventory, new_inventory, cset, reverse)
1187
get_inventory_change(inventory, new_inventory, cset)
1133
1188
tree.update_source_inventory(new_entries, remove_entries)
1136
def get_inventory_change(inventory, new_inventory, cset, reverse=False):
1191
def get_inventory_change(inventory, new_inventory, cset):
1137
1192
new_entries = {}
1138
1193
remove_entries = []
1139
1194
for entry in cset.entries.itervalues():
1159
1214
print entry.summarize_name(cset)
1161
class CompositionFailure(Exception):
1162
def __init__(self, old_entry, new_entry, problem):
1163
msg = "Unable to conpose entries.\n %s" % problem
1164
Exception.__init__(self, msg)
1166
class IDMismatch(CompositionFailure):
1167
def __init__(self, old_entry, new_entry):
1168
problem = "Attempt to compose entries with different ids: %s and %s" %\
1169
(old_entry.id, new_entry.id)
1170
CompositionFailure.__init__(self, old_entry, new_entry, problem)
1172
def compose_changesets(old_cset, new_cset):
1173
"""Combine two changesets into one. This works well for exact patching.
1174
Otherwise, not so well.
1176
:param old_cset: The first changeset that would be applied
1177
:type old_cset: `Changeset`
1178
:param new_cset: The second changeset that would be applied
1179
:type new_cset: `Changeset`
1180
:return: A changeset that combines the changes in both changesets
1183
composed = Changeset()
1184
for old_entry in old_cset.entries.itervalues():
1185
new_entry = new_cset.entries.get(old_entry.id)
1186
if new_entry is None:
1187
composed.add_entry(old_entry)
1189
composed_entry = compose_entries(old_entry, new_entry)
1190
if composed_entry.parent is not None or\
1191
composed_entry.new_parent is not None:
1192
composed.add_entry(composed_entry)
1193
for new_entry in new_cset.entries.itervalues():
1194
if not old_cset.entries.has_key(new_entry.id):
1195
composed.add_entry(new_entry)
1198
def compose_entries(old_entry, new_entry):
1199
"""Combine two entries into one.
1201
:param old_entry: The first entry that would be applied
1202
:type old_entry: ChangesetEntry
1203
:param old_entry: The second entry that would be applied
1204
:type old_entry: ChangesetEntry
1205
:return: A changeset entry combining both entries
1206
:rtype: `ChangesetEntry`
1208
if old_entry.id != new_entry.id:
1209
raise IDMismatch(old_entry, new_entry)
1210
output = ChangesetEntry(old_entry.id, old_entry.parent, old_entry.path)
1212
if (old_entry.parent != old_entry.new_parent or
1213
new_entry.parent != new_entry.new_parent):
1214
output.new_parent = new_entry.new_parent
1216
if (old_entry.path != old_entry.new_path or
1217
new_entry.path != new_entry.new_path):
1218
output.new_path = new_entry.new_path
1220
output.contents_change = compose_contents(old_entry, new_entry)
1221
output.metadata_change = compose_metadata(old_entry, new_entry)
1224
def compose_contents(old_entry, new_entry):
1225
"""Combine the contents of two changeset entries. Entries are combined
1226
intelligently where possible, but the fallback behavior returns an
1229
:param old_entry: The first entry that would be applied
1230
:type old_entry: `ChangesetEntry`
1231
:param new_entry: The second entry that would be applied
1232
:type new_entry: `ChangesetEntry`
1233
:return: A combined contents change
1234
:rtype: anything supporting the apply(reverse=False) method
1236
old_contents = old_entry.contents_change
1237
new_contents = new_entry.contents_change
1238
if old_entry.contents_change is None:
1239
return new_entry.contents_change
1240
elif new_entry.contents_change is None:
1241
return old_entry.contents_change
1242
elif isinstance(old_contents, ReplaceContents) and \
1243
isinstance(new_contents, ReplaceContents):
1244
if old_contents.old_contents == new_contents.new_contents:
1247
return ReplaceContents(old_contents.old_contents,
1248
new_contents.new_contents)
1249
elif isinstance(old_contents, ApplySequence):
1250
output = ApplySequence(old_contents.changes)
1251
if isinstance(new_contents, ApplySequence):
1252
output.changes.extend(new_contents.changes)
1254
output.changes.append(new_contents)
1256
elif isinstance(new_contents, ApplySequence):
1257
output = ApplySequence((old_contents.changes,))
1258
output.extend(new_contents.changes)
1261
return ApplySequence((old_contents, new_contents))
1263
def compose_metadata(old_entry, new_entry):
1264
old_meta = old_entry.metadata_change
1265
new_meta = new_entry.metadata_change
1266
if old_meta is None:
1268
elif new_meta is None:
1270
elif isinstance(old_meta, ChangeUnixPermissions) and \
1271
isinstance(new_meta, ChangeUnixPermissions):
1272
return ChangeUnixPermissions(old_meta.old_mode, new_meta.new_mode)
1274
return ApplySequence(old_meta, new_meta)
1277
def changeset_is_null(changeset):
1278
for entry in changeset.entries.itervalues():
1279
if not entry.is_boring():
1283
class UnsuppportedFiletype(Exception):
1284
def __init__(self, full_path, stat_result):
1285
msg = "The file \"%s\" is not a supported filetype." % full_path
1217
class UnsupportedFiletype(Exception):
1218
def __init__(self, kind, full_path):
1219
msg = "The file \"%s\" is a %s, which is not a supported filetype." \
1286
1221
Exception.__init__(self, msg)
1287
1222
self.full_path = full_path
1288
self.stat_result = stat_result
1290
1226
def generate_changeset(tree_a, tree_b, interesting_ids=None):
1291
1227
return ChangesetGenerator(tree_a, tree_b, interesting_ids)()
1293
1230
class ChangesetGenerator(object):
1294
1231
def __init__(self, tree_a, tree_b, interesting_ids=None):
1295
1232
object.__init__(self)
1380
1317
return self.make_entry(id, only_interesting=False)
1382
1319
return cs_entry
1385
1321
def make_entry(self, id, only_interesting=True):
1386
1322
cs_entry = self.make_basic_entry(id, only_interesting)
1388
1324
if cs_entry is None:
1327
cs_entry.metadata_change = self.make_exec_flag_change(id)
1390
1329
if id in self.tree_a and id in self.tree_b:
1391
1330
a_sha1 = self.tree_a.get_file_sha1(id)
1392
1331
b_sha1 = self.tree_b.get_file_sha1(id)
1393
1332
if None not in (a_sha1, b_sha1) and a_sha1 == b_sha1:
1394
1333
return cs_entry
1396
full_path_a = self.tree_a.readonly_path(id)
1397
full_path_b = self.tree_b.readonly_path(id)
1398
stat_a = self.lstat(full_path_a)
1399
stat_b = self.lstat(full_path_b)
1401
cs_entry.metadata_change = self.make_mode_change(stat_a, stat_b)
1402
cs_entry.contents_change = self.make_contents_change(full_path_a,
1335
cs_entry.contents_change = self.make_contents_change(id)
1406
1336
return cs_entry
1408
def make_mode_change(self, stat_a, stat_b):
1410
if stat_a is not None and not stat.S_ISLNK(stat_a.st_mode):
1411
mode_a = stat_a.st_mode & 0777
1413
if stat_b is not None and not stat.S_ISLNK(stat_b.st_mode):
1414
mode_b = stat_b.st_mode & 0777
1415
if mode_a == mode_b:
1417
return ChangeUnixPermissions(mode_a, mode_b)
1419
def make_contents_change(self, full_path_a, stat_a, full_path_b, stat_b):
1420
if stat_a is None and stat_b is None:
1422
if None not in (stat_a, stat_b) and stat.S_ISDIR(stat_a.st_mode) and\
1423
stat.S_ISDIR(stat_b.st_mode):
1425
if None not in (stat_a, stat_b) and stat.S_ISREG(stat_a.st_mode) and\
1426
stat.S_ISREG(stat_b.st_mode):
1427
if stat_a.st_ino == stat_b.st_ino and \
1428
stat_a.st_dev == stat_b.st_dev:
1431
a_contents = self.get_contents(stat_a, full_path_a)
1432
b_contents = self.get_contents(stat_b, full_path_b)
1338
def make_exec_flag_change(self, file_id):
1339
exec_flag_a = exec_flag_b = None
1340
if file_id in self.tree_a and self.tree_a.kind(file_id) == "file":
1341
exec_flag_a = self.tree_a.is_executable(file_id)
1343
if file_id in self.tree_b and self.tree_b.kind(file_id) == "file":
1344
exec_flag_b = self.tree_b.is_executable(file_id)
1346
if exec_flag_a == exec_flag_b:
1348
return ChangeExecFlag(exec_flag_a, exec_flag_b)
1350
def make_contents_change(self, file_id):
1351
a_contents = get_contents(self.tree_a, file_id)
1352
b_contents = get_contents(self.tree_b, file_id)
1433
1353
if a_contents == b_contents:
1435
1355
return ReplaceContents(a_contents, b_contents)
1437
def get_contents(self, stat_result, full_path):
1438
if stat_result is None:
1440
elif stat.S_ISREG(stat_result.st_mode):
1441
return FileCreate(file(full_path, "rb").read())
1442
elif stat.S_ISDIR(stat_result.st_mode):
1444
elif stat.S_ISLNK(stat_result.st_mode):
1445
return SymlinkCreate(os.readlink(full_path))
1447
raise UnsupportedFiletype(full_path, stat_result)
1449
def lstat(self, full_path):
1451
if full_path is not None:
1453
stat_result = os.lstat(full_path)
1455
if e.errno != errno.ENOENT:
1358
def get_contents(tree, file_id):
1359
"""Return the appropriate contents to create a copy of file_id from tree"""
1360
if file_id not in tree:
1362
kind = tree.kind(file_id)
1364
return TreeFileCreate(tree, file_id)
1365
elif kind in ("directory", "root_directory"):
1367
elif kind == "symlink":
1368
return SymlinkCreate(tree.get_symlink_target(file_id))
1370
raise UnsupportedFiletype(kind, tree.id2path(file_id))
1460
1373
def full_path(entry, tree):
1461
return os.path.join(tree.root, entry.path)
1374
return pathjoin(tree.basedir, entry.path)
1463
1377
def new_delete_entry(entry, tree, inventory, delete):
1464
1378
if entry.path == "":