@comet/upgrade 1.47.0 → 1.48.0

package/lib/v7/replace-exception-interceptor-with-exception-filter.js

@@ -1,19 +1,45 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const promises_1 = require("fs/promises");
-const format_code_util_1 = require("../util/format-code.util");
+const ts_morph_1 = require("ts-morph");
 /**
  * Replaces the old ExceptionInterceptor with the new ExceptionFilter
  */
 async function replaceExceptionInterceptorWithExceptionFilter() {
-    const filePath = "api/src/main.ts";
-    let fileContent = (await (0, promises_1.readFile)(filePath)).toString();
-    if (!fileContent.includes("ExceptionInterceptor")) {
-        console.log("ExceptionInterceptor not found in main.ts. Make sure that you use the new ExceptionFilter.");
+    const project = new ts_morph_1.Project({ tsConfigFilePath: "./api/tsconfig.json" });
+    const sourceFile = project.getSourceFile("api/src/main.ts");
+    if (!sourceFile) {
         return;
     }
-    fileContent = fileContent.replace("app.useGlobalInterceptors(new ExceptionInterceptor(config.debug));", "app.useGlobalFilters(new ExceptionFilter(config.debug));");
-    fileContent = fileContent.replace("ExceptionInterceptor", "ExceptionFilter");
-    await (0, promises_1.writeFile)(filePath, await (0, format_code_util_1.formatCode)(fileContent, filePath));
+    const cmsApiImport = sourceFile.getImportDeclaration((declaration) => declaration.getModuleSpecifierValue() === "@comet/cms-api");
+    if (!cmsApiImport) {
+        return;
+    }
+    const exceptionInterceptorImport = cmsApiImport.getNamedImports().find((namedImport) => namedImport.getName() === "ExceptionInterceptor");
+    if (!exceptionInterceptorImport) {
+        return;
+    }
+    const bootstrap = sourceFile.getFirstDescendantByKind(ts_morph_1.SyntaxKind.FunctionDeclaration);
+    if (!bootstrap) {
+        return;
+    }
+    const useGlobalInterceptors = sourceFile
+        .getDescendantsOfKind(ts_morph_1.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(ts_morph_1.SyntaxKind.NewExpression)
+        .find((node) => node.getText().includes("ExceptionInterceptor"));
+    if (exceptionInterceptor) {
+        useGlobalInterceptors.getFirstDescendantByKindOrThrow(ts_morph_1.SyntaxKind.CallExpression).removeArgument(exceptionInterceptor);
+        if (useGlobalInterceptors.getDescendantsOfKind(ts_morph_1.SyntaxKind.NewExpression).length === 0) {
+            useGlobalInterceptors.remove();
+        }
+    }
+    exceptionInterceptorImport.remove();
+    cmsApiImport.addNamedImport("ExceptionFilter");
+    await sourceFile.save();
 }
 exports.default = replaceExceptionInterceptorWithExceptionFilter;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@comet/upgrade",
-    "version": "1.47.0",
+    "version": "1.48.0",
     "description": "Upgrade scripts for Comet DXP",
     "homepage": "https://github.com/vivid-planet/comet-upgrade#readme",
     "bugs": {