devbonzai 1.7.0 → 1.7.1

package/cli.js

@@ -418,12 +418,20 @@ function extractJavaScriptFunctions(filePath) {
         return content.substring(node.start, node.end);
       };
       
+      // Track visited nodes to avoid duplicates
+      const visitedNodes = new Set();
+      
       // Traverse AST
-      function traverse(node) {
+      function traverse(node, parentType = null) {
         if (!node) return;
         
+        // Skip if already visited (avoid processing same node twice)
+        if (visitedNodes.has(node)) return;
+        visitedNodes.add(node);
+        
         // Function declarations: function myFunc() {}
-        if (node.type === 'FunctionDeclaration' && node.id) {
+        // Skip if inside ExportNamedDeclaration (will be handled below)
+        if (node.type === 'FunctionDeclaration' && node.id && parentType !== 'ExportNamedDeclaration') {
           functions.push({
             name: node.id.name,
             content: getCode(node),
@@ -437,7 +445,6 @@ function extractJavaScriptFunctions(filePath) {
             node.init && 
             (node.init.type === 'ArrowFunctionExpression' || node.init.type === 'FunctionExpression') &&
             node.id && node.id.type === 'Identifier') {
-          const funcNode = node.init;
           const funcContent = getCode(node);
           functions.push({
             name: node.id.name,
@@ -448,7 +455,8 @@ function extractJavaScriptFunctions(filePath) {
         }
         
         // Class declarations: class User { ... }
-        if (node.type === 'ClassDeclaration' && node.id) {
+        // Skip if inside ExportNamedDeclaration (will be handled below)
+        if (node.type === 'ClassDeclaration' && node.id && parentType !== 'ExportNamedDeclaration') {
           const className = node.id.name;
           const methods = [];
           
@@ -491,6 +499,8 @@ function extractJavaScriptFunctions(filePath) {
               endLine: node.declaration.loc ? node.declaration.loc.end.line : 0,
               isExported: true
             });
+            // Mark as visited to avoid duplicate processing
+            visitedNodes.add(node.declaration);
           } else if (node.declaration.type === 'ClassDeclaration' && node.declaration.id) {
             const className = node.declaration.id.name;
             const methods = [];
@@ -522,6 +532,8 @@ function extractJavaScriptFunctions(filePath) {
               endLine: node.declaration.loc ? node.declaration.loc.end.line : 0,
               isExported: true
             });
+            // Mark as visited to avoid duplicate processing
+            visitedNodes.add(node.declaration);
           }
         }
         
@@ -530,9 +542,9 @@ function extractJavaScriptFunctions(filePath) {
           if (key === 'parent' || key === 'leadingComments' || key === 'trailingComments') continue;
           const child = node[key];
           if (Array.isArray(child)) {
-            child.forEach(traverse);
+            child.forEach(c => traverse(c, node.type));
           } else if (child && typeof child === 'object' && child.type) {
-            traverse(child);
+            traverse(child, node.type);
           }
         }
       }
@@ -610,11 +622,20 @@ function extractVueFunctions(filePath) {
         return scriptContent.substring(node.start, node.end);
       };
       
+      // Track visited nodes to avoid duplicates
+      const visitedNodes = new Set();
+      
       // Traverse AST (same logic as JavaScript parser)
-      function traverse(node) {
+      function traverse(node, parentType = null) {
         if (!node) return;
         
-        if (node.type === 'FunctionDeclaration' && node.id) {
+        // Skip if already visited (avoid processing same node twice)
+        if (visitedNodes.has(node)) return;
+        visitedNodes.add(node);
+        
+        // Function declarations: function myFunc() {}
+        // Skip if inside ExportNamedDeclaration (will be handled below)
+        if (node.type === 'FunctionDeclaration' && node.id && parentType !== 'ExportNamedDeclaration') {
           functions.push({
             name: node.id.name,
             content: getCode(node),
@@ -623,6 +644,7 @@ function extractVueFunctions(filePath) {
           });
         }
         
+        // Arrow functions: const myFunc = () => {}
         if (node.type === 'VariableDeclarator' && 
             node.init && 
             (node.init.type === 'ArrowFunctionExpression' || node.init.type === 'FunctionExpression') &&
@@ -636,7 +658,9 @@ function extractVueFunctions(filePath) {
           });
         }
         
-        if (node.type === 'ClassDeclaration' && node.id) {
+        // Class declarations: class User { ... }
+        // Skip if inside ExportNamedDeclaration (will be handled below)
+        if (node.type === 'ClassDeclaration' && node.id && parentType !== 'ExportNamedDeclaration') {
           const className = node.id.name;
           const methods = [];
           
@@ -668,6 +692,7 @@ function extractVueFunctions(filePath) {
           });
         }
         
+        // Export declarations: export function, export default
         if (node.type === 'ExportNamedDeclaration' && node.declaration) {
           if (node.declaration.type === 'FunctionDeclaration' && node.declaration.id) {
             functions.push({
@@ -677,6 +702,8 @@ function extractVueFunctions(filePath) {
               endLine: node.declaration.loc ? node.declaration.loc.end.line : 0,
               isExported: true
             });
+            // Mark as visited to avoid duplicate processing
+            visitedNodes.add(node.declaration);
           } else if (node.declaration.type === 'ClassDeclaration' && node.declaration.id) {
             const className = node.declaration.id.name;
             const methods = [];
@@ -708,16 +735,19 @@ function extractVueFunctions(filePath) {
               endLine: node.declaration.loc ? node.declaration.loc.end.line : 0,
               isExported: true
             });
+            // Mark as visited to avoid duplicate processing
+            visitedNodes.add(node.declaration);
           }
         }
         
+        // Recursively traverse children
         for (const key in node) {
           if (key === 'parent' || key === 'leadingComments' || key === 'trailingComments') continue;
           const child = node[key];
           if (Array.isArray(child)) {
-            child.forEach(traverse);
+            child.forEach(c => traverse(c, node.type));
           } else if (child && typeof child === 'object' && child.type) {
-            traverse(child);
+            traverse(child, node.type);
           }
         }
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "devbonzai",
-  "version": "1.7.0",
+  "version": "1.7.1",
   "description": "Quickly set up a local file server in any repository for browser-based file access",
   "main": "cli.js",
   "bin": {