fix(schematics): improve utility getNamedImportReferences

taiga-family · Mar 13, 2024 · 920ab1a · 920ab1a
1 parent 522d94b
commit 920ab1a
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/projects/cdk/schematics/utils/get-named-import-references.ts b/projects/cdk/schematics/utils/get-named-import-references.ts
@@ -21,6 +21,17 @@ export function getNamedImportReferences(
     );
 
     return arrayFlat(
-        namedImports.map(specifier => specifier?.findReferencesAsNodes() || []),
+        namedImports.map(
+            specifier =>
+                specifier?.findReferencesAsNodes().filter(
+                    /**
+                     * Otherwise, each `findReferencesAsNodes` will return references across THE WHOLE project.
+                     * It will cause a lot of duplicates in the result and significantly slow down the process.
+                     */
+                    ref =>
+                        ref.getSourceFile().getFilePath() ===
+                        specifier?.getSourceFile().getFilePath(),
+                ) || [],
+        ),
     );
 }