feat: integrate Pi backend and Pi Web

2026-10-03 02:59:35 +00:00 · 2026-07-30 19:37:53 +08:00
commit 7392ab9dd7
1390 changed files with 337197 additions and 0 deletions
@@ -0,0 +1,139 @@
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { pollOAuthDeviceCodeFlow } from "../src/auth/oauth/device-code.ts";
+
+describe("OAuth device-code polling", () => {
+	afterEach(() => {
+		vi.useRealTimers();
+	});
+
+	it("polls immediately and returns the completed value", async () => {
+		vi.useFakeTimers();
+		vi.setSystemTime(new Date("2026-03-09T00:00:00Z"));
+
+		const pollTimes: number[] = [];
+		const poll = vi.fn(async () => {
+			pollTimes.push(Date.now());
+			return pollTimes.length === 1
+				? { status: "pending" as const }
+				: { status: "complete" as const, value: "token" };
+		});
+
+		const resultPromise = pollOAuthDeviceCodeFlow({
+			intervalSeconds: 2,
+			expiresInSeconds: 30,
+			poll,
+		});
+
+		await vi.advanceTimersByTimeAsync(0);
+		expect(pollTimes).toEqual([new Date("2026-03-09T00:00:00Z").getTime()]);
+
+		await vi.advanceTimersByTimeAsync(1999);
+		expect(pollTimes).toEqual([new Date("2026-03-09T00:00:00Z").getTime()]);
+
+		await vi.advanceTimersByTimeAsync(1);
+		await expect(resultPromise).resolves.toBe("token");
+		expect(pollTimes).toEqual([
+			new Date("2026-03-09T00:00:00Z").getTime(),
+			new Date("2026-03-09T00:00:02Z").getTime(),
+		]);
+	});
+
+	it("can wait before the first poll", async () => {
+		vi.useFakeTimers();
+		vi.setSystemTime(new Date("2026-03-09T00:00:00Z"));
+
+		const pollTimes: number[] = [];
+		const resultPromise = pollOAuthDeviceCodeFlow({
+			intervalSeconds: 2,
+			expiresInSeconds: 30,
+			waitBeforeFirstPoll: true,
+			poll: async () => {
+				pollTimes.push(Date.now());
+				return { status: "complete" as const, value: "token" };
+			},
+		});
+
+		await vi.advanceTimersByTimeAsync(1999);
+		expect(pollTimes).toEqual([]);
+
+		await vi.advanceTimersByTimeAsync(1);
+		await expect(resultPromise).resolves.toBe("token");
+		expect(pollTimes).toEqual([new Date("2026-03-09T00:00:02Z").getTime()]);
+	});
+
+	it("increases the interval by 5 seconds after slow_down without a server interval", async () => {
+		vi.useFakeTimers();
+		vi.setSystemTime(new Date("2026-03-09T00:00:00Z"));
+		const startTime = Date.now();
+
+		const pollTimes: number[] = [];
+		const results = [{ status: "slow_down" as const }, { status: "complete" as const, value: "token" }];
+		const resultPromise = pollOAuthDeviceCodeFlow({
+			intervalSeconds: 2,
+			expiresInSeconds: 900,
+			poll: async () => {
+				pollTimes.push(Date.now());
+				const result = results.shift();
+				if (!result) throw new Error("Unexpected extra poll");
+				return result;
+			},
+		});
+
+		await vi.advanceTimersByTimeAsync(0);
+		expect(pollTimes).toEqual([startTime]);
+
+		await vi.advanceTimersByTimeAsync(6999);
+		expect(pollTimes).toEqual([startTime]);
+
+		await vi.advanceTimersByTimeAsync(1);
+		await expect(resultPromise).resolves.toBe("token");
+		expect(pollTimes).toEqual([startTime, startTime + 7000]);
+	});
+
+	it("honors a server-provided slow_down interval", async () => {
+		vi.useFakeTimers();
+		vi.setSystemTime(new Date("2026-03-09T00:00:00Z"));
+		const startTime = Date.now();
+
+		const pollTimes: number[] = [];
+		const results = [
+			{ status: "slow_down" as const, intervalSeconds: 30 },
+			{ status: "complete" as const, value: "token" },
+		];
+		const resultPromise = pollOAuthDeviceCodeFlow({
+			intervalSeconds: 2,
+			expiresInSeconds: 900,
+			poll: async () => {
+				pollTimes.push(Date.now());
+				const result = results.shift();
+				if (!result) throw new Error("Unexpected extra poll");
+				return result;
+			},
+		});
+
+		await vi.advanceTimersByTimeAsync(0);
+		expect(pollTimes).toEqual([startTime]);
+
+		await vi.advanceTimersByTimeAsync(29999);
+		expect(pollTimes).toEqual([startTime]);
+
+		await vi.advanceTimersByTimeAsync(1);
+		await expect(resultPromise).resolves.toBe("token");
+		expect(pollTimes).toEqual([startTime, startTime + 30000]);
+	});
+
+	it("cancels an in-flight wait", async () => {
+		vi.useFakeTimers();
+		const controller = new AbortController();
+
+		const resultPromise = pollOAuthDeviceCodeFlow({
+			intervalSeconds: 5,
+			expiresInSeconds: 30,
+			poll: async () => ({ status: "pending" }),
+			signal: controller.signal,
+		});
+
+		controller.abort();
+		await expect(resultPromise).rejects.toThrow("Login cancelled");
+	});
+});