feat: integrate Pi backend and Pi Web

2026-10-03 11:09:34 +00:00 · 2026-07-30 19:37:53 +08:00
commit 7392ab9dd7
1390 changed files with 337197 additions and 0 deletions
@@ -0,0 +1,147 @@
+import { setKeybindings } from "@earendil-works/pi-tui";
+import { beforeAll, beforeEach, describe, expect, it } from "vitest";
+import { KeybindingsManager } from "../src/core/keybindings.ts";
+import { OAuthSelectorComponent } from "../src/modes/interactive/components/oauth-selector.ts";
+import { InteractiveMode } from "../src/modes/interactive/interactive-mode.ts";
+import { initTheme } from "../src/modes/interactive/theme/theme.ts";
+import { stripAnsi } from "../src/utils/ansi.ts";
+
+describe("OAuthSelectorComponent", () => {
+	beforeAll(() => {
+		initTheme("dark");
+	});
+
+	beforeEach(() => {
+		setKeybindings(new KeybindingsManager());
+	});
+
+	it("projects provider-owned auth options without provider-specific filtering", () => {
+		const getLoginProviderOptions = (
+			InteractiveMode as unknown as {
+				prototype: {
+					getLoginProviderOptions(
+						this: object,
+						authType?: "oauth" | "api_key",
+					): Array<{ id: string; name: string; authType: string; method?: { name: string; login?: unknown } }>;
+				};
+			}
+		).prototype.getLoginProviderOptions;
+		const providers = [
+			{
+				id: "anthropic",
+				name: "Anthropic",
+				auth: {
+					oauth: { name: "Anthropic (Claude Pro/Max)", login: async () => ({}) },
+					apiKey: { name: "Anthropic API key", login: async () => ({}) },
+				},
+			},
+			{
+				id: "google-vertex",
+				name: "Google Vertex AI",
+				auth: { apiKey: { name: "Google Cloud credentials" } },
+			},
+		];
+		const fakeThis = {
+			session: {
+				modelRuntime: {
+					getProviders: () => providers,
+					getProviderAuthStatus: () => ({ configured: false }),
+					isUsingOAuth: () => false,
+				},
+			},
+		};
+
+		const apiKeyOptions = getLoginProviderOptions.call(fakeThis, "api_key");
+		expect(apiKeyOptions).toMatchObject([
+			{
+				id: "anthropic",
+				name: "Anthropic",
+				authType: "api_key",
+				method: { name: "Anthropic API key" },
+			},
+			{
+				id: "google-vertex",
+				name: "Google Vertex AI",
+				authType: "api_key",
+				method: { name: "Google Cloud credentials" },
+			},
+		]);
+		expect(getLoginProviderOptions.call(fakeThis, "oauth")).toMatchObject([
+			{ id: "anthropic", name: "Anthropic", authType: "oauth" },
+		]);
+	});
+
+	it("renders an option without compiled auth status as unconfigured", () => {
+		const selector = new OAuthSelectorComponent(
+			"login",
+			[{ id: "google", name: "Google", authType: "api_key", status: undefined }],
+			() => {},
+			() => {},
+		);
+
+		const output = stripAnsi(selector.render(120).join("\n"));
+		expect(output).toContain("unconfigured");
+		expect(output).not.toContain("✓ configured");
+	});
+
+	it("shows OAuth auth distinctly in the API key selector", () => {
+		const selector = new OAuthSelectorComponent(
+			"login",
+			[{ id: "anthropic", name: "Anthropic", authType: "api_key", status: { type: "oauth", source: "OAuth" } }],
+			() => {},
+			() => {},
+		);
+
+		const output = stripAnsi(selector.render(120).join("\n"));
+		expect(output).toContain("subscription configured");
+	});
+
+	it("shows environment API key auth as configured", () => {
+		const selector = new OAuthSelectorComponent(
+			"login",
+			[{ id: "openai", name: "OpenAI", authType: "api_key", status: { type: "api_key", source: "OPENAI_API_KEY" } }],
+			() => {},
+			() => {},
+		);
+
+		const output = stripAnsi(selector.render(120).join("\n"));
+		expect(output).toContain("✓ env: OPENAI_API_KEY");
+		expect(output).not.toContain("unconfigured");
+	});
+
+	it("shows models.json API key auth as configured", () => {
+		const selector = new OAuthSelectorComponent(
+			"login",
+			[
+				{
+					id: "local-proxy",
+					name: "local-proxy",
+					authType: "api_key",
+					status: { type: "api_key", source: "key in models.json" },
+				},
+			],
+			() => {},
+			() => {},
+		);
+
+		expect(stripAnsi(selector.render(120).join("\n"))).toContain("✓ key in models.json");
+	});
+
+	it("shows models.json command auth as configured", () => {
+		const selector = new OAuthSelectorComponent(
+			"login",
+			[
+				{
+					id: "op-proxy",
+					name: "op-proxy",
+					authType: "api_key",
+					status: { type: "api_key", source: "command in models.json" },
+				},
+			],
+			() => {},
+			() => {},
+		);
+
+		expect(stripAnsi(selector.render(120).join("\n"))).toContain("✓ command in models.json");
+	});
+});