@hayasaka7/haya-pet 0.3.6 → 0.3.8

package/packages/cli-core/test/run-state.test.mjs

@@ -3,7 +3,8 @@ import { mkdtempSync, readFileSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
 import { test } from "../../../test/harness.mjs";
-import { parseStateArgs, runStateCommand } from "../src/run-state.js";
+import { extractTranscriptPath, parseStateArgs, runStateCommand } from "../src/run-state.js";
+import { readSessionTranscriptLink } from "../src/session-transcript-link.js";
 
 test("runStateCommand appends a debug line when HAYA_PET_HOOK_DEBUG is set", async () => {
   const logPath = join(mkdtempSync(join(tmpdir(), "haya-dbg-")), "hooks.jsonl");
@@ -21,6 +22,51 @@ test("runStateCommand appends a debug line when HAYA_PET_HOOK_DEBUG is set", asy
   assert.deepEqual(line, { ts: 7, state: "waiting_approval", sessionId: "s1", summary: "approval" });
 });
 
+test("extractTranscriptPath pulls transcript_path out of a Claude hook payload", () => {
+  assert.equal(
+    extractTranscriptPath(JSON.stringify({ session_id: "x", transcript_path: "/p/a.jsonl", cwd: "/p" })),
+    "/p/a.jsonl"
+  );
+  // Defensive: junk, missing field, wrong type, and empty all yield undefined.
+  assert.equal(extractTranscriptPath("{not json"), undefined);
+  assert.equal(extractTranscriptPath(JSON.stringify({ session_id: "x" })), undefined);
+  assert.equal(extractTranscriptPath(JSON.stringify({ transcript_path: 42 })), undefined);
+  assert.equal(extractTranscriptPath(""), undefined);
+  assert.equal(extractTranscriptPath(undefined), undefined);
+});
+
+test("runStateCommand records the session->transcript link when given a transcript path", async () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  await runStateCommand(
+    { command: "state", state: "thinking", summary: undefined, session: "sess_link" },
+    {
+      now: () => 1,
+      sessionDir,
+      transcriptPath: "/p/.claude/projects/D--p/abc.jsonl",
+      createIpcClient: async () => ({ send: async () => {}, close: async () => {} })
+    }
+  );
+
+  assert.equal(
+    readSessionTranscriptLink({ sessionDir, sessionId: "sess_link" }),
+    "/p/.claude/projects/D--p/abc.jsonl"
+  );
+});
+
+test("runStateCommand writes no link when no transcript path is supplied", async () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  await runStateCommand(
+    { command: "state", state: "thinking", summary: undefined, session: "sess_nolink" },
+    {
+      now: () => 1,
+      sessionDir,
+      createIpcClient: async () => ({ send: async () => {}, close: async () => {} })
+    }
+  );
+
+  assert.equal(readSessionTranscriptLink({ sessionDir, sessionId: "sess_nolink" }), undefined);
+});
+
 test("parseStateArgs reads state, summary, and session", () => {
   assert.deepEqual(parseStateArgs(["thinking"]), {
     command: "state",

package/packages/cli-core/test/session-transcript-link.test.mjs

@@ -0,0 +1,67 @@
+import assert from "node:assert/strict";
+import { existsSync, mkdtempSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { test } from "../../../test/harness.mjs";
+import {
+  readSessionTranscriptLink,
+  removeSessionTranscriptLink,
+  sessionLinkPath,
+  writeSessionTranscriptLink
+} from "../src/session-transcript-link.js";
+
+test("write then read round-trips a session's transcript path", () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  const transcriptPath = "D:\\proj\\.claude\\projects\\D--proj\\abc.jsonl";
+
+  const wrote = writeSessionTranscriptLink({ sessionDir, sessionId: "sess_a", transcriptPath });
+  assert.equal(wrote, true);
+  assert.equal(readSessionTranscriptLink({ sessionDir, sessionId: "sess_a" }), transcriptPath);
+});
+
+test("two sessions in the same dir keep separate, independent links", () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  writeSessionTranscriptLink({ sessionDir, sessionId: "sess_a", transcriptPath: "/p/a.jsonl" });
+  writeSessionTranscriptLink({ sessionDir, sessionId: "sess_b", transcriptPath: "/p/b.jsonl" });
+
+  // The core of the bug fix: each session resolves ONLY its own transcript.
+  assert.equal(readSessionTranscriptLink({ sessionDir, sessionId: "sess_a" }), "/p/a.jsonl");
+  assert.equal(readSessionTranscriptLink({ sessionDir, sessionId: "sess_b" }), "/p/b.jsonl");
+});
+
+test("reading a session with no link returns undefined (no guessing)", () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  assert.equal(readSessionTranscriptLink({ sessionDir, sessionId: "missing" }), undefined);
+});
+
+test("write is a no-op without the required fields", () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  assert.equal(writeSessionTranscriptLink({ sessionDir, sessionId: "s" }), false);
+  assert.equal(writeSessionTranscriptLink({ sessionDir, transcriptPath: "/p/a.jsonl" }), false);
+  assert.equal(writeSessionTranscriptLink({ sessionId: "s", transcriptPath: "/p/a.jsonl" }), false);
+});
+
+test("remove deletes the link and is safe when it is already gone", () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  writeSessionTranscriptLink({ sessionDir, sessionId: "sess_a", transcriptPath: "/p/a.jsonl" });
+  const path = sessionLinkPath(sessionDir, "sess_a");
+  assert.equal(existsSync(path), true);
+
+  removeSessionTranscriptLink({ sessionDir, sessionId: "sess_a" });
+  assert.equal(existsSync(path), false);
+  // Second removal must not throw.
+  removeSessionTranscriptLink({ sessionDir, sessionId: "sess_a" });
+});
+
+test("a corrupt link file reads as undefined rather than throwing", () => {
+  const sessionDir = mkdtempSync(join(tmpdir(), "sess-"));
+  writeSessionTranscriptLink(
+    { sessionDir, sessionId: "sess_a", transcriptPath: "/p/a.jsonl" },
+    { writeFileSync: () => {} } // pretend write; nothing on disk
+  );
+  // Inject a reader returning junk to simulate a partially-written file.
+  assert.equal(
+    readSessionTranscriptLink({ sessionDir, sessionId: "sess_a" }, { readFileSync: () => "{not json" }),
+    undefined
+  );
+});

package/packages/platform-core/src/paths.js

@@ -30,6 +30,7 @@ function getWindowsPaths(env, homeDir) {
     statePath: joinWindows(localAppData, "haya-pet", "state.json"),
     configPath: joinWindows(appData, "haya-pet", "config.json"),
     logDir: joinWindows(localAppData, "haya-pet", "logs"),
+    sessionDir: joinWindows(localAppData, "haya-pet", "sessions"),
     petSearchPaths: [
       joinWindows(homeDir, ".codex", "pets"),
       joinWindows(localAppData, "haya-pet", "pets")
@@ -44,6 +45,7 @@ function getUnixPaths(homeDir) {
     statePath: joinUnix(homeDir, ".haya-pet", "state.json"),
     configPath: joinUnix(homeDir, ".haya-pet", "config.json"),
     logDir: joinUnix(homeDir, ".haya-pet", "logs"),
+    sessionDir: joinUnix(homeDir, ".haya-pet", "sessions"),
     petSearchPaths: [
       joinUnix(homeDir, ".codex", "pets"),
       joinUnix(homeDir, ".haya-pet", "pets")
@@ -58,6 +60,7 @@ function getUnsupportedPaths(homeDir) {
     statePath: joinUnix(homeDir, ".haya-pet", "state.json"),
     configPath: joinUnix(homeDir, ".haya-pet", "config.json"),
     logDir: joinUnix(homeDir, ".haya-pet", "logs"),
+    sessionDir: joinUnix(homeDir, ".haya-pet", "sessions"),
     petSearchPaths: [
       joinUnix(homeDir, ".codex", "pets"),
       joinUnix(homeDir, ".haya-pet", "pets")