vibe-usage-cli 0.0.0-stage → 0.1.1

package/build/launchers/discover-antigravity.ps1

@@ -0,0 +1,68 @@
+# Internal transport only. Stdout includes a CSRF capability and must be consumed
+# in memory by the main process. Do not invoke this script in a visible console.
+# Windows PowerShell 5 compatible; no application or network request is started.
+[CmdletBinding()]
+param()
+
+$ErrorActionPreference = 'Stop'
+$ProgressPreference = 'SilentlyContinue'
+[Console]::OutputEncoding = New-Object System.Text.UTF8Encoding($false)
+$result = [ordered]@{ outcome = 'app_closed'; servers = @() }
+
+try {
+    $currentIdentity = [Security.Principal.WindowsIdentity]::GetCurrent()
+    $currentSid = $currentIdentity.User.Value
+    if ([string]::IsNullOrWhiteSpace($currentSid)) { throw 'identity-unavailable' }
+    $programs = Join-Path $env:LOCALAPPDATA 'Programs'
+    $appServer = Join-Path $programs 'Antigravity\resources\bin\language_server.exe'
+    $ideRoot = (Join-Path $programs 'Antigravity IDE\resources\app\extensions\antigravity\bin') + '\'
+    $processes = @(Get-CimInstance Win32_Process -Filter "Name LIKE 'language_server%' OR Name LIKE 'language-server%'")
+    $servers = New-Object System.Collections.Generic.List[object]
+    $candidateFound = $false
+    foreach ($entry in $processes) {
+        if ($servers.Count -ge 4) { break }
+        $executable = [string]$entry.ExecutablePath
+        if ([string]::IsNullOrWhiteSpace($executable)) { continue }
+        $isApp = [string]::Equals($executable, $appServer, [StringComparison]::OrdinalIgnoreCase)
+        $isIDE = $executable.StartsWith($ideRoot, [StringComparison]::OrdinalIgnoreCase) -and
+            [IO.Path]::GetFileName($executable) -match '^language[-_]server(?:[-_][a-zA-Z0-9]+)*\.exe$'
+        if (-not $isApp -and -not $isIDE) { continue }
+        # A matching installation path alone does not establish ownership.
+        # Never inspect another Windows user's CSRF capability or sockets.
+        $owner = Invoke-CimMethod -InputObject $entry -MethodName GetOwnerSid -ErrorAction SilentlyContinue
+        if ($null -eq $owner -or $owner.ReturnValue -ne 0 -or
+            -not [string]::Equals([string]$owner.Sid, $currentSid, [StringComparison]::Ordinal)) { continue }
+        $candidateFound = $true
+        $match = [regex]::Match([string]$entry.CommandLine, '(?:^|\s)--csrf_token(?:=|\s+)(?:"([^"\r\n]+)"|([^\s]+))')
+        if (-not $match.Success) { continue }
+        $csrfToken = if ($match.Groups[1].Success) { $match.Groups[1].Value } else { $match.Groups[2].Value }
+        if ($csrfToken.Length -gt 512 -or $csrfToken -match '[\x00-\x20\x7f]') { continue }
+        $ports = @(Get-NetTCPConnection -State Listen -OwningProcess $entry.ProcessId -ErrorAction SilentlyContinue |
+            Where-Object { $_.LocalAddress -in @('127.0.0.1', '0.0.0.0') } |
+            Select-Object -ExpandProperty LocalPort -Unique | Sort-Object | Select-Object -First 4)
+        if ($ports.Count -eq 0) { continue }
+        # Recheck identity after querying sockets so a recycled PID cannot inherit
+        # this server's capability. Only this exact process's listening ports pass.
+        $current = Get-CimInstance Win32_Process -Filter ('ProcessId = ' + [int]$entry.ProcessId)
+        if ($null -eq $current -or $current.CreationDate -ne $entry.CreationDate -or
+            -not [string]::Equals([string]$current.ExecutablePath, $executable, [StringComparison]::OrdinalIgnoreCase)) { continue }
+        $servers.Add([pscustomobject]@{ pid = [int]$entry.ProcessId; csrfToken = $csrfToken; ports = @($ports) })
+    }
+    $result.servers = @($servers.ToArray())
+    if ($servers.Count -gt 0) { $result.outcome = 'success' }
+    elseif ($candidateFound) { $result.outcome = 'local_service_unavailable' }
+} catch {
+    $result.outcome = 'local_service_unavailable'
+    $result.servers = @()
+} finally {
+    $csrfToken = $null
+    $match = $null
+    $entry = $null
+    $current = $null
+    $processes = $null
+    $owner = $null
+    $currentSid = $null
+    if ($null -ne $currentIdentity) { $currentIdentity.Dispose() }
+    $currentIdentity = $null
+}
+[Console]::WriteLine(($result | ConvertTo-Json -Depth 5 -Compress))

package/build/launchers/terminal.ps1

@@ -0,0 +1,78 @@
+# Runs one command-line tool in this (new, visible) console window.
+# Arguments arrive as base64-encoded, NUL-separated UTF-8 so that no shell ever re-parses
+# them (NUL cannot occur inside a process argument). The environment is prepared by the
+# caller. Windows PowerShell 5 compatible.
+#
+# With -Handoff, the tab stays with the session: when the tool exits and the app has written
+# the next command to that file (the session moved to another account), it runs here too.
+param(
+    [Parameter(Mandatory = $true)][string]$WorkingDirectory,
+    [Parameter(Mandatory = $true)][string]$Executable,
+    [string]$Title,
+    [string]$ArgumentsBase64,
+    [string]$Handoff,
+    [switch]$NewWindow
+)
+$ErrorActionPreference = 'Stop'
+# Each argument quoted the way CommandLineToArgvW reads it back: backslashes doubled before a
+# quote and at the end, the quote escaped. Windows PowerShell does not do this itself (it joins
+# -ArgumentList with spaces, and runs `& tool @args` without escaping quotes inside an
+# argument, which would let a prompt's quotes split it into options), so every command line
+# this script starts is built here.
+function ConvertTo-ArgumentList([string[]]$Values) {
+    @($Values | ForEach-Object { '"' + (($_ -replace '(\\*)"', '$1$1\"') -replace '(\\+)$', '$1$1') + '"' })
+}
+# The app starts this hidden, without a console a tool could use; Start-Process opens this
+# script again in a new, visible console window.
+if ($NewWindow) {
+    $relaunch = @('-NoLogo', '-NoProfile', '-NoExit', '-ExecutionPolicy', 'Bypass', '-File', $PSCommandPath)
+    foreach ($name in 'WorkingDirectory', 'Executable', 'Title', 'ArgumentsBase64', 'Handoff') {
+        $value = Get-Variable -Name $name -ValueOnly
+        if ($value) { $relaunch += "-$name", $value }
+    }
+    $quoted = ConvertTo-ArgumentList $relaunch
+    $powershell = Join-Path $PSHOME 'powershell.exe'
+    # With Windows Terminal, the tool opens as a tab of the window last used rather than as a
+    # window of its own, so several sessions continued at once stay together. Windows Terminal
+    # reads `;` as the end of a command, so it is escaped. Its `wt.exe` alias lives in the
+    # WindowsApps folder, which is not always on PATH.
+    $terminal = (Get-Command 'wt.exe' -CommandType Application -ErrorAction SilentlyContinue | Select-Object -First 1).Source
+    $alias = if ($env:LOCALAPPDATA) { Join-Path $env:LOCALAPPDATA 'Microsoft\WindowsApps\wt.exe' }
+    if (-not $terminal -and $alias -and (Test-Path -LiteralPath $alias)) { $terminal = $alias }
+    if ($terminal) {
+        $tab = @('-w', '0', 'new-tab', ('"' + $powershell + '"')) + $quoted
+        Start-Process -FilePath $terminal -ArgumentList (($tab | ForEach-Object { $_ -replace ';', '\;' }) -join ' ')
+    } else {
+        Start-Process -FilePath $powershell -ArgumentList ($quoted -join ' ')
+    }
+    exit
+}
+while ($true) {
+    Set-Location -LiteralPath $WorkingDirectory
+    if ($Title) {
+        $Host.UI.RawUI.WindowTitle = $Title
+        Write-Host $Title
+    }
+    [string[]]$arguments = @()
+    if ($ArgumentsBase64) {
+        $arguments = [Text.Encoding]::UTF8.GetString([Convert]::FromBase64String($ArgumentsBase64)).Split([char]0)
+    }
+    # Started with the command line built above, in this console, and waited for.
+    $start = New-Object System.Diagnostics.ProcessStartInfo
+    $start.FileName = $Executable
+    $start.Arguments = (ConvertTo-ArgumentList $arguments) -join ' '
+    $start.WorkingDirectory = $WorkingDirectory
+    $start.UseShellExecute = $false
+    $tool = [System.Diagnostics.Process]::Start($start)
+    $tool.WaitForExit()
+    if (-not $Handoff -or -not (Test-Path -LiteralPath $Handoff)) { break }
+    # The next command, taken once: the file is removed before it runs.
+    $next = Get-Content -LiteralPath $Handoff -Raw -Encoding UTF8 | ConvertFrom-Json
+    Remove-Item -LiteralPath $Handoff -Force
+    foreach ($name in @($next.unset)) { if ($name) { Remove-Item -LiteralPath "Env:$name" -ErrorAction SilentlyContinue } }
+    if ($next.set) { foreach ($entry in $next.set.PSObject.Properties) { Set-Item -LiteralPath "Env:$($entry.Name)" -Value $entry.Value } }
+    $WorkingDirectory = $next.workingDirectory
+    $Executable = $next.executable
+    $Title = $next.title
+    $ArgumentsBase64 = $next.argumentsBase64
+}