From 8fe1d622b5b46dee51db6581282ec1fba76782b1 Mon Sep 17 00:00:00 2001
From: Ori Bernstein <ori@eigenstate.org>
Date: Mon, 21 Sep 2020 09:40:42 -0700
Subject: diff: support unified diff via -u

The format produced by `diff -u` is inferior to that
produced by `diff -c`, but it's what ape/patch and
unix patch expect, so it's useful to generate it.

This patch adds `diff -u`.
---
 sys/src/cmd/diff/diff.h    |  1 +
 sys/src/cmd/diff/diffio.c  | 34 ++++++++++++++++++++++++----------
 sys/src/cmd/diff/diffreg.c |  1 +
 sys/src/cmd/diff/main.c    |  1 +
 4 files changed, 27 insertions(+), 10 deletions(-)

(limited to 'sys/src/cmd/diff')

diff --git a/sys/src/cmd/diff/diff.h b/sys/src/cmd/diff/diff.h
index 76fe050df..34a5dc010 100644
--- a/sys/src/cmd/diff/diff.h
+++ b/sys/src/cmd/diff/diff.h
@@ -22,5 +22,6 @@ Biobuf *prepare(int, char *);
 void panic(int, char *, ...);
 void check(Biobuf *, Biobuf *);
 void change(int, int, int, int);
+void fileheader(void);
 void flushchanges(void);
 
diff --git a/sys/src/cmd/diff/diffio.c b/sys/src/cmd/diff/diffio.c
index 640f2d2ba..d937b1fef 100644
--- a/sys/src/cmd/diff/diffio.c
+++ b/sys/src/cmd/diff/diffio.c
@@ -303,6 +303,7 @@ change(int a, int b, int c, int d)
 		break;
 	case 'c':
 	case 'a':
+	case 'u':
 		if(nchanges%1024 == 0)
 			changes = erealloc(changes, (nchanges+1024)*sizeof(changes[0]));
 		ch = &changes[nchanges++];
@@ -338,6 +339,15 @@ changeset(int i)
 	return nchanges;
 }
 
+void
+fileheader(void)
+{
+	if(mode != 'u')
+		return;
+	Bprint(&stdout, "--- %s\n", file1);
+	Bprint(&stdout, "+++ %s\n", file2);
+}
+
 void
 flushchanges(void)
 {
@@ -368,20 +378,24 @@ flushchanges(void)
 			d = len[1];
 			j = nchanges;
 		}
-		Bprint(&stdout, "%s:", file1);
-		range(a, b, ",");
-		Bprint(&stdout, " - ");
-		Bprint(&stdout, "%s:", file2);
-		range(c, d, ",");
-		Bputc(&stdout, '\n');
+		if(mode == 'u'){
+			Bprint(&stdout, "@@ -%d,%d +%d,%d @@\n", a, b-a+1, c, d-c+1);
+		}else{
+			Bprint(&stdout, "%s:", file1);
+			range(a, b, ",");
+			Bprint(&stdout, " - ");
+			Bprint(&stdout, "%s:", file2);
+			range(c, d, ",");
+			Bputc(&stdout, '\n');
+		}
 		at = a;
 		for(; i<j; i++){
-			fetch(ixold, at, changes[i].a-1, input[0], "  ");
-			fetch(ixold, changes[i].a, changes[i].b, input[0], "- ");
-			fetch(ixnew, changes[i].c, changes[i].d, input[1], "+ ");
+			fetch(ixold, at, changes[i].a-1, input[0], mode == 'u' ? " " : "  ");
+			fetch(ixold, changes[i].a, changes[i].b, input[0], mode == 'u' ? "-" : "- ");
+			fetch(ixnew, changes[i].c, changes[i].d, input[1], mode == 'u' ? "+" : "- ");
 			at = changes[i].b+1;
 		}
-		fetch(ixold, at, b, input[0], "  ");
+		fetch(ixold, at, b, input[0], mode == 'u' ? " " : "  ");
 	}
 	nchanges = 0;
 }
diff --git a/sys/src/cmd/diff/diffreg.c b/sys/src/cmd/diff/diffreg.c
index 7ef855994..6f6f144fc 100644
--- a/sys/src/cmd/diff/diffreg.c
+++ b/sys/src/cmd/diff/diffreg.c
@@ -285,6 +285,7 @@ output(void)
 	m = len[0];
 	J[0] = 0;
 	J[m+1] = len[1]+1;
+	fileheader();
 	if (mode != 'e') {
 		for (i0 = 1; i0 <= m; i0 = i1+1) {
 			while (i0 <= m && J[i0] == J[i0-1]+1)
diff --git a/sys/src/cmd/diff/main.c b/sys/src/cmd/diff/main.c
index 054f4c878..a25168cec 100644
--- a/sys/src/cmd/diff/main.c
+++ b/sys/src/cmd/diff/main.c
@@ -192,6 +192,7 @@ main(int argc, char *argv[])
 			case 'n':
 			case 'c':
 			case 'a':
+			case 'u':
 				mode = *p;
 				break;
 
-- 
cgit v1.2.3