From e249533d99fa06f6c1c6dc35e33565576c4a0e6f Mon Sep 17 00:00:00 2001
From: Jonas 'Sortie' Termansen <sortie@maxsi.org>
Date: Tue, 16 Apr 2013 10:30:38 +0200
Subject: [PATCH] Add home directory syntax to shell.

---
 utils/mxsh.cpp | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/utils/mxsh.cpp b/utils/mxsh.cpp
index d33d5e95..739bc69f 100644
--- a/utils/mxsh.cpp
+++ b/utils/mxsh.cpp
@@ -136,7 +136,7 @@ readcmd:
 	if ( strcmp(argv[0], "cd") == 0 )
 	{
 		internal = true;
-		const char* newdir = "/";
+		const char* newdir = getenv_safe("HOME", "/root");
 		if ( argv[1] ) { newdir = argv[1]; }
 		if ( chdir(newdir) )
 		{
@@ -282,8 +282,21 @@ int get_and_run_command(FILE* fp, const char* fpname, bool interactive,
 		const char* print_username = getenv_safe("USERNAME", "root");
 		const char* print_hostname = getenv_safe("HOSTNAME", "sortix");
 		const char* print_dir = getenv_safe("PWD", "?");
-		printf("\e[32m%s@%s \e[36m%s #\e[37m ", print_username, print_hostname,
-		       print_dir);
+		const char* home_dir = getenv_safe("HOME", "/root");
+		size_t home_dir_len = strlen(home_dir);
+		printf("\e[32m");
+		printf("%s", print_username);
+		printf("@");
+		printf("%s", print_hostname);
+		printf(" ");
+		printf("\e[36m");
+		if ( home_dir_len && strncmp(print_dir, home_dir, home_dir_len) == 0 )
+			printf("~%s", print_dir + home_dir_len);
+		else
+			printf("%s", print_dir);
+		printf(" ");
+		printf("#");
+		printf("\e[37m ");
 		fflush(stdout);
 	}