feat(cli): find nodes dynamically using syntax kind instead of index

ithaka · Oct 27, 2023 · e848da7 · e848da7
1 parent c0616c7
commit e848da7
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/packages/pharos-cli/src/utils/initComponentsUtil.ts b/packages/pharos-cli/src/utils/initComponentsUtil.ts
@@ -16,13 +16,22 @@ export async function addNewComponentToInitComponents(
     SyntaxKind.ImportDeclaration
   );
 
-  importStatementsDeclarations[0].addNamedImport(componentName);
+  importStatementsDeclarations.forEach((dec) => {
+    const importClause = dec.getImportClause();
+    if (importClause && importClause.getNamedImports().length > 0) {
+      dec.addNamedImport(componentName);
+    }
+  });
 
-  const registerComponentsExpressions = sourceFile.getDescendantsOfKind(
-    SyntaxKind.ArrayLiteralExpression
-  );
+  const registerComponentsExpressions = sourceFile.getDescendantsOfKind(SyntaxKind.CallExpression);
 
-  registerComponentsExpressions[0].addElement(componentName);
+  registerComponentsExpressions.forEach((callExp) => {
+    callExp.forEachChild((expChild) => {
+      if (expChild.isKind(SyntaxKind.ArrayLiteralExpression)) {
+        expChild.addElement(componentName);
+      }
+    });
+  });
 
   await sourceFile.save();
 }