feat(amazonq): conversation persistence, view/search chat history #6893

## Problem
Users lose all chats when they close VSCode, and there's no way to
browse through chat history. Users also cant export their conversations
to an easily shareable format.

## Solution
Automatically persist conversations to JSON files in
`~/.aws/amazonq/history`, one for each workspace where Amazon Q chats
occur. Add chat history and chat export buttons to top of Amazon Q
toolbar. Clicking on the chat history button allows users to browse and
search through chat history. Users click on an old conversation to open
it back up (currently open conversations are in bold). Clicking on chat
export button allows users to save chat transcript as a markdown or
html.

Note: persistence + history is only for Q Chat Tabs (not /dev, /doc,
/transform, etc.)

Author: avi-alpert

package-lock.json

@@ -0,0 +1,4 @@
+{
+	"type": "Feature",
+	"description": "Amazon Q chat: View and search chat history"
+}

packages/amazonq/.changes/next-release/Feature-2cf2317a-aa6e-4413-aa57-e0d8905ab109.json

@@ -0,0 +1,4 @@
+{
+	"type": "Feature",
+	"description": "Amazon Q chat: Automatically persist chats between IDE sessions"
+}

packages/amazonq/.changes/next-release/Feature-80e49868-8e04-4648-aace-ad34ef57eba2.json

@@ -0,0 +1,4 @@
+{
+	"type": "Feature",
+	"description": "Amazon Q chat: Click share icon to export chat to Markdown or HTML"
+}

packages/amazonq/.changes/next-release/Feature-ed1e1b2e-0073-4de0-99b8-e417298328e6.json

@@ -455,6 +455,7 @@
         "@types/js-yaml": "^4.0.5",
         "@types/jsdom": "^21.1.6",
         "@types/lodash": "^4.14.180",
+        "@types/lokijs": "^1.5.14",
         "@types/markdown-it": "^13.0.2",
         "@types/mime-types": "^2.1.4",
         "@types/mocha": "^10.0.6",
@@ -555,6 +556,7 @@
         "js-yaml": "^4.1.0",
         "jsonc-parser": "^3.2.0",
         "lodash": "^4.17.21",
+        "lokijs": "^1.5.12",
         "markdown-it": "^13.0.2",
         "mime-types": "^2.1.32",
         "node-fetch": "^2.7.0",

packages/core/package.json

@@ -3,12 +3,21 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { ChatItem, ChatItemAction, ChatItemType, FeedbackPayload, QuickActionCommand } from '@aws/mynah-ui'
+import {
+    ChatItem,
+    ChatItemAction,
+    ChatItemType,
+    DetailedList,
+    FeedbackPayload,
+    QuickActionCommand,
+} from '@aws/mynah-ui'
 import { ExtensionMessage } from '../commands'
 import { CodeReference } from './amazonqCommonsConnector'
 import { TabOpenType, TabsStorage, TabType } from '../storages/tabsStorage'
 import { FollowUpGenerator } from '../followUps/generator'
 import { CWCChatItem } from '../connector'
+import { DetailedListSheetProps } from '@aws/mynah-ui/dist/components/detailed-list/detailed-list-sheet'
+import { DetailedListConnector, DetailedListType } from '../detailedList/detailedListConnector'
 
 interface ChatPayload {
     chatMessage: string
@@ -23,6 +32,15 @@ export interface BaseConnectorProps {
     onError: (tabID: string, message: string, title: string) => void
     onWarning: (tabID: string, message: string, title: string) => void
     onOpenSettingsMessage: (tabID: string) => void
+    onNewTab: (tabType: TabType, chats?: ChatItem[]) => string | undefined
+    onOpenDetailedList: (data: DetailedListSheetProps) => {
+        update: (data: DetailedList) => void
+        close: () => void
+        changeTarget: (direction: 'up' | 'down', snapOnLastAndFirst?: boolean) => void
+        getTargetElementId: () => string | undefined
+    }
+    onSelectTab: (tabID: string, eventID: string) => void
+    onExportChat: (tabId: string, format: 'html' | 'markdown') => string
     tabsStorage: TabsStorage
 }
 
@@ -32,8 +50,13 @@ export abstract class BaseConnector {
     protected readonly onWarning
     protected readonly onChatAnswerReceived
     protected readonly onOpenSettingsMessage
+    protected readonly onNewTab
+    protected readonly onOpenDetailedList
+    protected readonly onExportChat
+    protected readonly onSelectTab
     protected readonly followUpGenerator: FollowUpGenerator
     protected readonly tabsStorage
+    protected historyConnector
 
     abstract getTabType(): TabType
 
@@ -43,8 +66,17 @@ export abstract class BaseConnector {
         this.onWarning = props.onWarning
         this.onError = props.onError
         this.onOpenSettingsMessage = props.onOpenSettingsMessage
+        this.onNewTab = props.onNewTab
         this.tabsStorage = props.tabsStorage
+        this.onOpenDetailedList = props.onOpenDetailedList
+        this.onExportChat = props.onExportChat
+        this.onSelectTab = props.onSelectTab
         this.followUpGenerator = new FollowUpGenerator()
+        this.historyConnector = new DetailedListConnector(
+            DetailedListType.history,
+            this.sendMessageToExtension,
+            this.onOpenDetailedList
+        )
     }
 
     onResponseBodyLinkClick = (tabID: string, messageId: string, link: string): void => {
@@ -296,5 +328,53 @@ export abstract class BaseConnector {
             await this.processOpenSettingsMessage(messageData)
             return
         }
+
+        if (messageData.type === 'restoreTabMessage') {
+            const newTabId = this.onNewTab(messageData.tabType, messageData.chats)
+            this.sendMessageToExtension({
+                command: 'tab-restored',
+                historyId: messageData.historyId,
+                newTabId,
+                tabType: this.getTabType(),
+                exportTab: messageData.exportTab,
+            })
+            return
+        }
+
+        if (messageData.type === 'updateDetailedListMessage') {
+            if (messageData.listType === DetailedListType.history) {
+                this.historyConnector.updateList(messageData.detailedList)
+            }
+            return
+        }
+
+        if (messageData.type === 'closeDetailedListMessage') {
+            if (messageData.listType === DetailedListType.history) {
+                this.historyConnector.closeList()
+            }
+            return
+        }
+
+        if (messageData.type === 'openDetailedListMessage') {
+            if (messageData.listType === DetailedListType.history) {
+                this.historyConnector.openList(messageData)
+            }
+            return
+        }
+
+        if (messageData.type === 'exportChatMessage') {
+            const serializedChat = this.onExportChat(messageData.tabID, messageData.format)
+            this.sendMessageToExtension({
+                command: 'save-chat',
+                uri: messageData.uri,
+                serializedChat,
+                tabType: 'cwc',
+            })
+            return
+        }
+
+        if (messageData.type === 'selectTabMessage') {
+            this.onSelectTab(messageData.tabID, messageData.eventID)
+        }
     }
 }

packages/core/src/amazonq/webview/ui/apps/baseConnector.ts

@@ -23,7 +23,6 @@ export interface ConnectorProps extends BaseConnectorProps {
     onUpdatePromptProgress: (tabID: string, progressField: ProgressField) => void
     onChatInputEnabled: (tabID: string, enabled: boolean) => void
     onUpdateAuthentication: (featureDevEnabled: boolean, authenticatingTabIDs: string[]) => void
-    onNewTab: (tabType: TabType) => void
 }
 
 export class Connector extends BaseConnector {
@@ -32,7 +31,6 @@ export class Connector extends BaseConnector {
     private readonly updatePlaceholder
     private readonly chatInputEnabled
     private readonly onUpdateAuthentication
-    private readonly onNewTab
     private readonly updatePromptProgress
 
     override getTabType(): TabType {
@@ -46,7 +44,6 @@ export class Connector extends BaseConnector {
         this.updatePlaceholder = props.onUpdatePlaceholder
         this.chatInputEnabled = props.onChatInputEnabled
         this.onUpdateAuthentication = props.onUpdateAuthentication
-        this.onNewTab = props.onNewTab
         this.updatePromptProgress = props.onUpdatePromptProgress
     }
 

packages/core/src/amazonq/webview/ui/apps/docChatConnector.ts

@@ -30,7 +30,6 @@ export interface ConnectorProps extends BaseConnectorProps {
     onUpdatePlaceholder: (tabID: string, newPlaceholder: string) => void
     onChatInputEnabled: (tabID: string, enabled: boolean) => void
     onUpdateAuthentication: (featureDevEnabled: boolean, authenticatingTabIDs: string[]) => void
-    onNewTab: (tabType: TabType) => void
 }
 
 export class Connector extends BaseConnector {
@@ -40,7 +39,6 @@ export class Connector extends BaseConnector {
     private readonly updatePlaceholder
     private readonly chatInputEnabled
     private readonly onUpdateAuthentication
-    private readonly onNewTab
 
     override getTabType(): TabType {
         return 'featuredev'
@@ -53,7 +51,6 @@ export class Connector extends BaseConnector {
         this.updatePlaceholder = props.onUpdatePlaceholder
         this.chatInputEnabled = props.onChatInputEnabled
         this.onUpdateAuthentication = props.onUpdateAuthentication
-        this.onNewTab = props.onNewTab
         this.onChatAnswerUpdated = props.onChatAnswerUpdated
     }
 

packages/core/src/amazonq/webview/ui/apps/featureDevChatConnector.ts

@@ -44,5 +44,12 @@ type MessageCommand =
     | 'update-welcome-count'
     | 'quick-command-group-action-click'
     | 'context-selected'
+    | 'tab-restored'
+    | 'tab-bar-button-clicked'
+    | 'export-chat'
+    | 'save-chat'
+    | 'detailed-list-filter-change'
+    | 'detailed-list-item-select'
+    | 'detailed-list-action-click'
 
 export type ExtensionMessage = Record<string, any> & { command: MessageCommand }

packages/core/src/amazonq/webview/ui/commands.ts

@@ -16,6 +16,7 @@ import {
     QuickActionCommand,
     ChatItemFormItem,
     ChatItemButton,
+    DetailedList,
 } from '@aws/mynah-ui'
 import { Connector as CWChatConnector } from './apps/cwChatConnector'
 import { Connector as FeatureDevChatConnector } from './apps/featureDevChatConnector'
@@ -30,6 +31,7 @@ import { WelcomeFollowupType } from './apps/amazonqCommonsConnector'
 import { AuthFollowUpType } from './followUps/generator'
 import { DiffTreeFileInfo } from './diffTree/types'
 import { UserIntent } from '@amzn/codewhisperer-streaming'
+import { DetailedListSheetProps } from '@aws/mynah-ui/dist/components/detailed-list/detailed-list-sheet'
 
 export interface CodeReference {
     licenseName?: string
@@ -94,7 +96,7 @@ export interface ConnectorProps {
     onUpdatePromptProgress: (tabID: string, progressField: ProgressField) => void
     onChatInputEnabled: (tabID: string, enabled: boolean) => void
     onUpdateAuthentication: (featureDevEnabled: boolean, authenticatingTabIDs: string[]) => void
-    onNewTab: (tabType: TabType) => void
+    onNewTab: (tabType: TabType, chats?: ChatItem[]) => string | undefined
     onFileActionClick: (tabID: string, messageId: string, filePath: string, actionName: string) => void
     handleCommand: (chatPrompt: ChatPrompt, tabId: string) => void
     sendStaticMessages: (tabID: string, messages: ChatItem[]) => void
@@ -106,6 +108,14 @@ export interface ConnectorProps {
         title?: string,
         description?: string
     ) => void
+    onOpenDetailedList: (data: DetailedListSheetProps) => {
+        update: (data: DetailedList) => void
+        close: () => void
+        changeTarget: (direction: 'up' | 'down', snapOnLastAndFirst?: boolean) => void
+        getTargetElementId: () => string | undefined
+    }
+    onSelectTab: (tabID: string, eventID: string) => void
+    onExportChat: (tabID: string, format: 'markdown' | 'html') => string
     tabsStorage: TabsStorage
 }
 
@@ -291,6 +301,7 @@ export class Connector {
         this.tabsStorage.updateTabLastCommand(messageData.tabID, '')
     }
 
+    // Run when user opens new tab in UI
     onTabAdd = (tabID: string): void => {
         this.tabsStorage.addTab({
             id: tabID,
@@ -684,6 +695,16 @@ export class Connector {
         return false
     }
 
+    onTabBarButtonClick = async (tabId: string, buttonId: string, eventId?: string) => {
+        this.sendMessageToExtension({
+            command: 'tab-bar-button-clicked',
+            buttonId,
+            type: '',
+            tabID: tabId,
+            tabType: 'cwc',
+        })
+    }
+
     onCustomFormAction = (
         tabId: string,
         messageId: string | undefined,