自定义PowerShell控制台提示符风格的方法

powershell控制台的默认命令行提示为ps+（当前路径）+箭头，形如：

ps c:\users\非苔>

前面加ps是powershell的标志，为了和传统的cmd控制台区别开来。后面加当前路径和箭头是为了和一般的命令行保持一致。这样的设计本来无可厚非，但是如果我在写博客时，需要执行某个命令然后连同结果一起复制出来，麻烦来了。比如我要演示get-date，打开控制台，敲入命令：

ps c:\users\非苔> get-date

2014年4月16日 14:48:47

结果正确，但是看起来不爽，这个路径是多余的，它会分散读者的注意力。有人说，你可以定位到c盘根目录下：

ps c:\> get-date

2014年4月16日 14:52:56

这样把路径弱化了，好多了，但是仍旧不爽。“c:\”还是多余的，我是不是有点疑似强迫症啊，如果改成这样呢？

ps> get-date

2014年4月16日 14:57:07

这样好，既能体现powershell的风格，又无冗余信息。甚至还能继续简化，改成：

>get-date

2014年4月16日 14:59:21

 get-date

2014年4月16日 14:59:48

以上讨论的关键其实就是控制台提示风格，下面讨论如何自定义。

知识点

控制台的提示风格由自动化函数prompt控制。函数的定义默认为：

"ps $($executioncontext.sessionstate.path.currentlocation)$('>' * ($nestedpromptlevel + 1)) "
# .link
# http://go.microsoft.com/fwlink/?linkid=225750
# .externalhelp system.management.automation.dll-help.xml

我们所要做的就是覆盖这个函数。

1.最好的更改应当是：提示支持多种风格，可以来回切换。既然要切换，就得有个变量来存储状态。这个变量最好是全局的，并且是只读的。
2.对于powershell内置的自动化函数和自动化变量的更改只有在当前控制台会话有效，所以最好将更改存放在profile配置文件中。

效果图：

实现脚本：

function prompt()
{
    switch ($pspromptmode)
    {
        'cmd'
        {
          "$($executioncontext.sessionstate.path.currentlocation)$('>' * ($nestedpromptlevel + 1)) "
        }

        'arrow'
        {
          '> '
        }

        'none'
        {
          ' '
        }
        'simple'
        {
         'ps> '
        }
        default
        {
         "ps $($executioncontext.sessionstate.path.currentlocation)$('>' * ($nestedpromptlevel + 1)) "
        }
    }
}

<#
.synopsis
   设置控制台提示风格
.description
   设置控制台提示风格，支持五种风格：normal，cmd，arrow，simple，none
#>
function set-prompt
{
    param(
    [parameter(mandatory=$true)]
    [validateset('normal','cmd','arrow','simple', 'none',ignorecase = $true)]
    $mode
    )
    $varpspromptmode = (get-variable 'pspromptmode'  -ea silentlycontinue)
    #配置变量不存在则新建
    if ( $varpspromptmode -eq $null)
    {
        new-variable -name 'pspromptmode' -value $mode -scope 'global'
        $varpspromptmode = get-variable -name 'pspromptmode'
        $varpspromptmode.description = '提示函数配置变量'

        #限制配置变量的取值集合
        $varpspromptmodeatt = new-object system.management.automation.validatesetattribute('normal','cmd','arrow','simple','none')
        $varpspromptmode.attributes.add($varpspromptmodeatt)

        #限制配置变量为只读并且可以贯穿所有作用域ls
        $varpspromptmode.options = 'readonly, allscope'

    }
    #更新配置
    #只读变量可以通过-force选项更新值
    set-variable -name pspromptmode -value $mode -force
}