优化Powershell的启动时间

1
$baseDir = Split-Path -Parent $PROFILE
2
$locksDir = Join-Path $baseDir 'Locks'
3
$completionsDir = Join-Path $baseDir 'Completions'
4

5
New-Item -ItemType Directory -Force -Path $locksDir, $completionsDir | Out-Null
6

113 collapsed lines
7
function Import-CompletionScript {
8
    param(
9
        [Parameter(Mandatory)]
10
        [string]$OutputFileName
11
    )
12

13
    $target = [System.IO.Path]::Combine($completionsDir, $OutputFileName)
14
    if (-not [System.IO.File]::Exists($target)) {
15
        return $false
16
    }
17

18
    . $target
19
    return $true
20
}
21

22
Import-CompletionScript 'starship.ps1' | Out-Null
23
Import-CompletionScript 'zoxide.ps1'   | Out-Null
24

25
Register-EngineEvent -SourceIdentifier PowerShell.OnIdle -Action {
26
    Import-CompletionScript 'pixi.ps1'     | Out-Null
27
    Import-CompletionScript 'uv.ps1'       | Out-Null
28
    Import-CompletionScript 'uvx.ps1'      | Out-Null
29
    Import-CompletionScript 'skim.ps1'     | Out-Null
30
    Import-CompletionScript 'rustup.ps1'   | Out-Null
31
    Import-CompletionScript "$env:HOMEDRIVE$env:HOMEPATH\xmake\scripts\profile-win.ps1" | Out-Null
32
} -MaxTriggerCount 1 -SupportEvent | Out-Null
33

34
Register-EngineEvent -SourceIdentifier PowerShell.OnIdle -Action {
35
    # 新时代的补全方法
36
    # 不生成大量 shell scripts，而是动态向应用程序询问
37
    # 但是目前还没有普及，只能说未来可期
38
    Register-ArgumentCompleter -Native -CommandName winget -ScriptBlock {
39
        param($wordToComplete, $commandAst, $cursorPosition)
40

41
        [Console]::InputEncoding = [System.Text.UTF8Encoding]::new()
42
        [Console]::OutputEncoding = [System.Text.UTF8Encoding]::new()
43
        $OutputEncoding = [System.Text.UTF8Encoding]::new()
44

45
        $word = $wordToComplete.Replace('"', '""')
46
        $ast = $commandAst.ToString().Replace('"', '""')
47

48
        winget complete --word="$word" --commandline "$ast" --position $cursorPosition |
49
        ForEach-Object {
50
            [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
51
        }
52
    }
53
} -MaxTriggerCount 1 -SupportEvent | Out-Null
54

55
# update scripts
56
Register-EngineEvent -SourceIdentifier PowerShell.OnIdle -Action {
57
    $baseDir = Split-Path -Parent $PROFILE
58

59
    Start-ThreadJob -Name 'completion-cache-update' -ArgumentList $baseDir -ScriptBlock {
60
        param($baseDir)
61

62
        $ErrorActionPreference = 'Stop'
63

64
        $locksDir = Join-Path $baseDir 'Locks'
65
        $completionsDir = Join-Path $baseDir 'Completions'
66
        New-Item -ItemType Directory -Force -Path $locksDir, $completionsDir | Out-Null
67

68
        function Update-CompletionScriptFile {
69
            param(
70
                [Parameter(Mandatory)][string]$CheckCommand,
71
                [Parameter(Mandatory)][string[]]$CheckArgs,
72
                [Parameter(Mandatory)][string]$GenerateCommand,
73
                [Parameter(Mandatory)][string[]]$GenerateArgs,
74
                [Parameter(Mandatory)][string]$OutputFileName
75
            )
76

77
            $target = Join-Path $completionsDir $OutputFileName
78
            $lockFile = Join-Path $locksDir ($OutputFileName + '.version')
79

80
            $current = (& $CheckCommand @CheckArgs 2>&1 | Out-String).Trim()
81
            if (-not $current) { throw "Empty version output for $OutputFileName" }
82

83
            $needsUpdate = $true
84
            if ((Test-Path $lockFile -PathType Leaf) -and (Test-Path $target -PathType Leaf)) {
85
                $prev = (Get-Content $lockFile -Raw).Trim()
86
                if ($prev -eq $current) {
87
                    $needsUpdate = $false
88
                }
89
            }
90

91
            if (-not $needsUpdate) { return }
92

93
            $body = (& $GenerateCommand @GenerateArgs 2>&1 | Out-String)
94
            if (-not $body.Trim()) { throw "Empty body for $OutputFileName" }
95

96
            $header = @(
97
                "# generated: $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')"
98
                ""
99
            ) -join [Environment]::NewLine
100

101
            $tmpTarget = "$target.tmp"
102
            $tmpLock = "$lockFile.tmp"
103

104
            Set-Content -Path $tmpTarget -Value ($header + $body) -Encoding utf8
105
            Set-Content -Path $tmpLock   -Value $current           -Encoding utf8
106

107
            Move-Item -Path $tmpTarget -Destination $target   -Force
108
            Move-Item -Path $tmpLock   -Destination $lockFile -Force
109
        }
110

111
        Update-CompletionScriptFile 'starship' @('--version') 'starship' @('init', 'powershell', '--print-full-init') 'starship.ps1'
112
        Update-CompletionScriptFile 'zoxide'   @('--version') 'zoxide'   @('init', 'powershell')                     'zoxide.ps1'
113
        Update-CompletionScriptFile 'pixi'     @('--version') 'pixi'     @('completion', '--shell', 'powershell')   'pixi.ps1'
114
        Update-CompletionScriptFile 'rustup'   @('--version') 'rustup'   @('completions', 'powershell')             'rustup.ps1'
115
        Update-CompletionScriptFile 'uv'       @('--version') 'uv'       @('generate-shell-completion', 'powershell') 'uv.ps1'
116
        Update-CompletionScriptFile 'uvx'      @('--version') 'uvx'      @('--generate-shell-completion', 'powershell') 'uvx.ps1'
117
        Update-CompletionScriptFile 'sk'       @('-V')        'sk'       @('--shell', 'power-shell')                'skim.ps1'
118
    } | Out-Null
119
} -MaxTriggerCount 1 -SupportEvent | Out-Null