Savapitech/42sh
1
0
Fork 0
mirror of https://github.com/Savapitech/42sh.git synced 2026-03-18 21:50:35 +01:00
Code Issues Packages Projects Releases Wiki Activity

Change the prompt if the user is root

Browse Source
This commit is contained in:
savalet
2025-05-21 23:30:27 +02:00
parent fce7d767b8
commit 60a65c20d7
1 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/repl/repl.c
View File

@@ -38,10 +38,14 @@ void print_second_shell_prompt(exec_ctx_t *ec)
{ {
env_t *env_ptr = ec->env; env_t *env_ptr = ec->env;
char const *ps1 = get_env_value(env_ptr, "PS1"); char const *ps1 = get_env_value(env_ptr, "PS1");
char const *username = get_env_value(env_ptr, "USER");
if (username == nullptr)
username = "?";
if (ps1 == nullptr) { if (ps1 == nullptr) {
printf(BLUE "└─[" PURPLE "%s$" BLUE "] " RESET, printf(BLUE "└─[" PURPLE "%s%s" BLUE "] " RESET,
ec->history->last_exit_code == 0 ? "" : RED); ec->history->last_exit_code == 0 ? "" : RED,
strcmp(username, "root") == 0 ? "#" : "$");
ec->prompt_len = 6; ec->prompt_len = 6;
} else { } else {
printf("%s", ps1); printf("%s", ps1);
@@ -52,15 +56,20 @@ void print_second_shell_prompt(exec_ctx_t *ec)
static static
void print_prompt(env_t *env_ptr, char *hostname, exec_ctx_t *ec) void print_prompt(env_t *env_ptr, char *hostname, exec_ctx_t *ec)
{ {
char const *username = get_env_value(env_ptr, "USER");
if (username == nullptr)
username = "?";
printf(BLUE PROMPT_HEADER GREEN "%s" RESET "@" CYAN "%s" BLUE "] " printf(BLUE PROMPT_HEADER GREEN "%s" RESET "@" CYAN "%s" BLUE "] "
RESET "-" BLUE " [" RESET "%s" BLUE RESET "-" BLUE " [" RESET "%s" BLUE
"] " RESET "-" BLUE " [" YELLOW "%d" BLUE "] " RESET "-" BLUE " [" YELLOW "%d" BLUE
"]\n└─[" PURPLE "%s$" BLUE "] " RESET, "]\n└─[" PURPLE "%s%s" BLUE "] " RESET,
get_env_value(env_ptr, "USER"), username,
hostname, hostname,
get_env_value(env_ptr, "PWD"), get_env_value(env_ptr, "PWD"),
ec->history_command->sz + 1, ec->history_command->sz + 1,
ec->history->last_exit_code == 0 ? "" : RED); ec->history->last_exit_code == 0 ? "" : RED,
strcmp(username, "root") == 0 ? "#" : "$");
} }
void print_shell_prompt(exec_ctx_t *ec) void print_shell_prompt(exec_ctx_t *ec)