npm - kanna-code - Versions diffs - 0.26.7 → 0.26.9 - Mend

kanna-code 0.26.7 → 0.26.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (5) hide show

package/dist/client/assets/{index-CH3ZbXse.js → index-C6vlAZYX.js} +178 -178
package/dist/client/index.html +1 -1
package/package.json +1 -1
package/src/server/event-store.ts +30 -0
package/src/server/ws-router.ts +135 -9

package/dist/client/index.html CHANGED Viewed

@@ -5,7 +5,7 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <link rel="icon" type="image/png" href="/favicon.png" />
     <title>Kanna</title>
-    <script type="module" crossorigin src="/assets/index-CH3ZbXse.js"></script>
+    <script type="module" crossorigin src="/assets/index-C6vlAZYX.js"></script>
     <link rel="stylesheet" crossorigin href="/assets/index-C5RBxQW4.css">
   </head>
   <body>

package/package.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "name": "kanna-code",
   "type": "module",
-  "version": "0.26.7",
+  "version": "0.26.9",
   "description": "A beautiful web UI for Claude Code",
   "license": "MIT",
   "keywords": [

package/src/server/event-store.ts CHANGED Viewed

@@ -19,6 +19,22 @@ import { resolveLocalPath } from "./paths"
 const COMPACTION_THRESHOLD_BYTES = 2 * 1024 * 1024
 const STALE_EMPTY_CHAT_MAX_AGE_MS = 30 * 60 * 1000
+function isSendToStartingProfilingEnabled() {
+  return process.env.KANNA_PROFILE_SEND_TO_STARTING === "1"
+}
+function logSendToStartingProfile(stage: string, details?: Record<string, unknown>) {
+  if (!isSendToStartingProfilingEnabled()) {
+    return
+  }
+  console.log("[kanna/send->starting][server]", JSON.stringify({
+    stage,
+    ...details,
+  }))
+}
 interface LegacyTranscriptStats {
   hasLegacyData: boolean
   sources: Array<"snapshot" | "messages_log">
@@ -589,13 +605,27 @@ export class EventStore {
     this.requireChat(chatId)
     const payload = `${JSON.stringify(entry)}\n`
     const transcriptPath = this.transcriptPath(chatId)
+    const queuedAt = performance.now()
     this.writeChain = this.writeChain.then(async () => {
+      const startedAt = performance.now()
+      const queueDelayMs = Number((startedAt - queuedAt).toFixed(1))
       await mkdir(this.transcriptsDir, { recursive: true })
+      const beforeAppendAt = performance.now()
       await appendFile(transcriptPath, payload, "utf8")
+      const afterAppendAt = performance.now()
       this.applyMessageMetadata(chatId, entry)
       if (this.cachedTranscript?.chatId === chatId) {
         this.cachedTranscript.entries.push({ ...entry })
       }
+      logSendToStartingProfile("event_store.append_message", {
+        chatId,
+        entryId: entry._id,
+        kind: entry.kind,
+        payloadBytes: payload.length,
+        queueDelayMs,
+        appendMs: Number((afterAppendAt - beforeAppendAt).toFixed(1)),
+        totalMs: Number((afterAppendAt - queuedAt).toFixed(1)),
+      })
     })
     return this.writeChain
   }

package/src/server/ws-router.ts CHANGED Viewed

@@ -37,6 +37,53 @@ function logSendToStartingProfile(
   }))
 }
+function countSubscriptionsByTopic(ws: ServerWebSocket<ClientState>) {
+  let sidebar = 0
+  let chat = 0
+  let projectGit = 0
+  let localProjects = 0
+  let update = 0
+  let keybindings = 0
+  let terminal = 0
+  for (const topic of ws.data.subscriptions.values()) {
+    switch (topic.type) {
+      case "sidebar":
+        sidebar += 1
+        break
+      case "chat":
+        chat += 1
+        break
+      case "project-git":
+        projectGit += 1
+        break
+      case "local-projects":
+        localProjects += 1
+        break
+      case "update":
+        update += 1
+        break
+      case "keybindings":
+        keybindings += 1
+        break
+      case "terminal":
+        terminal += 1
+        break
+    }
+  }
+  return {
+    total: ws.data.subscriptions.size,
+    sidebar,
+    chat,
+    projectGit,
+    localProjects,
+    update,
+    keybindings,
+    terminal,
+  }
+}
 export interface ClientState {
   subscriptions: Map<string, SubscriptionTopic>
   snapshotSignatures: Map<string, string>
@@ -56,7 +103,9 @@ interface CreateWsRouterArgs {
 }
 function send(ws: ServerWebSocket<ClientState>, message: ServerEnvelope) {
-  ws.send(JSON.stringify(message))
+  const payload = JSON.stringify(message)
+  ws.send(payload)
+  return payload.length
 }
 function ensureSnapshotSignatures(ws: ServerWebSocket<ClientState>) {
@@ -100,9 +149,13 @@ export function createWsRouter({
   }
   function getProtectedChatIds() {
+    const activeStatuses = agent.getActiveStatuses()
+    const drainingChatIds = typeof agent.getDrainingChatIds === "function"
+      ? agent.getDrainingChatIds()
+      : new Set<string>()
     return new Set([
-      ...agent.getActiveStatuses().keys(),
-      ...agent.getDrainingChatIds().values(),
+      ...activeStatuses.keys(),
+      ...drainingChatIds.values(),
     ])
   }
@@ -125,21 +178,48 @@ export function createWsRouter({
   }
   async function maybePruneStaleEmptyChats(extraSockets?: Iterable<ServerWebSocket<ClientState>>) {
-    await store.pruneStaleEmptyChats?.({
-      activeChatIds: getProtectedChatIds(),
-      protectedChatIds: getProtectedDraftChatIds(extraSockets),
+    const startedAt = performance.now()
+    const activeChatIds = getProtectedChatIds()
+    const protectedDraftChatIds = getProtectedDraftChatIds(extraSockets)
+    const prunedChatIds = await store.pruneStaleEmptyChats?.({
+      activeChatIds,
+      protectedChatIds: protectedDraftChatIds,
     })
+    if (isSendToStartingProfilingEnabled()) {
+      console.log("[kanna/send->starting][server]", JSON.stringify({
+        stage: "ws.prune_stale_empty_chats",
+        elapsedMs: Number((performance.now() - startedAt).toFixed(1)),
+        activeChatCount: activeChatIds.size,
+        protectedDraftChatCount: protectedDraftChatIds.size,
+        prunedCount: prunedChatIds?.length ?? 0,
+        totalChatCount: store.state.chatsById.size,
+        totalProjectCount: store.state.projectsById.size,
+      }))
+    }
   }
   function createEnvelope(id: string, topic: SubscriptionTopic): ServerEnvelope {
     if (topic.type === "sidebar") {
+      const startedAt = performance.now()
+      const data = deriveSidebarData(store.state, agent.getActiveStatuses())
+      if (isSendToStartingProfilingEnabled()) {
+        const totalChats = data.projectGroups.reduce((count, group) => count + group.chats.length, 0)
+        console.log("[kanna/send->starting][server]", JSON.stringify({
+          stage: "ws.sidebar_snapshot_built",
+          elapsedMs: Number((performance.now() - startedAt).toFixed(1)),
+          projectGroupCount: data.projectGroups.length,
+          chatCount: totalChats,
+          totalChatCount: store.state.chatsById.size,
+          totalProjectCount: store.state.projectsById.size,
+        }))
+      }
       return {
         v: PROTOCOL_VERSION,
         type: "snapshot",
         id,
         snapshot: {
           type: "sidebar",
-          data: deriveSidebarData(store.state, agent.getActiveStatuses()),
+          data,
         },
       }
     }
@@ -235,15 +315,22 @@ export function createWsRouter({
   }
   async function pushSnapshots(ws: ServerWebSocket<ClientState>, options?: { skipPrune?: boolean }) {
+    const pushStartedAt = performance.now()
     if (!options?.skipPrune) {
       await maybePruneStaleEmptyChats([ws])
     }
     const snapshotSignatures = ensureSnapshotSignatures(ws)
+    let sentCount = 0
+    let skippedCount = 0
     for (const [id, topic] of ws.data.subscriptions.entries()) {
+      const envelopeStartedAt = performance.now()
       const envelope = createEnvelope(id, topic)
+      const createdAt = performance.now()
       if (envelope.type !== "snapshot") continue
       const signature = JSON.stringify(envelope.snapshot)
+      const signatureReadyAt = performance.now()
       if (snapshotSignatures.get(id) === signature) {
+        skippedCount += 1
         continue
       }
       snapshotSignatures.set(id, signature)
@@ -253,17 +340,52 @@ export function createWsRouter({
           chatId: topic.chatId,
           status: envelope.snapshot.data.runtime.status,
           messageCount: envelope.snapshot.data.messages.length,
+          buildMs: Number((createdAt - envelopeStartedAt).toFixed(1)),
+          signatureMs: Number((signatureReadyAt - createdAt).toFixed(1)),
+          signatureBytes: signature.length,
+        })
+      }
+      const payloadBytes = send(ws, envelope)
+      sentCount += 1
+      if (topic.type === "chat" && envelope.snapshot.type === "chat" && envelope.snapshot.data?.runtime.status === "starting") {
+        const profile = agent.getActiveTurnProfile(topic.chatId)
+        logSendToStartingProfile(profile?.traceId, profile?.startedAt, "ws.snapshot_send_completed", {
+          chatId: topic.chatId,
+          payloadBytes,
         })
       }
-      send(ws, envelope)
+    }
+    if (isSendToStartingProfilingEnabled()) {
+      console.log("[kanna/send->starting][server]", JSON.stringify({
+        stage: "ws.push_snapshots_completed",
+        elapsedMs: Number((performance.now() - pushStartedAt).toFixed(1)),
+        skipPrune: Boolean(options?.skipPrune),
+        sentCount,
+        skippedCount,
+        ...countSubscriptionsByTopic(ws),
+      }))
     }
   }
   async function broadcastSnapshots() {
+    const startedAt = performance.now()
     await maybePruneStaleEmptyChats()
+    const afterPruneAt = performance.now()
+    let socketCount = 0
     for (const ws of sockets) {
+      socketCount += 1
       await pushSnapshots(ws, { skipPrune: true })
     }
+    if (isSendToStartingProfilingEnabled()) {
+      console.log("[kanna/send->starting][server]", JSON.stringify({
+        stage: "ws.broadcast_snapshots_completed",
+        elapsedMs: Number((performance.now() - startedAt).toFixed(1)),
+        pruneMs: Number((afterPruneAt - startedAt).toFixed(1)),
+        socketCount,
+        totalChatCount: store.state.chatsById.size,
+        totalProjectCount: store.state.projectsById.size,
+      }))
+    }
   }
   function broadcastError(message: string) {
@@ -473,7 +595,11 @@ export function createWsRouter({
           logSendToStartingProfile(profile?.traceId ?? command.clientTraceId, profile?.startedAt, "ws.chat_send_ack", {
             chatId: result.chatId ?? null,
           })
-          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          const payloadBytes = send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result })
+          logSendToStartingProfile(profile?.traceId ?? command.clientTraceId, profile?.startedAt, "ws.chat_send_ack_completed", {
+            chatId: result.chatId ?? null,
+            payloadBytes,
+          })
           break
         }
         case "chat.refreshDiffs": {