IDEA 通过脚本配置终端提示符样式的方法

function prompt {
    $dir = (Get-Location).Path;
    if ($null -eq $env:WORK) {
        $env:WORK = $dir
    }
    $name = $env:USERNAME.ToLower()
    # 定义路径映射：每个元素是路径表达式和对应简称的元组
    $PathAliases = @(
        @{ Path = $env:USERPROFILE; Alias = 'home' }
        @{ Path = $env:WORK; Alias = '~' }
        @{ Path = [Environment]::GetFolderPath('Desktop'); Alias = 'Desktop' }
        @{ Path = [Environment]::GetFolderPath('MyDocuments'); Alias = 'Documents' }
        @{ 
            # Personal 文件夹且不等于 MyDocuments（避免重复）
            Path = { [Environment]::GetFolderPath('Personal') }; 
            Condition = { $dir -eq $_.Invoke() -and $dir -ne [Environment]::GetFolderPath('MyDocuments') }
            Alias = 'Documents' 
        }
        @{ Path = [Environment]::GetFolderPath('Startup'); Alias = 'Startup' }
        @{ Path = [Environment]::GetFolderPath('StartMenu'); Alias = 'StartMenu' }
        @{ Path = [Environment]::GetFolderPath('CommonStartup'); Alias = 'Startup' }
        @{ Path = [Environment]::GetFolderPath('CommonStartMenu'); Alias = 'StartMenu' }
        @{ Path = [Environment]::GetFolderPath('Favorites'); Alias = 'Favorites' }
        @{ Path = [Environment]::GetFolderPath([Environment+SpecialFolder]::ApplicationData); Alias = 'Roaming' }
        @{ Path = [Environment]::GetFolderPath('LocalApplicationData'); Alias = 'LocalAppData' }
        @{ Path = ${env:ProgramFiles(x86)}; Alias = 'ProgramFiles(x86)' }
        @{ Path = [Environment]::GetFolderPath('Templates'); Alias = 'Templates' }
        @{ Path = [Environment]::GetFolderPath('Recent'); Alias = 'Recent' }
        @{ Path = [Environment]::GetFolderPath('SendTo'); Alias = 'SendTo' }
        @{ Path = [Environment]::GetFolderPath([Environment+SpecialFolder]::NetworkShortcuts); Alias = 'NetHood' }
        @{ Path = [Environment]::GetFolderPath([Environment+SpecialFolder]::PrinterShortcuts); Alias = 'PrintHood' }
        @{ Path = [Environment]::GetFolderPath([Environment+SpecialFolder]::MyMusic); Alias = 'Music' }
        @{ Path = [Environment]::GetFolderPath([Environment+SpecialFolder]::MyPictures); Alias = 'Pictures' }
        @{ Path = [Environment]::GetFolderPath([Environment+SpecialFolder]::MyVideos); Alias = 'Videos' }
        @{ 
            Path = { (New-Object -ComObject Shell.Application).Namespace('shell:Downloads').Self.Path }
            Alias = 'Downloads'
        }
        @{ Path = "$env:USERPROFILE\OneDrive"; Alias = 'OneDrive' }
        @{ Path = "$env:PROGRAMFILES\WindowsApps"; Alias = 'WindowsApps' }
        @{ Path = [Environment]::GetFolderPath('System'); Alias = 'System' }
        @{ Path = [Environment]::GetFolderPath('CommonApplicationData'); Alias = 'ProgramData' }
        @{ Path = [Environment]::GetFolderPath('ProgramFiles'); Alias = 'ProgramFiles' }
        @{ Path = $env:WINDIR; Alias = 'Windows' }
    )
    $flag = 0
    # 遍历数组查找匹配项
    foreach ($item in $PathAliases) {
        $targetPath = if ($item.Path -is [scriptblock]) { $item.Path.Invoke() } else { $item.Path }
        if ($dir.StartsWith($targetPath, [System.StringComparison]::OrdinalIgnoreCase)) {
            if ($item.Condition) {
                if($item.Condition.Invoke()){}
            }
            $relative = $( $suffix = $dir.Substring($targetPath.Length).TrimStart('\'); "/$suffix".TrimEnd('/') )
            $dir = $item.Alias + $relative
            $result = if ($item.Alias -eq '~') { '' } else { '/' }
            $dir = $result + $dir.Replace('\', '/').Replace(':', '/') -replace '/+', '/'
            $flag = 1
            break
        }
    }
    $dir = $dir.ToLower()
    if ($flag -eq 0) {
        # $dir = '/' + $dir.ToLower().Replace('\', '/').Replace(':', '/') -replace '/+', '/'
        $dir = '/' + $dir.Replace('\', '/').Replace(':', '/') -replace '/+', '/'
        $dir = $dir -replace '/$',''
    }
    # ✅ ANSI 控制字符
    $ESC = [char]27
    $GREEN_BOLD = "$ESC[1;32m"   # 加粗 + 绿色
    $BLUE       = "$ESC[34m"     # 蓝色
    $RESET      = "$ESC[0m"      # 重置（关闭所有样式）
    $isAdmin = ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
    $userPart     = "root@$name"
    $pathPart     = $dir
    $promptChar   = "# "
    if (!$isAdmin) {
        $userPart     = "$name@$name"
        $promptChar   = "$ "
    }
    # 组合输出：绿色加粗用户名 + 重置后加冒号 + 蓝色路径 + 重置 + 提示符
    "$GREEN_BOLD$userPart$RESET`:$BLUE$pathPart$RESET$promptChar"
    # "root@$name`:$dir# " 
}
# todo 启动一个终端并应用上面的函数 %SystemRoot%\System32\WindowsPowerShell\v1.0\powershell.exe -NoExit chcp 65001

c:\windows\system32\windowspowershell\v1.0\powershell.exe  -NoExit -Command "chcp 65001; . 'ztp_function_idea'"