基于文件系统位置的Shell提示符

Question

我必须在根文件系统下的三个主目录中工作，即home/username、project和scratch。我希望我的shell提示显示我所在的这些顶级目录之一。

以下是我的尝试：

top_level_dir ()
{
    if [[ "${PWD}" == *home* ]]
    then
        echo "home";
    elif [[ "${PWD}" == *scratch* ]]
    then
        echo "scratch";
    elif [[ "${PWD}" == *project* ]]
    then
        echo "project";
    fi

}

然后，我将PS1导出为：

export PS1='$(top_level_dir) : '

很遗憾，这并不像我想要的那样工作。当我在我的主目录中时，我的提示会显示home :，但如果我切换到scratch或projects，则提示不会改变。我对bash脚本不是很了解，所以我希望能得到任何帮助来纠正我的代码。

- deepak

---

切换到Scratch/项目后，您是否重新执行了export PS1=...命令？ - Kent

可以运行。你使用的bash版本是什么？ - choroba

@choroba 在我的bash 4.2.42上也起作用了。但在4.1.2上不行。这有道理吗？ - deepak

Answer 1

每次更改工作目录时，您可以钩入cd以更改提示。我经常问自己如何钩入cd，但现在我想我找到了一个解决方案。将以下内容添加到您的~/.bashrc中如何？：

#
# Wrapper function that is called if cd is invoked
# by the current shell
#
function cd {
    # call builtin cd. change to the new directory
    builtin cd $@
    # call a hook function that can use the new working directory
    # to decide what to do
    color_prompt
}

#
# Changes the color of the prompt depending
# on the current working directory
#
function color_prompt {
    pwd=$(pwd)
    if [[ "$pwd/" =~ ^/home/ ]] ; then
        PS1='\[\033[01;32m\]\u@\h:\w\[\033[00m\]\$ '
    elif [[ "$pwd/" =~ ^/etc/ ]] ; then
        PS1='\[\033[01;34m\]\u@\h:\w\[\033[00m\]\$ '
    elif [[ "$pwd/" =~ ^/tmp/ ]] ; then
        PS1='\[\033[01;33m\]\u@\h:\w\[\033[00m\]\$ '
    else
        PS1='\u@\h:\w\\$ '
    fi
    export PS1
}


# checking directory and setting prompt on shell startup
color_prompt

- hek2mgl

---

我已经成功修复了我的代码问题，请查看我对konsolebox答案的评论。不过，我很喜欢你使用钩子改变颜色的解决方案，我会采用它！ - deepak

@deepak 好的。我稍微修改了我的答案。会研究一下你的解决方案。不过，你的问题很好。你曾经想过这是不可能的吗？;) - hek2mgl

Answer 2

请尝试使用此方法，并告诉我们它的效果，例如您的提示符如何在主目录、项目或临时目录以及其他目录中更改。同时告诉我们您看到了哪些错误消息。问题就出在这里。

还要告诉我您是如何运行它的，是通过脚本、直接执行还是通过启动脚本（如~/.bashrc）运行的。

top_level_dir ()
{
    __DIR=$PWD
    case "$__DIR" in
    *home*)
        echo home
        ;;
    *scratch*)
        echo scratch
        ;;
    *project*)
        echo project
        ;;
    *)
        echo "$__DIR"
        ;;
    esac
}

export PS1='$(top_level_dir) : '
export -f top_level_dir

如果它不起作用，请尝试将__DIR=$PWD更改为__DIR=$(pwd)，并告诉我们是否有所帮助。我还想确认您是否真的在运行bash。请注意，像bash、zsh、ksh和dash这样的sh变体很多，安装和使用的默认版本取决于每个系统。要确认您正在使用Bash，请执行echo "$BASH_VERSION"并查看是否显示消息。

您还应确保使用单引号而不是双引号来运行export PS1='$(top_level_dir) : '：export PS1="$(top_level_dir) : "。

- konsolebox

---

你关于检查单引号的观点是正确的。谢谢！我之前使用的是双引号。在更改为单引号后，我的代码成功运行了。现在它可以在4.2和4.1版本中工作。 - deepak