@codebolt/codeboltjs 5.0.0 → 5.0.1

package/dist/core/Codebolt.d.ts

@@ -392,8 +392,16 @@ declare class Codebolt {
      * @private
      */
     private handleIncomingMessage;
+    /**
+     * Sets up a background listener for all messageResponse messages from the socket.
+     * This ensures that getMessage() promises are always resolved even if onMessage() is not called.
+     * @private
+     */
+    private setupMessageListener;
     /**
      * Sets up a listener for incoming messages with a direct handler function.
+     * Note: Message extraction and resolver handling is done by setupMessageListener.
+     * This method only adds the custom handler logic and sends processStoped response.
      * @param {Function} handler - The handler function to call when a message is received.
      * @returns {void}
      */

package/dist/core/Codebolt.js

@@ -79,6 +79,7 @@ class Codebolt {
         this.userMessage = user_message_utilities_1.userMessageUtilities;
         console.log("Codebolt Agent initialized");
         this.readyPromise = this.initializeConnection();
+        this.setupMessageListener();
     }
     /**
      * @method initializeConnection
@@ -154,7 +155,11 @@ class Codebolt {
      * Otherwise, waits for the next message to arrive.
      * @returns {Promise<FlatUserMessage>} A promise that resolves with the next message
      */
-    getMessage() {
+    async getMessage() {
+        // Wait for WebSocket to be ready first
+        console.log('[Codebolt] getMessage ');
+        await this.waitForReady();
+        console.log('[Codebolt] getMessage ');
         // If there are queued messages, return the first one
         if (this.messageQueue.length > 0) {
             const message = this.messageQueue.shift();
@@ -181,8 +186,52 @@ class Codebolt {
             this.messageQueue.push(message);
         }
     }
+    /**
+     * Sets up a background listener for all messageResponse messages from the socket.
+     * This ensures that getMessage() promises are always resolved even if onMessage() is not called.
+     * @private
+     */
+    setupMessageListener() {
+        this.waitForReady().then(() => {
+            const handleSocketMessage = async (response) => {
+                var _a, _b;
+                if (response.type === "messageResponse") {
+                    // Extract user-facing message from internal socket message
+                    const userMessage = {
+                        userMessage: response.message.userMessage,
+                        currentFile: response.message.currentFile,
+                        mentionedFiles: response.message.mentionedFiles || [],
+                        mentionedFullPaths: response.message.mentionedFullPaths || [],
+                        mentionedFolders: response.message.mentionedFolders || [],
+                        uploadedImages: response.message.uploadedImages || [],
+                        mentionedMCPs: response.message.mentionedMCPs || [],
+                        selectedAgent: {
+                            id: ((_a = response.message.selectedAgent) === null || _a === void 0 ? void 0 : _a.id) || '',
+                            name: ((_b = response.message.selectedAgent) === null || _b === void 0 ? void 0 : _b.name) || ''
+                        },
+                        messageId: response.message.messageId,
+                        threadId: response.message.threadId,
+                        selection: response.message.selection,
+                        remixPrompt: response.message.remixPrompt,
+                        mentionedAgents: response.message.mentionedAgents || [],
+                        activeFile: response.message.activeFile,
+                        openedFiles: response.message.activeFile
+                    };
+                    // Automatically save the user message globally
+                    user_message_manager_1.userMessageManager.saveMessage(userMessage);
+                    // Handle the message in the queue system for getMessage() resolvers
+                    this.handleIncomingMessage(userMessage);
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleSocketMessage);
+        }).catch(error => {
+            console.error('Failed to set up background message listener:', error);
+        });
+    }
     /**
      * Sets up a listener for incoming messages with a direct handler function.
+     * Note: Message extraction and resolver handling is done by setupMessageListener.
+     * This method only adds the custom handler logic and sends processStoped response.
      * @param {Function} handler - The handler function to call when a message is received.
      * @returns {void}
      */
@@ -215,11 +264,7 @@ class Codebolt {
                             activeFile: response.message.activeFile,
                             openedFiles: response.message.activeFile
                         };
-                        // Automatically save the user message globally
-                        user_message_manager_1.userMessageManager.saveMessage(userMessage);
-                        // Handle the message in the queue system
-                        this.handleIncomingMessage(userMessage);
-                        // Call the original handler
+                        // Call the custom handler
                         const result = await handler(userMessage);
                         // Send processStoped with optional message
                         const message = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codebolt/codeboltjs",
-  "version": "5.0.0",
+  "version": "5.0.1",
   "description": "",
   "keywords": [],
   "author": "",
@@ -23,6 +23,7 @@
     "access": "public"
   },
   "dependencies": {
+    "@codebolt/types": "latest",
     "@types/uuid": "^10.0.0",
     "buffer": "^6.0.3",
     "execa": "^9.5.2",
@@ -34,8 +35,7 @@
     "util": "^0.12.5",
     "uuid": "^11.1.0",
     "ws": "^8.18.3",
-    "yargs": "^17.7.2",
-    "@codebolt/types": "1.0.22"
+    "yargs": "^17.7.2"
   },
   "devDependencies": {
     "@types/events": "^3.0.3",