[ATA] Remove duplicate modules from getReferencesForModule (#3391)

Author: ChaseMalik

.changeset/khaki-spoons-double.md

@@ -0,0 +1,5 @@
+---
+"@typescript/ata": patch
+---
+
+Remove duplicate modules from getReferencesForModule

packages/ata/src/index.ts

@@ -173,19 +173,21 @@ export const getReferencesForModule = (ts: typeof import("typescript"), code: st
     .filter(f => !isDtsFile(f.fileName))
     .filter(d => !libMap.has(d.fileName))
 
-  return references.map(r => {
-    let version = undefined
-    if (!r.fileName.startsWith(".")) {
-      version = "latest"
-      const line = code.slice(r.end).split("\n")[0]!
-      if (line.includes("// types:")) version = line.split("// types: ")[1]!.trim()
-    }
+  return references
+    .map(r => {
+      let version = undefined
+      if (!r.fileName.startsWith(".")) {
+        version = "latest"
+        const line = code.slice(r.end).split("\n")[0]!
+        if (line.includes("// types:")) version = line.split("// types: ")[1]!.trim()
+      }
 
-    return {
-      module: r.fileName,
-      version,
-    }
-  })
+      return {
+        module: r.fileName,
+        version,
+      }
+    })
+    .filter((r, index, self) => self.findIndex(m => m.module === r.module && m.version === r.version) === index)
 }
 
 /** A list of modules from the current sourcefile which we don't have existing files for */

packages/ata/tests/ata.spec.ts

@@ -16,6 +16,11 @@ describe(getReferencesForModule, () => {
     const code = "import {asda} from '123' // types: 1.2.3"
     expect(getReferencesForModule(ts, code)[0]).toEqual({ module: "123", version: "1.2.3" })
   })
+
+  it("removes duplicate imports", () => {
+    const code = "import 'abc'; import {asda} from 'abc'"
+    expect(getReferencesForModule(ts, code).map(m => m.module)).toEqual(["abc"])
+  })
 })
 
 describe("ignores lib references", () => {