fix: retrive sourceFile when text change range is expected to conflict

lib/util/removeUnusedExport.ts

@@ -263,11 +263,25 @@ const getUpdatedExportDeclaration = (
 
 const getTextChanges = (
   languageService: ts.LanguageService,
-  sourceFile: ts.SourceFile,
+  file: string,
   editTracker: EditTracker,
 ) => {
+  const sourceFile = languageService.getProgram()?.getSourceFile(file);
+
+  if (!sourceFile) {
+    throw new Error('source file not found');
+  }
+
   const changes: ts.TextChange[] = [];
+  // usually we want to remove all unused exports in one pass, but there are some cases where we need to do multiple passes
+  // for example, when we have multiple export specifiers in one export declaration, we want to remove them one by one because the text change range will conflict
+  let aborted = false;
+
   for (const node of getUnusedExports(languageService, sourceFile)) {
+    if (aborted === true) {
+      break;
+    }
+
     if (ts.isExportSpecifier(node)) {
       const specifierCount = Array.from(node.parent.elements).length;
 
@@ -289,6 +303,7 @@ const getTextChanges = (
         continue;
       }
 
+      aborted = true;
       changes.push({
         newText: getUpdatedExportDeclaration(node.parent.parent, node),
         span: {
@@ -381,7 +396,7 @@ const getTextChanges = (
     });
   }
 
-  return changes;
+  return { changes, done: !aborted };
 };
 
 const disabledEditTracker: EditTracker = {
@@ -431,40 +446,54 @@ export const removeUnusedExport = ({
       }
     }
 
-    const changes = getTextChanges(languageService, sourceFile, editTracker);
+    let content = fileService.get(file);
 
-    editTracker.end(file);
+    do {
+      const { changes, done } = getTextChanges(
+        languageService,
+        file,
+        editTracker,
+      );
+
+      content = applyTextChanges(content, changes);
 
-    const oldContent = fileService.get(file);
-    let newContent = applyTextChanges(oldContent, changes);
+      fileService.set(file, content);
+
+      if (done) {
+        break;
+      }
+      // eslint-disable-next-line no-constant-condition
+    } while (true);
+
+    editTracker.end(file);
 
     if (enableCodeFix) {
       // eslint-disable-next-line no-constant-condition
       while (true) {
-        fileService.set(file, newContent);
+        fileService.set(file, content);
 
         const result = applyCodeFix({
           fixId: fixIdDelete,
           fileName: file,
           languageService,
         });
 
-        if (result === newContent) {
+        if (result === content) {
           break;
         }
 
-        newContent = result;
+        content = result;
       }
 
-      fileService.set(file, newContent);
+      fileService.set(file, content);
 
-      newContent = applyCodeFix({
+      content = applyCodeFix({
         fixId: fixIdDeleteImports,
         fileName: file,
         languageService,
       });
     }
 
-    fileService.set(file, newContent);
+    fileService.set(file, content);
   }
 };