在上一个命令失败后创建带有红色 $ 的 Bash 命令提示符

Question

我是 Bash 编程新手，正在致力于创建自定义 Bash 命令提示符。我的目标是创建一个提示，仅在登录名和主机名与我通常使用的不同时显示它们。当位于 Git 版本控制下的目录中时，我还希望将当前的 Git 分支附加到命令提示符。

我想将登录名和主机名部分设置为绿色，目录路径设置为蓝色，Git 分支部分设置为粉色，分隔符（: 和 $ 字符）设置为白色。但是，当先前执行的命令返回零以外的任何内容时，我想将 $ 分隔符设置为红色。没有颜色的一般格式如下所示：

loginname@hostname:~/current/path:branchname$

唯一必需的部分是目录路径和 $ 字符。这是我为 .bashrc 文件编写的代码：

MYNAME="gwen"
MYHOST="gwen-laptop"

RED="\[\033[31m\]"
WHITE="\[\033[0m\]"
GREEN="\[\033[01;32m\]"
BLUE="\[\033[01;34m\]"
PINK="\[\033[01;35m\]"

DOLLAR="if [ \$? = 0 ]; then echo ${WHITE}\$; else echo ${RED}\$${NORMAL}; fi"
GITBRN='$(__git_ps1 "\033[0m:\033[01;35m%s")'

USERNM="if [ \u != ${MYNAME} ]; then whoami; fi;"
HOSTNM="if [ \h != ${MYHOST} ]; then echo -n @; hostname; fi;"
COLONM="if [ \u != ${MYNAME} ] || [ \h != ${MYHOST} ]; then echo -n :; fi;"

PS1="${GREEN}\`${USERNM}\`\`${HOSTNM}\`${WHITE}\`${COLONM}\`${BLUE}\w${GITBRN}\`${DOLLAR}\` "

此代码满足我的所有要求，除了它始终将 $ 字符保留为白色，并且不会在适当的时间。 （我怀疑问题是DOLLAR中的“\$？”引用了之前执行的命令，但是DOLLAR是在构建PS1时最后执行的，所以之前的执行语句不再是PS1构建开始之前运行的命令；这是一些东西这是为了创建命令提示符而执行的。）我不知道如何解决这个问题。

这段代码很丑陋，需要重构。我试图将所有颜色代码移到它们自己的变量中，但是当我在 GITBRN 代码中使用这些颜色变量时，事情变得混乱，所以我最终在那里使用了文字颜色。

我花了一整天的时间试图让这段代码正常工作，但我想此时我无处可去。任何关于如何在适当的时间将美元符号变成红色的建议将不胜感激。我也愿意接受有关重构代码以使其更清晰、更具可读性的建议。

PS 我是 Ubuntu Linux (Lucid Lynx) 用户。

Answer 1

使用 PROMPT_COMMAND 变量，根据 bash 手册页，该变量在每个主提示符之前执行。

例如（这还不起作用，我试图让它正常工作，但我认为这是可能的）：

PROMPT_COMMAND="if [ \$? = 0 ]; then DOLLAR="${WHITE}\${NORMAL}"; else DOLLAR="${RED}\${NORMAL}"; fi"

---

编辑：由于在 PS1 内执行命令和非打印字符时遇到挫折 （\[ 和 \] 有时会按字面意思打印出来，而不是用作 PS1 的提示），我来了与此一起（将 ... 替换为无论您在提示中想要什么）：

PROMPT_COMMAND='if [ $? = 0 ]; then DOLLAR_COLOR="\033[0m"; else DOLLAR_COLOR="\033[31m"; fi'
PS1='...\[$(echo -ne $DOLLAR_COLOR)\]$\[\033[m\] '

当然，使用 $() 您可以输入您喜欢在 PS1 中使用其中的部分内容，而不是使用 PROMPT_COMMAND，我只是喜欢这种方式，以便 PROMPT_COMMAND 包含逻辑和 PS1 包含显示命令。

Answer 2

我让它工作：

PROMPT_COMMAND='if [ $? = 0 ]; then PS1="\[\e[32;1m\]\u@\[\e[0m\e[30;47m\]\H\[\e[0m\]:\[\e[34;1m\]\w\[\e[0m\]$ "; else PS1="\[\e[31;1m\]\u@\[\e[0m\e[31;47m\]\H\[\e[0m\]:\[\e[31;1m\]\w\[\e[0m\]$ "; fi'

站在 @jtbandes 的支持上。 @jtbandes 是这个想法的原始作者。

Answer 3

这是 Alexandr 的答案，修改为仅在命令失败时显示红色，而不是在空命令行上按 Enter 时显示红色，正如斯蒂芬所要求的。

trap 'PREVIOUS_COMMAND=$THIS_COMMAND; THIS_COMMAND=$BASH_COMMAND' DEBUG
read -r -d '' PROMPT_COMMAND << 'END'
    if [ $? = 0 -o $? == 130 -o "$PREVIOUS_COMMAND" = ": noop" ]; then
        PS1='\[\e[32;1m\]\u@\[\e[0m\e[30;47m\]\H\[\e[0m\]:\[\e[34;1m\]\w\[\e[0m\]$ '
    else
        PS1='\[\e[31;1m\]\u@\[\e[0m\e[31;47m\]\H\[\e[0m\]:\[\e[31;1m\]\w\[\e[0m\]$ '
    fi
    : noop
END