enhance the transformComponentReferenceToJSX.js module to support com…

…ponent reference inside a ()
wpdas · Apr 19, 2024 · 27b4c04 · 27b4c04
1 parent a391466
commit 27b4c04
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/lib/parsers/transformComponentReferenceToJSX.js b/lib/parsers/transformComponentReferenceToJSX.js
@@ -50,12 +50,20 @@ function transformComponentReferenceToJSX(code, componentName) {
 }
 
 function isValidJSXContext(path) {
-  // Checks if the identifier is used in an array, or as an object property value, or directly assigned to a variable
-  return (
+  // Expand the check to include situations where the identifier is an argument in any function call
+  if (
     path.parentPath.isObjectProperty({ value: path.node }) ||
     path.parentPath.isArrayExpression() ||
     path.parentPath.isVariableDeclarator({ init: path.node })
-  );
+  ) {
+    return true;
+  }
+
+  if (path.parentPath.isCallExpression()) {
+    return path.parentPath.node.arguments.includes(path.node);
+  }
+
+  return false;
 }
 
 module.exports = transformComponentReferenceToJSX;