14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
from __future__ import absolute_import
23
from bzrlib.lazy_import import lazy_import
25
from .lazy_import import lazy_import
24
26
lazy_import(globals(), """
31
branch as _mod_branch,
43
from bzrlib.workingtree import WorkingTree
43
from breezy.workingtree import WorkingTree
44
from breezy.i18n import gettext
46
from bzrlib.registry import (
47
from .registry import (
49
from bzrlib.symbol_versioning import (
53
from bzrlib.trace import mutter, note, warning
50
from .sixish import text_type
51
from .trace import mutter, note, warning
52
from .tree import FileTimestampUnavailable
55
DEFAULT_CONTEXT_AMOUNT = 3
56
58
class AtTemplate(string.Template):
86
88
# In the meantime we at least make sure the patch isn't
90
# Special workaround for Python2.3, where difflib fails if
91
# both sequences are empty.
92
if not oldlines and not newlines:
95
91
if allow_binary is False:
96
92
textfile.check_text_lines(oldlines)
97
93
textfile.check_text_lines(newlines)
99
95
if sequence_matcher is None:
100
96
sequence_matcher = patiencediff.PatienceSequenceMatcher
101
ud = patiencediff.unified_diff(oldlines, newlines,
102
fromfile=old_filename.encode(path_encoding),
103
tofile=new_filename.encode(path_encoding),
104
sequencematcher=sequence_matcher)
97
ud = unified_diff_bytes(
99
fromfile=old_label.encode(path_encoding, 'replace'),
100
tofile=new_label.encode(path_encoding, 'replace'),
101
n=context_lines, sequencematcher=sequence_matcher)
107
if len(ud) == 0: # Identical contents, nothing to do
104
if len(ud) == 0: # Identical contents, nothing to do
109
106
# work-around for difflib being too smart for its own good
110
107
# if /dev/null is "1,0", patch won't recognize it as /dev/null
112
ud[2] = ud[2].replace('-1,0', '-0,0')
109
ud[2] = ud[2].replace(b'-1,0', b'-0,0')
113
110
elif not newlines:
114
ud[2] = ud[2].replace('+1,0', '+0,0')
111
ud[2] = ud[2].replace(b'+1,0', b'+0,0')
117
114
to_file.write(line)
118
if not line.endswith('\n'):
119
to_file.write("\n\\ No newline at end of file\n")
115
if not line.endswith(b'\n'):
116
to_file.write(b"\n\\ No newline at end of file\n")
120
def unified_diff_bytes(a, b, fromfile=b'', tofile=b'', fromfiledate=b'',
121
tofiledate=b'', n=3, lineterm=b'\n', sequencematcher=None):
123
Compare two sequences of lines; generate the delta as a unified diff.
125
Unified diffs are a compact way of showing line changes and a few
126
lines of context. The number of context lines is set by 'n' which
129
By default, the diff control lines (those with ---, +++, or @@) are
130
created with a trailing newline. This is helpful so that inputs
131
created from file.readlines() result in diffs that are suitable for
132
file.writelines() since both the inputs and outputs have trailing
135
For inputs that do not have trailing newlines, set the lineterm
136
argument to "" so that the output will be uniformly newline free.
138
The unidiff format normally has a header for filenames and modification
139
times. Any or all of these may be specified using strings for
140
'fromfile', 'tofile', 'fromfiledate', and 'tofiledate'. The modification
141
times are normally expressed in the format returned by time.ctime().
145
>>> for line in bytes_unified_diff(b'one two three four'.split(),
146
... b'zero one tree four'.split(), b'Original', b'Current',
147
... b'Sat Jan 26 23:30:50 1991', b'Fri Jun 06 10:20:52 2003',
150
--- Original Sat Jan 26 23:30:50 1991
151
+++ Current Fri Jun 06 10:20:52 2003
160
if sequencematcher is None:
161
sequencematcher = difflib.SequenceMatcher
164
fromfiledate = b'\t' + bytes(fromfiledate)
166
tofiledate = b'\t' + bytes(tofiledate)
169
for group in sequencematcher(None, a, b).get_grouped_opcodes(n):
171
yield b'--- %s%s%s' % (fromfile, fromfiledate, lineterm)
172
yield b'+++ %s%s%s' % (tofile, tofiledate, lineterm)
174
i1, i2, j1, j2 = group[0][1], group[-1][2], group[0][3], group[-1][4]
175
yield b"@@ -%d,%d +%d,%d @@%s" % (i1 + 1, i2 - i1, j1 + 1, j2 - j1, lineterm)
176
for tag, i1, i2, j1, j2 in group:
178
for line in a[i1:i2]:
181
if tag == 'replace' or tag == 'delete':
182
for line in a[i1:i2]:
184
if tag == 'replace' or tag == 'insert':
185
for line in b[j1:j2]:
123
189
def _spawn_external_diff(diffcmd, capture_errors=True):
124
"""Spawn the externall diff process, and return the child handle.
190
"""Spawn the external diff process, and return the child handle.
126
192
:param diffcmd: The command list to spawn
127
193
:param capture_errors: Capture stderr as well as setting LANG=C
160
def external_diff(old_filename, oldlines, new_filename, newlines, to_file,
226
# diff style options as of GNU diff v3.2
227
style_option_list = ['-c', '-C', '--context',
229
'-f', '--forward-ed',
233
'-u', '-U', '--unified',
234
'-y', '--side-by-side',
238
def default_style_unified(diff_opts):
239
"""Default to unified diff style if alternative not specified in diff_opts.
241
diff only allows one style to be specified; they don't override.
242
Note that some of these take optargs, and the optargs can be
243
directly appended to the options.
244
This is only an approximate parser; it doesn't properly understand
247
:param diff_opts: List of options for external (GNU) diff.
248
:return: List of options with default style=='unified'.
250
for s in style_option_list:
258
diff_opts.append('-u')
262
def external_diff(old_label, oldlines, new_label, newlines, to_file,
162
264
"""Display a diff by calling out to the external diff program."""
163
265
# make sure our own output is properly ordered before the diff
166
oldtmp_fd, old_abspath = tempfile.mkstemp(prefix='bzr-diff-old-')
167
newtmp_fd, new_abspath = tempfile.mkstemp(prefix='bzr-diff-new-')
268
oldtmp_fd, old_abspath = tempfile.mkstemp(prefix='brz-diff-old-')
269
newtmp_fd, new_abspath = tempfile.mkstemp(prefix='brz-diff-new-')
168
270
oldtmpf = os.fdopen(oldtmp_fd, 'wb')
169
271
newtmpf = os.fdopen(newtmp_fd, 'wb')
187
289
if sys.platform == 'win32':
188
290
# Popen doesn't do the proper encoding for external commands
189
291
# Since we are dealing with an ANSI api, use mbcs encoding
190
old_filename = old_filename.encode('mbcs')
191
new_filename = new_filename.encode('mbcs')
292
old_label = old_label.encode('mbcs')
293
new_label = new_label.encode('mbcs')
192
294
diffcmd = ['diff',
193
'--label', old_filename,
295
'--label', old_label,
195
'--label', new_filename,
297
'--label', new_label,
200
# diff only allows one style to be specified; they don't override.
201
# note that some of these take optargs, and the optargs can be
202
# directly appended to the options.
203
# this is only an approximate parser; it doesn't properly understand
205
for s in ['-c', '-u', '-C', '-U',
210
'-y', '--side-by-side',
302
diff_opts = default_style_unified(diff_opts)
222
305
diffcmd.extend(diff_opts)
224
307
pipe = _spawn_external_diff(diffcmd, capture_errors=True)
225
out,err = pipe.communicate()
308
out, err = pipe.communicate()
226
309
rc = pipe.returncode
228
311
# internal_diff() adds a trailing newline, add one here for consistency
231
314
# 'diff' gives retcode == 2 for all sorts of errors
232
315
# one of those is 'Binary files differ'.
239
322
out, err = pipe.communicate()
241
324
# Write out the new i18n diff response
242
to_file.write(out+'\n')
325
to_file.write(out + b'\n')
243
326
if pipe.returncode != 2:
244
327
raise errors.BzrError(
245
'external diff failed with exit code 2'
246
' when run with LANG=C and LC_ALL=C,'
247
' but not when run natively: %r' % (diffcmd,))
328
'external diff failed with exit code 2'
329
' when run with LANG=C and LC_ALL=C,'
330
' but not when run natively: %r' % (diffcmd,))
249
first_line = lang_c_out.split('\n', 1)[0]
332
first_line = lang_c_out.split(b'\n', 1)[0]
250
333
# Starting with diffutils 2.8.4 the word "binary" was dropped.
251
m = re.match('^(binary )?files.*differ$', first_line, re.I)
334
m = re.match(b'^(binary )?files.*differ$', first_line, re.I)
253
336
raise errors.BzrError('external diff failed with exit code 2;'
254
337
' command: %r' % (diffcmd,))
267
350
msg = 'exit code %d' % rc
269
352
raise errors.BzrError('external diff failed with %s; command: %r'
274
356
oldtmpf.close() # and delete
276
# Clean up. Warn in case the files couldn't be deleted
277
# (in case windows still holds the file open, but not
278
# if the files have already been deleted)
280
os.remove(old_abspath)
282
if e.errno not in (errno.ENOENT,):
283
warning('Failed to delete temporary file: %s %s',
286
os.remove(new_abspath)
288
if e.errno not in (errno.ENOENT,):
289
warning('Failed to delete temporary file: %s %s',
293
@deprecated_function(deprecated_in((2, 2, 0)))
294
def get_trees_and_branches_to_diff(path_list, revision_specs, old_url, new_url,
296
"""Get the trees and specific files to diff given a list of paths.
298
This method works out the trees to be diff'ed and the files of
299
interest within those trees.
302
the list of arguments passed to the diff command
303
:param revision_specs:
304
Zero, one or two RevisionSpecs from the diff command line,
305
saying what revisions to compare.
307
The url of the old branch or tree. If None, the tree to use is
308
taken from the first path, if any, or the current working tree.
310
The url of the new branch or tree. If None, the tree to use is
311
taken from the first path, if any, or the current working tree.
313
if True and a view is set, apply the view or check that the paths
316
a tuple of (old_tree, new_tree, old_branch, new_branch,
317
specific_files, extra_trees) where extra_trees is a sequence of
318
additional trees to search in for file-ids. The trees and branches
321
op = cleanup.OperationWithCleanups(get_trees_and_branches_to_diff_locked)
322
return op.run_simple(path_list, revision_specs, old_url, new_url,
323
op.add_cleanup, apply_view=apply_view)
360
# Warn in case the file couldn't be deleted (in case windows still
361
# holds the file open, but not if the files have already been
366
if e.errno not in (errno.ENOENT,):
367
warning('Failed to delete temporary file: %s %s', path, e)
326
373
def get_trees_and_branches_to_diff_locked(
327
path_list, revision_specs, old_url, new_url, add_cleanup, apply_view=True):
374
path_list, revision_specs, old_url, new_url, exit_stack, apply_view=True):
328
375
"""Get the trees and specific files to diff given a list of paths.
330
377
This method works out the trees to be diff'ed and the files of
408
453
new_url = default_location
409
454
if new_url != old_url:
410
455
working_tree, branch, relpath = \
411
bzrdir.BzrDir.open_containing_tree_or_branch(new_url)
456
controldir.ControlDir.open_containing_tree_or_branch(new_url)
412
457
lock_tree_or_branch(working_tree, branch)
413
458
if consider_relpath and relpath != '':
414
459
if working_tree is not None and apply_view:
415
460
views.check_path_in_view(working_tree, relpath)
416
461
specific_files.append(relpath)
417
462
new_tree = _get_tree_to_diff(new_revision_spec, working_tree, branch,
418
basis_is_default=working_tree is None)
463
basis_is_default=working_tree is None)
419
464
new_branch = branch
421
466
# Get the specific files (all files is None, no files is [])
422
467
if make_paths_wt_relative and working_tree is not None:
424
from bzrlib.builtins import safe_relpath_files
425
other_paths = safe_relpath_files(working_tree, other_paths,
468
other_paths = working_tree.safe_relpath_files(
426
470
apply_view=apply_view)
427
except errors.FileInWrongBranch:
428
raise errors.BzrCommandError("Files are in different branches")
429
471
specific_files.extend(other_paths)
430
472
if len(specific_files) == 0:
431
473
specific_files = None
432
if (working_tree is not None and working_tree.supports_views()
474
if (working_tree is not None and working_tree.supports_views() and
434
476
view_files = working_tree.views.lookup_view()
436
478
specific_files = view_files
437
479
view_str = views.view_display_str(view_files)
438
note("*** Ignoring files outside view. View is %s" % view_str)
480
note(gettext("*** Ignoring files outside view. View is %s") % view_str)
440
482
# Get extra trees that ought to be searched for file-ids
441
483
extra_trees = None
442
484
if working_tree is not None and working_tree not in (old_tree, new_tree):
443
485
extra_trees = (working_tree,)
444
return old_tree, new_tree, old_branch, new_branch, specific_files, extra_trees
486
return (old_tree, new_tree, old_branch, new_branch,
487
specific_files, extra_trees)
447
490
def _get_tree_to_diff(spec, tree=None, branch=None, basis_is_default=True):
464
507
extra_trees=None,
465
508
path_encoding='utf8',
511
context=DEFAULT_CONTEXT_AMOUNT):
468
512
"""Show in text form the changes from one tree to another.
470
514
:param to_file: The output stream.
471
:param specific_files:Include only changes to these files - None for all
515
:param specific_files: Include only changes to these files - None for all
473
:param external_diff_options: If set, use an external GNU diff and pass
517
:param external_diff_options: If set, use an external GNU diff and pass
475
519
:param extra_trees: If set, more Trees to use for looking up file ids
476
:param path_encoding: If set, the path will be encoded as specified,
520
:param path_encoding: If set, the path will be encoded as specified,
477
521
otherwise is supposed to be utf8
478
522
:param format_cls: Formatter class (DiffTree subclass)
525
context = DEFAULT_CONTEXT_AMOUNT
480
526
if format_cls is None:
481
527
format_cls = DiffTree
528
with cleanup.ExitStack() as exit_stack:
529
exit_stack.enter_context(old_tree.lock_read())
484
530
if extra_trees is not None:
485
531
for tree in extra_trees:
489
differ = format_cls.from_trees_options(old_tree, new_tree, to_file,
491
external_diff_options,
492
old_label, new_label, using)
493
return differ.show_diff(specific_files, extra_trees)
496
if extra_trees is not None:
497
for tree in extra_trees:
503
def _patch_header_date(tree, file_id, path):
532
exit_stack.enter_context(tree.lock_read())
533
exit_stack.enter_context(new_tree.lock_read())
534
differ = format_cls.from_trees_options(old_tree, new_tree, to_file,
536
external_diff_options,
537
old_label, new_label, using,
538
context_lines=context)
539
return differ.show_diff(specific_files, extra_trees)
542
def _patch_header_date(tree, path):
504
543
"""Returns a timestamp suitable for use in a patch header."""
506
mtime = tree.get_file_mtime(file_id, path)
507
except errors.FileTimestampUnavailable:
545
mtime = tree.get_file_mtime(path)
546
except FileTimestampUnavailable:
509
548
return timestamp.format_patch_date(mtime)
512
551
def get_executable_change(old_is_x, new_is_x):
513
descr = { True:"+x", False:"-x", None:"??" }
552
descr = {True: b"+x", False: b"-x", None: b"??"}
514
553
if old_is_x != new_is_x:
515
return ["%s to %s" % (descr[old_is_x], descr[new_is_x],)]
554
return [b"%s to %s" % (descr[old_is_x], descr[new_is_x],)]
587
625
if None in (old_kind, new_kind):
588
626
return DiffPath.CANNOT_DIFF
589
result = DiffPath._diff_many(self.differs, file_id, old_path,
590
new_path, old_kind, None)
627
result = DiffPath._diff_many(
628
self.differs, old_path, new_path, old_kind, None)
591
629
if result is DiffPath.CANNOT_DIFF:
593
return DiffPath._diff_many(self.differs, file_id, old_path, new_path,
631
return DiffPath._diff_many(
632
self.differs, old_path, new_path, None, new_kind)
635
class DiffTreeReference(DiffPath):
637
def diff(self, old_path, new_path, old_kind, new_kind):
638
"""Perform comparison between two tree references. (dummy)
641
if 'tree-reference' not in (old_kind, new_kind):
642
return self.CANNOT_DIFF
643
if old_kind not in ('tree-reference', None):
644
return self.CANNOT_DIFF
645
if new_kind not in ('tree-reference', None):
646
return self.CANNOT_DIFF
597
650
class DiffDirectory(DiffPath):
599
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
652
def diff(self, old_path, new_path, old_kind, new_kind):
600
653
"""Perform comparison between two directories. (dummy)
654
709
EPOCH_DATE = '1970-01-01 00:00:00 +0000'
656
711
def __init__(self, old_tree, new_tree, to_file, path_encoding='utf-8',
657
old_label='', new_label='', text_differ=internal_diff):
712
old_label='', new_label='', text_differ=internal_diff,
713
context_lines=DEFAULT_CONTEXT_AMOUNT):
658
714
DiffPath.__init__(self, old_tree, new_tree, to_file, path_encoding)
659
715
self.text_differ = text_differ
660
716
self.old_label = old_label
661
717
self.new_label = new_label
662
718
self.path_encoding = path_encoding
719
self.context_lines = context_lines
664
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
721
def diff(self, old_path, new_path, old_kind, new_kind):
665
722
"""Compare two files in unified diff format
667
:param file_id: The file_id of the file to compare
668
724
:param old_path: Path of the file in the old tree
669
725
:param new_path: Path of the file in the new tree
670
726
:param old_kind: Old file-kind of the file
673
729
if 'file' not in (old_kind, new_kind):
674
730
return self.CANNOT_DIFF
675
from_file_id = to_file_id = file_id
676
731
if old_kind == 'file':
677
old_date = _patch_header_date(self.old_tree, file_id, old_path)
732
old_date = _patch_header_date(self.old_tree, old_path)
678
733
elif old_kind is None:
679
734
old_date = self.EPOCH_DATE
682
736
return self.CANNOT_DIFF
683
737
if new_kind == 'file':
684
new_date = _patch_header_date(self.new_tree, file_id, new_path)
738
new_date = _patch_header_date(self.new_tree, new_path)
685
739
elif new_kind is None:
686
740
new_date = self.EPOCH_DATE
689
742
return self.CANNOT_DIFF
690
from_label = '%s%s\t%s' % (self.old_label, old_path, old_date)
691
to_label = '%s%s\t%s' % (self.new_label, new_path, new_date)
692
return self.diff_text(from_file_id, to_file_id, from_label, to_label,
743
from_label = '%s%s\t%s' % (self.old_label, old_path,
745
to_label = '%s%s\t%s' % (self.new_label, new_path,
747
return self.diff_text(old_path, new_path, from_label, to_label)
695
def diff_text(self, from_file_id, to_file_id, from_label, to_label,
696
from_path=None, to_path=None):
749
def diff_text(self, from_path, to_path, from_label, to_label):
697
750
"""Diff the content of given files in two trees
699
:param from_file_id: The id of the file in the from tree. If None,
752
:param from_path: The path in the from tree. If None,
700
753
the file is not present in the from tree.
701
:param to_file_id: The id of the file in the to tree. This may refer
702
to a different file from from_file_id. If None,
754
:param to_path: The path in the to tree. This may refer
755
to a different file from from_path. If None,
703
756
the file is not present in the to tree.
704
:param from_path: The path in the from tree or None if unknown.
705
:param to_path: The path in the to tree or None if unknown.
707
def _get_text(tree, file_id, path):
708
if file_id is not None:
709
return tree.get_file(file_id, path).readlines()
758
def _get_text(tree, path):
762
return tree.get_file_lines(path)
763
except errors.NoSuchFile:
713
from_text = _get_text(self.old_tree, from_file_id, from_path)
714
to_text = _get_text(self.new_tree, to_file_id, to_path)
766
from_text = _get_text(self.old_tree, from_path)
767
to_text = _get_text(self.new_tree, to_path)
715
768
self.text_differ(from_label, from_text, to_label, to_text,
769
self.to_file, path_encoding=self.path_encoding,
770
context_lines=self.context_lines)
717
771
except errors.BinaryFile:
718
772
self.to_file.write(
719
("Binary files %s and %s differ\n" %
720
(from_label, to_label)).encode(self.path_encoding))
773
("Binary files %s and %s differ\n" %
774
(from_label, to_label)).encode(self.path_encoding, 'replace'))
721
775
return self.CHANGED
742
def make_from_diff_tree(klass, command_string):
796
def make_from_diff_tree(klass, command_string, external_diff_options=None):
743
797
def from_diff_tree(diff_tree):
744
return klass.from_string(command_string, diff_tree.old_tree,
798
full_command_string = [command_string]
799
if external_diff_options is not None:
800
full_command_string += ' ' + external_diff_options
801
return klass.from_string(full_command_string, diff_tree.old_tree,
745
802
diff_tree.new_tree, diff_tree.to_file)
746
803
return from_diff_tree
748
805
def _get_command(self, old_path, new_path):
749
806
my_map = {'old_path': old_path, 'new_path': new_path}
750
return [AtTemplate(t).substitute(my_map) for t in
751
self.command_template]
807
command = [AtTemplate(t).substitute(my_map) for t in
808
self.command_template]
809
if sys.platform == 'win32': # Popen doesn't accept unicode on win32
812
if isinstance(c, text_type):
813
command_encoded.append(c.encode('mbcs'))
815
command_encoded.append(c)
816
return command_encoded
753
820
def _execute(self, old_path, new_path):
754
821
command = self._get_command(old_path, new_path)
756
823
proc = subprocess.Popen(command, stdout=subprocess.PIPE,
759
826
if e.errno == errno.ENOENT:
760
827
raise errors.ExecutableMissing(command[0])
763
830
self.to_file.write(proc.stdout.read())
764
832
return proc.wait()
766
834
def _try_symlink_root(self, tree, prefix):
767
if (getattr(tree, 'abspath', None) is None
768
or not osutils.host_os_dereferences_symlinks()):
835
if (getattr(tree, 'abspath', None) is None or
836
not osutils.host_os_dereferences_symlinks()):
771
839
os.symlink(tree.abspath(''), osutils.pathjoin(self._root, prefix))
773
841
if e.errno != errno.EEXIST:
777
def _write_file(self, file_id, tree, prefix, relpath, force_temp=False,
847
"""Returns safe encoding for passing file path to diff tool"""
848
if sys.platform == 'win32':
851
# Don't fallback to 'utf-8' because subprocess may not be able to
852
# handle utf-8 correctly when locale is not utf-8.
853
return sys.getfilesystemencoding() or 'ascii'
855
def _is_safepath(self, path):
856
"""Return true if `path` may be able to pass to subprocess."""
859
return path == path.encode(fenc).decode(fenc)
863
def _safe_filename(self, prefix, relpath):
864
"""Replace unsafe character in `relpath` then join `self._root`,
865
`prefix` and `relpath`."""
867
# encoded_str.replace('?', '_') may break multibyte char.
868
# So we should encode, decode, then replace(u'?', u'_')
869
relpath_tmp = relpath.encode(fenc, 'replace').decode(fenc, 'replace')
870
relpath_tmp = relpath_tmp.replace(u'?', u'_')
871
return osutils.pathjoin(self._root, prefix, relpath_tmp)
873
def _write_file(self, relpath, tree, prefix, force_temp=False,
778
874
allow_write=False):
779
875
if not force_temp and isinstance(tree, WorkingTree):
780
return tree.abspath(tree.id2path(file_id))
782
full_path = osutils.pathjoin(self._root, prefix, relpath)
876
full_path = tree.abspath(relpath)
877
if self._is_safepath(full_path):
880
full_path = self._safe_filename(prefix, relpath)
783
881
if not force_temp and self._try_symlink_root(tree, prefix):
785
883
parent_dir = osutils.dirname(full_path)
787
885
os.makedirs(parent_dir)
789
887
if e.errno != errno.EEXIST:
791
source = tree.get_file(file_id, relpath)
793
target = open(full_path, 'wb')
795
osutils.pumpfile(source, target)
801
mtime = tree.get_file_mtime(file_id)
802
except errors.FileTimestampUnavailable:
889
with tree.get_file(relpath) as source, \
890
open(full_path, 'wb') as target:
891
osutils.pumpfile(source, target)
893
mtime = tree.get_file_mtime(relpath)
894
except FileTimestampUnavailable:
805
897
os.utime(full_path, (mtime, mtime))
807
899
osutils.make_readonly(full_path)
810
def _prepare_files(self, file_id, old_path, new_path, force_temp=False,
902
def _prepare_files(self, old_path, new_path, force_temp=False,
811
903
allow_write_new=False):
812
old_disk_path = self._write_file(file_id, self.old_tree, 'old',
813
old_path, force_temp)
814
new_disk_path = self._write_file(file_id, self.new_tree, 'new',
815
new_path, force_temp,
816
allow_write=allow_write_new)
904
old_disk_path = self._write_file(
905
old_path, self.old_tree, 'old', force_temp)
906
new_disk_path = self._write_file(
907
new_path, self.new_tree, 'new', force_temp,
908
allow_write=allow_write_new)
817
909
return old_disk_path, new_disk_path
819
911
def finish(self):
821
913
osutils.rmtree(self._root)
823
915
if e.errno != errno.ENOENT:
824
916
mutter("The temporary directory \"%s\" was not "
825
"cleanly removed: %s." % (self._root, e))
917
"cleanly removed: %s." % (self._root, e))
827
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
919
def diff(self, old_path, new_path, old_kind, new_kind):
828
920
if (old_kind, new_kind) != ('file', 'file'):
829
921
return DiffPath.CANNOT_DIFF
830
922
(old_disk_path, new_disk_path) = self._prepare_files(
831
file_id, old_path, new_path)
832
924
self._execute(old_disk_path, new_disk_path)
834
def edit_file(self, file_id):
926
def edit_file(self, old_path, new_path):
835
927
"""Use this tool to edit a file.
837
929
A temporary copy will be edited, and the new contents will be
840
:param file_id: The id of the file to edit.
841
932
:return: The new contents of the file.
843
old_path = self.old_tree.id2path(file_id)
844
new_path = self.new_tree.id2path(file_id)
845
new_abs_path = self._prepare_files(file_id, old_path, new_path,
846
allow_write_new=True,
848
command = self._get_command(osutils.pathjoin('old', old_path),
849
osutils.pathjoin('new', new_path))
934
old_abs_path, new_abs_path = self._prepare_files(
935
old_path, new_path, allow_write_new=True, force_temp=True)
936
command = self._get_command(old_abs_path, new_abs_path)
850
937
subprocess.call(command, cwd=self._root)
851
new_file = open(new_abs_path, 'r')
938
with open(new_abs_path, 'rb') as new_file:
853
939
return new_file.read()
858
942
class DiffTree(object):
915
1001
:param using: Commandline to use to invoke an external diff tool
917
1003
if using is not None:
918
extra_factories = [DiffFromTool.make_from_diff_tree(using)]
1004
extra_factories = [DiffFromTool.make_from_diff_tree(
1005
using, external_diff_options)]
920
1007
extra_factories = []
921
1008
if external_diff_options:
922
1009
opts = external_diff_options.split()
923
def diff_file(olab, olines, nlab, nlines, to_file):
1011
def diff_file(olab, olines, nlab, nlines, to_file, path_encoding=None, context_lines=None):
1012
""":param path_encoding: not used but required
1013
to match the signature of internal_diff.
924
1015
external_diff(olab, olines, nlab, nlines, to_file, opts)
926
1017
diff_file = internal_diff
927
1018
diff_text = DiffText(old_tree, new_tree, to_file, path_encoding,
928
old_label, new_label, diff_file)
1019
old_label, new_label, diff_file, context_lines=context_lines)
929
1020
return klass(old_tree, new_tree, to_file, path_encoding, diff_text,
930
1021
extra_factories)
945
1036
# TODO: Generation of pseudo-diffs for added/deleted files could
946
1037
# be usefully made into a much faster special case.
947
1038
iterator = self.new_tree.iter_changes(self.old_tree,
948
specific_files=specific_files,
949
extra_trees=extra_trees,
950
require_versioned=True)
1039
specific_files=specific_files,
1040
extra_trees=extra_trees,
1041
require_versioned=True)
952
1044
def changes_key(change):
953
1045
old_path, new_path = change[1]
955
1047
if path is None:
958
1051
def get_encoded_path(path):
959
1052
if path is not None:
960
1053
return path.encode(self.path_encoding, "replace")
961
for (file_id, paths, changed_content, versioned, parent, name, kind,
962
executable) in sorted(iterator, key=changes_key):
1054
for change in sorted(iterator, key=changes_key):
963
1055
# The root does not get diffed, and items with no known kind (that
964
1056
# is, missing) in both trees are skipped as well.
965
if parent == (None, None) or kind == (None, None):
967
oldpath, newpath = paths
968
oldpath_encoded = get_encoded_path(paths[0])
969
newpath_encoded = get_encoded_path(paths[1])
970
old_present = (kind[0] is not None and versioned[0])
971
new_present = (kind[1] is not None and versioned[1])
972
renamed = (parent[0], name[0]) != (parent[1], name[1])
1057
if change.parent_id == (None, None) or change.kind == (None, None):
1059
if change.kind[0] == 'symlink' and not self.new_tree.supports_symlinks():
1061
'Ignoring "%s" as symlinks are not '
1062
'supported on this filesystem.' % (change.path[0],))
1064
oldpath, newpath = change.path
1065
oldpath_encoded = get_encoded_path(change.path[0])
1066
newpath_encoded = get_encoded_path(change.path[1])
1067
old_present = (change.kind[0] is not None and change.versioned[0])
1068
new_present = (change.kind[1] is not None and change.versioned[1])
1069
executable = change.executable
1071
renamed = (change.parent_id[0], change.name[0]) != (change.parent_id[1], change.name[1])
974
1073
properties_changed = []
975
properties_changed.extend(get_executable_change(executable[0], executable[1]))
1074
properties_changed.extend(
1075
get_executable_change(executable[0], executable[1]))
977
1077
if properties_changed:
978
prop_str = " (properties changed: %s)" % (", ".join(properties_changed),)
1078
prop_str = b" (properties changed: %s)" % (
1079
b", ".join(properties_changed),)
982
1083
if (old_present, new_present) == (True, False):
983
self.to_file.write("=== removed %s '%s'\n" %
984
(kind[0], oldpath_encoded))
1084
self.to_file.write(b"=== removed %s '%s'\n" %
1085
(kind[0].encode('ascii'), oldpath_encoded))
985
1086
newpath = oldpath
986
1087
elif (old_present, new_present) == (False, True):
987
self.to_file.write("=== added %s '%s'\n" %
988
(kind[1], newpath_encoded))
1088
self.to_file.write(b"=== added %s '%s'\n" %
1089
(kind[1].encode('ascii'), newpath_encoded))
989
1090
oldpath = newpath
991
self.to_file.write("=== renamed %s '%s' => '%s'%s\n" %
992
(kind[0], oldpath_encoded, newpath_encoded, prop_str))
1092
self.to_file.write(b"=== renamed %s '%s' => '%s'%s\n" %
1093
(kind[0].encode('ascii'), oldpath_encoded, newpath_encoded, prop_str))
994
1095
# if it was produced by iter_changes, it must be
995
1096
# modified *somehow*, either content or execute bit.
996
self.to_file.write("=== modified %s '%s'%s\n" % (kind[0],
997
newpath_encoded, prop_str))
999
self._diff(file_id, oldpath, newpath, kind[0], kind[1])
1097
self.to_file.write(b"=== modified %s '%s'%s\n" % (kind[0].encode('ascii'),
1098
newpath_encoded, prop_str))
1099
if change.changed_content:
1100
self._diff(oldpath, newpath, kind[0], kind[1])
1000
1101
has_changes = 1
1002
1103
has_changes = 1
1003
1104
return has_changes
1005
def diff(self, file_id, old_path, new_path):
1106
def diff(self, old_path, new_path):
1006
1107
"""Perform a diff of a single file
1008
:param file_id: file-id of the file
1009
1109
:param old_path: The path of the file in the old tree
1010
1110
:param new_path: The path of the file in the new tree
1013
old_kind = self.old_tree.kind(file_id)
1014
except (errors.NoSuchId, errors.NoSuchFile):
1112
if old_path is None:
1015
1113
old_kind = None
1017
new_kind = self.new_tree.kind(file_id)
1018
except (errors.NoSuchId, errors.NoSuchFile):
1115
old_kind = self.old_tree.kind(old_path)
1116
if new_path is None:
1019
1117
new_kind = None
1020
self._diff(file_id, old_path, new_path, old_kind, new_kind)
1023
def _diff(self, file_id, old_path, new_path, old_kind, new_kind):
1024
result = DiffPath._diff_many(self.differs, file_id, old_path,
1025
new_path, old_kind, new_kind)
1119
new_kind = self.new_tree.kind(new_path)
1120
self._diff(old_path, new_path, old_kind, new_kind)
1122
def _diff(self, old_path, new_path, old_kind, new_kind):
1123
result = DiffPath._diff_many(
1124
self.differs, old_path, new_path, old_kind, new_kind)
1026
1125
if result is DiffPath.CANNOT_DIFF:
1027
1126
error_path = new_path
1028
1127
if error_path is None: