@codebolt/codeboltjs 1.1.34 → 1.1.36

package/index.d.ts

@@ -131,10 +131,7 @@ declare class Codebolt {
         executeCommandWithStream(command: string): EventEmitter<[never]>;
     };
     codeutils: {
-        getJsTree: (filePath: string) => Promise<import("@codebolt/types").GetJsTreeResponse>; /**
-         * @constructor
-         * @description Initializes the websocket connection.
-         */
+        getJsTree: (filePath?: string | undefined) => Promise<any>;
         getAllFilesAsMarkDown: () => Promise<string>;
         performMatch: (matcherDefinition: object, problemPatterns: any[], problems: any[]) => Promise<import("@codebolt/types").MatchProblemResponse>;
         getMatcherList: () => Promise<import("@codebolt/types").GetMatcherListTreeResponse>;

package/modules/codeutils.d.ts

@@ -1,4 +1,4 @@
-import { GetJsTreeResponse, MatchProblemResponse, GetMatcherListTreeResponse, getMatchDetail } from '@codebolt/types';
+import { MatchProblemResponse, GetMatcherListTreeResponse, getMatchDetail } from '@codebolt/types';
 /**
  * A utility module for working with code.
  */
@@ -8,7 +8,7 @@ declare 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>;
+    getJsTree: (filePath?: string) => Promise<any>;
     /**
      * Retrieves all files as Markdown.
      * @returns {Promise<string>} A promise that resolves with the Markdown content of all files.

package/modules/codeutils.js

@@ -1,9 +1,36 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const websocket_1 = __importDefault(require("./websocket"));
+const fs = __importStar(require("fs"));
+const path_1 = __importDefault(require("path"));
+const tree_sitter_1 = __importDefault(require("tree-sitter"));
+const tree_sitter_javascript_1 = __importDefault(require("tree-sitter-javascript"));
 /**
  * A utility module for working with code.
  */
@@ -14,20 +41,70 @@ const cbcodeutils = {
      * @returns {Promise<GetJsTreeResponse>} A promise that resolves with the JS tree response.
      */
     getJsTree: (filePath) => {
-        return new Promise((resolve, reject) => {
+        return new Promise(async (resolve, reject) => {
             websocket_1.default.getWebsocket.send(JSON.stringify({
-                "type": "codeEvent",
-                "action": "getJsTree",
-                payload: {
-                    filePath
-                }
+                "type": "settingEvent",
+                "action": "getProjectPath"
             }));
             websocket_1.default.getWebsocket.on('message', (data) => {
                 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 tree_sitter_1.default();
+                        // Initialize the parser with the JavaScript language
+                        parser.setLanguage(tree_sitter_javascript_1.default);
+                        const trees = [];
+                        const functionNodes = [];
+                        const processDirectory = (directory) => {
+                            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_1.default.join(directory, file.name)); // Recursive call for subdirectories
+                                    }
+                                }
+                                else if (path_1.default.extname(file.name) === '.js') {
+                                    const code = fs.readFileSync(path_1.default.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_1.default.extname(pathInput) === '.js') {
+                            // Read a single JavaScript file
+                            const code = fs.readFileSync(pathInput, 'utf-8');
+                            let tree = parser.parse(code);
+                            trees.push(tree);
+                        }
+                        resolve({ event: 'GetJsTreeResponse', payload: trees }); // Return an array of abstract syntax trees (ASTs)
+                    }
+                    catch (error) {
+                        console.error('An error occurred:', error);
+                        return { event: 'GetJsTreeResponse', payload: 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
+            //     } 
+            // });
         });
     },
     /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codebolt/codeboltjs",
-  "version": "1.1.34",
+  "version": "1.1.36",
   "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"
   },

package/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<any> => {
+        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);
+                        }
+            
+                        resolve({ event: 'GetJsTreeResponse',payload:trees}); // Return an array of abstract syntax trees (ASTs)
+                    } catch (error) {
+                        console.error('An error occurred:', error);
+                        return { event: 'GetJsTreeResponse',payload: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
+            //     } 
+            // });
         });
     },