powershell - 如何在 PowerShell 远程处理 session 的提示中为机器名称添加颜色？

Question

为了让我在远程到实时/生产服务器时更加明显，我认为在使用远程 PowerShell session 时能够为我连接的机器名称着色会很方便。

但是，我看不到这样做的方法......服务器名称前缀似乎独立于 Prompt 函数，即使我可以使用它，我也不确定如何定义一个新的 Prompt 仅用于 session 的持续时间。

有没有办法自定义这个？注意:我不想将所有服务器名称都涂成相同的颜色，我想区分本地/生产服务器。

Answer 1

经过一番搜索，您似乎是正确的，没有内置的钩子(Hook)可以覆盖预先提示 [computername]:标签。

幸运的是，我有一个可以为您工作的 hacky 解决方法!

要获得颜色，我们可以使用 Write-Host . Write-Host prompt 的输出函数将完全左对齐，这就是我们想要的。不幸的是，默认 [computername]:之后直接插入标签。这导致计算机名称在提示中重复，一次有颜色，一次没有。

我们通过返回一个包含退格字符的字符串来解决这个问题，因此未着色的 [computername]:将被覆盖。这是正常的提示字符串，通常是当前路径。

最后，如果正常提示字符串很短并且没有完全覆盖未着色的 [computername]:标记，我们需要通过添加虚拟空格字符来做一些最后的清理。但是，这可能会推出插入符号，因此我们需要添加更多的退格来将插入符号返回到正确的位置。

总而言之，在您的远程机器上使用它:

# put your logic here for getting prompt color by machine name
function GetMachineColor($computerName)
{
   [ConsoleColor]::Green
}

function GetComputerName
{
  # if you want FQDN
  $ipProperties = [System.Net.NetworkInformation.IPGlobalProperties]::GetIPGlobalProperties()
  "{0}.{1}" -f $ipProperties.HostName, $ipProperties.DomainName

  # if you want host name only
  # $env:computername
}

function prompt
{
  $cn = GetComputerName

  # write computer name with color
  Write-Host "[${cn}]: " -Fore (GetMachineColor $cn) -NoNew

  # generate regular prompt you would be showing
  $defaultPrompt = "PS $($executionContext.SessionState.Path.CurrentLocation)$('>' * ($nestedPromptLevel + 1)) "

  # generate backspaces to cover [computername]: pre-prompt printed by powershell
  $backspaces = "`b" * ($cn.Length + 4)

  # compute how much extra, if any, needs to be cleaned up at the end
  $remainingChars = [Math]::Max(($cn.Length + 4) - $defaultPrompt.Length, 0)
  $tail = (" " * $remainingChars) + ("`b" * $remainingChars)

  "${backspaces}${defaultPrompt}${tail}"
}