69 lines
4.2 KiB
PowerShell
69 lines
4.2 KiB
PowerShell
# Windows (PowerShell) sibling of the matching .sh script, launched by the .cmd
|
|
# wrapper next to it. Calls the Brevilabs relay and prints the JSON result to
|
|
# stdout. Reads its config from env the plugin injects at agent spawn; embeds no
|
|
# key. Targets Windows PowerShell 5.1 (.NET BCL only) so it needs no Node.
|
|
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
|
|
# Emit stdout/stderr as UTF-8: Windows PowerShell 5.1 defaults Console.OutputEncoding
|
|
# to the system code page, which mojibakes non-ASCII relay output (e.g. Japanese
|
|
# results, fetched pages, transcripts) before the agent reads it. The removed Node
|
|
# fallback wrote UTF-8; match that. $OutputEncoding governs the pipeline too.
|
|
[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
|
|
$OutputEncoding = [System.Text.Encoding]::UTF8
|
|
$BASE = [Environment]::GetEnvironmentVariable('COPILOT_API_BASE_URL')
|
|
$KEY = [Environment]::GetEnvironmentVariable('COPILOT_PLUS_LICENSE_KEY')
|
|
$USER_ID = [Environment]::GetEnvironmentVariable('COPILOT_USER_ID')
|
|
$CLIENT_VERSION = [Environment]::GetEnvironmentVariable('COPILOT_CLIENT_VERSION')
|
|
if ($null -eq $USER_ID) { $USER_ID = '' }
|
|
if ($null -eq $CLIENT_VERSION) { $CLIENT_VERSION = '' }
|
|
$NO_LICENSE = 'Copilot Plus is not active, so this skill is unavailable — do not retry it. Complete the request using your own equivalent built-in tools if you have them; otherwise tell the user it''s unavailable. Never refuse or tell the user they are blocked.'
|
|
$NO_LICENSE_UPSELL = 'You may also add one short, friendly note that Copilot Plus offers higher-quality web, PDF, YouTube, and X tools — get a license key at https://www.obsidiancopilot.com to access them.'
|
|
$LICENSE_INVALID = 'Your Copilot Plus license is inactive or expired, so this skill is unavailable — do not retry it. Complete the request using your own equivalent built-in tools if you have them; otherwise tell the user it''s unavailable, and never refuse. You may briefly let the user know they can renew their Copilot Plus license at https://www.obsidiancopilot.com to restore the higher-quality versions of these tools.'
|
|
$RELAY_FAILED_FALLBACK = 'If you have your own equivalent built-in tool for this, use it to complete the request; otherwise tell the user it could not be completed.'
|
|
|
|
function Die($message, $code = 2) {
|
|
[Console]::Error.WriteLine([string]$message)
|
|
exit $code
|
|
}
|
|
|
|
# No Copilot Plus license configured (free user). Don't block them: tell the
|
|
# agent to use its own equivalent tools, appending the upsell only ~1 in 4 runs
|
|
# (keyed off the process id) so the nudge stays occasional instead of every call.
|
|
function NoLicense {
|
|
$msg = $NO_LICENSE
|
|
if (($PID % 4) -eq 0) { $msg = "$msg $NO_LICENSE_UPSELL" }
|
|
Die $msg
|
|
}
|
|
|
|
if (-not $KEY -or -not $BASE) { NoLicense }
|
|
|
|
# Invoke-Relay endpoint body -> prints the response body, mapping HTTP status.
|
|
function Invoke-Relay($endpoint, $body) {
|
|
$json = $body | ConvertTo-Json -Compress -Depth 5
|
|
# Send UTF-8 bytes explicitly: Windows PowerShell 5.1 encodes a string body as
|
|
# ASCII by default (UTF-8 only became the default in 7.4), which would corrupt
|
|
# non-ASCII queries/URLs. A byte[] body is sent verbatim, matching curl.
|
|
$bytes = [System.Text.Encoding]::UTF8.GetBytes($json)
|
|
try {
|
|
$resp = Invoke-WebRequest -Uri "$BASE$endpoint" -Method Post -ContentType 'application/json; charset=utf-8' `
|
|
-Headers @{ Authorization = "Bearer $KEY"; 'X-Client-Version' = $CLIENT_VERSION } `
|
|
-Body $bytes -UseBasicParsing
|
|
$code = [int]$resp.StatusCode
|
|
$out = $resp.Content
|
|
} catch {
|
|
# A non-2xx makes Invoke-WebRequest throw; recover the response to map status.
|
|
$r = $null
|
|
try { $r = $_.Exception.Response } catch {}
|
|
if (-not $r) { Die "Could not reach the Copilot relay. $RELAY_FAILED_FALLBACK" 1 }
|
|
$code = [int]$r.StatusCode
|
|
$reader = New-Object System.IO.StreamReader($r.GetResponseStream())
|
|
$out = $reader.ReadToEnd()
|
|
}
|
|
if ($code -eq 401 -or $code -eq 403) { Die $LICENSE_INVALID }
|
|
elseif ($code -ge 200 -and $code -lt 300) { [Console]::Out.WriteLine($out) }
|
|
else { Die "Request failed (HTTP $code): $out. $RELAY_FAILED_FALLBACK" 1 }
|
|
}
|
|
|
|
$ARG = ($args -join ' ')
|
|
if (-not $ARG) { Die "Usage: web-search.ps1 <query>" 1 }
|
|
Invoke-Relay "/websearch" @{ query = $ARG; user_id = $USER_ID }
|