msg: working-partially;

- Added chat=wrapper support to handle title
- Should ideally be controller but no title property in chatMessage

Author: sacrodge

src/components/chat-item/chat-wrapper.ts

@@ -54,12 +54,13 @@ export class ChatWrapper {
   private lastStreamingChatItemCard: ChatItemCard | null;
   private lastStreamingChatItemMessageId: string | null;
   private allRenderedChatItems: Record<string, ChatItemCard> = {};
+  private allRenderedModifiedFileChatItems: Record<string, ChatItem> = {};
   render: ExtendedHTMLElement;
   private readonly dragOverlayContent: HTMLElement;
   private readonly dragBlurOverlay: HTMLElement;
   private dragOverlayVisibility: boolean = true;
   private imageContextFeatureEnabled: boolean = false;
-  private readonly modifiedFilesTracker: ModifiedFilesTracker;
+  private modifiedFilesTracker: ModifiedFilesTracker;
 
   constructor (props: ChatWrapperProps) {
     StyleLoader.getInstance().load('components/chat/_chat-wrapper.scss');
@@ -392,6 +393,18 @@ export class ChatWrapper {
     // Check if messageId contains "modified-files-" prefix
     if (chatItem.messageId != null && chatItem.messageId !== '' && chatItem.messageId.includes('modified-files-')) {
       // Forward only to ModifiedFilesTracker, skip normal flow
+      if (chatItem.fileList !== null) {
+        this.allRenderedModifiedFileChatItems[chatItem.messageId] = chatItem;
+        const size = Object.keys(this.allRenderedModifiedFileChatItems).length;
+        chatItem.title = size === 1 ? '1 file modified!' : `${size} files modified!`;
+      } else if (chatItem.fileList === null || chatItem.fileList === undefined) {
+        // This will mean that it is reset signal i.e. new chat;
+        // Ideally should be sent through controller
+        chatItem.title = 'working ...';
+        // remove all items from allRenderedModifiedFileChatItems
+        this.allRenderedModifiedFileChatItems = {};
+        this.modifiedFilesTracker = new ModifiedFilesTracker({ tabId: this.props.tabId });
+      }
       this.modifiedFilesTracker.addChatItem(chatItem);
       return;
     }

src/components/modified-files-tracker.ts

@@ -46,10 +46,7 @@ export class ModifiedFilesTracker {
       type: 'div',
       classNames: [ 'mynah-modified-files-tracker-wrapper' ],
       testId: testIds.modifiedFilesTracker.container,
-      children: [ this.collapsibleContent.render ],
-      attributes: {
-        style: 'display: block !important; visibility: visible !important;'
-      }
+      children: [ this.collapsibleContent.render ]
     });
 
     console.log('[ModifiedFilesTracker] Render element created:', this.render);
@@ -160,7 +157,16 @@ export class ModifiedFilesTracker {
     return messageId.startsWith('modified-files-') ? messageId.replace('modified-files-', '') : messageId;
   }
 
+  private updateTitleText (chatItem: ChatItem): void {
+    // check if chatItem
+    if (chatItem.title !== undefined && chatItem.title !== '') {
+      this.titleText = chatItem.title;
+      this.collapsibleContent.updateTitle(this.titleText);
+    }
+  }
+
   public addChatItem (chatItem: ChatItem): void {
+    this.updateTitleText(chatItem);
     if (chatItem.header?.fileList != null) {
       // Store the current chatItem for button handling
       this.props.chatItem = chatItem;