@rayburst/cli 0.2.3 → 0.2.5

package/dist/{chunk-2VW2AH2U.js → chunk-JA66HKP7.js}

@@ -309,10 +309,135 @@ function getUniqueNodeId(filePath, nodeName, gitHash, usedIds, idCounters) {
   return id;
 }
 function isReactComponent(func) {
+  try {
+    let name = "";
+    if (typeof func.getName === "function") {
+      name = func.getName() || "";
+    }
+    if (!name && func.getParent) {
+      const parent = func.getParent();
+      if (parent && typeof parent.asKind === "function") {
+        const varDecl = parent.asKind(SyntaxKind.VariableDeclaration);
+        if (varDecl && typeof varDecl.getName === "function") {
+          name = varDecl.getName() || "";
+        }
+      }
+    }
+    const hasComponentName = name && /^[A-Z]/.test(name);
+    const returnType = func.getReturnType()?.getText() || "";
+    const hasJsxReturnType = returnType.includes("JSX.Element") || returnType.includes("React.ReactElement") || returnType.includes("ReactElement") || returnType.includes("React.ReactNode") || returnType.includes("ReactNode");
+    const body = func.getBody();
+    if (!body) return false;
+    const hasJsxElements = body.getDescendantsOfKind(SyntaxKind.JsxElement).length > 0 || body.getDescendantsOfKind(SyntaxKind.JsxSelfClosingElement).length > 0 || body.getDescendantsOfKind(SyntaxKind.JsxFragment).length > 0;
+    const hasHooks = body.getDescendantsOfKind(SyntaxKind.CallExpression).some((call) => {
+      const expr = call.getExpression().getText();
+      return expr.startsWith("use") && /^use[A-Z]/.test(expr);
+    });
+    if (hasJsxElements) return true;
+    if (hasJsxReturnType) return true;
+    if (hasComponentName && hasHooks) return true;
+    return false;
+  } catch (error) {
+    return false;
+  }
+}
+function analyzeReturnStatements(func) {
   const body = func.getBody();
-  if (!body) return false;
-  const text = body.getText();
-  return text.includes("return") && (text.includes("</") || text.includes("<>") || text.includes("jsx"));
+  if (!body) {
+    return {
+      hasJsxReturn: false,
+      hasNullReturn: false,
+      hasUndefinedReturn: false,
+      returnValueSummary: "void",
+      primaryReturnType: "undefined"
+    };
+  }
+  const returnStatements = body.getDescendantsOfKind(SyntaxKind.ReturnStatement);
+  let hasJsxReturn = false;
+  let hasNullReturn = false;
+  let hasUndefinedReturn = false;
+  const returnTypes = [];
+  for (const returnStmt of returnStatements) {
+    const expr = returnStmt.getExpression();
+    if (!expr) {
+      hasUndefinedReturn = true;
+      returnTypes.push("undefined");
+      continue;
+    }
+    if (Node.isJsxElement(expr) || Node.isJsxSelfClosingElement(expr) || Node.isJsxFragment(expr)) {
+      hasJsxReturn = true;
+      let jsxName = "JSX";
+      if (Node.isJsxElement(expr)) {
+        jsxName = expr.getOpeningElement().getTagNameNode().getText();
+      } else if (Node.isJsxSelfClosingElement(expr)) {
+        jsxName = expr.getTagNameNode().getText();
+      } else {
+        jsxName = "<>";
+      }
+      returnTypes.push(`<${jsxName}>`);
+      continue;
+    }
+    if (expr.getKind() === SyntaxKind.NullKeyword) {
+      hasNullReturn = true;
+      returnTypes.push("null");
+      continue;
+    }
+    if (expr.getKind() === SyntaxKind.Identifier && expr.getText() === "undefined") {
+      hasUndefinedReturn = true;
+      returnTypes.push("undefined");
+      continue;
+    }
+    if (Node.isConditionalExpression(expr)) {
+      const whenTrue = expr.getWhenTrue();
+      const whenFalse = expr.getWhenFalse();
+      const trueIsJsx = Node.isJsxElement(whenTrue) || Node.isJsxSelfClosingElement(whenTrue);
+      const falseIsJsx = Node.isJsxElement(whenFalse) || Node.isJsxSelfClosingElement(whenFalse);
+      if (trueIsJsx || falseIsJsx) hasJsxReturn = true;
+      if (whenTrue.getText() === "null") hasNullReturn = true;
+      if (whenFalse.getText() === "null") hasNullReturn = true;
+      const trueText = whenTrue.getText().length > 20 ? whenTrue.getText().substring(0, 17) + "..." : whenTrue.getText();
+      const falseText = whenFalse.getText().length > 20 ? whenFalse.getText().substring(0, 17) + "..." : whenFalse.getText();
+      returnTypes.push(`${trueText} : ${falseText}`);
+      continue;
+    }
+    const exprText = expr.getText();
+    const truncated = exprText.length > 50 ? exprText.substring(0, 47) + "..." : exprText;
+    returnTypes.push(truncated);
+  }
+  let primaryReturnType = "mixed";
+  if (hasJsxReturn && !hasNullReturn && !hasUndefinedReturn) {
+    primaryReturnType = "jsx";
+  } else if (hasNullReturn && !hasJsxReturn && !hasUndefinedReturn) {
+    primaryReturnType = "null";
+  } else if (hasUndefinedReturn && !hasJsxReturn && !hasNullReturn) {
+    primaryReturnType = "undefined";
+  } else if (returnTypes.length === 1) {
+    primaryReturnType = "primitive";
+  }
+  const uniqueReturns = Array.from(new Set(returnTypes));
+  const returnValueSummary = uniqueReturns.join(" | ") || "void";
+  return {
+    hasJsxReturn,
+    hasNullReturn,
+    hasUndefinedReturn,
+    returnValueSummary,
+    primaryReturnType
+  };
+}
+function getReturnDescription(returnAnalysis) {
+  if (returnAnalysis.hasJsxReturn && returnAnalysis.hasNullReturn) {
+    return "Returns JSX or null";
+  }
+  if (returnAnalysis.hasJsxReturn) {
+    return "Returns JSX";
+  }
+  if (returnAnalysis.hasNullReturn) {
+    return "Returns null";
+  }
+  if (returnAnalysis.hasUndefinedReturn) {
+    return "Returns void";
+  }
+  return "Returns value";
 }
 function extractComponents(sourceFile, relativePath, gitHash, baseX, startY, nodes, nodeMap, usedIds, idCounters) {
   let count = 0;
@@ -325,6 +450,7 @@ function extractComponents(sourceFile, relativePath, gitHash, baseX, startY, nod
       const params = func.getParameters();
       const propsParam = params[0];
       const propsType = propsParam ? propsParam.getType().getText() : "any";
+      const returnAnalysis = analyzeReturnStatements(func);
       const node = {
         id,
         type: "component",
@@ -333,7 +459,12 @@ function extractComponents(sourceFile, relativePath, gitHash, baseX, startY, nod
           componentName: name,
           props: propsType,
           label: name,
-          description: `Component in ${relativePath}`
+          description: `Component in ${relativePath}`,
+          returnValueSummary: returnAnalysis.returnValueSummary,
+          returnDescription: getReturnDescription(returnAnalysis),
+          hasJsxReturn: returnAnalysis.hasJsxReturn,
+          hasNullReturn: returnAnalysis.hasNullReturn,
+          primaryReturnType: returnAnalysis.primaryReturnType
         }
       };
       nodes.push(node);
@@ -353,6 +484,7 @@ function extractComponents(sourceFile, relativePath, gitHash, baseX, startY, nod
         const params = init.getParameters();
         const propsParam = params[0];
         const propsType = propsParam ? propsParam.getType().getText() : "any";
+        const returnAnalysis = analyzeReturnStatements(init);
         const node = {
           id,
           type: "component",
@@ -361,7 +493,12 @@ function extractComponents(sourceFile, relativePath, gitHash, baseX, startY, nod
             componentName: name,
             props: propsType,
             label: name,
-            description: `Component in ${relativePath}`
+            description: `Component in ${relativePath}`,
+            returnValueSummary: returnAnalysis.returnValueSummary,
+            returnDescription: getReturnDescription(returnAnalysis),
+            hasJsxReturn: returnAnalysis.hasJsxReturn,
+            hasNullReturn: returnAnalysis.hasNullReturn,
+            primaryReturnType: returnAnalysis.primaryReturnType
           }
         };
         nodes.push(node);
@@ -389,6 +526,7 @@ function extractFunctions(sourceFile, relativePath, gitHash, baseX, startY, node
         return `${paramName}: ${paramType}`;
       }).join(", ");
       const returnType = func.getReturnType().getText();
+      const returnAnalysis = analyzeReturnStatements(func);
       const node = {
         id,
         type: "function",
@@ -398,7 +536,12 @@ function extractFunctions(sourceFile, relativePath, gitHash, baseX, startY, node
           parameters: params,
           returnType,
           label: name,
-          description: `Function in ${relativePath}`
+          description: `Function in ${relativePath}`,
+          returnValueSummary: returnAnalysis.returnValueSummary,
+          returnDescription: getReturnDescription(returnAnalysis),
+          hasJsxReturn: returnAnalysis.hasJsxReturn,
+          hasNullReturn: returnAnalysis.hasNullReturn,
+          primaryReturnType: returnAnalysis.primaryReturnType
         }
       };
       nodes.push(node);

package/dist/index.js

@@ -9,7 +9,7 @@ import {
   readLocalMeta,
   writeLocalAnalysis,
   writeLocalMeta
-} from "./chunk-2VW2AH2U.js";
+} from "./chunk-JA66HKP7.js";
 export {
   addGitignoreEntry,
   analyzeProject,

package/dist/vite-plugin.js

@@ -1,6 +1,6 @@
 import {
   rayburstPlugin
-} from "./chunk-2VW2AH2U.js";
+} from "./chunk-JA66HKP7.js";
 export {
   rayburstPlugin
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rayburst/cli",
-  "version": "0.2.3",
+  "version": "0.2.5",
   "description": "Rayburst - Automatic code analysis for TypeScript/JavaScript projects via Vite plugin",
   "type": "module",
   "main": "./dist/index.js",
@@ -42,12 +42,12 @@
     }
   },
   "dependencies": {
-    "@rayburst/types": "^0.1.3",
     "chalk": "^5.3.0",
     "ts-morph": "^21.0.1",
     "zod": "^4.2.0"
   },
   "devDependencies": {
+    "@rayburst/types": "^0.1.4",
     "@types/node": "^25.0.2",
     "tsup": "^8.5.1",
     "typescript": "^5.9.3",