Files
learnharness/tests/auth-keys.test.ts
T
amosblomqvist ec38f91a2b feat(keys): free-form model-provider keys — pi-ai provider table on /api/keys
Provider-keys interlude (owner spec, 2026-09-19): any pi-compatible
provider's key is manageable, not just the curated set.

GET /api/keys now also returns providers: {id, envVar, label}[] — every
pi-ai provider that takes a plain api-key env var (33 today), derived
from pi-ai's own provider→env-var table. No-drift mechanism: the table
is generated from pi-ai (builtinProviders + findEnvKeys over a
truthy-for-every-var proxy env) and pinned by
tests/auth-provider-keys.test.ts, which re-derives the truth at test
time and asserts EXACT equality — a pi-ai upgrade fails the gate loudly.

Owner-spec exclusions: github-copilot (OAuth), amazon-bedrock (AWS),
google-vertex (ADC), azure-openai-responses (base URL), the cloudflare
pair; anthropic keeps ANTHROPIC_API_KEY only. OPENROUTER_API_KEY
graduates from the curated catalog to the provider table (dedupe — the
UI pins it always-visible); curated stays Deepgram + the Google search
pair. Statuses: curated + present vars + unknown file vars — absent
provider keys never spam rows; the dropdown is discovery.

Write side: PUT/DELETE accept provider env vars (generic truncation
guard — trim, no whitespace, ≥12; family checks stay on the curated
three). Present-in-env provider vars surface as status rows so
env-wins/shadow semantics stay honest.
2026-09-19 22:52:40 +03:00

106 lines
4.6 KiB
TypeScript

/**
* The curated auth-key catalog + shape rules (auth-keys wave, 2026-09-19)
* — PURE module, so this file owns no fs. Owner decisions being pinned:
* - the catalog is exactly the four known keys, ids = env-var names;
* - validation is SHAPE-ONLY (no live pings) and bad pastes are 400s;
* - masking is last-4 + length — plaintext never leaves the store.
*/
import assert from "node:assert/strict";
import { describe, it } from "node:test";
import { AUTH_KEYS, genericProviderKeyGuard, isKnownAuthKey, maskKey, validateKeyShape } from "../src/core/auth-keys.ts";
describe("the catalog", () => {
it("is exactly the three CURATED app-service keys (provider keys live in the provider table)", () => {
assert.deepEqual(
AUTH_KEYS.map((k) => k.id),
["DEEPGRAM_API_KEY", "GOOGLE_SEARCH_API_KEY", "GOOGLE_CSE_ID"],
);
for (const k of AUTH_KEYS) {
assert.ok(k.label.length > 0, `${k.id} has a label`);
assert.ok(k.description.length > 0, `${k.id} has a description`);
assert.ok(k.docsUrl.startsWith("https://"), `${k.id} has a docs link`);
}
// OPENROUTER_API_KEY graduated to the provider table (owner decision,
// provider-keys interlude) — not curated anymore.
assert.ok(!isKnownAuthKey("OPENROUTER_API_KEY"));
});
it("isKnownAuthKey: catalog yes, everything else no", () => {
assert.ok(isKnownAuthKey("DEEPGRAM_API_KEY"));
assert.ok(isKnownAuthKey("GOOGLE_CSE_ID"));
assert.ok(!isKnownAuthKey("OPENROUTER_API_KEY"));
assert.ok(!isKnownAuthKey("OPENAI_API_KEY"), "provider vars are a separate surface");
assert.ok(!isKnownAuthKey("FUTURE_PLANS_TOKEN"));
assert.ok(!isKnownAuthKey(""));
});
});
describe("validateKeyShape (the 400 gate — curated keys)", () => {
it("Google API key: AIza prefix required", () => {
assert.ok(validateKeyShape("GOOGLE_SEARCH_API_KEY", "AIzaSyD-abcdef1234").ok);
assert.ok(!validateKeyShape("GOOGLE_SEARCH_API_KEY", "sk-or-v1-abcdefghij").ok, "wrong family");
assert.ok(!validateKeyShape("GOOGLE_SEARCH_API_KEY", "AIza").ok, "prefix alone fails");
});
it("CSE ID: legacy digits:digits OR plain token form", () => {
assert.ok(validateKeyShape("GOOGLE_CSE_ID", "017576662512468239146:omuauf_lfve").ok);
assert.ok(validateKeyShape("GOOGLE_CSE_ID", "a1b2c3d4e5f6g7h8j9k0lmnop").ok);
assert.ok(!validateKeyShape("GOOGLE_CSE_ID", "17:short").ok, "short legacy form fails");
assert.ok(!validateKeyShape("GOOGLE_CSE_ID", "has whitespace here").ok);
assert.ok(!validateKeyShape("GOOGLE_CSE_ID", "sk-or-v1-abcdefg").ok, "wrong family");
});
it("Deepgram: no public format — a ≥20-char truncation guard, no whitespace", () => {
assert.ok(validateKeyShape("DEEPGRAM_API_KEY", "a".repeat(40)).ok);
assert.ok(!validateKeyShape("DEEPGRAM_API_KEY", "a".repeat(19)).ok, "truncated paste");
assert.ok(!validateKeyShape("DEEPGRAM_API_KEY", "a b c d e f g h i j k l m n o p q r s").ok, "whitespace");
});
it("trims surrounding paste whitespace before checking", () => {
assert.ok(validateKeyShape("DEEPGRAM_API_KEY", ` ${"d".repeat(40)} `).ok);
});
it("unknown ids are rejected with the supported list", () => {
const r = validateKeyShape("SOMETHING_ELSE", "whatever");
assert.ok(!r.ok);
if (!r.ok) assert.match(r.error, /SOMETHING_ELSE.*DEEPGRAM_API_KEY/);
});
});
describe("genericProviderKeyGuard (model-provider keys — paste hygiene only)", () => {
it("accepts any ≥12-char whitespace-free value (families vary)", () => {
assert.deepEqual(genericProviderKeyGuard("OPENROUTER_API_KEY", "sk-or-v1-abc123"), { ok: true });
assert.deepEqual(genericProviderKeyGuard("OPENAI_API_KEY", "whaterver-value-here"), { ok: true });
assert.deepEqual(genericProviderKeyGuard("HF_TOKEN", "hf_xxxxxxxx9999"), { ok: true });
});
it("rejects truncations and whitespace — the only guards", () => {
const short = genericProviderKeyGuard("ZAI_API_KEY", "short");
assert.ok(!short.ok);
if (!short.ok) assert.match(short.error, /at least 12/);
assert.ok(!genericProviderKeyGuard("OPENAI_API_KEY", "has whitespace now").ok);
assert.ok(!genericProviderKeyGuard("OPENAI_API_KEY", " ").ok);
});
it("trims before checking", () => {
assert.ok(genericProviderKeyGuard("OPENROUTER_API_KEY", " sk-or-v1-xyz999 ").ok);
});
});
describe("maskKey (plaintext never leaves)", () => {
it("shows the last 4 only", () => {
assert.equal(maskKey("sk-or-v1-abcdefgh1234"), "…1234");
});
it("never reveals a degenerate short value in full", () => {
assert.equal(maskKey("ab"), "…ab");
assert.equal(maskKey("abcd"), "…abcd");
assert.equal(maskKey("abcde"), "…bcde");
});
it("trims before masking", () => {
assert.equal(maskKey(" sk-or-v1-xyzw9 "), "…yzw9");
});
});