fix: add session cursor for tool output

src/shared/session-cursor.ts

@@ -0,0 +1,84 @@
+type MessageTime =
+  | { created?: number | string }
+  | number
+  | string
+  | undefined
+
+type MessageInfo = {
+  id?: string
+  time?: MessageTime
+}
+
+export type CursorMessage = {
+  info?: MessageInfo
+}
+
+interface CursorState {
+  lastKey?: string
+  lastCount: number
+}
+
+const sessionCursors = new Map<string, CursorState>()
+
+function buildMessageKey(message: CursorMessage, index: number): string {
+  const id = message.info?.id
+  if (id) return `id:${id}`
+
+  const time = message.info?.time
+  if (typeof time === "number" || typeof time === "string") {
+    return `t:${time}:${index}`
+  }
+
+  const created = time?.created
+  if (typeof created === "number") {
+    return `t:${created}:${index}`
+  }
+  if (typeof created === "string") {
+    return `t:${created}:${index}`
+  }
+
+  return `i:${index}`
+}
+
+export function getNewMessages<T extends CursorMessage>(
+  sessionID: string | undefined,
+  messages: T[]
+): T[] {
+  if (!sessionID) return messages
+
+  const keys = messages.map((message, index) => buildMessageKey(message, index))
+  const cursor = sessionCursors.get(sessionID)
+  let startIndex = 0
+
+  if (cursor) {
+    if (cursor.lastCount > messages.length) {
+      startIndex = 0
+    } else if (cursor.lastKey) {
+      const lastIndex = keys.lastIndexOf(cursor.lastKey)
+      if (lastIndex >= 0) {
+        startIndex = lastIndex + 1
+      } else if (cursor.lastCount <= messages.length) {
+        startIndex = cursor.lastCount
+      }
+    }
+  }
+
+  if (messages.length === 0) {
+    sessionCursors.delete(sessionID)
+  } else {
+    sessionCursors.set(sessionID, {
+      lastKey: keys[keys.length - 1],
+      lastCount: messages.length,
+    })
+  }
+
+  return messages.slice(startIndex)
+}
+
+export function resetMessageCursor(sessionID?: string): void {
+  if (sessionID) {
+    sessionCursors.delete(sessionID)
+    return
+  }
+  sessionCursors.clear()
+}