changes

Author: rawatSingh2022

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codebolt/codeboltjs",
-  "version": "1.1.40",
+  "version": "1.1.41",
   "description": "",
   "keywords": [],
   "author": "",

src/modules/chat.ts

@@ -37,12 +37,18 @@ const cbchat = {
      * Sets up a listener for incoming WebSocket messages and emits a custom event when a message is received.
      * @returns {EventEmitter} The event emitter used for emitting custom events.
      */
-    userMessageListener: () => {
+    onActionMessage: () => {
         if (!cbws.getWebsocket) return;
         cbws.getWebsocket.on('message', (data: string) => {
             const response = JSON.parse(data);
             if (response.type === "messageResponse") {
-                eventEmitter.emit("userMessage", response.response);
+                // Pass a callback function as an argument to the emit method
+                eventEmitter.emit("userMessage", response, (message: string) => {
+                    console.log("Callback function invoked with message:", message);
+                    cbws.getWebsocket.send(JSON.stringify({
+                        "type": "processStoped"
+                    }));
+                });
             } 
         });
         return eventEmitter;

src/modules/crawler.ts

@@ -98,11 +98,23 @@ const cbcrawler = {
     /**
      * Initiates a crawl process.
      */
-    crawl: () => {
-       cbws.getWebsocket.send(JSON.stringify({
-            "type": "crawlerEvent",
-            action: 'crawl'
-        }));
+    crawl: (query:string) => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "crawlerEvent",
+                "action": 'crawl',
+                "message":{
+                    query
+                }
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                const response = JSON.parse(data);
+                if (response.type === "crawlResponse") {
+                    resolve(response); // Resolve the Promise with the response data
+                } 
+            });
+        });
+  
     }
 };
 

src/modules/websocket.ts

@@ -22,10 +22,10 @@ class cbws {
     private getUniqueConnectionId(): string {
         try {
             let fileContents = fs.readFileSync('./codebotagent.yml', 'utf8');
-            let data = yaml.load(fileContents);
+            let data:any = yaml.load(fileContents);
             return data.uniqueConnectionId;
         } catch (e) {
-            console.log(e);
+            console.error('Unable to locate codebotagent.yml file.');
             return '';
         }
     }