feat: add Pi management interface

2026-08-05 05:53:28 +02:00
parent 4422568f61
commit 7a8bcacbfe
6 changed files with 606 additions and 10 deletions
@@ -0,0 +1,67 @@
+import { http, HttpResponse } from "msw";
+import { server } from "../test/msw";
+import {
+  asPiManagementApiError,
+  getPiManagementLogs,
+  getPiManagementOptions,
+  getPiManagementStatus,
+  runPiManagementTest,
+  savePiManagementConfig,
+} from "./pi-management";
+
+test("Pi management client calls only the sanctioned sanitized endpoints", async () => {
+  const calls: Array<{ method: string; path: string; body?: unknown }> = [];
+  server.use(
+    http.get("/api/pi-management/status", ({ request }) => {
+      calls.push({ method: request.method, path: new URL(request.url).pathname });
+      return HttpResponse.json({ ready: true, version: "0.80.3", config: {}, checkedAt: "2026-08-05T10:00:00.000Z" });
+    }),
+    http.get("/api/pi-management/options", ({ request }) => {
+      calls.push({ method: request.method, path: new URL(request.url).pathname });
+      return HttpResponse.json({ providers: ["zai"], models: [{ provider: "zai", id: "glm-5.2" }], reasoning: ["low", "medium", "high"], checkedAt: "2026-08-05T10:00:00.000Z" });
+    }),
+    http.put("/api/pi-management/config", async ({ request }) => {
+      calls.push({ method: request.method, path: new URL(request.url).pathname, body: await request.json() });
+      return HttpResponse.json({ provider: "zai", model: "glm-5.2", reasoning: "high", updatedAt: "2026-08-05T10:00:00.000Z" });
+    }),
+    http.post("/api/pi-management/test", ({ request }) => {
+      calls.push({ method: request.method, path: new URL(request.url).pathname });
+      return HttpResponse.json({ ready: true, checkedAt: "2026-08-05T10:00:00.000Z" });
+    }),
+    http.get("/api/pi-management/logs", ({ request }) => {
+      calls.push({ method: request.method, path: new URL(request.url).pathname });
+      return HttpResponse.json({ lines: ["Pi smoke check succeeded"], checkedAt: "2026-08-05T10:00:00.000Z" });
+    }),
+  );
+
+  await expect(getPiManagementStatus()).resolves.toMatchObject({ version: "0.80.3", ready: true });
+  await expect(getPiManagementOptions()).resolves.toMatchObject({ providers: ["zai"] });
+  await expect(savePiManagementConfig({ provider: "zai", model: "glm-5.2", reasoning: "high" })).resolves.toMatchObject({ reasoning: "high" });
+  await expect(runPiManagementTest()).resolves.toMatchObject({ ready: true });
+  await expect(getPiManagementLogs()).resolves.toMatchObject({ lines: ["Pi smoke check succeeded"] });
+
+  expect(calls).toEqual([
+    { method: "GET", path: "/api/pi-management/status" },
+    { method: "GET", path: "/api/pi-management/options" },
+    { method: "PUT", path: "/api/pi-management/config", body: { provider: "zai", model: "glm-5.2", reasoning: "high" } },
+    { method: "POST", path: "/api/pi-management/test" },
+    { method: "GET", path: "/api/pi-management/logs" },
+  ]);
+  expect(calls.some((call) => /update|terminal|shell/i.test(call.path))).toBe(false);
+});
+
+test("Pi management client exposes the stable forbidden message without raw response text", async () => {
+  server.use(http.get("/api/pi-management/status", () =>
+    HttpResponse.json({ code: "pi_management_forbidden", error: "Pi management is not permitted", raw: "token=do-not-show" }, { status: 403 }),
+  ));
+
+  await expect(getPiManagementStatus()).rejects.toSatisfy((error: unknown) => {
+    expect(asPiManagementApiError(error)).toEqual({
+      status: 403,
+      code: "pi_management_forbidden",
+      message: "Pi management is not permitted",
+    });
+    expect(asPiManagementApiError(error)?.message).not.toContain("token");
+    return true;
+  });
+});