← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/changeset.py

Fixed inner merges in status

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/changeset.py
17
17
import errno
18
18
import patch
19
19
import stat
 
20
from tempfile import mkdtemp
 
21
from shutil import rmtree
20
22
from bzrlib.trace import mutter
 
23
from bzrlib.osutils import rename, sha_file
 
24
import bzrlib
 
25
from itertools import izip
21
26
 
22
27
# XXX: mbp: I'm not totally convinced that we should handle conflicts
23
28
# as part of changeset application, rather than only in the merge
42
47
        newdict[value] = key
43
48
    return newdict
44
49
 
45
 
 
46
50
       
47
 
class ChangeUnixPermissions(object):
 
51
class ChangeExecFlag(object):
48
52
    """This is two-way change, suitable for file modification, creation,
49
53
    deletion"""
50
 
    def __init__(self, old_mode, new_mode):
51
 
        self.old_mode = old_mode
52
 
        self.new_mode = new_mode
 
54
    def __init__(self, old_exec_flag, new_exec_flag):
 
55
        self.old_exec_flag = old_exec_flag
 
56
        self.new_exec_flag = new_exec_flag
53
57
 
54
58
    def apply(self, filename, conflict_handler, reverse=False):
55
59
        if not reverse:
56
 
            from_mode = self.old_mode
57
 
            to_mode = self.new_mode
 
60
            from_exec_flag = self.old_exec_flag
 
61
            to_exec_flag = self.new_exec_flag
58
62
        else:
59
 
            from_mode = self.new_mode
60
 
            to_mode = self.old_mode
 
63
            from_exec_flag = self.new_exec_flag
 
64
            to_exec_flag = self.old_exec_flag
61
65
        try:
62
 
            current_mode = os.stat(filename).st_mode &0777
 
66
            current_exec_flag = bool(os.stat(filename).st_mode & 0111)
63
67
        except OSError, e:
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":
66
70
                    return
67
71
                else:
68
 
                    current_mode = from_mode
 
72
                    current_exec_flag = from_exec_flag
69
73
 
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":
73
77
                return
74
78
 
75
 
        if to_mode is not None:
 
79
        if to_exec_flag is not None:
 
80
            current_mode = os.stat(filename).st_mode
 
81
            if to_exec_flag:
 
82
                umask = os.umask(0)
 
83
                os.umask(umask)
 
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
 
90
            else:
 
91
                to_mode = current_mode & ~0111
76
92
            try:
77
93
                os.chmod(filename, to_mode)
78
94
            except IOError, e:
79
95
                if e.errno == errno.ENOENT:
80
 
                    conflict_handler.missing_for_chmod(filename)
 
96
                    conflict_handler.missing_for_exec_flag(filename)
81
97
 
82
98
    def __eq__(self, other):
83
 
        if not isinstance(other, ChangeUnixPermissions):
84
 
            return False
85
 
        elif self.old_mode != other.old_mode:
86
 
            return False
87
 
        elif self.new_mode != other.new_mode:
88
 
            return False
89
 
        else:
90
 
            return True
 
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)
91
102
 
92
103
    def __ne__(self, other):
93
104
        return not (self == other)
94
105
 
 
106
 
95
107
def dir_create(filename, conflict_handler, reverse):
96
108
    """Creates the directory, or deletes it if reverse is true.  Intended to be
97
109
    used with ReplaceContents.
117
129
        try:
118
130
            os.rmdir(filename)
119
131
        except OSError, e:
120
 
            if e.errno != 39:
 
132
            if e.errno != errno.ENOTEMPTY:
121
133
                raise
122
134
            if conflict_handler.rmdir_non_empty(filename) == "skip":
123
135
                return
124
136
            os.rmdir(filename)
125
137
 
126
 
                
127
 
            
128
138
 
129
139
class SymlinkCreate(object):
130
140
    """Creates or deletes a symlink (for use with ReplaceContents)"""
136
146
        """
137
147
        self.target = contents
138
148
 
 
149
    def __repr__(self):
 
150
        return "SymlinkCreate(%s)" % self.target
 
151
 
139
152
    def __call__(self, filename, conflict_handler, reverse):
140
153
        """Creates or destroys the symlink.
141
154
 
223
236
 
224
237
                    
225
238
 
 
239
class TreeFileCreate(object):
 
240
    """Create or delete a file (for use with ReplaceContents)"""
 
241
    def __init__(self, tree, file_id):
 
242
        """Constructor
 
243
 
 
244
        :param contents: The contents of the file to write
 
245
        :type contents: str
 
246
        """
 
247
        self.tree = tree
 
248
        self.file_id = file_id
 
249
 
 
250
    def __repr__(self):
 
251
        return "TreeFileCreate(%s)" % self.file_id
 
252
 
 
253
    def __eq__(self, other):
 
254
        if not isinstance(other, TreeFileCreate):
 
255
            return False
 
256
        return self.tree.get_file_sha1(self.file_id) == \
 
257
            other.tree.get_file_sha1(other.file_id)
 
258
 
 
259
    def __ne__(self, other):
 
260
        return not (self == other)
 
261
 
 
262
    def write_file(self, filename):
 
263
        outfile = file(filename, "wb")
 
264
        for line in self.tree.get_file(self.file_id):
 
265
            outfile.write(line)
 
266
 
 
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)
 
270
 
 
271
    def __call__(self, filename, conflict_handler, reverse):
 
272
        """Create or delete a file
 
273
 
 
274
        :param filename: The name of the file to create
 
275
        :type filename: str
 
276
        :param reverse: Delete the file instead of creating it
 
277
        :type reverse: bool
 
278
        """
 
279
        if not reverse:
 
280
            try:
 
281
                self.write_file(filename)
 
282
            except IOError, e:
 
283
                if e.errno == errno.ENOENT:
 
284
                    if conflict_handler.missing_parent(filename)=="continue":
 
285
                        self.write_file(filename)
 
286
                else:
 
287
                    raise
 
288
 
 
289
        else:
 
290
            try:
 
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":
 
295
                        return
 
296
                os.unlink(filename)
 
297
            except IOError, e:
 
298
                if e.errno != errno.ENOENT:
 
299
                    raise
 
300
                if conflict_handler.missing_for_rm(filename, undo) == "skip":
 
301
                    return
 
302
 
 
303
                    
 
304
 
226
305
def reversed(sequence):
227
306
    max = len(sequence) - 1
228
307
    for i in range(len(sequence)):
295
374
            if mode is not None:
296
375
                os.chmod(filename, mode)
297
376
 
 
377
    def is_creation(self):
 
378
        return self.new_contents is not None and self.old_contents is None
 
379
 
 
380
    def is_deletion(self):
 
381
        return self.old_contents is not None and self.new_contents is None
 
382
 
298
383
class ApplySequence(object):
299
384
    def __init__(self, changes=None):
300
385
        self.changes = []
326
411
 
327
412
 
328
413
class Diff3Merge(object):
 
414
    history_based = False
329
415
    def __init__(self, file_id, base, other):
330
416
        self.file_id = file_id
331
417
        self.base = base
332
418
        self.other = other
333
419
 
 
420
    def is_creation(self):
 
421
        return False
 
422
 
 
423
    def is_deletion(self):
 
424
        return False
 
425
 
334
426
    def __eq__(self, other):
335
427
        if not isinstance(other, Diff3Merge):
336
428
            return False
340
432
    def __ne__(self, other):
341
433
        return not (self == other)
342
434
 
 
435
    def dump_file(self, temp_dir, name, tree):
 
436
        out_path = os.path.join(temp_dir, name)
 
437
        out_file = file(out_path, "wb")
 
438
        in_file = tree.get_file(self.file_id)
 
439
        for line in in_file:
 
440
            out_file.write(line)
 
441
        return out_path
 
442
 
343
443
    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)
347
 
        if not reverse:
348
 
            base = base_file
349
 
            other = other_file
350
 
        else:
351
 
            base = other_file
352
 
            other = base_file
353
 
        status = patch.diff3(new_file, filename, base, other)
354
 
        if status == 0:
355
 
            os.chmod(new_file, os.stat(filename).st_mode)
356
 
            os.rename(new_file, filename)
357
 
            return
358
 
        else:
359
 
            assert(status == 1)
360
 
            def get_lines(filename):
361
 
                my_file = file(base, "rb")
362
 
                lines = my_file.readlines()
363
 
                my_file.close()
364
 
            base_lines = get_lines(base)
365
 
            other_lines = get_lines(other)
366
 
            conflict_handler.merge_conflict(new_file, filename, base_lines, 
367
 
                                            other_lines)
 
444
        temp_dir = mkdtemp(prefix="bzr-")
 
445
        try:
 
446
            new_file = filename+".new"
 
447
            base_file = self.dump_file(temp_dir, "base", self.base)
 
448
            other_file = self.dump_file(temp_dir, "other", self.other)
 
449
            if not reverse:
 
450
                base = base_file
 
451
                other = other_file
 
452
            else:
 
453
                base = other_file
 
454
                other = base_file
 
455
            status = patch.diff3(new_file, filename, base, other)
 
456
            if status == 0:
 
457
                os.chmod(new_file, os.stat(filename).st_mode)
 
458
                rename(new_file, filename)
 
459
                return
 
460
            else:
 
461
                assert(status == 1)
 
462
                def get_lines(filename):
 
463
                    my_file = file(filename, "rb")
 
464
                    lines = my_file.readlines()
 
465
                    my_file.close()
 
466
                    return lines
 
467
                base_lines = get_lines(base)
 
468
                other_lines = get_lines(other)
 
469
                conflict_handler.merge_conflict(new_file, filename, base_lines, 
 
470
                                                other_lines)
 
471
        finally:
 
472
            rmtree(temp_dir)
368
473
 
369
474
 
370
475
def CreateDir():
403
508
    """
404
509
    return ReplaceContents(FileCreate(contents), None)
405
510
 
406
 
def ReplaceFileContents(old_contents, new_contents):
 
511
def ReplaceFileContents(old_tree, new_tree, file_id):
407
512
    """Convenience fucntion to replace the contents of a file.
408
513
    
409
514
    :param old_contents: The contents of the file to replace 
413
518
    :return: A ReplaceContents that will replace the contents of a file a file 
414
519
    :rtype: `ReplaceContents`
415
520
    """
416
 
    return ReplaceContents(FileCreate(old_contents), FileCreate(new_contents))
 
521
    return ReplaceContents(TreeFileCreate(old_tree, file_id), 
 
522
                           TreeFileCreate(new_tree, file_id))
417
523
 
418
524
def CreateSymlink(target):
419
525
    """Convenience fucntion to create a symlink.
582
688
        :param reverse: if true, the changeset is being applied in reverse
583
689
        :rtype: bool
584
690
        """
585
 
        return ((self.new_parent is None and not reverse) or 
586
 
                (self.parent is None and reverse))
 
691
        return self.is_creation(not reverse)
587
692
 
588
693
    def is_creation(self, reverse):
589
694
        """Return true if applying the entry would create a file/directory.
591
696
        :param reverse: if true, the changeset is being applied in reverse
592
697
        :rtype: bool
593
698
        """
594
 
        return ((self.parent is None and not reverse) or 
595
 
                (self.new_parent is None and reverse))
 
699
        if self.contents_change is None:
 
700
            return False
 
701
        if reverse:
 
702
            return self.contents_change.is_deletion()
 
703
        else:
 
704
            return self.contents_change.is_creation()
596
705
 
597
706
    def is_creation_or_deletion(self):
598
707
        """Return true if applying the entry would create or delete a 
600
709
 
601
710
        :rtype: bool
602
711
        """
603
 
        return self.parent is None or self.new_parent is None
 
712
        return self.is_creation(False) or self.is_deletion(False)
604
713
 
605
714
    def get_cset_path(self, mod=False):
606
715
        """Determine the path of the entry according to the changeset.
778
887
    :rtype: (List, List)
779
888
    """
780
889
    source_entries = [x for x in changeset.entries.itervalues() 
781
 
                      if x.needs_rename()]
 
890
                      if x.needs_rename() or x.is_creation_or_deletion()]
782
891
    # these are done from longest path to shortest, to avoid deleting a
783
892
    # parent before its children are deleted/renamed 
784
893
    def longest_to_shortest(entry):
825
934
            entry.apply(path, conflict_handler, reverse)
826
935
            temp_name[entry.id] = None
827
936
 
828
 
        else:
 
937
        elif entry.needs_rename():
829
938
            to_name = os.path.join(temp_dir, str(i))
830
939
            src_path = inventory.get(entry.id)
831
940
            if src_path is not None:
832
941
                src_path = os.path.join(dir, src_path)
833
942
                try:
834
 
                    os.rename(src_path, to_name)
 
943
                    rename(src_path, to_name)
835
944
                    temp_name[entry.id] = to_name
836
945
                except OSError, e:
837
946
                    if e.errno != errno.ENOENT:
838
947
                        raise
839
 
                    if conflict_handler.missing_for_rename(src_path) == "skip":
 
948
                    if conflict_handler.missing_for_rename(src_path, to_name) \
 
949
                        == "skip":
840
950
                        continue
841
951
 
842
952
    return temp_name
863
973
            continue
864
974
        new_path = os.path.join(dir, new_tree_path)
865
975
        old_path = changed_inventory.get(entry.id)
866
 
        if os.path.exists(new_path):
 
976
        if bzrlib.osutils.lexists(new_path):
867
977
            if conflict_handler.target_exists(entry, new_path, old_path) == \
868
978
                "skip":
869
979
                continue
870
980
        if entry.is_creation(reverse):
871
981
            entry.apply(new_path, conflict_handler, reverse)
872
982
            changed_inventory[entry.id] = new_tree_path
873
 
        else:
 
983
        elif entry.needs_rename():
874
984
            if old_path is None:
875
985
                continue
876
986
            try:
877
 
                os.rename(old_path, new_path)
 
987
                rename(old_path, new_path)
878
988
                changed_inventory[entry.id] = new_tree_path
879
989
            except OSError, e:
880
990
                raise Exception ("%s is missing" % new_path)
915
1025
        Exception.__init__(self, "Conflict applying changes to %s" % this_path)
916
1026
        self.this_path = this_path
917
1027
 
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
924
 
this: %o
925
 
base: %o
926
 
other: %o
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)
932
 
 
933
1028
class WrongOldContents(Exception):
934
1029
    def __init__(self, filename):
935
1030
        msg = "Contents mismatch deleting %s" % filename
936
1031
        self.filename = filename
937
1032
        Exception.__init__(self, msg)
938
1033
 
939
 
class WrongOldPermissions(Exception):
940
 
    def __init__(self, filename, old_perms, new_perms):
941
 
        msg = "Permission missmatch on %s:\n" \
942
 
        "Expected 0%o, got 0%o." % (filename, old_perms, new_perms)
 
1034
class WrongOldExecFlag(Exception):
 
1035
    def __init__(self, filename, old_exec_flag, new_exec_flag):
 
1036
        msg = "Executable flag missmatch on %s:\n" \
 
1037
        "Expected %s, got %s." % (filename, old_exec_flag, new_exec_flag)
943
1038
        self.filename = filename
944
1039
        Exception.__init__(self, msg)
945
1040
 
963
1058
        Exception.__init__(self, msg)
964
1059
        self.filename = filename
965
1060
 
966
 
class MissingPermsFile(Exception):
 
1061
class MissingForSetExec(Exception):
967
1062
    def __init__(self, filename):
968
1063
        msg = "Attempt to change permissions on  %s, which does not exist" %\
969
1064
            filename
978
1073
 
979
1074
 
980
1075
class MissingForRename(Exception):
981
 
    def __init__(self, filename):
982
 
        msg = "Attempt to move missing path %s" % (filename)
 
1076
    def __init__(self, filename, to_path):
 
1077
        msg = "Attempt to move missing path %s to %s" % (filename, to_path)
983
1078
        Exception.__init__(self, msg)
984
1079
        self.filename = filename
985
1080
 
988
1083
        msg = "Conflicting contents for new file %s" % (filename)
989
1084
        Exception.__init__(self, msg)
990
1085
 
 
1086
class WeaveMergeConflict(Exception):
 
1087
    def __init__(self, filename):
 
1088
        msg = "Conflicting contents for file %s" % (filename)
 
1089
        Exception.__init__(self, msg)
 
1090
 
 
1091
class ThreewayContentsConflict(Exception):
 
1092
    def __init__(self, filename):
 
1093
        msg = "Conflicting contents for file %s" % (filename)
 
1094
        Exception.__init__(self, msg)
 
1095
 
991
1096
 
992
1097
class MissingForMerge(Exception):
993
1098
    def __init__(self, filename):
1003
1108
    descend from this class if they have a better way to handle some or
1004
1109
    all types of conflict.
1005
1110
    """
1006
 
    def __init__(self, dir):
1007
 
        self.dir = dir
1008
 
    
1009
1111
    def missing_parent(self, pathname):
1010
1112
        parent = os.path.dirname(pathname)
1011
1113
        raise Exception("Parent directory missing for %s" % pathname)
1029
1131
        os.unlink(new_file)
1030
1132
        raise MergeConflict(this_path)
1031
1133
 
1032
 
    def permission_conflict(self, this_path, base_path, other_path):
1033
 
        raise MergePermissionConflict(this_path, base_path, other_path)
1034
 
 
1035
1134
    def wrong_old_contents(self, filename, expected_contents):
1036
1135
        raise WrongOldContents(filename)
1037
1136
 
1038
1137
    def rem_contents_conflict(self, filename, this_contents, base_contents):
1039
1138
        raise RemoveContentsConflict(filename)
1040
1139
 
1041
 
    def wrong_old_perms(self, filename, old_perms, new_perms):
1042
 
        raise WrongOldPermissions(filename, old_perms, new_perms)
 
1140
    def wrong_old_exec_flag(self, filename, old_exec_flag, new_exec_flag):
 
1141
        raise WrongOldExecFlag(filename, old_exec_flag, new_exec_flag)
1043
1142
 
1044
1143
    def rmdir_non_empty(self, filename):
1045
1144
        raise DeletingNonEmptyDirectory(filename)
1050
1149
    def patch_target_missing(self, filename, contents):
1051
1150
        raise PatchTargetMissing(filename)
1052
1151
 
1053
 
    def missing_for_chmod(self, filename):
1054
 
        raise MissingPermsFile(filename)
 
1152
    def missing_for_exec_flag(self, filename):
 
1153
        raise MissingForExecFlag(filename)
1055
1154
 
1056
1155
    def missing_for_rm(self, filename, change):
1057
1156
        raise MissingForRm(filename)
1058
1157
 
1059
 
    def missing_for_rename(self, filename):
1060
 
        raise MissingForRename(filename)
 
1158
    def missing_for_rename(self, filename, to_path):
 
1159
        raise MissingForRename(filename, to_path)
1061
1160
 
1062
1161
    def missing_for_merge(self, file_id, other_path):
1063
1162
        raise MissingForMerge(other_path)
1065
1164
    def new_contents_conflict(self, filename, other_contents):
1066
1165
        raise NewContentsConflict(filename)
1067
1166
 
1068
 
    def finalize():
 
1167
    def weave_merge_conflict(self, filename, weave, other_i, out_file):
 
1168
        raise WeaveMergeConflict(filename)
 
1169
 
 
1170
    def threeway_contents_conflict(self, filename, this_contents,
 
1171
                                   base_contents, other_contents):
 
1172
        raise ThreewayContentsConflict(filename)
 
1173
 
 
1174
    def finalize(self):
1069
1175
        pass
1070
1176
 
1071
1177
def apply_changeset(changeset, inventory, dir, conflict_handler=None, 
1084
1190
    :rtype: Dictionary
1085
1191
    """
1086
1192
    if conflict_handler is None:
1087
 
        conflict_handler = ExceptionConflictHandler(dir)
 
1193
        conflict_handler = ExceptionConflictHandler()
1088
1194
    temp_dir = os.path.join(dir, "bzr-tree-change")
1089
1195
    try:
1090
1196
        os.mkdir(temp_dir)
1101
1207
    
1102
1208
    #apply changes that don't affect filenames
1103
1209
    for entry in changeset.entries.itervalues():
1104
 
        if not entry.is_creation_or_deletion():
 
1210
        if not entry.is_creation_or_deletion() and not entry.is_boring():
1105
1211
            path = os.path.join(dir, inventory[entry.id])
1106
1212
            entry.apply(path, conflict_handler, reverse)
1107
1213
 
1126
1232
    r_inventory = {}
1127
1233
    for entry in tree.source_inventory().itervalues():
1128
1234
        inventory[entry.id] = entry.path
1129
 
    new_inventory = apply_changeset(cset, r_inventory, tree.root,
 
1235
    new_inventory = apply_changeset(cset, r_inventory, tree.basedir,
1130
1236
                                    reverse=reverse)
1131
1237
    new_entries, remove_entries = \
1132
1238
        get_inventory_change(inventory, new_inventory, cset, reverse)
1267
1373
        return new_meta
1268
1374
    elif new_meta is None:
1269
1375
        return old_meta
1270
 
    elif isinstance(old_meta, ChangeUnixPermissions) and \
1271
 
        isinstance(new_meta, ChangeUnixPermissions):
1272
 
        return ChangeUnixPermissions(old_meta.old_mode, new_meta.new_mode)
 
1376
    elif (isinstance(old_meta, ChangeExecFlag) and
 
1377
          isinstance(new_meta, ChangeExecFlag)):
 
1378
        return ChangeExecFlag(old_meta.old_exec_flag, new_meta.new_exec_flag)
1273
1379
    else:
1274
1380
        return ApplySequence(old_meta, new_meta)
1275
1381
 
1280
1386
            return False
1281
1387
    return True
1282
1388
 
1283
 
class UnsuppportedFiletype(Exception):
1284
 
    def __init__(self, full_path, stat_result):
1285
 
        msg = "The file \"%s\" is not a supported filetype." % full_path
 
1389
class UnsupportedFiletype(Exception):
 
1390
    def __init__(self, kind, full_path):
 
1391
        msg = "The file \"%s\" is a %s, which is not a supported filetype." \
 
1392
            % (full_path, kind)
1286
1393
        Exception.__init__(self, msg)
1287
1394
        self.full_path = full_path
1288
 
        self.stat_result = stat_result
 
1395
        self.kind = kind
1289
1396
 
1290
1397
def generate_changeset(tree_a, tree_b, interesting_ids=None):
1291
1398
    return ChangesetGenerator(tree_a, tree_b, interesting_ids)()
1292
1399
 
 
1400
 
1293
1401
class ChangesetGenerator(object):
1294
1402
    def __init__(self, tree_a, tree_b, interesting_ids=None):
1295
1403
        object.__init__(self)
1331
1439
    def get_entry(self, file_id, tree):
1332
1440
        if not tree.has_or_had_id(file_id):
1333
1441
            return None
1334
 
        return tree.tree.inventory[file_id]
 
1442
        return tree.inventory[file_id]
1335
1443
 
1336
1444
    def get_entry_parent(self, entry):
1337
1445
        if entry is None:
1387
1495
 
1388
1496
        if cs_entry is None:
1389
1497
            return None
 
1498
 
 
1499
        cs_entry.metadata_change = self.make_exec_flag_change(id)
 
1500
 
1390
1501
        if id in self.tree_a and id in self.tree_b:
1391
1502
            a_sha1 = self.tree_a.get_file_sha1(id)
1392
1503
            b_sha1 = self.tree_b.get_file_sha1(id)
1393
1504
            if None not in (a_sha1, b_sha1) and a_sha1 == b_sha1:
1394
1505
                return cs_entry
1395
1506
 
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)
1400
 
        
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,
1403
 
                                                             stat_a, 
1404
 
                                                             full_path_b, 
1405
 
                                                             stat_b)
 
1507
        cs_entry.contents_change = self.make_contents_change(id)
1406
1508
        return cs_entry
1407
1509
 
1408
 
    def make_mode_change(self, stat_a, stat_b):
1409
 
        mode_a = None
1410
 
        if stat_a is not None and not stat.S_ISLNK(stat_a.st_mode):
1411
 
            mode_a = stat_a.st_mode & 0777
1412
 
        mode_b = None
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:
1416
 
            return None
1417
 
        return ChangeUnixPermissions(mode_a, mode_b)
1418
 
 
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:
1421
 
            return 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):
1424
 
            return None
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:
1429
 
                return None
1430
 
 
1431
 
        a_contents = self.get_contents(stat_a, full_path_a)
1432
 
        b_contents = self.get_contents(stat_b, full_path_b)
 
1510
    def make_exec_flag_change(self, file_id):
 
1511
        exec_flag_a = exec_flag_b = None
 
1512
        if file_id in self.tree_a and self.tree_a.kind(file_id) == "file":
 
1513
            exec_flag_a = self.tree_a.is_executable(file_id)
 
1514
 
 
1515
        if file_id in self.tree_b and self.tree_b.kind(file_id) == "file":
 
1516
            exec_flag_b = self.tree_b.is_executable(file_id)
 
1517
 
 
1518
        if exec_flag_a == exec_flag_b:
 
1519
            return None
 
1520
        return ChangeExecFlag(exec_flag_a, exec_flag_b)
 
1521
 
 
1522
    def make_contents_change(self, file_id):
 
1523
        a_contents = get_contents(self.tree_a, file_id)
 
1524
        b_contents = get_contents(self.tree_b, file_id)
1433
1525
        if a_contents == b_contents:
1434
1526
            return None
1435
1527
        return ReplaceContents(a_contents, b_contents)
1436
1528
 
1437
 
    def get_contents(self, stat_result, full_path):
1438
 
        if stat_result is None:
1439
 
            return 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):
1443
 
            return dir_create
1444
 
        elif stat.S_ISLNK(stat_result.st_mode):
1445
 
            return SymlinkCreate(os.readlink(full_path))
1446
 
        else:
1447
 
            raise UnsupportedFiletype(full_path, stat_result)
1448
1529
 
1449
 
    def lstat(self, full_path):
1450
 
        stat_result = None
1451
 
        if full_path is not None:
1452
 
            try:
1453
 
                stat_result = os.lstat(full_path)
1454
 
            except OSError, e:
1455
 
                if e.errno != errno.ENOENT:
1456
 
                    raise
1457
 
        return stat_result
 
1530
def get_contents(tree, file_id):
 
1531
    """Return the appropriate contents to create a copy of file_id from tree"""
 
1532
    if file_id not in tree:
 
1533
        return None
 
1534
    kind = tree.kind(file_id)
 
1535
    if kind == "file":
 
1536
        return TreeFileCreate(tree, file_id)
 
1537
    elif kind in ("directory", "root_directory"):
 
1538
        return dir_create
 
1539
    elif kind == "symlink":
 
1540
        return SymlinkCreate(tree.get_symlink_target(file_id))
 
1541
    else:
 
1542
        raise UnsupportedFiletype(kind, tree.id2path(file_id))
1458
1543
 
1459
1544
 
1460
1545
def full_path(entry, tree):
1461
 
    return os.path.join(tree.root, entry.path)
 
1546
    return os.path.join(tree.basedir, entry.path)
1462
1547
 
1463
1548
def new_delete_entry(entry, tree, inventory, delete):
1464
1549
    if entry.path == "":