oh-my-opencode/src/hooks/atlas/idle-event-lineage.test.ts

import { afterEach, beforeEach, describe, it } from "bun:test"
import assert from "node:assert/strict"
import { randomUUID } from "node:crypto"
import { existsSync, mkdirSync, rmSync, writeFileSync } from "node:fs"
import { tmpdir } from "node:os"
import { join } from "node:path"
import { clearBoulderState, readBoulderState, writeBoulderState } from "../../features/boulder-state"
import type { BoulderState } from "../../features/boulder-state"
import { _resetForTesting, registerAgentName, setSessionAgent, subagentSessions } from "../../features/claude-code-session-state"
import { unsafeTestValue } from "../../../test-support/unsafe-test-value"

const { createAtlasHook } = await import("./index")

describe("atlas hook idle-event session lineage", () => {
  const MAIN_SESSION_ID = "main-session-123"

  let testDirectory = ""
  let promptCalls: Array<unknown> = []

  function writeIncompleteBoulder(overrides: Partial<BoulderState> = {}): void {
    const planPath = join(testDirectory, "test-plan.md")
    writeFileSync(planPath, "# Plan\n- [ ] Task 1\n- [ ] Task 2")

    const state: BoulderState = {
      active_plan: planPath,
      started_at: "2026-01-02T10:00:00Z",
      session_ids: [MAIN_SESSION_ID],
      plan_name: "test-plan",
      ...overrides,
    }

    writeBoulderState(testDirectory, state)
  }

  function createHook(parentSessionIDs?: Record<string, string | undefined>) {
    return createAtlasHook(unsafeTestValue<Parameters<typeof createAtlasHook>[0]>({
      directory: testDirectory,
      client: {
        session: {
          get: async (input: { path: { id: string } }) => ({
            data: {
              parentID: parentSessionIDs?.[input.path.id],
            },
          }),
          messages: async () => ({ data: [] }),
          prompt: async (input: unknown) => {
            promptCalls.push(input)
            return { data: {} }
          },
          promptAsync: async (input: unknown) => {
            promptCalls.push(input)
            return { data: {} }
          },
        },
      },
    }))
  }

  beforeEach(() => {
    testDirectory = join(tmpdir(), `atlas-idle-lineage-${randomUUID()}`)
    if (!existsSync(testDirectory)) {
      mkdirSync(testDirectory, { recursive: true })
    }

    promptCalls = []
    clearBoulderState(testDirectory)
    _resetForTesting()
    registerAgentName("atlas")
    registerAgentName("sisyphus")
    subagentSessions.clear()
  })

  afterEach(() => {
    clearBoulderState(testDirectory)
    if (existsSync(testDirectory)) {
      rmSync(testDirectory, { recursive: true, force: true })
    }

    _resetForTesting()
  })

  it("does not append unrelated subagent sessions during idle", async () => {
    const unrelatedSubagentSessionID = "subagent-session-unrelated"
    const unrelatedParentSessionID = "unrelated-parent-session"

    writeIncompleteBoulder()
    subagentSessions.add(unrelatedSubagentSessionID)

    const hook = createHook({
      [unrelatedSubagentSessionID]: unrelatedParentSessionID,
    })

    await hook.handler({
      event: {
        type: "session.idle",
        properties: { sessionID: unrelatedSubagentSessionID },
      },
    })

    assert.equal(readBoulderState(testDirectory)?.session_ids.includes(unrelatedSubagentSessionID), false)
    assert.equal(promptCalls.length, 0)
  })

  it("does not append lineage-only subagent sessions during idle even when lineage reaches tracked session", async () => {
    const subagentSessionID = "subagent-session-456"
    const intermediateParentSessionID = "subagent-parent-789"

    writeIncompleteBoulder()
    subagentSessions.add(subagentSessionID)
    setSessionAgent(subagentSessionID, "atlas")

    const hook = createHook({
      [subagentSessionID]: intermediateParentSessionID,
      [intermediateParentSessionID]: MAIN_SESSION_ID,
    })

    await hook.handler({
      event: {
        type: "session.idle",
        properties: { sessionID: subagentSessionID },
      },
    })

    assert.equal(readBoulderState(testDirectory)?.session_ids.includes(subagentSessionID), false)
    assert.equal(promptCalls.length, 0)
  })

  it("does not inject continuation for lineage-only subagent with non-matching agent", async () => {
    const subagentSessionID = "subagent-session-agent-mismatch"

    writeIncompleteBoulder({ agent: "atlas" })
    subagentSessions.add(subagentSessionID)
    setSessionAgent(subagentSessionID, "sisyphus-junior")

    const hook = createHook({
      [subagentSessionID]: MAIN_SESSION_ID,
    })

    await hook.handler({
      event: {
        type: "session.idle",
        properties: { sessionID: subagentSessionID },
      },
    })

    assert.equal(readBoulderState(testDirectory)?.session_ids.includes(subagentSessionID), false)
    assert.equal(promptCalls.length, 0)
  })

  it("does not inject continuation for lineage-only subagent with matching agent until explicitly tracked", async () => {
    const subagentSessionID = "subagent-session-agent-match"

    writeIncompleteBoulder({ agent: "atlas" })
    subagentSessions.add(subagentSessionID)
    setSessionAgent(subagentSessionID, "atlas")

    const hook = createHook({
      [subagentSessionID]: MAIN_SESSION_ID,
    })

    await hook.handler({
      event: {
        type: "session.idle",
        properties: { sessionID: subagentSessionID },
      },
    })

    assert.equal(readBoulderState(testDirectory)?.session_ids.includes(subagentSessionID), false)
    assert.equal(promptCalls.length, 0)
  })

  it("injects continuation for explicitly tracked boulder session regardless of agent", async () => {
    writeIncompleteBoulder({ agent: "atlas" })
    setSessionAgent(MAIN_SESSION_ID, "hephaestus")

    const hook = createHook()

    await hook.handler({
      event: {
        type: "session.idle",
        properties: { sessionID: MAIN_SESSION_ID },
      },
    })

    assert.equal(promptCalls.length, 1)
  })
})