changes in codebolt

Author: rawatSingh2022

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codebolt/codeboltjs",
-  "version": "1.1.34",
+  "version": "1.1.35",
   "description": "",
   "keywords": [],
   "author": "",
@@ -19,6 +19,8 @@
   "homepage": "https://codeboltai.github.io",
   "dependencies": {
     "@codebolt/types": "^1.0.9",
+    "tree-sitter": "^0.21.1",
+    "tree-sitter-javascript": "^0.21.2",
     "typedoc-plugin-missing-exports": "^2.2.0",
     "ws": "^8.17.0"
   },

src/modules/codeutils.ts

@@ -1,4 +1,9 @@
 import cbws from './websocket';
+import * as fs from 'fs';
+import path from 'path';
+import  Parser  from 'tree-sitter';
+import  JavaScript  from 'tree-sitter-javascript';
+
 import { GetJsTreeResponse, MatchProblemResponse, GetMatcherListTreeResponse, getMatchDetail } from '@codebolt/types';
 
 /**
@@ -11,21 +16,73 @@ const cbcodeutils = {
      * @param {string} filePath - The path of the file to retrieve the JS tree for.
      * @returns {Promise<GetJsTreeResponse>} A promise that resolves with the JS tree response.
      */
-    getJsTree: (filePath: string): Promise<GetJsTreeResponse> => {
-        return new Promise((resolve, reject) => {
+    getJsTree: (filePath?: string): Promise<GetJsTreeResponse> => {
+        return new Promise( async (resolve, reject) => {
             cbws.getWebsocket.send(JSON.stringify({
-                "type": "codeEvent",
-                "action":"getJsTree",
-                payload:{
-                    filePath
-                }
+                "type": "settingEvent",
+                "action": "getProjectPath"
             }));
             cbws.getWebsocket.on('message', (data: string) => {
                 const response = JSON.parse(data);
-                if (response.type === "getJsTreeResponse") {
-                    resolve(response); // Resolve the Promise with the response data
-                } 
+                if (response.type === "getProjectPathResponse") {
+                    // resolve(response);
+                    try {
+                        let pathInput= response.projectPath;
+                        let parser= new Parser();
+                        // Initialize the parser with the JavaScript language
+                        parser.setLanguage(JavaScript);
+                        const trees = [];
+                        const functionNodes = [];
+                        const processDirectory = (directory:any) => {
+                            console.log("isdir")
+                            // Read all files in the directory
+                            const files = fs.readdirSync(directory, { withFileTypes: true });
+            
+                            files.forEach(file => {
+                                if (file.isDirectory()) {
+                                    if (file.name !== 'node_modules') { // Ignore node_modules directory
+                                        processDirectory(path.join(directory, file.name)); // Recursive call for subdirectories
+                                    }
+                                } else if (path.extname(file.name) === '.js') {
+                                    const code = fs.readFileSync(path.join(directory, file.name), 'utf-8');
+                                    console.log(code);
+                                    const tree = parser.parse(code);
+                                    trees.push(tree);
+                                }
+                            });
+                        };
+            
+                        if (fs.lstatSync(pathInput).isDirectory()) {
+                            processDirectory(pathInput);
+                        } else if (path.extname(pathInput) === '.js') {
+                            // Read a single JavaScript file
+                            const code = fs.readFileSync(pathInput, 'utf-8');
+                            let tree = parser.parse(code);
+                          
+                            trees.push(tree);
+                        }
+            
+                        return trees; // Return an array of abstract syntax trees (ASTs)
+                    } catch (error) {
+                        console.error('An error occurred:', error);
+                        return null; // Return null in case of error
+                    }
+                }
             });
+
+            // cbws.getWebsocket.send(JSON.stringify({
+            //     "type": "codeEvent",
+            //     "action":"getJsTree",
+            //     payload:{
+            //         filePath
+            //     }
+            // }));
+            // cbws.getWebsocket.on('message', (data: string) => {
+            //     const response = JSON.parse(data);
+            //     if (response.type === "getJsTreeResponse") {
+            //         resolve(response); // Resolve the Promise with the response data
+            //     } 
+            // });
         });
     },