1
@echo off
2
:: ==================== 配置区 ====================
3
@set SHOW_VERSION=0   :: 0=不显示 Windows 版本，1=显示
4
@set SHOW_GIT=1       :: 0=不显示 Git 信息，1=显示（分支+状态）
5
@set NEW_LINE=1        :: 0=单行提示符，1=路径后换行
6
:: =============================================
7

8
:: ---------- 1. SHOW_VERSION ----------
9
if %SHOW_VERSION%==1 (
10
    @ver
11
    @echo (c) Microsoft Corporation。保留所有权利。
12
)
13

14
:: ---------- 2. NEW_LINE ----------
15
if %NEW_LINE%==1 (
16
    @set "NL=$_$ - "
17
) else (
18
    @set "NL= "
19
)
20

21
:: ---------- 3. 取得 ESC 字符 ----------
22
@for /f "delims=#" %%i in ('prompt #$E#^&echo on^&for %%a in ^(1^) do rem') do @set "ESC=%%i"
23

24
:: ---------- 4. Git 信息（仅当 SHOW_GIT=1 时计算） ----------
25
@set "GIT_INFO="
26
if %SHOW_GIT%==1 (
27
    git rev-parse --is-inside-work-tree >nul 2>&1 || goto :git_end
28
    :: 当前分支
29
    for /f "delims=" %%B in ('git branch --show-current 2^>nul') do set "BRANCH=%%B"
30
    if not defined BRANCH (
31
        for /f "delims=" %%H in ('git symbolic-ref --short HEAD 2^>nul') do set "BRANCH=%%H"
32
    )
33
    if not defined BRANCH (
34
        for /f "delims=" %%S in ('git rev-parse --short HEAD 2^>nul') do set "BRANCH=detached@%%S"
35
    )
36
    :: 状态指示符
37
    set "ST="
38
    for /f "delims=" %%L in ('git status --porcelain 2^>nul') do set "PORC=%%L"& call :parse_porcelain
39
    set "GIT_INFO= (%BRANCH%%ST%)"
40
)
41
:git_end
42
@goto :prompt_set
43
:parse_porcelain
44
@rem ?? → 未跟踪
45
@echo %PORC%|findstr /b /c:"??" >nul && if not defined ST set "ST=%"
46
@rem 第二列 M/D → 已修改
47
@echo %PORC%|findstr /r /c:"^.M" /c:"^.D" >nul && set "ST=%ST%*"
48
@rem 第一列 M/A/D/R → 已暂存
49
@echo %PORC%|findstr /r /c:"^M." /c:"^A." /c:"^D." /c:"^R." >nul && set "ST=%ST%+"
50
@goto :eof
51

52
:: ---------- 5. 动态生成 Prompt ----------
53
:prompt_set
54
@set "GIT_PART="
55
@if %SHOW_GIT%==1 @set "GIT_PART=%ESC%[38;2;78;154;6m%GIT_INFO%%ESC%[m"
56

57
@set "PROMPT_LINE=%ESC%[38;2;88;110;117m[$T%ESC%[m %ESC%[38;2;101;123;131m%USERNAME%%ESC%[m%ESC%[38;2;108;113;196m@%COMPUTERNAME%%ESC%[m %ESC%[38;2;203;75;22m$P%ESC%[m%GIT_PART%%NL%$%ESC%[m$"
58

59
@prompt %PROMPT_LINE%

1
# 用法: . .\Set-Prompt.ps1 -ShowGit $true -NewLine $true
2

3
param(
4
    [bool]$ShowGit = $true,
5
    [bool]$NewLine = $true,
6
    [bool]$ShowVersion = $false
7
)
8

9
# ==================== Git 状态函数 ====================
10
function Get-GitStatus {
11
    if (-not (Get-Command git -ErrorAction SilentlyContinue)) { return "" }
12

13
    $gitDir = git rev-parse --git-dir 2>$null
14
    if (-not $gitDir) { return "" }
15

16
    # 分支名
17
    $branch = git branch --show-current 2>$null
18
    if (-not $branch) {
19
        $head = git symbolic-ref --short HEAD 2>$null
20
        if ($head) { $branch = $head }
21
        else {
22
            $sha = git rev-parse --short HEAD 2>$null
23
            $branch = "detached@$sha"
24
        }
25
    }
26

27
    # 状态指示符
28
    $status = ""
29
    $porcelain = git status --porcelain 2>$null
30
    if ($porcelain) {
31
        if ($porcelain -match '^\?\?') { $status += "%" }
32
        if ($porcelain -match '^.[MD]') { $status += "*" }
33
        if ($porcelain -match '^[MADR]') { $status += "+" }
34
    }
35

36
    # 前后提交（可选）
37
    $branchInfo = git status --porcelain --branch 2>$null
38
    if ($branchInfo -match '\[ahead (\d+)\]') { $status += ">$($Matches[1])" }
39
    if ($branchInfo -match '\[behind (\d+)\]') { $status += "<$($Matches[1])" }
40

41
    if ($branch) {
42
        return " ($branch$status)"
43
    }
44
    return ""
45
}
46

47
# ==================== 颜色定义 ====================
48
$colors = @{
49
    time     = "38;2;88;110;117"
50
    user     = "38;2;101;123;131"
51
    at       = "38;2;108;113;196"
52
    path     = "38;2;203;75;22"
53
    git      = "38;2;78;154;6"
54
    reset    = "0"
55
}
56

57
function Write-Colored {
58
    param([string]$text, [string]$color)
59
    Write-Host $text -NoNewline -ForegroundColor $color
60
}
61

62
# ==================== 主提示符函数 ====================
63
function global:prompt {
64
    # 1. chroot 前缀
65
    if ($env:DEBIAN_CHROOT) {
66
        Write-Host "($env:DEBIAN_CHROOT) " -NoNewline -ForegroundColor Cyan
67
    }
68

69
    # 2. 时间
70
    $time = (Get-Date -Format "yyyy-MM-dd HH:mm:ss")
71
    Write-Host "[$time" -NoNewline
72
    Write-Colored " " $colors.time
73
    Write-Host "]" -NoNewline
74

75
    # 3. 用户@主机
76
    Write-Colored " $($env:USERNAME)" $colors.user
77
    Write-Colored "@$($env:COMPUTERNAME)" $colors.at
78

79
    # 4. 路径
80
    $path = (Get-Location).Path
81
    Write-Colored " $path" $colors.path
82

83
    # 5. Git 信息
84
    if ($ShowGit) {
85
        $git = Get-GitStatus
86
        if ($git) {
87
            Write-Colored $git $colors.git
88
        }
89
    }
90

91
    # 6. 换行或空格
92
    if ($NewLine) {
93
        Write-Host ""
94
        Write-Host "$" -NoNewline -ForegroundColor Gray
95
    } else {
96
        Write-Host " $" -NoNewline -ForegroundColor Gray
97
    }
98

99
    return " "
100
}
101

102
# ==================== 可选：显示 PowerShell 版本 ====================
103
if ($ShowVersion) {
104
    "PowerShell $($PSVersionTable.PSVersion) on $($PSVersionTable.OS)"
105
}
106

107
# 自动加载提示符
108
Write-Host "PowerShell 提示符已加载 (Git=$ShowGit, NewLine=$NewLine)" -ForegroundColor Green