Usando variáveis ​​para armazenar códigos de cores de terminal para PS1?

No meu .bashrc, uso códigos de cores do terminal ANSI para colorir vários bits. Se parece com isso:

PS1='\u@\h:\w\[\033[33m\]$(virtual_env)\[\033[32m\]$(git_branch)\[\033[0m\]$ '

onde virtual_enve git_branchsão funções bash que produzem coisas no stdout.

Agora, para facilitar a leitura e a modificação, eu gostaria de armazenar os códigos de cores nas variáveis ​​e consultá-las, em vez de incorporá-las diretamente PS1. Então, eu tenho um monte de variáveis ​​como esta:

GREEN="\[\033[32m\]"
YELLOW="\[\033[33m\]"
RESET="\[\033[0m\]"

Eu gostaria de poder escrever algo como:

PS1='\u@\h:\w${YELLOW}$(virtual_env)${GREEN}$(git_branch)${RESET}$ '

Mas isso não funciona - os códigos de cores aparecem no prompt, como se tivessem escapado. As cores funcionarão corretamente se eu usar aspas duplas em vez disso PS1, mas o prompt somente será alterado quando o fizer source ~/.bashrc.

Eu tentei outras coisas que vi pessoas fazerem - usando printf, usando aspas simples para as cores, colocando a variável in \[e \]ao PS1invés da variável de cor, mas nada parece funcionar.

Como posso usar variáveis ​​para os códigos de cores?

A solução é fazer com que o shell substitua as variáveis ​​de cor ao definir o prompt, mas não as funções. Para fazer isso, use aspas duplas como você havia tentado originalmente, mas escape dos comandos para que não sejam avaliados até que o prompt seja desenhado.

PS1="\u@\h:\w${YELLOW}\$(virtual_env)${GREEN}\$(git_branch)${RESET}$ "

Observe o \ antes do $()em cada comando.

Se ecoarmos isso, vemos:

echo "$PS1"
\u@\h:\w\[\033[33m\]$(virtual_env)\[\033[32m\]$(git_branch)\[\033[0m\]$ 

Como você pode ver, as variáveis ​​de cores foram substituídas, mas não os comandos.

O problema é que sua variável GREENcontém a cadeia literal que consiste em "barra invertida barra invertida zero três três" e assim por diante. Ele não contém, por exemplo, um caractere de escape ASCII, conforme necessário, para que o terminal mude de cor.

Você pode colocar caracteres de controle em GREEN(e YELLOWe RESET) manualmente, mas uma opção muito melhor é usar tputem primeiro lugar para que você não precise codificar nada e suporte qualquer tipo de terminal.

GREEN="$(tput setaf 2)"
YELLOW="$(tput setaf 3)"
RESET="$(tput setaf 0)"

A razão pela qual o mundo quando você coloca "barra invertida zero três três" etc ... diretamente PS1é que a interpretação de certas seqüências de barra invertida é um recurso do prompt do bash (consulte a seção PROMPTING no manual. Esta substituição ocorre antes da expansão do parâmetro, comando substituição, expansão aritmética e remoção de cotação, portanto, não é aplicada aos resultados de todas as outras operações.

Mude a maneira como você preenche $ VERDE, $ AMARELO e $ RESET:

GREEN="$(echo -e "\033[32m")"
YELLOW="$(echo -e "\033[33m")"
RESET="$(echo -e "\033[0m")"

PS1='\u@\h:\w${YELLOW}$(virtual_env)${GREEN}$(git_branch)${RESET}$ '