@axhub/genie 0.2.5 → 0.2.7

package/shared/conversationEvents.test.js

@@ -0,0 +1,403 @@
+import assert from 'node:assert/strict';
+import test from 'node:test';
+
+import {
+  applyConversationEventToTimelineMessages,
+  CONVERSATION_EVENT_KINDS,
+  createConversationEvent,
+  extractAcpSessionMetadataFromConversationEvents,
+  normalizeLegacyHistoryEntries,
+  normalizeRealtimePayloadToConversationEvents
+} from './conversationEvents.js';
+
+test('createConversationEvent returns normalized event payload', () => {
+  const event = createConversationEvent({
+    kind: CONVERSATION_EVENT_KINDS.USER_MESSAGE,
+    provider: 'claude',
+    sessionId: 'session-1',
+    payload: {
+      text: 'hello'
+    }
+  });
+
+  assert.equal(event.kind, CONVERSATION_EVENT_KINDS.USER_MESSAGE);
+  assert.equal(event.provider, 'claude');
+  assert.equal(event.sessionId, 'session-1');
+  assert.equal(event.payload.text, 'hello');
+  assert.match(event.eventId, /^user_message:/);
+});
+
+test('applyConversationEventToTimelineMessages builds assistant and tool messages', () => {
+  const events = [
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_START,
+      provider: 'claude',
+      sessionId: 'session-2',
+      payload: { messageId: 'assistant:1' }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_DELTA,
+      provider: 'claude',
+      sessionId: 'session-2',
+      payload: { messageId: 'assistant:1', text: 'Hello from ACP.' }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_END,
+      provider: 'claude',
+      sessionId: 'session-2',
+      payload: { messageId: 'assistant:1' }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.TOOL_CALL_START,
+      provider: 'claude',
+      sessionId: 'session-2',
+      payload: {
+        toolCallId: 'tool:1',
+        toolName: 'WriteFile',
+        title: 'Write file',
+        kind: 'edit',
+        status: 'in_progress'
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.TOOL_CALL_INPUT,
+      provider: 'claude',
+      sessionId: 'session-2',
+      payload: {
+        toolCallId: 'tool:1',
+        input: { path: 'README.md' },
+        rawInput: { path: 'README.md' },
+        locations: [{ path: 'README.md', line: 1 }]
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.TOOL_CALL_END,
+      provider: 'claude',
+      sessionId: 'session-2',
+      payload: {
+        toolCallId: 'tool:1',
+        toolName: 'WriteFile',
+        status: 'completed'
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.TOOL_RESULT,
+      provider: 'claude',
+      sessionId: 'session-2',
+      payload: {
+        toolCallId: 'tool:1',
+        content: 'updated README',
+        contentBlocks: [
+          {
+            type: 'diff',
+            path: 'README.md',
+            oldText: 'old',
+            newText: 'new'
+          }
+        ],
+        isError: false
+      }
+    })
+  ];
+
+  const messages = events.reduce(
+    (timeline, event) => applyConversationEventToTimelineMessages(timeline, event, 'claude'),
+    []
+  );
+
+  assert.equal(messages.length, 2);
+  assert.equal(messages[0].content, 'Hello from ACP.');
+  assert.equal(messages[0].isStreaming, false);
+  assert.equal(messages[1].isToolUse, true);
+  assert.equal(messages[1].toolName, 'WriteFile');
+  assert.equal(messages[1].toolResult.content, 'updated README');
+  assert.equal(messages[1].acpToolCall.kind, 'edit');
+  assert.equal(messages[1].acpToolCall.status, 'completed');
+  assert.equal(messages[1].acpToolCall.locations[0].path, 'README.md');
+  assert.equal(messages[1].acpToolCall.content[0].type, 'diff');
+});
+
+test('applyConversationEventToTimelineMessages marks tool call end even without a tool result', () => {
+  const events = [
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.TOOL_CALL_START,
+      provider: 'codex',
+      sessionId: 'session-tool-end',
+      timestamp: '2026-03-30T09:10:34.879Z',
+      payload: {
+        toolCallId: 'tool:web:1',
+        toolName: 'Searching the Web'
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.TOOL_CALL_INPUT,
+      provider: 'codex',
+      sessionId: 'session-tool-end',
+      timestamp: '2026-03-30T09:10:35.725Z',
+      payload: {
+        toolCallId: 'tool:web:1',
+        toolName: 'Searching for: weather: Guangzhou',
+        input: {
+          query: 'weather: Guangzhou'
+        }
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.TOOL_CALL_END,
+      provider: 'codex',
+      sessionId: 'session-tool-end',
+      timestamp: '2026-03-30T09:10:36.142Z',
+      payload: {
+        toolCallId: 'tool:web:1',
+        toolName: 'Searching for: weather: Guangzhou'
+      }
+    })
+  ];
+
+  const messages = events.reduce(
+    (timeline, event) => applyConversationEventToTimelineMessages(timeline, event, 'codex'),
+    []
+  );
+
+  assert.equal(messages.length, 1);
+  assert.equal(messages[0].isToolUse, true);
+  assert.equal(messages[0].toolResult, undefined);
+  assert.equal(messages[0].toolEndedAt, '2026-03-30T09:10:36.142Z');
+});
+
+test('applyConversationEventToTimelineMessages merges echoed user events into optimistic messages', () => {
+  const optimisticMessage = {
+    type: 'user',
+    content: '广州天气怎么样?',
+    timestamp: '2026-03-30T09:00:00.000Z',
+    provider: 'codex',
+    clientRequestId: 'req-user-1',
+    contentBlocks: [
+      {
+        type: 'resource_link',
+        uri: '/tmp/weather.md',
+        name: 'weather.md'
+      }
+    ]
+  };
+
+  const echoedEvent = createConversationEvent({
+    kind: CONVERSATION_EVENT_KINDS.USER_MESSAGE,
+    provider: 'codex',
+    sessionId: 'session-user',
+    timestamp: '2026-03-30T09:00:01.000Z',
+    payload: {
+      text: '广州天气怎么样?',
+      contentBlocks: [
+        {
+          type: 'resource_link',
+          uri: '/tmp/weather.md',
+          name: 'weather.md'
+        }
+      ]
+    },
+    extensions: {
+      clientRequestId: 'req-user-1'
+    }
+  });
+
+  const messages = applyConversationEventToTimelineMessages([optimisticMessage], echoedEvent, 'codex');
+
+  assert.equal(messages.length, 1);
+  assert.equal(messages[0].content, '广州天气怎么样?');
+  assert.equal(messages[0].clientRequestId, 'req-user-1');
+  assert.equal(messages[0].eventId, echoedEvent.eventId);
+  assert.equal(messages[0].provider, 'codex');
+});
+
+test('applyConversationEventToTimelineMessages replaces ACP plan entries in-place', () => {
+  const firstPlan = createConversationEvent({
+    kind: CONVERSATION_EVENT_KINDS.PLAN_UPDATE,
+    provider: 'claude',
+    sessionId: 'session-plan',
+    payload: {
+      entries: [
+        { content: 'Read files', status: 'completed', priority: 'medium' },
+        { content: 'Implement UI', status: 'in_progress', priority: 'high' }
+      ],
+      message: 'Implementation plan updated'
+    }
+  });
+
+  const secondPlan = createConversationEvent({
+    kind: CONVERSATION_EVENT_KINDS.PLAN_UPDATE,
+    provider: 'claude',
+    sessionId: 'session-plan',
+    payload: {
+      entries: [
+        { content: 'Read files', status: 'completed', priority: 'medium' },
+        { content: 'Implement UI', status: 'completed', priority: 'high' }
+      ],
+      message: 'Implementation plan updated'
+    }
+  });
+
+  let messages = applyConversationEventToTimelineMessages([], firstPlan, 'claude');
+  messages = applyConversationEventToTimelineMessages(messages, secondPlan, 'claude');
+
+  assert.equal(messages.length, 1);
+  assert.equal(messages[0].isPlanUpdate, true);
+  assert.equal(messages[0].planEntries.length, 2);
+  assert.equal(messages[0].planEntries[1].status, 'completed');
+});
+
+test('applyConversationEventToTimelineMessages keeps assistant content blocks on the same message', () => {
+  const events = [
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.ASSISTANT_CONTENT_BLOCK,
+      provider: 'claude',
+      sessionId: 'session-content',
+      payload: {
+        messageId: 'assistant:content:1',
+        contentBlock: {
+          type: 'resource_link',
+          uri: '/tmp/spec.md',
+          name: 'spec.md'
+        }
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_START,
+      provider: 'claude',
+      sessionId: 'session-content',
+      payload: {
+        messageId: 'assistant:content:1'
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_DELTA,
+      provider: 'claude',
+      sessionId: 'session-content',
+      payload: {
+        messageId: 'assistant:content:1',
+        text: 'Attached the spec for review.'
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_END,
+      provider: 'claude',
+      sessionId: 'session-content',
+      payload: {
+        messageId: 'assistant:content:1'
+      }
+    })
+  ];
+
+  const messages = events.reduce(
+    (timeline, event) => applyConversationEventToTimelineMessages(timeline, event, 'claude'),
+    []
+  );
+
+  assert.equal(messages.length, 1);
+  assert.equal(messages[0].content, 'Attached the spec for review.');
+  assert.equal(messages[0].contentBlocks.length, 1);
+  assert.equal(messages[0].contentBlocks[0].type, 'resource_link');
+});
+
+test('extractAcpSessionMetadataFromConversationEvents keeps latest modes and commands', () => {
+  const metadata = extractAcpSessionMetadataFromConversationEvents([
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.MODE_UPDATE,
+      provider: 'claude',
+      sessionId: 'session-meta',
+      payload: {
+        availableModes: [
+          { id: 'ask', name: 'Ask', description: 'Answer questions' },
+          { id: 'code', name: 'Code', description: 'Edit files' }
+        ],
+        currentModeId: 'ask'
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.AVAILABLE_COMMANDS_UPDATE,
+      provider: 'claude',
+      sessionId: 'session-meta',
+      payload: {
+        availableCommands: [
+          { name: 'create_plan', description: 'Create a plan' }
+        ]
+      }
+    }),
+    createConversationEvent({
+      kind: CONVERSATION_EVENT_KINDS.MODE_UPDATE,
+      provider: 'claude',
+      sessionId: 'session-meta',
+      payload: {
+        currentModeId: 'code'
+      }
+    })
+  ]);
+
+  assert.equal(metadata.modeState.currentModeId, 'code');
+  assert.equal(metadata.modeState.availableModes.length, 2);
+  assert.equal(metadata.availableCommands[0].name, 'create_plan');
+  assert.equal(metadata.availableCommands[0].source, 'acp');
+});
+
+test('normalizeLegacyHistoryEntries keeps user and assistant content across providers', () => {
+  const rawMessages = [
+    {
+      timestamp: '2026-03-30T10:00:00.000Z',
+      message: {
+        role: 'user',
+        content: 'Please summarize the diff.'
+      }
+    },
+    {
+      timestamp: '2026-03-30T10:00:01.000Z',
+      message: {
+        role: 'assistant',
+        content: 'Summary ready.'
+      }
+    }
+  ];
+
+  for (const provider of ['claude', 'codex', 'gemini', 'opencode']) {
+    const events = normalizeLegacyHistoryEntries(rawMessages, provider, `session-${provider}`);
+    assert.equal(events[0].kind, CONVERSATION_EVENT_KINDS.USER_MESSAGE);
+    assert.equal(events[0].payload.text, 'Please summarize the diff.');
+    assert.equal(events[1].kind, CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_START);
+    assert.equal(events[2].kind, CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_DELTA);
+    assert.equal(events[2].payload.text, 'Summary ready.');
+    assert.equal(events[3].kind, CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_END);
+  }
+});
+
+test('normalizeRealtimePayloadToConversationEvents maps realtime payloads into conversation events', () => {
+  const queuedEvents = normalizeRealtimePayloadToConversationEvents({
+    type: 'session-created',
+    provider: 'claude',
+    sessionId: 'session-3',
+    modes: {
+      availableModes: [
+        { id: 'ask', name: 'Ask' }
+      ],
+      currentModeId: 'ask'
+    }
+  }, 'claude');
+  assert.equal(queuedEvents[0].kind, CONVERSATION_EVENT_KINDS.SESSION_STATE_CHANGED);
+  assert.equal(queuedEvents[0].payload.state, 'queued');
+  assert.equal(queuedEvents[1].kind, CONVERSATION_EVENT_KINDS.MODE_UPDATE);
+  assert.equal(queuedEvents[1].payload.currentModeId, 'ask');
+
+  const deltaEvents = normalizeRealtimePayloadToConversationEvents({
+    type: 'claude-response',
+    provider: 'claude',
+    sessionId: 'session-3',
+    data: {
+      type: 'content_block_delta',
+      delta: {
+        text: 'stream chunk'
+      }
+    }
+  }, 'claude');
+
+  assert.equal(deltaEvents[0].kind, CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_START);
+  assert.equal(deltaEvents[1].kind, CONVERSATION_EVENT_KINDS.ASSISTANT_TEXT_DELTA);
+  assert.equal(deltaEvents[1].payload.text, 'stream chunk');
+});