fix: respect Pi session directories by cwd

src/server/sessions/piSessionManagerGateway.test.ts

@@ -0,0 +1,85 @@
+import { mkdir, mkdtemp, rm, writeFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import { createPiSessionManagerGateway, defaultPiSessionDir, defaultPiSessionsRoot, SessionDirResolver } from "./piSessionManagerGateway.js";
+import type { PiSessionManager } from "./piSessionService.js";
+
+let tempDir: string;
+let agentDir: string;
+let cwd: string;
+
+beforeEach(async () => {
+  tempDir = await mkdtemp(join(tmpdir(), "pi-web-session-gateway-test-"));
+  agentDir = join(tempDir, "agent");
+  cwd = join(tempDir, "workspace");
+  await mkdir(cwd, { recursive: true });
+});
+
+afterEach(async () => {
+  await rm(tempDir, { recursive: true, force: true });
+});
+
+describe("SessionDirResolver", () => {
+  it("uses Pi default session storage when no Pi override is configured", () => {
+    const resolver = new SessionDirResolver({ agentDir, env: {} });
+
+    expect(resolver.resolve(cwd)).toMatchObject({ source: "pi-default", sessionDir: defaultPiSessionDir(cwd, agentDir), usesConfiguredSessionDir: false });
+    expect(defaultPiSessionsRoot(agentDir)).toBe(join(agentDir, "sessions"));
+  });
+
+  it("uses Pi sessionDir settings and resolves relative paths against the session cwd", async () => {
+    await mkdir(agentDir, { recursive: true });
+    await writeFile(join(agentDir, "settings.json"), `${JSON.stringify({ sessionDir: ".pi/sessions" }, null, 2)}\n`, "utf8");
+
+    const resolver = new SessionDirResolver({ agentDir, env: {} });
+
+    expect(resolver.resolve(cwd)).toMatchObject({ source: "settings", sessionDir: join(cwd, ".pi", "sessions"), usesConfiguredSessionDir: true });
+  });
+
+  it("lets project-local Pi sessionDir settings override global Pi settings for that cwd", async () => {
+    await mkdir(agentDir, { recursive: true });
+    await mkdir(join(cwd, ".pi"), { recursive: true });
+    await writeFile(join(agentDir, "settings.json"), `${JSON.stringify({ sessionDir: join(tempDir, "global-sessions") }, null, 2)}\n`, "utf8");
+    await writeFile(join(cwd, ".pi", "settings.json"), `${JSON.stringify({ sessionDir: ".workspace-sessions" }, null, 2)}\n`, "utf8");
+
+    const resolver = new SessionDirResolver({ agentDir, env: {} });
+
+    expect(resolver.resolve(cwd)).toMatchObject({ source: "settings", sessionDir: join(cwd, ".workspace-sessions"), usesConfiguredSessionDir: true });
+  });
+
+  it("lets the Pi sessionDir environment override Pi settings", async () => {
+    const envDir = join(tempDir, "env-sessions");
+    await mkdir(agentDir, { recursive: true });
+    await writeFile(join(agentDir, "settings.json"), `${JSON.stringify({ sessionDir: join(tempDir, "settings-sessions") }, null, 2)}\n`, "utf8");
+
+    const resolver = new SessionDirResolver({ agentDir, env: { PI_CODING_AGENT_SESSION_DIR: envDir } });
+
+    expect(resolver.resolve(cwd)).toMatchObject({ source: "env", sessionDir: envDir, usesConfiguredSessionDir: true });
+  });
+});
+
+describe("Pi session manager gateway", () => {
+  it("lists only sessions for the requested cwd when a custom Pi sessionDir is shared", async () => {
+    const sharedSessionDir = join(tempDir, "shared-sessions");
+    const otherCwd = join(tempDir, "other-workspace");
+    await writeSessionFile(sharedSessionDir, "session-a", cwd);
+    await writeSessionFile(sharedSessionDir, "session-b", otherCwd);
+    const gateway = createPiSessionManagerGateway({ agentDir, env: { PI_CODING_AGENT_SESSION_DIR: sharedSessionDir } });
+
+    await expect(gateway.list(cwd)).resolves.toMatchObject([{ id: "session-a", cwd }]);
+    const created = gateway.create(cwd);
+    expect(hasSessionDir(created)).toBe(true);
+    if (!hasSessionDir(created)) throw new Error("Expected SDK session manager");
+    expect(created.getSessionDir()).toBe(sharedSessionDir);
+  });
+});
+
+function hasSessionDir(manager: PiSessionManager): manager is PiSessionManager & { getSessionDir(): string } {
+  return "getSessionDir" in manager && typeof manager.getSessionDir === "function";
+}
+
+async function writeSessionFile(dir: string, id: string, sessionCwd: string): Promise<void> {
+  await mkdir(dir, { recursive: true });
+  await writeFile(join(dir, `${id}.jsonl`), `${JSON.stringify({ type: "session", version: 3, id, timestamp: "2026-01-01T00:00:00.000Z", cwd: sessionCwd })}\n`, "utf8");
+}