fix(ide): prevent race condition when diff accepted through CLI (#7633)

packages/vscode-ide-companion/src/diff-manager.ts

@@ -132,7 +132,7 @@ export class DiffManager {
   /**
    * Closes an open diff view for a specific file.
    */
-  async closeDiff(filePath: string) {
+  async closeDiff(filePath: string, suppressNotification = false) {
     let uriToClose: vscode.Uri | undefined;
     for (const [uriString, diffInfo] of this.diffDocuments.entries()) {
       if (diffInfo.originalFilePath === filePath) {
@@ -145,6 +145,7 @@ export class DiffManager {
       const rightDoc = await vscode.workspace.openTextDocument(uriToClose);
       const modifiedContent = rightDoc.getText();
       await this.closeDiffEditor(uriToClose);
+      if (!suppressNotification) {
         this.onDidChangeEmitter.fire(
           IdeDiffClosedNotificationSchema.parse({
             jsonrpc: '2.0',
@@ -155,6 +156,7 @@ export class DiffManager {
             },
           }),
         );
+      }
       return modifiedContent;
     }
     return;

packages/vscode-ide-companion/src/ide-server.ts

@@ -356,10 +356,20 @@ const createMcpServer = (diffManager: DiffManager) => {
       description: '(IDE Tool) Close an open diff view for a specific file.',
       inputSchema: z.object({
         filePath: z.string(),
+        suppressNotification: z.boolean().optional(),
       }).shape,
     },
-    async ({ filePath }: { filePath: string }) => {
+    async ({
-      const content = await diffManager.closeDiff(filePath);
+      filePath,
+      suppressNotification,
+    }: {
+      filePath: string;
+      suppressNotification?: boolean;
+    }) => {
+      const content = await diffManager.closeDiff(
+        filePath,
+        suppressNotification,
+      );
       const response = { content: content ?? undefined };
       return {
         content: [