added onReady Function

Author: utkarshx

src/core/Codebolt.ts

@@ -35,6 +35,7 @@ class Codebolt {
     websocket: WebSocket | null = null;
     private isReady: boolean = false;
     private readyPromise: Promise<void>;
+    private readyHandlers: Array<() => void | Promise<void>> = [];
 
     /**
      * @constructor
@@ -56,6 +57,15 @@ class Codebolt {
             this.websocket = cbws.getWebsocket;
             this.isReady = true;
             console.log("Codebolt WebSocket connection established");
+            
+            // Execute all registered ready handlers
+            for (const handler of this.readyHandlers) {
+                try {
+                    await handler();
+                } catch (error) {
+                    console.error('Error executing ready handler:', error);
+                }
+            }
         } catch (error) {
             console.error('Failed to initialize WebSocket connection:', error);
             throw error;
@@ -106,6 +116,31 @@ class Codebolt {
     utils = cbutils;
     notify = notificationFunctions;
 
+    /**
+     * Sets up a handler function to be executed when the WebSocket connection is established.
+     * If the connection is already established, the handler will be executed immediately.
+     * @param {Function} handler - The handler function to call when the connection is ready.
+     * @returns {void}
+     */
+    onReady(handler: () => void | Promise<void>) {
+        if (this.isReady) {
+            // If already ready, execute the handler immediately
+            try {
+                const result = handler();
+                if (result instanceof Promise) {
+                    result.catch(error => {
+                        console.error('Error in ready handler:', error);
+                    });
+                }
+            } catch (error) {
+                console.error('Error in ready handler:', error);
+            }
+        } else {
+            // If not ready yet, add to the list of handlers to execute when ready
+            this.readyHandlers.push(handler);
+        }
+    }
+
     /**
      * Sets up a listener for incoming messages with a direct handler function.
      * @param {Function} handler - The handler function to call when a message is received.

src/core/websocket.ts

@@ -66,8 +66,9 @@ class cbws {
         const parentAgentInstanceIdParam = process.env.parentAgentInstanceId ? `&parentAgentInstanceId=${process.env.parentAgentInstanceId}` : '';
         const agentTask = process.env.agentTask ? `&agentTask=${process.env.agentTask}` : '';
         const socketPort = process.env.SOCKET_PORT || '12345';
+        const serverUrl = process.env.CODEBOLT_SERVER_URL || 'localhost';
 
-        const wsUrl = `ws://localhost:${socketPort}/codebolt?id=${uniqueConnectionId}${agentIdParam}${parentIdParam}${parentAgentInstanceIdParam}${agentTask}${process.env.Is_Dev ? '&dev=true' : ''}`;
+        const wsUrl = `ws://${serverUrl}:${socketPort}/codebolt?id=${uniqueConnectionId}${agentIdParam}${parentIdParam}${parentAgentInstanceIdParam}${agentTask}${process.env.Is_Dev ? '&dev=true' : ''}`;
         console.log('[WebSocket] Connecting to:', wsUrl);
 
         this.websocket = new WebSocket(wsUrl);