msg: working-partially;

- Only one file item per component is displayed
- undo button per file shows up
- Fileclik and undo click are using same messageIDs as conterparts but with suffix
- Now need to handle those actions

Author: sacrodge

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

@@ -60,7 +60,7 @@ export class ChatWrapper {
   private dragOverlayVisibility: boolean = true;
   private imageContextFeatureEnabled: boolean = false;
   private readonly modifiedFilesTracker: ModifiedFilesTracker;
-  private modifiedFilesChatItem: ChatItem | null = null;
+
 
   constructor (props: ChatWrapperProps) {
     StyleLoader.getInstance().load('components/chat/_chat-wrapper.scss');
@@ -387,14 +387,17 @@ export class ChatWrapper {
   };
 
   private readonly insertChatItem = (chatItem: ChatItem): void => {
-    // Log chatItem details (only non-undefined properties)
-    const definedProps = Object.fromEntries(
-      Object.entries(chatItem).filter(([ _, value ]) => value !== undefined)
-    );
-    console.log('[ChatWrapper] insertChatItem received chatItem:', definedProps);
-
+    // Normal flow on initially opening ui requires the currentMessageId;
     this.removeEmptyCardsAndFollowups();
     const currentMessageId: string = (chatItem.messageId != null && chatItem.messageId !== '') ? chatItem.messageId : `TEMP_${generateUID()}`;
+    // 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
+      this.modifiedFilesTracker.addChatItem(chatItem);
+      return;
+    }
+
+    // Normal flow for all other chat items
     const chatItemCard = new ChatItemCard({
       tabId: this.props.tabId,
       chatItem: {
@@ -424,15 +427,6 @@ export class ChatWrapper {
     // Add to all rendered chat items map
     this.allRenderedChatItems[currentMessageId] = chatItemCard;
 
-    // Update ModifiedFilesTracker only if chatItem messageId contains suffix "modified-files-"
-    if (chatItem.messageId != null && chatItem.header?.fileList != null) {
-      if (chatItem.messageId.includes('modified-files-')) {
-        console.log('[ChatWrapper] Setting ModifiedFilesChatItem with fileList:', chatItem.header.fileList);
-        this.setModifiedFilesChatItem(chatItem);
-        return;
-      }
-    }
-
     if (chatItem.type === ChatItemType.PROMPT || chatItem.type === ChatItemType.SYSTEM_PROMPT) {
       // Make sure we align to top when there is a new prompt.
       // Only if it is a PROMPT!
@@ -562,23 +556,4 @@ export class ChatWrapper {
     this.dragOverlayContent.style.display = visible ? 'flex' : 'none';
     this.dragBlurOverlay.style.display = visible ? 'block' : 'none';
   }
-
-  public setModifiedFilesTrackerVisible (visible: boolean): void {
-    // No-op: component is always visible
-  }
-
-  public updateModifiedFilesTracker (): void {
-    if (this.modifiedFilesChatItem != null) {
-      this.modifiedFilesTracker.addChatItem(this.modifiedFilesChatItem);
-    }
-  }
-
-  public setModifiedFilesChatItem (chatItem: ChatItem): void {
-    console.log('[ChatWrapper] setModifiedFilesChatItem called with:', {
-      messageId: chatItem.messageId,
-      fileList: chatItem.header?.fileList
-    });
-    this.modifiedFilesChatItem = chatItem;
-    this.updateModifiedFilesTracker();
-  }
 }

src/main.ts

@@ -1346,7 +1346,7 @@ export class MynahUI {
    */
   public setFilesTrackerVisible = (tabId: string, visible: boolean): void => {
     if (this.chatWrappers[tabId] != null) {
-      this.chatWrappers[tabId].setModifiedFilesTrackerVisible(visible);
+      // this.chatWrappers[tabId].setModifiedFilesTrackerVisible(visible);
     }
   };