From 53b2bba26ebe461fa7f79221c38b8ce33c9b745d Mon Sep 17 00:00:00 2001
From: Richard Burke <rich.g.burke@gmail.com>
Date: Sun, 5 Jun 2016 21:20:50 +0100
Subject: [PATCH] Show tracking info in tig status

---
 src/status.c                             |  76 +++++++++++++++-
 test/status/on-branch-tracking-info-test | 105 +++++++++++++++++++++++
 2 files changed, 180 insertions(+), 1 deletion(-)
 create mode 100755 test/status/on-branch-tracking-info-test

diff --git a/src/status.c b/src/status.c
index 11e46159e..0112af485 100644
--- a/src/status.c
+++ b/src/status.c
@@ -199,6 +199,71 @@ status_restore(struct view *view)
 	clear_position(&view->prev_pos);
 }
 
+static bool
+status_branch_tracking_info(char *buf, size_t buf_len, const char *head,
+			    const char *remote)
+{
+	if (!string_nformat(buf, buf_len, NULL, "%s...%s",
+			    head, remote)) {
+		return false;
+	}
+
+	const char *tracking_info_argv[] = {
+		"git", "rev-list", "--left-right", buf, NULL
+	};
+
+	struct io io;
+
+	if (!io_run(&io, IO_RD, repo.cdup, NULL, tracking_info_argv)) {
+		return false;
+	}
+
+	struct buffer result = { 0 };
+	int ahead = 0;
+	int behind = 0;
+
+	while (io_get(&io, &result, '\n', true)) {
+		if (result.size > 0 && result.data) {
+			if (result.data[0] == '<') {
+				ahead++;
+			} else if (result.data[0] == '>') {
+				behind++;
+			}
+		}
+	}
+
+	bool io_failed = io_error(&io);
+	io_done(&io);
+
+	if (io_failed) {
+		return false;
+	}
+
+	if (ahead == 0 && behind == 0) {
+		return string_nformat(buf, buf_len, NULL,
+				      "Your branch is up-to-date with '%s'.",
+				      remote);
+	} else if (ahead > 0 && behind > 0) {
+		return string_nformat(buf, buf_len, NULL,
+				      "Your branch and '%s' have diverged, "
+				      "and have %d and %d different commits "
+				      "each, respectively",
+				      remote, ahead, behind);
+	} else if (ahead > 0) {
+		return string_nformat(buf, buf_len, NULL,
+				      "Your branch is ahead of '%s' by "
+				      "%d commit%s.", remote, ahead,
+				      ahead > 1 ? "s" : "");
+	} else if (behind > 0) {
+		return string_nformat(buf, buf_len, NULL,
+				      "Your branch is behind '%s' by "
+				      "%d commit%s.", remote, behind,
+				      behind > 1 ? "s" : "");
+	}
+
+	return false;
+}
+
 static void
 status_update_onbranch(void)
 {
@@ -224,6 +289,7 @@ status_update_onbranch(void)
 	for (i = 0; i < ARRAY_SIZE(paths); i++) {
 		const char *prefix = paths[i][2];
 		const char *head = repo.head;
+		const char *tracking_info = "";
 
 		if (!string_format(buf, "%s/%s", repo.git_dir, paths[i][0]) ||
 		    lstat(buf, &stat) < 0)
@@ -248,9 +314,17 @@ status_update_onbranch(void)
 
 			if (ref && strcmp(ref->name, "HEAD"))
 				head = ref->name;
+		} else if (!paths[i][1] && *repo.remote) {
+			if (status_branch_tracking_info(buf, sizeof(buf),
+							head, repo.remote)) {
+				tracking_info = buf;
+			}
 		}
 
-		if (!string_format(status_onbranch, "%s %s", prefix, head))
+		const char *fmt = *tracking_info == '\0' ? "%s %s" : "%s %s. %s";
+
+		if (!string_format(status_onbranch, fmt,
+				   prefix, head, tracking_info))
 			string_copy(status_onbranch, repo.head);
 		return;
 	}
diff --git a/test/status/on-branch-tracking-info-test b/test/status/on-branch-tracking-info-test
new file mode 100755
index 000000000..3454a6115
--- /dev/null
+++ b/test/status/on-branch-tracking-info-test
@@ -0,0 +1,105 @@
+#!/bin/sh
+
+. libtest.sh
+. libgit.sh
+
+export LINES=10
+
+test_setup_work_dir()
+{
+	create_repo_from_tgz "$base_dir/files/refs-repo.tgz" 
+
+	git checkout -b ahead
+	echo test > test.txt
+	git add test.txt
+	git_commit -m "Ahead branch"
+	git checkout master
+
+	git checkout -b diverged HEAD~2
+	echo test > test.txt
+	git add test.txt
+	git_commit -m "Diverged branch"
+	git checkout master
+}
+
+test_case up-to-date-with-remote-master \
+	--args='status' <<EOF
+On branch master. Your branch is up-to-date with 'origin/master'.               
+Changes to be committed:                                                        
+  (no files)                                                                    
+Changes not staged for commit:                                                  
+  (no files)                                                                    
+Untracked files:                                                                
+  (no files)                                                                    
+                                                                                
+[status] Nothing to update                                                  100%
+EOF
+
+test_case behind-remote-master \
+	--args='status' \
+	--before="git reset --hard HEAD~2" \
+	--after="git reset --hard origin/master" <<EOF
+On branch master. Your branch is behind 'origin/master' by 2 commits.           
+Changes to be committed:                                                        
+  (no files)                                                                    
+Changes not staged for commit:                                                  
+  (no files)                                                                    
+Untracked files:                                                                
+  (no files)                                                                    
+                                                                                
+[status] Nothing to update                                                  100%
+EOF
+
+test_case ahead-of-remote-master \
+	--args='status' \
+	--before="git merge ahead" \
+	--after="git reset --hard origin/master" <<EOF
+On branch master. Your branch is ahead of 'origin/master' by 1 commit.          
+Changes to be committed:                                                        
+  (no files)                                                                    
+Changes not staged for commit:                                                  
+  (no files)                                                                    
+Untracked files:                                                                
+  (no files)                                                                    
+                                                                                
+[status] Nothing to update                                                  100%
+EOF
+
+test_case diverged-from-remote-master \
+	--args='status' \
+	--before="git reset --hard diverged" \
+	--after="git reset --hard origin/master" <<EOF
+On branch master. Your branch and 'origin/master' have diverged, and have 1 and 
+Changes to be committed:                                                        
+  (no files)                                                                    
+Changes not staged for commit:                                                  
+  (no files)                                                                    
+Untracked files:                                                                
+  (no files)                                                                    
+                                                                                
+[status] Nothing to update                                                  100%
+EOF
+
+run_test_cases
+
+add_exec_prefix()
+{
+	code="$(cat "$1")"
+	if [ -n "$code" ]; then
+		echo "$code" | sed -e 's/^[ 	]*//' -e '/^$/d' -e 's/^/:exec @/'
+	fi
+}
+
+tig_script "all" "
+	$(for name in $(cat test-cases); do
+		add_exec_prefix "$name-before"
+		echo :save-display all-$name.screen
+		add_exec_prefix "$name-after"
+	done)
+"
+
+test_tig status
+
+for name in $(cat test-cases); do
+	assert_equals "all-$name.screen" < "$name.expected"
+done