kaizenai 0.2.2 → 0.3.0

package/src/server/event-store.ts

@@ -26,6 +26,7 @@ const FEATURE_METADATA_VERSION = 1 as const
 
 interface PersistedProjectFeature {
   v: typeof FEATURE_METADATA_VERSION
+  id: string
   title: string
   description: string
   browserState: FeatureBrowserState
@@ -81,6 +82,7 @@ export class EventStore {
     await this.loadSnapshot()
     await this.replayLogs()
     this.hydrateProjectWorktrees()
+    await this.reconcileAllProjectFeatureState()
     if (!(await this.hasLegacyTranscriptData()) && await this.shouldCompact()) {
       await this.compact()
     }
@@ -99,6 +101,13 @@ export class EventStore {
     }
   }
 
+  private async reconcileAllProjectFeatureState() {
+    for (const project of this.state.projectsById.values()) {
+      if (project.deletedAt) continue
+      await this.reconcileProjectFeatureState(project.id)
+    }
+  }
+
   private async ensureFile(filePath: string) {
     const file = Bun.file(filePath)
     if (!(await file.exists())) {
@@ -334,6 +343,12 @@ export class EventStore {
         const feature = this.state.featuresById.get(event.featureId)
         if (!feature) break
         feature.title = event.title
+        if (event.directoryRelativePath) {
+          feature.directoryRelativePath = event.directoryRelativePath
+        }
+        if (typeof event.overviewRelativePath === "string") {
+          feature.overviewRelativePath = event.overviewRelativePath
+        }
         feature.updatedAt = event.timestamp
         break
       }
@@ -377,6 +392,12 @@ export class EventStore {
         }
         break
       }
+      case "feature_overview_updated": {
+        const feature = this.state.featuresById.get(event.featureId)
+        if (!feature) break
+        feature.updatedAt = event.timestamp
+        break
+      }
       case "chat_created": {
         const chat = {
           id: event.chatId,
@@ -608,7 +629,7 @@ export class EventStore {
     await this.append(this.projectsLogPath, event)
   }
 
-  async createFeature(projectId: string, title: string, description = "") {
+  async createFeature(projectId: string, title: string, description = "", generateOverview = true) {
     const project = this.getProject(projectId)
     if (!project) {
       throw new Error("Project not found")
@@ -621,17 +642,19 @@ export class EventStore {
 
     const directoryName = this.generateUniqueFeatureDirectoryName(projectId, trimmedTitle)
     const directoryRelativePath = path.posix.join(PROJECT_METADATA_DIR_NAME, directoryName)
-    const overviewRelativePath = path.posix.join(directoryRelativePath, "overview.md")
+    const overviewRelativePath = generateOverview ? path.posix.join(directoryRelativePath, "overview.md") : ""
     const featureDirPath = path.join(project.localPath, directoryRelativePath)
-    const overviewPath = path.join(project.localPath, overviewRelativePath)
     const sortOrder = this.getNextFeatureSortOrder(projectId, "idea")
 
     await mkdir(featureDirPath, { recursive: true })
-    await writeFile(overviewPath, this.buildFeatureOverviewContent({
-      projectTitle: project.title,
-      featureTitle: trimmedTitle,
-      description: trimmedDescription,
-    }), "utf8")
+    if (generateOverview) {
+      const overviewPath = path.join(project.localPath, overviewRelativePath)
+      await writeFile(overviewPath, this.buildFeatureOverviewContent({
+        projectTitle: project.title,
+        featureTitle: trimmedTitle,
+        description: trimmedDescription,
+      }), "utf8")
+    }
 
     const featureId = crypto.randomUUID()
     const event: FeatureEvent = {
@@ -658,12 +681,33 @@ export class EventStore {
     if (!trimmedTitle) return
     const feature = this.requireFeature(featureId)
     if (feature.title === trimmedTitle) return
+
+    const project = this.getProject(feature.projectId)
+    if (!project) throw new Error("Project not found")
+
+    const newDirectoryName = this.generateUniqueFeatureDirectoryName(feature.projectId, trimmedTitle)
+    const newDirectoryRelativePath = path.posix.join(PROJECT_METADATA_DIR_NAME, newDirectoryName)
+    const newOverviewRelativePath = feature.overviewRelativePath
+      ? path.posix.join(newDirectoryRelativePath, "overview.md")
+      : ""
+
+    const oldDirPath = path.join(project.localPath, feature.directoryRelativePath)
+    const newDirPath = path.join(project.localPath, newDirectoryRelativePath)
+
+    try {
+      await rename(oldDirPath, newDirPath)
+    } catch (error: any) {
+      if (error.code !== "ENOENT") throw error
+    }
+
     const event: FeatureEvent = {
       v: STORE_VERSION,
       type: "feature_renamed",
       timestamp: Date.now(),
       featureId,
       title: trimmedTitle,
+      directoryRelativePath: newDirectoryRelativePath,
+      overviewRelativePath: newOverviewRelativePath,
     }
     await this.append(this.featuresLogPath, event)
     await this.syncProjectFeatureState(feature.projectId)
@@ -739,6 +783,57 @@ export class EventStore {
     await this.syncProjectFeatureState(feature.projectId)
   }
 
+  getFeatureOverviewSnapshot(featureId: string) {
+    const feature = this.getFeature(featureId)
+    if (!feature) return null
+    if (!feature.overviewRelativePath) return null
+    const project = this.getProject(feature.projectId)
+    if (!project) return null
+
+    const overviewPath = path.join(project.localPath, feature.overviewRelativePath)
+    const content = existsSync(overviewPath)
+      ? readFileSyncImmediate(overviewPath, "utf8")
+      : this.buildFeatureOverviewContent({
+          projectTitle: project.title,
+          featureTitle: feature.title,
+          description: feature.description,
+        })
+
+    return {
+      featureId: feature.id,
+      projectId: project.id,
+      title: feature.title,
+      directoryRelativePath: feature.directoryRelativePath,
+      overviewRelativePath: feature.overviewRelativePath,
+      localPath: overviewPath,
+      content,
+      updatedAt: feature.updatedAt,
+    }
+  }
+
+  async updateFeatureOverview(featureId: string, content: string) {
+    const feature = this.requireFeature(featureId)
+    if (!feature.overviewRelativePath) {
+      throw new Error("Feature overview is disabled for this feature")
+    }
+    const project = this.getProject(feature.projectId)
+    if (!project) {
+      throw new Error("Project not found")
+    }
+
+    const overviewPath = path.join(project.localPath, feature.overviewRelativePath)
+    await mkdir(path.dirname(overviewPath), { recursive: true })
+    await writeFile(overviewPath, content, "utf8")
+    await this.append(this.featuresLogPath, {
+      v: STORE_VERSION,
+      type: "feature_overview_updated",
+      timestamp: Date.now(),
+      featureId,
+    })
+    await this.syncProjectFeatureState(feature.projectId, true)
+    return this.getFeatureOverviewSnapshot(featureId)
+  }
+
   async hideProject(localPath: string) {
     const identity = resolveProjectRepositoryIdentity(localPath)
     if (this.state.hiddenProjectKeys.has(identity.repoKey)) return
@@ -945,19 +1040,11 @@ export class EventStore {
     }
 
     const persistedFeatures = await this.readProjectFeatureMetadata(project.localPath)
-    const persistedByDirectory = new Map(
-      persistedFeatures.map((feature) => [feature.directoryRelativePath, feature] as const)
-    )
     const existingFeatures = this.listFeaturesByProject(projectId)
     const existingByDirectory = new Map(
       existingFeatures.map((feature) => [feature.directoryRelativePath, feature] as const)
     )
 
-    for (const feature of existingFeatures) {
-      if (persistedByDirectory.has(feature.directoryRelativePath)) continue
-      await this.deleteFeature(feature.id)
-    }
-
     const chatsByKey = new Map(
       this.listChatsByProject(projectId)
         .map((chat) => {
@@ -973,7 +1060,7 @@ export class EventStore {
       if (existingFeature) {
         continue
       }
-      const featureId = crypto.randomUUID()
+      const featureId = persistedFeature.id
       const timestamp = Date.now()
       const event: FeatureEvent = {
         v: STORE_VERSION,
@@ -1006,6 +1093,7 @@ export class EventStore {
       }
     }
 
+    await this.syncProjectFeatureState(projectId, true)
     return importedFeatures
   }
 
@@ -1305,6 +1393,7 @@ export class EventStore {
     for (const feature of this.listFeaturesByProject(projectId)) {
       const persistedFeature: PersistedProjectFeature = {
         v: FEATURE_METADATA_VERSION,
+        id: feature.id,
         title: feature.title,
         description: feature.description,
         browserState: feature.browserState,
@@ -1319,6 +1408,17 @@ export class EventStore {
       }
       const featureMetadataPath = this.getFeatureMetadataPath(project.localPath, feature.directoryRelativePath)
       await mkdir(path.dirname(featureMetadataPath), { recursive: true })
+      if (feature.overviewRelativePath) {
+        const overviewPath = path.join(project.localPath, feature.overviewRelativePath)
+        await mkdir(path.dirname(overviewPath), { recursive: true })
+        if (!existsSync(overviewPath)) {
+          await writeFile(overviewPath, this.buildFeatureOverviewContent({
+            projectTitle: project.title,
+            featureTitle: feature.title,
+            description: feature.description,
+          }), "utf8")
+        }
+      }
       await writeFile(featureMetadataPath, JSON.stringify(persistedFeature, null, 2), "utf8")
     }
   }
@@ -1335,7 +1435,7 @@ export class EventStore {
         if (!raw?.trim()) continue
         const parsed = JSON.parse(raw) as Partial<PersistedProjectFeature>
         if (
-          parsed.v !== FEATURE_METADATA_VERSION
+          ((parsed.v as number | undefined) !== 1 && parsed.v !== FEATURE_METADATA_VERSION)
           || typeof parsed.title !== "string"
           || typeof parsed.description !== "string"
           || typeof parsed.directoryRelativePath !== "string"
@@ -1348,6 +1448,7 @@ export class EventStore {
         }
         features.push({
           v: FEATURE_METADATA_VERSION,
+          id: typeof parsed.id === "string" && parsed.id.trim() ? parsed.id : crypto.randomUUID(),
           title: parsed.title,
           description: parsed.description,
           browserState: (parsed.browserState ?? "OPEN") as FeatureBrowserState,

package/src/server/events.ts

@@ -109,6 +109,8 @@ export type FeatureEvent =
       timestamp: number
       featureId: string
       title: string
+      directoryRelativePath?: string
+      overviewRelativePath?: string
     }
   | {
       v: 3
@@ -138,6 +140,12 @@ export type FeatureEvent =
       timestamp: number
       featureId: string
     }
+  | {
+      v: 3
+      type: "feature_overview_updated"
+      timestamp: number
+      featureId: string
+    }
 
 export type ChatEvent =
   | {

package/src/server/ws-router.ts

@@ -174,6 +174,18 @@ export function createWsRouter({
       }
     }
 
+    if (topic.type === "feature-overview") {
+      return {
+        v: PROTOCOL_VERSION,
+        type: "snapshot",
+        id,
+        snapshot: {
+          type: "feature-overview",
+          data: store.getFeatureOverviewSnapshot(topic.featureId),
+        },
+      }
+    }
+
     return {
       v: PROTOCOL_VERSION,
       type: "snapshot",
@@ -234,6 +246,15 @@ export function createWsRouter({
     }
   }
 
+  function pushFeatureOverviewSnapshot(featureId: string) {
+    for (const ws of sockets) {
+      for (const [id, topic] of ws.data.subscriptions.entries()) {
+        if (topic.type !== "feature-overview" || topic.featureId !== featureId) continue
+        send(ws, createEnvelope(id, topic))
+      }
+    }
+  }
+
   function pushTerminalEvent(terminalId: string, event: Extract<ServerEnvelope, { type: "event" }>["event"]) {
     for (const ws of sockets) {
       for (const [id, topic] of ws.data.subscriptions.entries()) {
@@ -523,7 +544,12 @@ export function createWsRouter({
           break
         }
         case "feature.create": {
-          const feature = await store.createFeature(command.projectId, command.title, command.description ?? "")
+          const feature = await store.createFeature(
+            command.projectId,
+            command.title,
+            command.description ?? "",
+            command.generateOverview ?? true
+          )
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result: { featureId: feature.id } })
           break
         }
@@ -552,6 +578,13 @@ export function createWsRouter({
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id })
           break
         }
+        case "feature.updateOverview": {
+          const snapshot = await store.updateFeatureOverview(command.featureId, command.content)
+          pushFeatureOverviewSnapshot(command.featureId)
+          broadcastSidebarSnapshots()
+          send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result: snapshot })
+          break
+        }
         case "chat.create": {
           const chat = await store.createChat(command.projectId, command.featureId)
           send(ws, { v: PROTOCOL_VERSION, type: "ack", id, result: { chatId: chat.id } })

package/src/shared/protocol.ts

@@ -10,6 +10,7 @@ import type {
   LocalProjectsSnapshot,
   ModelOptions,
   SidebarData,
+  FeatureOverviewSnapshot,
   ThemeSettingsSnapshot,
   UpdateSnapshot,
 } from "./types"
@@ -29,6 +30,7 @@ export type SubscriptionTopic =
   | { type: "theme-settings" }
   | { type: "provider-settings" }
   | { type: "chat"; chatId: string }
+  | { type: "feature-overview"; featureId: string }
   | { type: "terminal"; terminalId: string }
 
 export interface TerminalSnapshot {
@@ -75,12 +77,13 @@ export type ClientCommand =
       editor?: EditorOpenSettings
     }
   | { type: "chat.create"; projectId: string; featureId?: string }
-  | { type: "feature.create"; projectId: string; title: string; description?: string }
+  | { type: "feature.create"; projectId: string; title: string; description?: string; generateOverview?: boolean }
   | { type: "feature.rename"; featureId: string; title: string }
   | { type: "feature.setBrowserState"; featureId: string; browserState: FeatureBrowserState }
   | { type: "feature.setStage"; featureId: string; stage: FeatureStage }
   | { type: "feature.reorder"; projectId: string; orderedFeatureIds: string[] }
   | { type: "feature.delete"; featureId: string }
+  | { type: "feature.updateOverview"; featureId: string; content: string }
   | { type: "chat.setFeature"; chatId: string; featureId: string | null }
   | { type: "chat.rename"; chatId: string; title: string }
   | { type: "chat.delete"; chatId: string }
@@ -121,6 +124,7 @@ export type ServerSnapshot =
   | { type: "theme-settings"; data: ThemeSettingsSnapshot }
   | { type: "provider-settings"; data: ProviderSettingsSnapshot }
   | { type: "chat"; data: ChatSnapshot | null }
+  | { type: "feature-overview"; data: FeatureOverviewSnapshot | null }
   | { type: "terminal"; data: TerminalSnapshot | null }
 
 export type ServerEnvelope =

package/src/shared/types.ts

@@ -467,6 +467,17 @@ export interface SidebarFeatureRow {
   chats: SidebarChatRow[]
 }
 
+export interface FeatureOverviewSnapshot {
+  featureId: string
+  projectId: string
+  title: string
+  directoryRelativePath: string
+  overviewRelativePath: string
+  localPath: string
+  content: string
+  updatedAt: number
+}
+
 export interface SidebarProjectGroup {
   groupKey: string
   title: string
@@ -888,6 +899,7 @@ export interface ExitPlanModeToolResult {
   clearContext?: boolean
   message?: string
   discarded?: boolean
+  attachments?: ChatAttachmentUpload[]
 }
 
 export type HydratedAskUserQuestionToolCall =