blob: fb616f4739154c5abb5b50d9c0e5644ce3aa7651 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#!/usr/bin/env pwsh
# Save original directory, restore after execution
$_load_original_dir = Get-Location
# Resolve script directory (works with dot-source: . ./load.ps1)
$scriptPath = Split-Path -Parent $MyInvocation.MyCommand.Path
if (-not $scriptPath) {
Write-Error "load.ps1: failed to resolve script directory"
return
}
# Load completion script mling_comp.ps1 from the .comp subdirectory
$compScript = Join-Path -Path $scriptPath -ChildPath ".comp" | Join-Path -ChildPath "mling_comp.ps1"
if (Test-Path $compScript) {
. $compScript
}
# Change to script directory
try {
Set-Location $scriptPath -ErrorAction Stop
} catch {
Write-Error "load.ps1: failed to cd to script directory"
return
}
# Add bin directories from all namespaces to PATH
$allDirs = Get-ChildItem -Directory
Get-ChildItem -Directory | Where-Object { Test-Path (Join-Path -Path $_.FullName -ChildPath "bin") } | ForEach-Object {
$binPath = Join-Path -Path $_.FullName -ChildPath "bin"
$env:PATH = "$binPath;$env:PATH"
}
# Helper function: execute script with appropriate shell
function _load_script {
param([string]$script)
if ($script -like "*.ps1") {
& $script 2>$null
}
}
# Iterate over all namespaces (top-level directories except .comp)
$nsDirs = Get-ChildItem -Directory -Exclude ".comp"
foreach ($_dir in $nsDirs) {
$ns = $_dir.Name
# Skip if UNTRUSTED marker exists
$untrustedMarker = Join-Path -Path $ns -ChildPath "UNTRUSTED"
if (Test-Path $untrustedMarker) { continue }
$compDir = Join-Path -Path $ns -ChildPath "comp"
if (-not (Test-Path $compDir -PathType Container)) { continue }
# Find all loadable scripts under comp
$scripts = Get-ChildItem -Path $compDir -Filter "*.ps1" -File -ErrorAction SilentlyContinue
if (-not $scripts) { continue }
$count = ($scripts | Measure-Object).Count
# If TRUSTED marker exists, load directly
$trustedMarker = Join-Path -Path $ns -ChildPath "TRUSTED"
if (Test-Path $trustedMarker) {
foreach ($_script in $scripts) { _load_script $_script.FullName }
continue
}
# No marker, ask user
$answer = Read-Host "'$ns' has $count completion script(s) to load, do you trust it? [Y/n] "
if ($answer -eq "" -or $answer -match "^(y|yes)$") {
# Mark as TRUSTED
New-Item -ItemType File -Path $trustedMarker -Force | Out-Null
# Ask whether to load immediately
$load_answer = Read-Host "Load it immediately? [Y/n] "
if ($load_answer -eq "" -or $load_answer -match "^(y|yes)$") {
foreach ($_script in $scripts) { _load_script $_script.FullName }
}
} else {
New-Item -ItemType File -Path $untrustedMarker -Force | Out-Null
}
}
# Restore original working directory
try {
Set-Location $_load_original_dir -ErrorAction Stop
} catch {}
# Cleanup
Remove-Variable -Name _load_original_dir -ErrorAction SilentlyContinue
Remove-Item Function:_load_script -ErrorAction SilentlyContinue
|
