Method backfillMessageTree

packages/node-runtime/src/ai/chats.ts:316–349

(db: Database.Database)

  }

  private backfillMessageTree(db: Database.Database): void {
    const aiChats = db.prepare('SELECT id FROM ai_chat').all() as Array<{ id: string }>
    const updateMessage = db.prepare(
      'UPDATE ai_message SET parent_id = ?, sibling_group_id = COALESCE(sibling_group_id, ?), branch_index = COALESCE(branch_index, 0) WHERE id = ?'
    )
    const updateAIChat = db.prepare('UPDATE ai_chat SET active_message_id = ? WHERE id = ?')

    const tx = db.transaction(() => {
      for (const aiChat of aiChats) {
        const messages = db
          .prepare(
            `SELECT id, parent_id as parentId, sibling_group_id as siblingGroupId
             FROM ai_message WHERE ai_chat_id = ? ORDER BY timestamp ASC, id ASC`
          )
          .all(aiChat.id) as Array<{ id: string; parentId: string | null; siblingGroupId: string | null }>

        let previousId: string | null = null
        for (const message of messages) {
          const parentId = message.parentId === undefined ? previousId : (message.parentId ?? previousId)
          updateMessage.run(parentId, message.siblingGroupId ?? message.id, message.id)
          previousId = message.id
        }

        const aiChatRow = db
          .prepare('SELECT active_message_id as activeMessageId FROM ai_chat WHERE id = ?')
          .get(aiChat.id) as { activeMessageId: string | null } | undefined
        if (!aiChatRow?.activeMessageId && previousId) {
          updateAIChat.run(previousId, aiChat.id)
        }
      }
    })

    tx()
  }

  private generateId(prefix: string): string {
    return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`