fix diffs containing embedded "\r".
authorVadim Gelfer <vadim.gelfer@gmail.com>
Wed, 10 May 2006 10:31:54 -0700
changeset 2248 b914f0557832
parent 2247 546c76e5a3e6
child 2249 3e5fbf001f9b
fix diffs containing embedded "\r". add test to make sure fix stays fixed.
mercurial/mdiff.py
tests/test-diff-newlines
tests/test-diff-newlines.out
--- a/mercurial/mdiff.py	Wed May 10 10:31:22 2006 -0700
+++ b/mercurial/mdiff.py	Wed May 10 10:31:54 2006 -0700
@@ -10,6 +10,20 @@
 demandload(globals(), "re")
 
 
+def splitnewlines(text, keepends=False):
+    '''like str.splitlines, but only split on newlines.'''
+    i = 0
+    lines = []
+    while True:
+        n = text.find('\n', i)
+        if n == -1:
+            last = text[i:]
+            if last:
+                lines.append(last)
+            return lines
+        lines.append(text[i:keepends and n+1 or n])
+        i = n + 1
+
 def unidiff(a, ad, b, bd, fn, r=None, text=False,
             showfunc=False, ignorews=False):
 
@@ -19,7 +33,7 @@
     if not text and (util.binary(a) or util.binary(b)):
         l = ['Binary file %s has changed\n' % fn]
     elif not a:
-        b = b.splitlines(1)
+        b = splitnewlines(b, keepends=True)
         if a is None:
             l1 = "--- %s\t%s\n" % ("/dev/null", epoch)
         else:
@@ -28,7 +42,7 @@
         l3 = "@@ -0,0 +1,%d @@\n" % len(b)
         l = [l1, l2, l3] + ["+" + e for e in b]
     elif not b:
-        a = a.splitlines(1)
+        a = splitnewlines(a, keepends=True)
         l1 = "--- %s\t%s\n" % ("a/" + fn, ad)
         if b is None:
             l2 = "+++ %s\t%s\n" % ("/dev/null", epoch)
@@ -37,8 +51,8 @@
         l3 = "@@ -1,%d +0,0 @@\n" % len(a)
         l = [l1, l2, l3] + ["-" + e for e in a]
     else:
-        al = a.splitlines(1)
-        bl = b.splitlines(1)
+        al = splitnewlines(a, keepends=True)
+        bl = splitnewlines(b, keepends=True)
         l = list(bunidiff(a, b, al, bl, "a/" + fn, "b/" + fn,
                           showfunc=showfunc, ignorews=ignorews))
         if not l: return ""
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test-diff-newlines	Wed May 10 10:31:54 2006 -0700
@@ -0,0 +1,8 @@
+#!/bin/sh
+
+hg init
+python -c 'print "confuse str.splitlines\nembedded\rnewline"' > a
+hg ci -Ama -d '1 0'
+echo clean diff >> a
+hg ci -mb -d '2 0'
+hg diff -r0 -r1
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test-diff-newlines.out	Wed May 10 10:31:54 2006 -0700
@@ -0,0 +1,8 @@
+adding a
+diff -r 107ba6f817b5 -r 310ce7989cdc a
+--- a/a	Thu Jan 01 00:00:01 1970 +0000
++++ b/a	Thu Jan 01 00:00:02 1970 +0000
+@@ -1,2 +1,3 @@ confuse str.splitlines
+ confuse str.splitlines
+ embedded
newline
++clean diff