From 43dc3459997d79d112a2776a8e95546580112460 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Thu, 29 Jun 2023 09:51:24 -0700
Subject: messaging of welcome step

---
 extension/src/continueIdeClient.ts | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

(limited to 'extension/src/continueIdeClient.ts')

diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index 8ab3e075..d983d93a 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -226,9 +226,18 @@ class IdeProtocolClient {
   }
 
   async getSessionId(): Promise<string> {
-    if (this.messenger === null) {
-      console.log("MESSENGER IS NULL");
-    }
+    await new Promise((resolve, reject) => {
+      // Repeatedly try to connect to the server
+      const interval = setInterval(() => {
+        if (
+          this.messenger &&
+          this.messenger.websocket.readyState === 1 // 1 => OPEN
+        ) {
+          clearInterval(interval);
+          resolve(null);
+        }
+      }, 1000);
+    });
     const resp = await this.messenger?.sendAndReceive("openGUI", {});
     const sessionId = resp.sessionId;
     console.log("New Continue session with ID: ", sessionId);
-- 
cgit v1.2.3-70-g09d2