mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-04 14:22:09 +03:00
feat(oauth): remote Antigravity login via local helper + paste-credentials (#5203)
Remote Antigravity login: local helper (omniroute login antigravity) + paste-credentials. Integrated into release/v3.8.39.
This commit is contained in:
committed by
GitHub
parent
dd5529929b
commit
c06fa4e8e1
@@ -10,6 +10,7 @@ _In development — bullets added per PR; finalized at release._
|
||||
|
||||
### ✨ New Features
|
||||
|
||||
- **feat(oauth): remote Antigravity login via local helper + paste-credentials** — Antigravity (and other Google "native/desktop" OAuth providers) use Google's `firstparty/nativeapp` consent, which only releases the auth code when the loopback redirect (`127.0.0.1:<port>`) is reachable from the approving browser. On a remote VPS install that loopback lives on the server, so the consent hangs forever and never emits a code — the "paste the callback URL" fallback has nothing to paste (a Google-side constraint, identical in upstream 9router). A new `omniroute login antigravity` CLI helper runs the OAuth on the user's **own** machine (where 127.0.0.1 works), exchanges the code, and prints a single-line `omniroute-cred-v1.…` credential blob; the dashboard's Antigravity Connect → Step 2 field now accepts that blob (alongside callback URLs) and persists the connection via a new `paste-credentials` action (server-side onboarding, provider-allowlisted, with the blob's embedded provider required to match the route). The SSH local-forward tunnel is documented as a zero-tooling alternative. See [`docs/guides/REMOTE-MODE.md`](docs/guides/REMOTE-MODE.md). ([#5203](https://github.com/diegosouzapw/OmniRoute/pull/5203))
|
||||
- **feat(agent-bridge): graceful cert-install fallback for containers / headless** — when the MITM root CA can't be installed into the system trust store automatically (Docker / headless / no sudo / read-only trust store), the Agent Bridge no longer hard-fails on start with a generic "Certificate install failed". It now starts in skip mode and the dashboard surfaces a platform-specific **manual-install guide** (plus a CA download link) so the operator can trust the certificate by hand. The trust-cert endpoints return a structured `{ skippable, manualGuide }` response (HTTP 200) for environment failures instead of a 500; an explicit user cancellation is still reported distinctly. ([#4546](https://github.com/diegosouzapw/OmniRoute/issues/4546) — thanks @phuchptty)
|
||||
|
||||
### 🔧 Bug Fixes
|
||||
|
||||
192
bin/cli/commands/login.mjs
Normal file
192
bin/cli/commands/login.mjs
Normal file
@@ -0,0 +1,192 @@
|
||||
import { createServer } from "node:http";
|
||||
import { randomUUID } from "node:crypto";
|
||||
|
||||
/**
|
||||
* `omniroute login antigravity` — local OAuth helper for remote installs.
|
||||
*
|
||||
* Why this exists: Google's `firstparty/nativeapp` consent for the embedded
|
||||
* Antigravity desktop client only releases the authorization code when the
|
||||
* loopback redirect (127.0.0.1:<port>) is REACHABLE. On a remote VPS install the
|
||||
* loopback is unreachable, so the consent hangs forever and never emits a code —
|
||||
* the dashboard's "paste the callback URL" fallback has nothing to paste. (The
|
||||
* same flow works locally and over an SSH tunnel, where the loopback IS reachable.)
|
||||
*
|
||||
* This command runs the OAuth on the user's OWN machine — where 127.0.0.1 works —
|
||||
* captures the code on a local loopback server, exchanges it for tokens, and
|
||||
* prints a single-line credential blob. The user pastes that blob into the remote
|
||||
* dashboard (Antigravity → "Paste credentials"), which decodes it, finalizes the
|
||||
* onboarding server-side, and persists the connection.
|
||||
*
|
||||
* It talks ONLY to Google (no OmniRoute server needed locally), so it works even
|
||||
* if the remote VPS is firewalled from the user's machine.
|
||||
*/
|
||||
|
||||
const PROVIDER = "antigravity";
|
||||
|
||||
/** Open the system browser; no-op if the optional `open` dependency is missing. */
|
||||
async function defaultOpenBrowser(url) {
|
||||
try {
|
||||
const { default: open } = await import("open");
|
||||
await open(url);
|
||||
} catch {
|
||||
// `open` not available — the caller already printed the URL to paste manually.
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Start a loopback HTTP server bound to 127.0.0.1 (NOT 0.0.0.0 — we never want to
|
||||
* expose the callback to the LAN). Resolves to { port, waitForCallback, close }.
|
||||
*/
|
||||
function defaultStartServer(preferredPort) {
|
||||
return new Promise((resolve, reject) => {
|
||||
let resolveCallback;
|
||||
const callbackPromise = new Promise((r) => {
|
||||
resolveCallback = r;
|
||||
});
|
||||
|
||||
const server = createServer((req, res) => {
|
||||
const url = new URL(req.url, "http://127.0.0.1");
|
||||
if (url.pathname !== "/callback" && url.pathname !== "/auth/callback") {
|
||||
res.writeHead(404).end();
|
||||
return;
|
||||
}
|
||||
const params = Object.fromEntries(url.searchParams.entries());
|
||||
res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
|
||||
res.end(
|
||||
"<!doctype html><meta charset=utf-8><title>OmniRoute</title>" +
|
||||
"<body style=\"font-family:system-ui;padding:2rem\">" +
|
||||
"<h2>✅ Authorization received</h2>" +
|
||||
"<p>Return to your terminal — you can close this tab.</p></body>"
|
||||
);
|
||||
resolveCallback(params);
|
||||
});
|
||||
|
||||
server.on("error", reject);
|
||||
server.listen(preferredPort || 0, "127.0.0.1", () => {
|
||||
const { port } = server.address();
|
||||
resolve({
|
||||
port,
|
||||
waitForCallback: () => callbackPromise,
|
||||
close: () => new Promise((r) => server.close(() => r())),
|
||||
});
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/** Lazy-load the antigravity provider + blob codec (TS source via tsx). */
|
||||
async function loadDeps() {
|
||||
const { antigravity } = await import("../../../src/lib/oauth/providers/antigravity.ts");
|
||||
const { encodeCredentialBlob } = await import("../../../src/lib/oauth/credentialBlob.ts");
|
||||
return { antigravity, encodeCredentialBlob };
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the Google authorization request for a given loopback port. Uses a plain
|
||||
* authorization_code grant (NO PKCE code_challenge) — matching the working flow:
|
||||
* a code_challenge here would force the exchange to require a code_verifier.
|
||||
*/
|
||||
export async function buildAntigravityAuthRequest(port, makeState = randomUUID) {
|
||||
const { antigravity } = await loadDeps();
|
||||
const redirectUri = `http://127.0.0.1:${port}/callback`;
|
||||
const state = makeState();
|
||||
const authUrl = antigravity.buildAuthUrl(antigravity.config, redirectUri, state);
|
||||
return { redirectUri, state, authUrl };
|
||||
}
|
||||
|
||||
/** Exchange the captured code for raw Google tokens (no code_verifier — no PKCE). */
|
||||
export async function exchangeAntigravityCode(code, redirectUri) {
|
||||
const { antigravity } = await loadDeps();
|
||||
return antigravity.exchangeToken(antigravity.config, code, redirectUri);
|
||||
}
|
||||
|
||||
/**
|
||||
* Orchestrate the local login. Dependencies are injectable for testing; the real
|
||||
* path uses a 127.0.0.1 loopback server, the system browser, and a live token
|
||||
* exchange against Google. Returns the credential blob string.
|
||||
*/
|
||||
export async function runAntigravityLogin(opts = {}, deps = {}) {
|
||||
const startServer = deps.startServer ?? defaultStartServer;
|
||||
const openBrowser = deps.openBrowser ?? defaultOpenBrowser;
|
||||
const exchange = deps.exchange ?? exchangeAntigravityCode;
|
||||
const makeState = deps.makeState ?? randomUUID;
|
||||
const print = deps.print ?? ((s) => process.stdout.write(s));
|
||||
const log = deps.log ?? ((s) => process.stderr.write(s));
|
||||
const { encodeCredentialBlob } = await loadDeps();
|
||||
|
||||
const server = await startServer(opts.port);
|
||||
const { redirectUri, state, authUrl } = await buildAntigravityAuthRequest(server.port, makeState);
|
||||
|
||||
log(`\nOpen this URL to authorize Antigravity (it will open automatically):\n\n ${authUrl}\n\n`);
|
||||
if (opts.browser !== false) await openBrowser(authUrl);
|
||||
log("Waiting for Google to redirect back to the local loopback...\n");
|
||||
|
||||
const timeoutMs = opts.timeout ?? 300000;
|
||||
let timer;
|
||||
let params;
|
||||
try {
|
||||
params = await Promise.race([
|
||||
server.waitForCallback(),
|
||||
new Promise((_, reject) => {
|
||||
timer = setTimeout(
|
||||
() => reject(new Error("Timed out waiting for the OAuth callback")),
|
||||
timeoutMs
|
||||
);
|
||||
// Don't keep the event loop alive solely for this timer.
|
||||
if (typeof timer.unref === "function") timer.unref();
|
||||
}),
|
||||
]);
|
||||
} finally {
|
||||
clearTimeout(timer);
|
||||
await server.close();
|
||||
}
|
||||
|
||||
if (params.error) {
|
||||
throw new Error(`Authorization failed: ${params.error_description || params.error}`);
|
||||
}
|
||||
if (params.state !== state) {
|
||||
throw new Error("State mismatch — aborting (possible CSRF). Please retry the login.");
|
||||
}
|
||||
if (!params.code) {
|
||||
throw new Error("No authorization code returned by Google.");
|
||||
}
|
||||
|
||||
const tokens = await exchange(params.code, redirectUri);
|
||||
const blob = encodeCredentialBlob({ provider: PROVIDER, tokens });
|
||||
|
||||
print(
|
||||
"\n" +
|
||||
"Antigravity authorized. Copy the line below and paste it into your remote\n" +
|
||||
"OmniRoute dashboard: Providers → Antigravity → Connect → \"Paste credentials\".\n" +
|
||||
"(This contains a refresh token — treat it like a password.)\n\n" +
|
||||
blob +
|
||||
"\n\n"
|
||||
);
|
||||
return blob;
|
||||
}
|
||||
|
||||
async function runLoginAntigravity(opts) {
|
||||
try {
|
||||
await runAntigravityLogin({
|
||||
browser: opts.browser,
|
||||
timeout: opts.timeout,
|
||||
port: opts.port,
|
||||
});
|
||||
} catch (err) {
|
||||
process.stderr.write(`\nLogin failed: ${err?.message || err}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
export function registerLogin(program) {
|
||||
const login = program
|
||||
.command("login")
|
||||
.description("Local OAuth helpers for remote OmniRoute installs (run on your own machine)");
|
||||
|
||||
login
|
||||
.command("antigravity")
|
||||
.description("Authorize Antigravity locally and print a credential blob to paste remotely")
|
||||
.option("--no-browser", "Do not auto-open the browser; print the URL instead")
|
||||
.option("--port <n>", "Fixed loopback port (default: OS-assigned)", (v) => parseInt(v, 10))
|
||||
.option("--timeout <ms>", "How long to wait for the callback", (v) => parseInt(v, 10), 300000)
|
||||
.action(runLoginAntigravity);
|
||||
}
|
||||
@@ -2,6 +2,7 @@ import { registerMemory } from "./memory.mjs";
|
||||
import { registerSkills } from "./skills.mjs";
|
||||
import { registerAudit } from "./audit.mjs";
|
||||
import { registerOAuth } from "./oauth.mjs";
|
||||
import { registerLogin } from "./login.mjs";
|
||||
import { registerCloud } from "./cloud.mjs";
|
||||
import { registerEval } from "./eval.mjs";
|
||||
import { registerWebhooks } from "./webhooks.mjs";
|
||||
@@ -83,6 +84,7 @@ export function registerCommands(program) {
|
||||
registerSkills(program);
|
||||
registerAudit(program);
|
||||
registerOAuth(program);
|
||||
registerLogin(program);
|
||||
registerCloud(program);
|
||||
registerEval(program);
|
||||
registerWebhooks(program);
|
||||
|
||||
@@ -100,6 +100,68 @@ A token with insufficient scope gets `403` with a clear message.
|
||||
|
||||
---
|
||||
|
||||
## Connecting Antigravity on a remote install
|
||||
|
||||
Antigravity (and other Google "native/desktop" OAuth providers such as
|
||||
`gemini-cli`) use Google's `firstparty/nativeapp` consent screen. Google only
|
||||
releases the authorization code when the **loopback redirect**
|
||||
(`http://127.0.0.1:<port>/callback`) is **reachable from the browser that
|
||||
approves the sign-in**. On a remote VPS install that loopback lives on the
|
||||
server, not on your machine, so the consent screen **hangs forever and never
|
||||
emits a code** — the normal "paste the callback URL" fallback has nothing to
|
||||
paste. (This is a Google-side constraint: the same hang happens in any proxy
|
||||
that uses the bundled Antigravity desktop client, not just OmniRoute.)
|
||||
|
||||
There are two supported ways to connect Antigravity to a remote OmniRoute.
|
||||
|
||||
### Option A — local login helper (recommended)
|
||||
|
||||
Run the OAuth on **your own computer**, where `127.0.0.1` is reachable, and paste
|
||||
the result into the remote dashboard. The helper talks only to Google — it does
|
||||
**not** need network access to your VPS, so it works even behind firewalls.
|
||||
|
||||
```bash
|
||||
# On your LOCAL machine (needs Node.js + a browser):
|
||||
npx omniroute login antigravity
|
||||
# ↳ opens the Google consent in your browser, captures the callback on a local
|
||||
# loopback port, exchanges it, and prints a one-line credential blob:
|
||||
#
|
||||
# omniroute-cred-v1.eyJ2IjoxLCJ...
|
||||
```
|
||||
|
||||
Then, in the **remote** dashboard: **Providers → Antigravity → Connect**, and
|
||||
paste the `omniroute-cred-v1.…` blob into the **Step 2** field (it accepts either
|
||||
a callback URL or a credential blob). OmniRoute decodes it, runs the Cloud Code
|
||||
onboarding server-side, and persists the connection.
|
||||
|
||||
> The blob contains a refresh token — treat it like a password. It is sent once
|
||||
> over your dashboard connection and stored encrypted at rest.
|
||||
|
||||
Flags: `--no-browser` (print the URL instead of auto-opening), `--port <n>`
|
||||
(pin the loopback port), `--timeout <ms>`.
|
||||
|
||||
### Option B — SSH local-forward tunnel
|
||||
|
||||
If you have SSH access to the VPS, forward the dashboard port so that the
|
||||
loopback callback resolves back to the server through the tunnel:
|
||||
|
||||
```bash
|
||||
# On your LOCAL machine:
|
||||
ssh -L 20128:localhost:20128 user@your-vps
|
||||
# then open http://localhost:20128 in your LOCAL browser and connect Antigravity
|
||||
# normally — the 127.0.0.1:20128/callback redirect now reaches the VPS via SSH.
|
||||
```
|
||||
|
||||
Because you reach the dashboard as `localhost:20128`, the Google consent
|
||||
completes and the callback is delivered to the server through the same tunnel —
|
||||
no blob needed. Keep the tunnel open until the connection shows as active.
|
||||
|
||||
> A fully headless alternative (no helper, no tunnel) is to configure your **own**
|
||||
> Google OAuth web credentials + a public base URL; see the provider's OAuth
|
||||
> environment variables. The two options above need no extra Google setup.
|
||||
|
||||
---
|
||||
|
||||
## Managing tokens
|
||||
|
||||
```bash
|
||||
|
||||
90
src/app/api/oauth/[provider]/paste-credentials/route.ts
Normal file
90
src/app/api/oauth/[provider]/paste-credentials/route.ts
Normal file
@@ -0,0 +1,90 @@
|
||||
import { NextResponse } from "next/server";
|
||||
import { finalizeTokens } from "@/lib/oauth/providers";
|
||||
import { persistOAuthConnection } from "@/lib/oauth/connectionPersistence";
|
||||
import { parsePastedCredentials } from "@/lib/oauth/pasteCredentials";
|
||||
import { oauthPasteCredentialsSchema } from "@/shared/validation/schemas";
|
||||
import { isValidationFailure, validateBody } from "@/shared/validation/helpers";
|
||||
import { isAuthRequired, isAuthenticated } from "@/shared/utils/apiAuth";
|
||||
import { sanitizeErrorMessage } from "@omniroute/open-sse/utils/error";
|
||||
|
||||
/**
|
||||
* POST /api/oauth/[provider]/paste-credentials
|
||||
*
|
||||
* Persist credentials produced by the local remote-login helper
|
||||
* (`omniroute login antigravity`). Google's `firstparty/nativeapp` consent only
|
||||
* releases the auth code when the loopback redirect is reachable, which never
|
||||
* happens on a remote VPS — so the helper runs the OAuth on the user's own
|
||||
* machine and prints a single-line credential blob. The dashboard pastes that
|
||||
* blob here; we decode + validate it (provider allowlist + match), finalize the
|
||||
* tokens (the Cloud Code onboarding runs here on the server, which CAN reach
|
||||
* Google's APIs), and persist the connection. Same finalize path as the
|
||||
* `device-complete` action. See src/lib/oauth/credentialBlob.ts.
|
||||
*
|
||||
* This lives in its own static route segment (not the dynamic `[action]` route)
|
||||
* so Next.js routes `/paste-credentials` here; static segments win over `[action]`.
|
||||
*/
|
||||
export async function POST(
|
||||
request: Request,
|
||||
{ params }: { params: Promise<{ provider: string }> }
|
||||
) {
|
||||
// Creating a connection is owner-only — gate behind dashboard auth.
|
||||
if ((await isAuthRequired(request)) && !(await isAuthenticated(request))) {
|
||||
return NextResponse.json({ error: "Unauthorized" }, { status: 401 });
|
||||
}
|
||||
|
||||
try {
|
||||
const { provider } = await params;
|
||||
|
||||
let rawBody: unknown;
|
||||
try {
|
||||
rawBody = await request.json();
|
||||
} catch {
|
||||
return NextResponse.json({ error: "Invalid JSON body" }, { status: 400 });
|
||||
}
|
||||
|
||||
const validation = validateBody(oauthPasteCredentialsSchema, rawBody);
|
||||
if (isValidationFailure(validation)) {
|
||||
return NextResponse.json({ error: validation.error }, { status: 400 });
|
||||
}
|
||||
const { blob, connectionId } = validation.data;
|
||||
|
||||
// Decode + gate (allowlist + blob-provider must match the route provider).
|
||||
let pasted;
|
||||
try {
|
||||
pasted = parsePastedCredentials(provider, blob);
|
||||
} catch (gateErr: any) {
|
||||
return NextResponse.json(
|
||||
{ success: false, error: sanitizeErrorMessage(gateErr?.message) || "Invalid credentials" },
|
||||
{ status: 400 }
|
||||
);
|
||||
}
|
||||
|
||||
let tokenData: any;
|
||||
try {
|
||||
tokenData = await finalizeTokens(provider, pasted.tokens);
|
||||
} catch (finalizeErr: any) {
|
||||
return NextResponse.json(
|
||||
{
|
||||
success: false,
|
||||
error: sanitizeErrorMessage(finalizeErr?.message) || "Failed to finalize tokens",
|
||||
},
|
||||
{ status: 500 }
|
||||
);
|
||||
}
|
||||
|
||||
const connection = await persistOAuthConnection(provider, tokenData, connectionId);
|
||||
|
||||
return NextResponse.json({
|
||||
success: true,
|
||||
connection: {
|
||||
id: connection.id,
|
||||
provider: connection.provider,
|
||||
email: connection.email,
|
||||
displayName: connection.displayName,
|
||||
},
|
||||
});
|
||||
} catch (error) {
|
||||
console.error("OAuth paste-credentials error:", error);
|
||||
return NextResponse.json({ error: "Internal server error" }, { status: 500 });
|
||||
}
|
||||
}
|
||||
@@ -7749,7 +7749,7 @@
|
||||
"deviceCodeVerificationUrl": "Verification URL",
|
||||
"deviceCodeYourCode": "Your code",
|
||||
"deviceCodeWaiting": "Waiting for authorization...",
|
||||
"googleOAuthWarning": "Remote access + Google OAuth: bundled credentials only accept loopback redirects like <code>127.0.0.1</code>. The browser that approves Google must be able to reach OmniRoute on that local port, usually by opening OmniRoute locally or using an SSH/local-forward tunnel. For fully remote use without this local callback, <a>configure your own OAuth credentials</a>.",
|
||||
"googleOAuthWarning": "Remote access + Google OAuth: bundled credentials only accept loopback redirects like <code>127.0.0.1</code>. The browser that approves Google must be able to reach OmniRoute on that local port, usually by opening OmniRoute locally or using an SSH/local-forward tunnel. Recommended for remote installs: on your own computer run <code>npx omniroute login antigravity</code> and paste the credential blob it prints into the field below. For fully remote use without this local callback, <a>configure your own OAuth credentials</a>.",
|
||||
"remoteAccessInfo": "Remote access: Since you're accessing OmniRoute remotely, after authorization you'll see an error page (localhost not found). This is normal — just copy the full URL from your browser address bar and paste it below.",
|
||||
"step1OpenUrl": "Step 1: Open this URL in your browser",
|
||||
"copy": "Copy",
|
||||
|
||||
105
src/lib/oauth/credentialBlob.ts
Normal file
105
src/lib/oauth/credentialBlob.ts
Normal file
@@ -0,0 +1,105 @@
|
||||
/**
|
||||
* Paste-safe credential blob codec for the remote OAuth login helper.
|
||||
*
|
||||
* Why this exists: Google's `firstparty/nativeapp` consent for embedded desktop
|
||||
* clients (Antigravity, gemini-cli) only releases the authorization code when the
|
||||
* loopback redirect (127.0.0.1:<port>) is reachable. On a remote VPS install the
|
||||
* loopback is unreachable, so the consent hangs and never emits a code — there is
|
||||
* nothing for the user to paste back. (The same flow works locally and over an SSH
|
||||
* tunnel because then the loopback IS reachable.)
|
||||
*
|
||||
* The workaround: a local helper (`omniroute login antigravity`) runs the OAuth on
|
||||
* the user's own machine (loopback reachable → consent completes → tokens), then
|
||||
* encodes the raw token response into a single-line blob with this codec. The user
|
||||
* pastes the blob into the remote dashboard, which decodes it and persists the
|
||||
* connection (running the provider's post-exchange/onboarding from the server,
|
||||
* which CAN reach Google's Cloud Code APIs).
|
||||
*
|
||||
* Format: `<prefix><base64url(JSON)>` — a recognizable prefix so humans and the
|
||||
* decoder can identify it, followed by a URL/shell-safe base64url payload (no
|
||||
* `+`, `/`, `=`, or whitespace) so it survives copy-paste through terminals.
|
||||
*/
|
||||
|
||||
/** Human-recognizable, copy-paste-safe prefix. The decoder requires it. */
|
||||
export const CREDENTIAL_BLOB_PREFIX = "omniroute-cred-v1.";
|
||||
|
||||
/** Current blob schema version (embedded in the payload as `v`). */
|
||||
const CREDENTIAL_BLOB_VERSION = 1;
|
||||
|
||||
export interface CredentialBlobTokens {
|
||||
access_token?: string;
|
||||
refresh_token?: string;
|
||||
id_token?: string;
|
||||
expires_in?: number;
|
||||
scope?: string;
|
||||
[key: string]: unknown;
|
||||
}
|
||||
|
||||
export interface CredentialBlob {
|
||||
provider: string;
|
||||
tokens: CredentialBlobTokens;
|
||||
}
|
||||
|
||||
/**
|
||||
* Encode a provider + raw OAuth token response into a single-line blob.
|
||||
* Throws if `provider` is missing — a blob with no provider cannot be routed.
|
||||
*/
|
||||
export function encodeCredentialBlob(input: CredentialBlob): string {
|
||||
if (!input || typeof input.provider !== "string" || !input.provider.trim()) {
|
||||
throw new Error("encodeCredentialBlob: a non-empty provider is required");
|
||||
}
|
||||
if (!input.tokens || typeof input.tokens !== "object") {
|
||||
throw new Error("encodeCredentialBlob: tokens object is required");
|
||||
}
|
||||
const payload = {
|
||||
v: CREDENTIAL_BLOB_VERSION,
|
||||
provider: input.provider.trim(),
|
||||
tokens: input.tokens,
|
||||
};
|
||||
const json = JSON.stringify(payload);
|
||||
const b64 = Buffer.from(json, "utf8").toString("base64url");
|
||||
return `${CREDENTIAL_BLOB_PREFIX}${b64}`;
|
||||
}
|
||||
|
||||
/**
|
||||
* Decode a credential blob produced by {@link encodeCredentialBlob}.
|
||||
* Validates the prefix, version, JSON shape, and the presence of an access_token.
|
||||
* Throws a descriptive error on any malformed/tampered/unsupported input.
|
||||
*/
|
||||
export function decodeCredentialBlob(blob: string): CredentialBlob {
|
||||
if (typeof blob !== "string" || !blob.startsWith(CREDENTIAL_BLOB_PREFIX)) {
|
||||
throw new Error(
|
||||
`decodeCredentialBlob: invalid format — must start with "${CREDENTIAL_BLOB_PREFIX}"`
|
||||
);
|
||||
}
|
||||
const b64 = blob.slice(CREDENTIAL_BLOB_PREFIX.length).trim();
|
||||
if (!/^[A-Za-z0-9_-]+$/.test(b64)) {
|
||||
throw new Error("decodeCredentialBlob: invalid payload — not base64url");
|
||||
}
|
||||
|
||||
let parsed: { v?: unknown; provider?: unknown; tokens?: unknown };
|
||||
try {
|
||||
const json = Buffer.from(b64, "base64url").toString("utf8");
|
||||
parsed = JSON.parse(json);
|
||||
} catch {
|
||||
throw new Error("decodeCredentialBlob: invalid payload — could not parse JSON");
|
||||
}
|
||||
|
||||
if (parsed.v !== CREDENTIAL_BLOB_VERSION) {
|
||||
throw new Error(
|
||||
`decodeCredentialBlob: unsupported blob version ${String(parsed.v)} (expected ${CREDENTIAL_BLOB_VERSION})`
|
||||
);
|
||||
}
|
||||
if (typeof parsed.provider !== "string" || !parsed.provider.trim()) {
|
||||
throw new Error("decodeCredentialBlob: invalid payload — missing provider");
|
||||
}
|
||||
const tokens = parsed.tokens as CredentialBlobTokens | undefined;
|
||||
if (!tokens || typeof tokens !== "object") {
|
||||
throw new Error("decodeCredentialBlob: invalid payload — missing tokens");
|
||||
}
|
||||
if (typeof tokens.access_token !== "string" || !tokens.access_token) {
|
||||
throw new Error("decodeCredentialBlob: invalid payload — missing access_token");
|
||||
}
|
||||
|
||||
return { provider: parsed.provider.trim(), tokens };
|
||||
}
|
||||
51
src/lib/oauth/pasteCredentials.ts
Normal file
51
src/lib/oauth/pasteCredentials.ts
Normal file
@@ -0,0 +1,51 @@
|
||||
/**
|
||||
* Server-side gate for the remote login helper's "paste credentials" flow.
|
||||
*
|
||||
* Google's `firstparty/nativeapp` consent for embedded desktop clients only
|
||||
* releases the authorization code when the loopback redirect is reachable, which
|
||||
* never happens on a remote VPS install. The remote login helper runs the OAuth
|
||||
* locally and emits a credential blob (see ./credentialBlob.ts); the dashboard
|
||||
* POSTs that blob to /api/oauth/<provider>/paste-credentials, which decodes it
|
||||
* and persists the connection via the same finalize path as `device-complete`.
|
||||
*
|
||||
* This module holds the pure, security-relevant gate: which providers may use
|
||||
* the paste path, and the requirement that the blob's embedded provider matches
|
||||
* the route provider (so a blob minted for one provider cannot be replayed
|
||||
* against another).
|
||||
*/
|
||||
|
||||
import { decodeCredentialBlob, type CredentialBlob } from "./credentialBlob";
|
||||
|
||||
/**
|
||||
* Providers eligible for the paste-credentials flow: Google native-loopback
|
||||
* clients whose consent cannot complete on a headless/remote host. `agy` is the
|
||||
* Antigravity alias. Codex is intentionally excluded — it has its own browser
|
||||
* device flow (`device-complete`) that works remotely.
|
||||
*/
|
||||
export const PASTE_CREDENTIAL_PROVIDERS = new Set(["antigravity", "agy", "gemini-cli"]);
|
||||
|
||||
/**
|
||||
* Validate + decode a pasted credential blob for a given route provider.
|
||||
* Throws a descriptive error if the provider is not allowlisted, if the blob's
|
||||
* embedded provider does not match, or if the blob itself is malformed.
|
||||
*/
|
||||
export function parsePastedCredentials(routeProvider: string, blob: string): CredentialBlob {
|
||||
if (!PASTE_CREDENTIAL_PROVIDERS.has(routeProvider)) {
|
||||
throw new Error(
|
||||
`paste-credentials not supported for provider: ${routeProvider}. ` +
|
||||
`Supported: ${[...PASTE_CREDENTIAL_PROVIDERS].join(", ")}`
|
||||
);
|
||||
}
|
||||
|
||||
// decodeCredentialBlob validates prefix/version/JSON shape + access_token.
|
||||
const decoded = decodeCredentialBlob(blob);
|
||||
|
||||
if (decoded.provider !== routeProvider) {
|
||||
throw new Error(
|
||||
`Pasted credential provider mismatch: blob is for "${decoded.provider}" ` +
|
||||
`but the route provider is "${routeProvider}"`
|
||||
);
|
||||
}
|
||||
|
||||
return decoded;
|
||||
}
|
||||
@@ -7,6 +7,7 @@ import Button from "./Button";
|
||||
import Input from "./Input";
|
||||
import { useCopyToClipboard } from "@/shared/hooks/useCopyToClipboard";
|
||||
import { parseResponseBody, getErrorMessage } from "@/shared/utils/api";
|
||||
import { isCredentialBlob, submitCredentialBlob } from "@/shared/components/oauthBlobSubmit";
|
||||
|
||||
const GOOGLE_OAUTH_PROVIDERS = new Set(["antigravity", "agy", "gemini-cli"]);
|
||||
|
||||
@@ -653,7 +654,10 @@ export default function OAuthModal({
|
||||
const handleManualSubmit = async () => {
|
||||
try {
|
||||
setError(null);
|
||||
|
||||
if (isCredentialBlob(callbackUrl)) {
|
||||
await submitCredentialBlob(provider, callbackUrl, reauthConnection, setStep, onSuccess);
|
||||
return;
|
||||
}
|
||||
if (!authData) {
|
||||
throw new Error(
|
||||
"OAuth session not initialized. Restart the connection flow and try again."
|
||||
@@ -908,7 +912,7 @@ export default function OAuthModal({
|
||||
<Button
|
||||
onClick={handleManualSubmit}
|
||||
fullWidth
|
||||
disabled={!callbackUrl || !authData}
|
||||
disabled={!callbackUrl || (!authData && !isCredentialBlob(callbackUrl))}
|
||||
>
|
||||
{t("connect")}
|
||||
</Button>
|
||||
|
||||
42
src/shared/components/oauthBlobSubmit.ts
Normal file
42
src/shared/components/oauthBlobSubmit.ts
Normal file
@@ -0,0 +1,42 @@
|
||||
import { parseResponseBody, getErrorMessage } from "@/shared/utils/api";
|
||||
import { CREDENTIAL_BLOB_PREFIX } from "@/lib/oauth/credentialBlob";
|
||||
|
||||
/**
|
||||
* Helpers for the remote-login "paste credentials" path in OAuthModal.
|
||||
*
|
||||
* Google's native-loopback consent can't complete on a remote install, so the
|
||||
* user runs `omniroute login antigravity` locally and pastes the credential blob
|
||||
* it prints into the modal's Step 2 field. Extracted from OAuthModal to keep that
|
||||
* (god-file) component within its frozen size budget.
|
||||
*/
|
||||
|
||||
/** True if the pasted text is a credential blob (vs a callback URL / auth code). */
|
||||
export function isCredentialBlob(value: string): boolean {
|
||||
return typeof value === "string" && value.trim().startsWith(CREDENTIAL_BLOB_PREFIX);
|
||||
}
|
||||
|
||||
/**
|
||||
* POST a pasted credential blob to the paste-credentials endpoint. On success it
|
||||
* advances the modal to the success step and fires onSuccess; on failure it
|
||||
* THROWS so the caller's existing try/catch surfaces the error (keeps the modal
|
||||
* call site to two lines). Decoding + finalize + persist happen server-side.
|
||||
*/
|
||||
export async function submitCredentialBlob(
|
||||
provider: string,
|
||||
blob: string,
|
||||
reauthConnection: { id?: string } | null | undefined,
|
||||
setStep: (s: string) => void,
|
||||
onSuccess?: () => void
|
||||
): Promise<void> {
|
||||
const res = await fetch(`/api/oauth/${provider}/paste-credentials`, {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify({ blob: blob.trim(), connectionId: reauthConnection?.id }),
|
||||
});
|
||||
const data = (await parseResponseBody(res)) as Record<string, unknown>;
|
||||
if (!res.ok) {
|
||||
throw new Error(getErrorMessage(data, res.status, "Failed to import credentials"));
|
||||
}
|
||||
setStep("success");
|
||||
onSuccess?.();
|
||||
}
|
||||
@@ -185,6 +185,19 @@ export const oauthDeviceCompleteSchema = z.object({
|
||||
connectionId: z.string().optional(),
|
||||
});
|
||||
|
||||
/**
|
||||
* Persist credentials obtained by the local remote-login helper. Google's
|
||||
* `firstparty/nativeapp` consent only releases the auth code when the loopback
|
||||
* redirect is reachable, which never happens on a remote VPS install — so the
|
||||
* helper (`omniroute login antigravity`) runs the OAuth on the user's own machine
|
||||
* and emits a single-line credential blob. The dashboard pastes that blob here;
|
||||
* the server decodes + finalizes + persists. See src/lib/oauth/credentialBlob.ts.
|
||||
*/
|
||||
export const oauthPasteCredentialsSchema = z.object({
|
||||
blob: z.string().trim().min(1, "credential blob is required"),
|
||||
connectionId: z.string().optional(),
|
||||
});
|
||||
|
||||
export const cursorImportSchema = z.object({
|
||||
accessToken: z.string().trim().min(1, "Access token is required"),
|
||||
machineId: z.string().trim().optional(),
|
||||
|
||||
107
tests/unit/cli-login-antigravity.test.ts
Normal file
107
tests/unit/cli-login-antigravity.test.ts
Normal file
@@ -0,0 +1,107 @@
|
||||
// Tests for the `omniroute login antigravity` local OAuth helper.
|
||||
//
|
||||
// The helper runs the OAuth on the user's own machine (where the Google
|
||||
// native-loopback consent can complete) and prints a credential blob to paste
|
||||
// into a remote install. We test the two pieces that, if wrong, silently break
|
||||
// the flow: the authorization request (must be a plain authorization_code grant
|
||||
// with a 127.0.0.1 loopback redirect and NO PKCE challenge) and the end-to-end
|
||||
// orchestration (state validation + exchange + blob emission), with the browser,
|
||||
// loopback server, and token exchange injected as fakes.
|
||||
|
||||
import test from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
import {
|
||||
buildAntigravityAuthRequest,
|
||||
runAntigravityLogin,
|
||||
} from "../../bin/cli/commands/login.mjs";
|
||||
import { decodeCredentialBlob } from "../../src/lib/oauth/credentialBlob.ts";
|
||||
|
||||
test("buildAntigravityAuthRequest: loopback redirect on 127.0.0.1 + no PKCE", async () => {
|
||||
const { authUrl, redirectUri, state } = await buildAntigravityAuthRequest(54321, () => "fixed");
|
||||
assert.equal(redirectUri, "http://127.0.0.1:54321/callback");
|
||||
assert.equal(state, "fixed");
|
||||
|
||||
const url = new URL(authUrl);
|
||||
assert.equal(url.origin, "https://accounts.google.com");
|
||||
assert.equal(url.searchParams.get("redirect_uri"), redirectUri);
|
||||
assert.equal(url.searchParams.get("response_type"), "code");
|
||||
assert.equal(url.searchParams.get("state"), "fixed");
|
||||
assert.equal(url.searchParams.get("code_challenge"), null, "must NOT carry a PKCE challenge");
|
||||
});
|
||||
|
||||
test("runAntigravityLogin: validates state, exchanges code, prints a decodable blob", async () => {
|
||||
let exchangedCode = null;
|
||||
let exchangedRedirect = null;
|
||||
|
||||
const blob = await runAntigravityLogin(
|
||||
{},
|
||||
{
|
||||
makeState: () => "S",
|
||||
openBrowser: async () => {},
|
||||
startServer: async () => ({
|
||||
port: 54321,
|
||||
waitForCallback: async () => ({ code: "the-code", state: "S" }),
|
||||
close: async () => {},
|
||||
}),
|
||||
exchange: async (code, redirectUri) => {
|
||||
exchangedCode = code;
|
||||
exchangedRedirect = redirectUri;
|
||||
return { access_token: "ya29.a", refresh_token: "1//r", expires_in: 3600, scope: "x" };
|
||||
},
|
||||
print: () => {},
|
||||
log: () => {},
|
||||
}
|
||||
);
|
||||
|
||||
assert.equal(exchangedCode, "the-code");
|
||||
assert.equal(exchangedRedirect, "http://127.0.0.1:54321/callback");
|
||||
|
||||
const decoded = decodeCredentialBlob(blob);
|
||||
assert.equal(decoded.provider, "antigravity");
|
||||
assert.equal(decoded.tokens.access_token, "ya29.a");
|
||||
assert.equal(decoded.tokens.refresh_token, "1//r");
|
||||
});
|
||||
|
||||
test("runAntigravityLogin: rejects a state mismatch (CSRF guard)", async () => {
|
||||
await assert.rejects(
|
||||
() =>
|
||||
runAntigravityLogin(
|
||||
{},
|
||||
{
|
||||
makeState: () => "expected",
|
||||
openBrowser: async () => {},
|
||||
startServer: async () => ({
|
||||
port: 1,
|
||||
waitForCallback: async () => ({ code: "c", state: "ATTACKER" }),
|
||||
close: async () => {},
|
||||
}),
|
||||
exchange: async () => ({ access_token: "x" }),
|
||||
print: () => {},
|
||||
log: () => {},
|
||||
}
|
||||
),
|
||||
/state mismatch|csrf/i
|
||||
);
|
||||
});
|
||||
|
||||
test("runAntigravityLogin: surfaces an OAuth error param", async () => {
|
||||
await assert.rejects(
|
||||
() =>
|
||||
runAntigravityLogin(
|
||||
{},
|
||||
{
|
||||
makeState: () => "S",
|
||||
openBrowser: async () => {},
|
||||
startServer: async () => ({
|
||||
port: 1,
|
||||
waitForCallback: async () => ({ error: "access_denied", state: "S" }),
|
||||
close: async () => {},
|
||||
}),
|
||||
exchange: async () => ({ access_token: "x" }),
|
||||
print: () => {},
|
||||
log: () => {},
|
||||
}
|
||||
),
|
||||
/access_denied|authorization failed/i
|
||||
);
|
||||
});
|
||||
84
tests/unit/oauth-credential-blob.test.ts
Normal file
84
tests/unit/oauth-credential-blob.test.ts
Normal file
@@ -0,0 +1,84 @@
|
||||
// Unit tests for the OAuth credential blob codec used by the remote login helper.
|
||||
//
|
||||
// Context (operator report 2026-06-27): Google's `firstparty/nativeapp` consent
|
||||
// for the embedded Antigravity desktop client only releases the authorization
|
||||
// code when the loopback redirect (127.0.0.1:<port>) is reachable. On a remote
|
||||
// VPS install the loopback is unreachable, so the consent hangs and never emits
|
||||
// a code — the normal "paste the callback URL" fallback has nothing to paste.
|
||||
//
|
||||
// The fix is a local helper (`omniroute login antigravity`) that runs the OAuth
|
||||
// on the user's own machine (loopback reachable → consent completes → tokens),
|
||||
// then prints a single-line, paste-safe credential blob. The user pastes it into
|
||||
// the remote dashboard, which decodes it and persists the connection.
|
||||
//
|
||||
// This codec is the contract between the helper (encoder) and the server/dashboard
|
||||
// (decoder). These tests pin: roundtrip fidelity, the human-recognizable prefix,
|
||||
// version + provider gating, and rejection of malformed/tampered input.
|
||||
|
||||
import test from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
import {
|
||||
CREDENTIAL_BLOB_PREFIX,
|
||||
encodeCredentialBlob,
|
||||
decodeCredentialBlob,
|
||||
} from "../../src/lib/oauth/credentialBlob.ts";
|
||||
|
||||
const SAMPLE = {
|
||||
provider: "antigravity",
|
||||
tokens: {
|
||||
access_token: "ya29.access",
|
||||
refresh_token: "1//refresh",
|
||||
id_token: "eyJ.id.token",
|
||||
expires_in: 3599,
|
||||
scope: "https://www.googleapis.com/auth/cloud-platform",
|
||||
},
|
||||
};
|
||||
|
||||
test("encode → decode roundtrips the provider and tokens", () => {
|
||||
const blob = encodeCredentialBlob(SAMPLE);
|
||||
const decoded = decodeCredentialBlob(blob);
|
||||
assert.equal(decoded.provider, "antigravity");
|
||||
assert.deepEqual(decoded.tokens, SAMPLE.tokens);
|
||||
});
|
||||
|
||||
test("blob is a single paste-safe line with the recognizable prefix", () => {
|
||||
const blob = encodeCredentialBlob(SAMPLE);
|
||||
assert.ok(blob.startsWith(CREDENTIAL_BLOB_PREFIX), "must carry the omniroute prefix");
|
||||
assert.ok(!/\s/.test(blob), "must contain no whitespace (single line, copy-paste safe)");
|
||||
// base64url only after the prefix — no +/= that break in URLs / shells.
|
||||
const payload = blob.slice(CREDENTIAL_BLOB_PREFIX.length);
|
||||
assert.match(payload, /^[A-Za-z0-9_-]+$/, "payload must be base64url");
|
||||
});
|
||||
|
||||
test("decode rejects a blob without the prefix", () => {
|
||||
const raw = Buffer.from(JSON.stringify({ v: 1, ...SAMPLE })).toString("base64url");
|
||||
assert.throws(() => decodeCredentialBlob(raw), /prefix|invalid|format/i);
|
||||
});
|
||||
|
||||
test("decode rejects an unsupported version", () => {
|
||||
// Hand-craft a v999 blob with the right prefix.
|
||||
const payload = Buffer.from(JSON.stringify({ v: 999, ...SAMPLE })).toString("base64url");
|
||||
assert.throws(() => decodeCredentialBlob(`${CREDENTIAL_BLOB_PREFIX}${payload}`), /version/i);
|
||||
});
|
||||
|
||||
test("decode rejects a blob missing an access_token", () => {
|
||||
const bad = encodeCredentialBlob({
|
||||
provider: "antigravity",
|
||||
tokens: { refresh_token: "only-refresh" },
|
||||
});
|
||||
assert.throws(() => decodeCredentialBlob(bad), /access_token|token/i);
|
||||
});
|
||||
|
||||
test("decode rejects tampered base64 (not valid JSON)", () => {
|
||||
assert.throws(
|
||||
() => decodeCredentialBlob(`${CREDENTIAL_BLOB_PREFIX}not-valid-base64-json!!!`),
|
||||
/invalid|parse|format/i
|
||||
);
|
||||
});
|
||||
|
||||
test("encode requires a provider", () => {
|
||||
assert.throws(
|
||||
() => encodeCredentialBlob({ tokens: { access_token: "x" } } as never),
|
||||
/provider/i
|
||||
);
|
||||
});
|
||||
79
tests/unit/oauth-paste-credentials-route.test.ts
Normal file
79
tests/unit/oauth-paste-credentials-route.test.ts
Normal file
@@ -0,0 +1,79 @@
|
||||
// Route-wiring tests for the paste-credentials OAuth action.
|
||||
//
|
||||
// These exercise the rejection paths only — they all return 400 BEFORE the
|
||||
// route touches finalizeTokens (Google APIs) or persistOAuthConnection (DB), so
|
||||
// no network is hit. The happy path's finalize+persist is the same IO as the
|
||||
// already-covered `device-complete` action; the new, security-relevant logic
|
||||
// (allowlist + provider match + blob validation) is unit-tested in
|
||||
// oauth-paste-credentials.test.ts and re-asserted through the HTTP boundary here.
|
||||
//
|
||||
// Auth is disabled via settings (requireLogin:false) so we reach the action
|
||||
// dispatch rather than a 401. DB handles are released in test.after (CLAUDE.md
|
||||
// learning: unreleased SQLite handles hang node:test).
|
||||
|
||||
import test from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
import fs from "node:fs";
|
||||
import os from "node:os";
|
||||
import path from "node:path";
|
||||
|
||||
const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-paste-creds-route-"));
|
||||
process.env.DATA_DIR = TEST_DATA_DIR;
|
||||
|
||||
const core = await import("../../src/lib/db/core.ts");
|
||||
const settingsDb = await import("../../src/lib/db/settings.ts");
|
||||
const route = await import("../../src/app/api/oauth/[provider]/paste-credentials/route.ts");
|
||||
const { encodeCredentialBlob } = await import("../../src/lib/oauth/credentialBlob.ts");
|
||||
|
||||
const tokens = { access_token: "ya29.x", refresh_token: "1//r", expires_in: 3599 };
|
||||
|
||||
test.before(async () => {
|
||||
await settingsDb.updateSettings({ requireLogin: false });
|
||||
});
|
||||
|
||||
test.after(async () => {
|
||||
core.resetDbInstance();
|
||||
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
async function postPaste(provider: string, body: unknown) {
|
||||
const request = new Request(`http://localhost:20128/api/oauth/${provider}/paste-credentials`, {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify(body),
|
||||
});
|
||||
const response = await route.POST(request, {
|
||||
params: Promise.resolve({ provider, action: "paste-credentials" }),
|
||||
} as never);
|
||||
return { status: response.status, body: await response.json() };
|
||||
}
|
||||
|
||||
test("paste-credentials: non-allowlisted provider is rejected with 400", async () => {
|
||||
const blob = encodeCredentialBlob({ provider: "openai", tokens });
|
||||
const { status, body } = await postPaste("openai", { blob });
|
||||
assert.equal(status, 400);
|
||||
assert.match(body.error, /not supported|supported/i);
|
||||
});
|
||||
|
||||
test("paste-credentials: malformed blob is rejected with 400", async () => {
|
||||
const { status, body } = await postPaste("antigravity", { blob: "totally-not-a-blob" });
|
||||
assert.equal(status, 400);
|
||||
assert.match(body.error, /invalid|format|prefix/i);
|
||||
});
|
||||
|
||||
test("paste-credentials: provider mismatch (blob for antigravity, route agy) → 400", async () => {
|
||||
const blob = encodeCredentialBlob({ provider: "antigravity", tokens });
|
||||
const { status, body } = await postPaste("agy", { blob });
|
||||
assert.equal(status, 400);
|
||||
assert.match(body.error, /match|mismatch|provider/i);
|
||||
});
|
||||
|
||||
test("paste-credentials: empty body fails schema validation with 400", async () => {
|
||||
const { status } = await postPaste("antigravity", {});
|
||||
assert.equal(status, 400);
|
||||
});
|
||||
|
||||
test("paste-credentials: error responses never leak a stack trace", async () => {
|
||||
const { body } = await postPaste("antigravity", { blob: "totally-not-a-blob" });
|
||||
assert.ok(!String(body.error).includes("at /"), "must not leak a stack trace");
|
||||
});
|
||||
49
tests/unit/oauth-paste-credentials.test.ts
Normal file
49
tests/unit/oauth-paste-credentials.test.ts
Normal file
@@ -0,0 +1,49 @@
|
||||
// Unit tests for the server-side paste-credentials gate.
|
||||
//
|
||||
// The remote login helper (`omniroute login antigravity`) prints a credential
|
||||
// blob; the dashboard POSTs it to /api/oauth/<provider>/paste-credentials. Before
|
||||
// the server finalizes + persists the tokens it MUST validate that (a) the route
|
||||
// provider is on the paste-credentials allowlist (only Google native-loopback
|
||||
// providers, never arbitrary providers) and (b) the blob's embedded provider
|
||||
// matches the route provider — otherwise a blob minted for one provider could be
|
||||
// replayed against another. This pins that gate; the finalize/persist IO is the
|
||||
// same path as the already-tested `device-complete` action.
|
||||
|
||||
import test from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
import { encodeCredentialBlob } from "../../src/lib/oauth/credentialBlob.ts";
|
||||
import {
|
||||
PASTE_CREDENTIAL_PROVIDERS,
|
||||
parsePastedCredentials,
|
||||
} from "../../src/lib/oauth/pasteCredentials.ts";
|
||||
|
||||
const tokens = { access_token: "ya29.x", refresh_token: "1//r", expires_in: 3599 };
|
||||
|
||||
test("allowlist contains antigravity and its agy alias, not codex", () => {
|
||||
assert.ok(PASTE_CREDENTIAL_PROVIDERS.has("antigravity"));
|
||||
assert.ok(PASTE_CREDENTIAL_PROVIDERS.has("agy"));
|
||||
assert.ok(!PASTE_CREDENTIAL_PROVIDERS.has("codex"), "codex uses its own device-complete path");
|
||||
});
|
||||
|
||||
test("accepts a matching antigravity blob and returns the tokens", () => {
|
||||
const blob = encodeCredentialBlob({ provider: "antigravity", tokens });
|
||||
const result = parsePastedCredentials("antigravity", blob);
|
||||
assert.equal(result.provider, "antigravity");
|
||||
assert.deepEqual(result.tokens, tokens);
|
||||
});
|
||||
|
||||
test("rejects a provider not on the allowlist", () => {
|
||||
const blob = encodeCredentialBlob({ provider: "openai", tokens });
|
||||
assert.throws(() => parsePastedCredentials("openai", blob), /not supported|allowlist|supported/i);
|
||||
});
|
||||
|
||||
test("rejects a blob whose embedded provider does not match the route provider", () => {
|
||||
// Blob minted for antigravity, replayed against the agy route → must reject.
|
||||
const blob = encodeCredentialBlob({ provider: "antigravity", tokens });
|
||||
assert.throws(() => parsePastedCredentials("agy", blob), /match|mismatch|provider/i);
|
||||
});
|
||||
|
||||
test("propagates codec validation errors (e.g. missing access_token)", () => {
|
||||
const blob = encodeCredentialBlob({ provider: "antigravity", tokens: { refresh_token: "r" } });
|
||||
assert.throws(() => parsePastedCredentials("antigravity", blob), /access_token|token/i);
|
||||
});
|
||||
Reference in New Issue
Block a user