Fix script to replace ExceptionInterceptor

src/v7/replace-exception-interceptor-with-exception-filter.ts

@@ -1,24 +1,59 @@
-import { readFile, writeFile } from "fs/promises";
-
-import { formatCode } from "../util/format-code.util";
+import { Project, SyntaxKind } from "ts-morph";
 
 /**
  * Replaces the old ExceptionInterceptor with the new ExceptionFilter
  */
 export default async function replaceExceptionInterceptorWithExceptionFilter() {
-    const filePath = "api/src/main.ts";
-    let fileContent = (await readFile(filePath)).toString();
+    const project = new Project({ tsConfigFilePath: "./api/tsconfig.json" });
+
+    const sourceFile = project.getSourceFile("api/src/main.ts");
+
+    if (!sourceFile) {
+        return;
+    }
+
+    const cmsApiImport = sourceFile.getImportDeclaration((declaration) => declaration.getModuleSpecifierValue() === "@comet/cms-api");
 
-    if (!fileContent.includes("ExceptionInterceptor")) {
-        console.log("ExceptionInterceptor not found in main.ts. Make sure that you use the new ExceptionFilter.");
+    if (!cmsApiImport) {
         return;
     }
 
-    fileContent = fileContent.replace(
-        "app.useGlobalInterceptors(new ExceptionInterceptor(config.debug));",
-        "app.useGlobalFilters(new ExceptionFilter(config.debug));",
-    );
-    fileContent = fileContent.replace("ExceptionInterceptor", "ExceptionFilter");
+    const exceptionInterceptorImport = cmsApiImport.getNamedImports().find((namedImport) => namedImport.getName() === "ExceptionInterceptor");
+
+    if (!exceptionInterceptorImport) {
+        return;
+    }
+
+    const bootstrap = sourceFile.getFirstDescendantByKind(SyntaxKind.FunctionDeclaration);
+
+    if (!bootstrap) {
+        return;
+    }
+
+    const useGlobalInterceptors = sourceFile
+        .getDescendantsOfKind(SyntaxKind.ExpressionStatement)
+        .find((node) => node.getText().includes("useGlobalInterceptors"));
+
+    if (!useGlobalInterceptors) {
+        return;
+    }
+
+    bootstrap.insertStatements(useGlobalInterceptors.getChildIndex() + 1, "app.useGlobalFilters(new ExceptionFilter(config.debug));");
+
+    const exceptionInterceptor = useGlobalInterceptors
+        .getDescendantsOfKind(SyntaxKind.NewExpression)
+        .find((node) => node.getText().includes("ExceptionInterceptor"));
+
+    if (exceptionInterceptor) {
+        useGlobalInterceptors.getFirstDescendantByKindOrThrow(SyntaxKind.CallExpression).removeArgument(exceptionInterceptor);
+
+        if (useGlobalInterceptors.getDescendantsOfKind(SyntaxKind.NewExpression).length === 0) {
+            useGlobalInterceptors.remove();
+        }
+    }
+
+    exceptionInterceptorImport.remove();
+    cmsApiImport.addNamedImport("ExceptionFilter");
 
-    await writeFile(filePath, await formatCode(fileContent, filePath));
+    await sourceFile.save();
 }