refactor: :loud_sound: show errors to IDE before GUI connected

1 files changed, 16 insertions, 0 deletions

diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index 498cf9de..a0ce009c 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -192,6 +192,8 @@ class IdeProtocolClient {
     });
   }
 
+  visibleMessages: Set<string> = new Set();
+
   async handleMessage(
     messageType: string,
     data: any,
@@ -254,6 +256,20 @@ class IdeProtocolClient {
         this.openFile(data.filepath);
         // TODO: Close file if False
         break;
+      case "showMessage":
+        if (!this.visibleMessages.has(data.message)) {
+          this.visibleMessages.add(data.message);
+          vscode.window
+            .showInformationMessage(data.message, "Copy Traceback", "View Logs")
+            .then((selection) => {
+              if (selection === "View Logs") {
+                vscode.commands.executeCommand("continue.viewLogs");
+              } else if (selection === "Copy Traceback") {
+                vscode.env.clipboard.writeText(data.message);
+              }
+            });
+        }
+        break;
       case "showVirtualFile":
         this.showVirtualFile(data.name, data.contents);
         break;