diff --git a/main.c b/main.c index 6d7893f..f08ebf3 100644 --- a/main.c +++ b/main.c @@ -98,7 +98,12 @@ int main() { char host[1024]; gethostname(host, sizeof(host)); char prompt[2051]; - snprintf(prompt, sizeof(prompt), "%s@%s %s$ ", user, host, dir); +char *home = getenv("HOME"); +if(strncmp(dir, home, strlen(home)) == 0) { + snprintf(prompt, sizeof(prompt), "%s@%s ~%s$ ", user, host, dir + strlen(home)); +} else { + snprintf(prompt, sizeof(prompt), "%s@%s %s$ ", user, host, dir); +} char *line = readline(prompt); if(line == NULL) return 0; if(line && *line) add_history(line);