add-nest-auth 1.0.5 → 1.0.6

package/dist/cli.js

@@ -489,13 +489,157 @@ var init_ast_updater = __esm({
   }
 });
 
+// src/installer/main-ts-updater.ts
+var import_ts_morph2, fs5, MainTsUpdater;
+var init_main_ts_updater = __esm({
+  "src/installer/main-ts-updater.ts"() {
+    "use strict";
+    init_cjs_shims();
+    import_ts_morph2 = require("ts-morph");
+    fs5 = __toESM(require("fs-extra"));
+    MainTsUpdater = class {
+      constructor(mainTsPath) {
+        this.mainTsPath = mainTsPath;
+        this.project = new import_ts_morph2.Project({
+          skipAddingFilesFromTsConfig: true
+        });
+      }
+      project;
+      sourceFile;
+      backupPath = null;
+      /**
+       * Update main.ts with global guards and validation pipe
+       */
+      async update() {
+        if (!await fs5.pathExists(this.mainTsPath)) {
+          throw new Error(`main.ts not found at ${this.mainTsPath}`);
+        }
+        await this.createBackup();
+        try {
+          this.sourceFile = this.project.addSourceFileAtPath(this.mainTsPath);
+          this.addImports();
+          this.addGlobalGuardsAndPipes();
+          await this.sourceFile.save();
+        } catch (error) {
+          await this.restoreBackup();
+          throw error;
+        }
+      }
+      /**
+       * Add necessary imports
+       */
+      addImports() {
+        if (!this.sourceFile) {
+          throw new Error("Source file not loaded");
+        }
+        this.addImport("@nestjs/core", ["Reflector"]);
+        this.addImport("@nestjs/common", ["ValidationPipe"]);
+        this.addImport("./auth/guards/jwt-auth.guard", ["JwtAuthGuard"]);
+      }
+      /**
+       * Add an import statement if it doesn't exist
+       */
+      addImport(moduleSpecifier, namedImports) {
+        if (!this.sourceFile) return;
+        const existingImport = this.sourceFile.getImportDeclarations().find((imp) => imp.getModuleSpecifierValue() === moduleSpecifier);
+        if (existingImport) {
+          const existingNames = existingImport.getNamedImports().map((ni) => ni.getName());
+          const missingImports = namedImports.filter(
+            (name) => !existingNames.includes(name)
+          );
+          if (missingImports.length > 0) {
+            existingImport.addNamedImports(missingImports);
+          }
+        } else {
+          this.sourceFile.addImportDeclaration({
+            moduleSpecifier,
+            namedImports
+          });
+        }
+      }
+      /**
+       * Add global guards and validation pipe to bootstrap function
+       */
+      addGlobalGuardsAndPipes() {
+        if (!this.sourceFile) return;
+        const bootstrapFunc = this.sourceFile.getFunction("bootstrap");
+        if (!bootstrapFunc) {
+          throw new Error("bootstrap function not found in main.ts");
+        }
+        const body = bootstrapFunc.getBody();
+        if (!body || !import_ts_morph2.Node.isBlock(body)) {
+          throw new Error("bootstrap function has no body");
+        }
+        const bodyText = body.getText();
+        if (bodyText.includes("useGlobalGuards") || bodyText.includes("JwtAuthGuard")) {
+          return;
+        }
+        const statements = body.getStatements();
+        let listenIndex = -1;
+        for (let i = 0; i < statements.length; i++) {
+          const text = statements[i].getText();
+          if (text.includes(".listen(") || text.includes(".listen (")) {
+            listenIndex = i;
+            break;
+          }
+        }
+        if (listenIndex === -1) {
+          listenIndex = statements.length;
+        }
+        const codeToInsert = [
+          "",
+          "  // Enable global validation pipe",
+          "  app.useGlobalPipes(",
+          "    new ValidationPipe({",
+          "      whitelist: true,",
+          "      forbidNonWhitelisted: true,",
+          "      transform: true,",
+          "    }),",
+          "  );",
+          "",
+          "  // Enable global JWT guard (all routes protected by default)",
+          "  // Use @Public() decorator on routes that should be accessible without auth",
+          "  const reflector = app.get(Reflector);",
+          "  app.useGlobalGuards(new JwtAuthGuard(reflector));",
+          ""
+        ].join("\n");
+        body.insertStatements(listenIndex, codeToInsert);
+      }
+      /**
+       * Create backup of main.ts
+       */
+      async createBackup() {
+        this.backupPath = `${this.mainTsPath}.backup`;
+        await fs5.copy(this.mainTsPath, this.backupPath);
+      }
+      /**
+       * Restore backup
+       */
+      async restoreBackup() {
+        if (this.backupPath && await fs5.pathExists(this.backupPath)) {
+          await fs5.copy(this.backupPath, this.mainTsPath, { overwrite: true });
+          await fs5.remove(this.backupPath);
+        }
+      }
+      /**
+       * Clean up backup
+       */
+      async cleanupBackup() {
+        if (this.backupPath && await fs5.pathExists(this.backupPath)) {
+          await fs5.remove(this.backupPath);
+        }
+      }
+    };
+  }
+});
+
 // src/installer/package-updater.ts
-var fs5, PackageUpdater;
+var fs6, PackageUpdater;
 var init_package_updater = __esm({
   "src/installer/package-updater.ts"() {
     "use strict";
     init_cjs_shims();
-    fs5 = __toESM(require("fs-extra"));
+    fs6 = __toESM(require("fs-extra"));
     PackageUpdater = class {
       constructor(packageJsonPath) {
         this.packageJsonPath = packageJsonPath;
@@ -507,7 +651,7 @@ var init_package_updater = __esm({
       async update(config) {
         await this.createBackup();
         try {
-          const packageJson = await fs5.readJSON(this.packageJsonPath);
+          const packageJson = await fs6.readJSON(this.packageJsonPath);
           const deps = this.getDependencies(config);
           packageJson.dependencies = {
             ...packageJson.dependencies,
@@ -521,7 +665,7 @@ var init_package_updater = __esm({
           packageJson.devDependencies = this.sortObject(
             packageJson.devDependencies
           );
-          await fs5.writeJSON(this.packageJsonPath, packageJson, { spaces: 2 });
+          await fs6.writeJSON(this.packageJsonPath, packageJson, { spaces: 2 });
         } catch (error) {
           await this.restoreBackup();
           throw error;
@@ -581,23 +725,23 @@ var init_package_updater = __esm({
        */
       async createBackup() {
         this.backupPath = `${this.packageJsonPath}.backup`;
-        await fs5.copy(this.packageJsonPath, this.backupPath);
+        await fs6.copy(this.packageJsonPath, this.backupPath);
       }
       /**
        * Restore backup
        */
       async restoreBackup() {
-        if (this.backupPath && await fs5.pathExists(this.backupPath)) {
-          await fs5.copy(this.backupPath, this.packageJsonPath, { overwrite: true });
-          await fs5.remove(this.backupPath);
+        if (this.backupPath && await fs6.pathExists(this.backupPath)) {
+          await fs6.copy(this.backupPath, this.packageJsonPath, { overwrite: true });
+          await fs6.remove(this.backupPath);
         }
       }
       /**
        * Clean up backup
        */
       async cleanupBackup() {
-        if (this.backupPath && await fs5.pathExists(this.backupPath)) {
-          await fs5.remove(this.backupPath);
+        if (this.backupPath && await fs6.pathExists(this.backupPath)) {
+          await fs6.remove(this.backupPath);
         }
       }
     };
@@ -650,6 +794,7 @@ var installer_exports = {};
 __export(installer_exports, {
   AppModuleUpdater: () => AppModuleUpdater,
   DependencyInstaller: () => DependencyInstaller,
+  MainTsUpdater: () => MainTsUpdater,
   PackageUpdater: () => PackageUpdater
 });
 var init_installer = __esm({
@@ -657,6 +802,7 @@ var init_installer = __esm({
     "use strict";
     init_cjs_shims();
     init_ast_updater();
+    init_main_ts_updater();
     init_package_updater();
     init_dependency_installer();
   }
@@ -730,6 +876,7 @@ var ProjectDetector = class {
       errors.push(`app.module.ts not found at ${sourceRoot}/app.module.ts`);
       return this.createInvalidProject(root, errors);
     }
+    const mainTsPath = path.join(root, sourceRoot, "main.ts");
     const orm = await detectORM(packageJson);
     const authModulePath = path.join(root, sourceRoot, "auth");
     if (await fs.pathExists(authModulePath)) {
@@ -739,6 +886,7 @@ var ProjectDetector = class {
       root,
       sourceRoot,
       appModulePath,
+      mainTsPath,
       packageJsonPath,
       nestCliConfigPath,
       orm,
@@ -781,6 +929,7 @@ var ProjectDetector = class {
       root,
       sourceRoot: "src",
       appModulePath: path.join(root, "src", "app.module.ts"),
+      mainTsPath: path.join(root, "src", "main.ts"),
       packageJsonPath: path.join(root, "package.json"),
       nestCliConfigPath: path.join(root, "nest-cli.json"),
       orm: "none",
@@ -1091,6 +1240,16 @@ async function run(cwd = process.cwd()) {
     );
     process.exit(1);
   }
+  const mainSpinner = createSpinner("Updating main.ts with global guards...").start();
+  try {
+    const { MainTsUpdater: MainTsUpdater2 } = await Promise.resolve().then(() => (init_installer(), installer_exports));
+    const mainUpdater = new MainTsUpdater2(projectInfo.mainTsPath);
+    await mainUpdater.update();
+    await mainUpdater.cleanupBackup();
+    mainSpinner.succeed("Updated main.ts with global JWT guard");
+  } catch (error) {
+    mainSpinner.warn("Could not auto-update main.ts (see main.ts.example for manual setup)");
+  }
   const pkgSpinner = createSpinner("Updating package.json...").start();
   try {
     const { PackageUpdater: PackageUpdater2 } = await Promise.resolve().then(() => (init_installer(), installer_exports));
@@ -1128,8 +1287,8 @@ async function run(cwd = process.cwd()) {
       refreshExpiration: config.features.refreshTokens ? config.jwt.refreshExpiration : void 0
     }
   });
-  console.log("\u{1F41B} Issues? https://github.com/yourusername/add-nest-auth/issues");
-  console.log("\u2B50 Like it? https://github.com/yourusername/add-nest-auth");
+  console.log("\u{1F41B} Issues? https://github.com/Islamawad132/add-nest-auth/issues");
+  console.log("\u2B50 Like it? https://github.com/Islamawad132/add-nest-auth");
   console.log();
 }