nest-authme 1.2.3 → 1.3.0

package/dist/index.js

@@ -343,8 +343,7 @@ var init_generator = __esm({
         } else if (config.orm === "prisma") {
           plan.push(
             { template: "prisma/prisma.service.ts.hbs", output: `${config.sourceRoot}/prisma/prisma.service.ts` },
-            { template: "prisma/prisma.module.ts.hbs", output: `${config.sourceRoot}/prisma/prisma.module.ts` },
-            { template: "prisma/schema.prisma.additions.hbs", output: "prisma-schema-additions.prisma" }
+            { template: "prisma/prisma.module.ts.hbs", output: `${config.sourceRoot}/prisma/prisma.module.ts` }
           );
         }
         if (config.features.emailService) {
@@ -777,13 +776,139 @@ var init_main_ts_updater = __esm({
   }
 });
 
+// src/installer/prisma-schema-updater.ts
+var fs6, import_execa, PrismaSchemaUpdater;
+var init_prisma_schema_updater = __esm({
+  "src/installer/prisma-schema-updater.ts"() {
+    "use strict";
+    init_cjs_shims();
+    fs6 = __toESM(require("fs-extra"));
+    import_execa = require("execa");
+    init_template_engine();
+    init_config_builder();
+    PrismaSchemaUpdater = class {
+      constructor(schemaPath) {
+        this.schemaPath = schemaPath;
+      }
+      backupPath = null;
+      /**
+       * Update prisma/schema.prisma with auth models
+       */
+      async update(config) {
+        const exists = await fs6.pathExists(this.schemaPath);
+        if (!exists) {
+          return {
+            updated: false,
+            message: 'prisma/schema.prisma not found. Run "npx prisma init" first, then re-run nest-authme.',
+            skippedModels: []
+          };
+        }
+        const existingContent = await fs6.readFile(this.schemaPath, "utf-8");
+        const skippedModels = [];
+        const hasUser = /^\s*model\s+User\s*\{/m.test(existingContent);
+        const hasRefreshToken = /^\s*model\s+RefreshToken\s*\{/m.test(existingContent);
+        if (hasUser) {
+          skippedModels.push("User");
+        }
+        if (hasRefreshToken && config.features.refreshTokens) {
+          skippedModels.push("RefreshToken");
+        }
+        const needsUser = !hasUser;
+        const needsRefreshToken = config.features.refreshTokens && !hasRefreshToken;
+        if (!needsUser && !needsRefreshToken) {
+          return {
+            updated: false,
+            message: `Models already exist in schema.prisma: ${skippedModels.join(", ")}. Please check that your existing models have all required auth fields.`,
+            skippedModels
+          };
+        }
+        const templateEngine = new TemplateEngine();
+        const context = buildTemplateContext(config);
+        let modelsContent = await templateEngine.render("prisma/schema.prisma.models.hbs", context);
+        if (hasUser) {
+          modelsContent = modelsContent.replace(/model\s+User\s*\{[^}]*\}\n?/s, "");
+        }
+        if (hasRefreshToken) {
+          modelsContent = modelsContent.replace(/model\s+RefreshToken\s*\{[^}]*\}\n?/s, "");
+        }
+        modelsContent = modelsContent.replace(/\n{3,}/g, "\n\n").trim();
+        if (!modelsContent) {
+          return {
+            updated: false,
+            message: "No new models to add.",
+            skippedModels
+          };
+        }
+        await this.createBackup();
+        try {
+          const separator = "\n\n// === Auth models (added by nest-authme) ===\n\n";
+          const updatedContent = existingContent.trimEnd() + separator + modelsContent + "\n";
+          await fs6.writeFile(this.schemaPath, updatedContent, "utf-8");
+          await this.tryPrismaFormat();
+          const addedModels = [];
+          if (needsUser) addedModels.push("User");
+          if (needsRefreshToken) addedModels.push("RefreshToken");
+          let message = `Added ${addedModels.join(", ")} model(s) to prisma/schema.prisma`;
+          if (skippedModels.length > 0) {
+            message += `. Skipped existing: ${skippedModels.join(", ")} (check for missing auth fields)`;
+          }
+          return {
+            updated: true,
+            message,
+            skippedModels
+          };
+        } catch (error) {
+          await this.restoreBackup();
+          throw error;
+        }
+      }
+      /**
+       * Try to run prisma format for consistent indentation
+       */
+      async tryPrismaFormat() {
+        try {
+          await (0, import_execa.execa)("npx", ["prisma", "format"], {
+            cwd: this.schemaPath.replace(/[/\\]prisma[/\\]schema\.prisma$/, ""),
+            timeout: 15e3
+          });
+        } catch {
+        }
+      }
+      /**
+       * Create backup of schema.prisma
+       */
+      async createBackup() {
+        this.backupPath = `${this.schemaPath}.backup`;
+        await fs6.copy(this.schemaPath, this.backupPath);
+      }
+      /**
+       * Restore backup on failure
+       */
+      async restoreBackup() {
+        if (this.backupPath && await fs6.pathExists(this.backupPath)) {
+          await fs6.copy(this.backupPath, this.schemaPath, { overwrite: true });
+          await fs6.remove(this.backupPath);
+        }
+      }
+      /**
+       * Clean up backup after success
+       */
+      async cleanupBackup() {
+        if (this.backupPath && await fs6.pathExists(this.backupPath)) {
+          await fs6.remove(this.backupPath);
+        }
+      }
+    };
+  }
+});
+
 // src/installer/package-updater.ts
-var fs6, PackageUpdater;
+var fs7, PackageUpdater;
 var init_package_updater = __esm({
   "src/installer/package-updater.ts"() {
     "use strict";
     init_cjs_shims();
-    fs6 = __toESM(require("fs-extra"));
+    fs7 = __toESM(require("fs-extra"));
     PackageUpdater = class {
       constructor(packageJsonPath) {
         this.packageJsonPath = packageJsonPath;
@@ -795,7 +920,7 @@ var init_package_updater = __esm({
       async update(config) {
         await this.createBackup();
         try {
-          const packageJson = await fs6.readJSON(this.packageJsonPath);
+          const packageJson = await fs7.readJSON(this.packageJsonPath);
           const deps = this.getDependencies(config);
           packageJson.dependencies = {
             ...packageJson.dependencies,
@@ -809,7 +934,7 @@ var init_package_updater = __esm({
           packageJson.devDependencies = this.sortObject(
             packageJson.devDependencies
           );
-          await fs6.writeJSON(this.packageJsonPath, packageJson, { spaces: 2 });
+          await fs7.writeJSON(this.packageJsonPath, packageJson, { spaces: 2 });
         } catch (error) {
           await this.restoreBackup();
           throw error;
@@ -891,23 +1016,23 @@ var init_package_updater = __esm({
        */
       async createBackup() {
         this.backupPath = `${this.packageJsonPath}.backup`;
-        await fs6.copy(this.packageJsonPath, this.backupPath);
+        await fs7.copy(this.packageJsonPath, this.backupPath);
       }
       /**
        * Restore backup
        */
       async restoreBackup() {
-        if (this.backupPath && await fs6.pathExists(this.backupPath)) {
-          await fs6.copy(this.backupPath, this.packageJsonPath, { overwrite: true });
-          await fs6.remove(this.backupPath);
+        if (this.backupPath && await fs7.pathExists(this.backupPath)) {
+          await fs7.copy(this.backupPath, this.packageJsonPath, { overwrite: true });
+          await fs7.remove(this.backupPath);
         }
       }
       /**
        * Clean up backup
        */
       async cleanupBackup() {
-        if (this.backupPath && await fs6.pathExists(this.backupPath)) {
-          await fs6.remove(this.backupPath);
+        if (this.backupPath && await fs7.pathExists(this.backupPath)) {
+          await fs7.remove(this.backupPath);
         }
       }
     };
@@ -915,12 +1040,12 @@ var init_package_updater = __esm({
 });
 
 // src/installer/dependency-installer.ts
-var import_execa, import_detect_package_manager, DependencyInstaller;
+var import_execa2, import_detect_package_manager, DependencyInstaller;
 var init_dependency_installer = __esm({
   "src/installer/dependency-installer.ts"() {
     "use strict";
     init_cjs_shims();
-    import_execa = require("execa");
+    import_execa2 = require("execa");
     import_detect_package_manager = require("detect-package-manager");
     DependencyInstaller = class {
       /**
@@ -930,7 +1055,7 @@ var init_dependency_installer = __esm({
         const packageManager = await this.detectPackageManager(cwd);
         console.log(`\u{1F4E6} Installing dependencies with ${packageManager}...`);
         try {
-          await (0, import_execa.execa)(packageManager, ["install"], {
+          await (0, import_execa2.execa)(packageManager, ["install"], {
             cwd,
             stdio: "inherit"
           });
@@ -947,7 +1072,7 @@ var init_dependency_installer = __esm({
       async installCapture(cwd) {
         const packageManager = await this.detectPackageManager(cwd);
         try {
-          const result = await (0, import_execa.execa)(packageManager, ["install"], {
+          const result = await (0, import_execa2.execa)(packageManager, ["install"], {
             cwd,
             stdio: "pipe"
           });
@@ -978,7 +1103,8 @@ __export(installer_exports, {
   AppModuleUpdater: () => AppModuleUpdater,
   DependencyInstaller: () => DependencyInstaller,
   MainTsUpdater: () => MainTsUpdater,
-  PackageUpdater: () => PackageUpdater
+  PackageUpdater: () => PackageUpdater,
+  PrismaSchemaUpdater: () => PrismaSchemaUpdater
 });
 var init_installer = __esm({
   "src/installer/index.ts"() {
@@ -986,6 +1112,7 @@ var init_installer = __esm({
     init_cjs_shims();
     init_ast_updater();
     init_main_ts_updater();
+    init_prisma_schema_updater();
     init_package_updater();
     init_dependency_installer();
   }
@@ -998,6 +1125,7 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 init_cjs_shims();
+var path5 = __toESM(require("path"));
 
 // src/analyzer/index.ts
 init_cjs_shims();
@@ -1476,10 +1604,16 @@ function showSuccess(stats) {
   console.log(import_chalk.default.gray("      # .env.example is also provided as a git-safe reference"));
   console.log();
   if (stats.orm === "prisma") {
-    console.log(import_chalk.default.cyan("   2. Add Prisma schema models (see prisma-schema-additions.prisma)"));
-    console.log(import_chalk.default.gray("      # Copy the models into your prisma/schema.prisma"));
-    console.log(import_chalk.default.gray("      npx prisma migrate dev --name add-auth-models"));
-    console.log(import_chalk.default.gray("      npx prisma generate"));
+    if (stats.prismaSchemaUpdated) {
+      console.log(import_chalk.default.cyan("   2. Run Prisma migration"));
+      console.log(import_chalk.default.gray("      npx prisma migrate dev --name add-auth-models"));
+      console.log(import_chalk.default.gray("      npx prisma generate"));
+    } else {
+      console.log(import_chalk.default.cyan("   2. Add Prisma schema models manually"));
+      console.log(import_chalk.default.gray("      # Check src/auth/README.md for the model definitions"));
+      console.log(import_chalk.default.gray("      npx prisma migrate dev --name add-auth-models"));
+      console.log(import_chalk.default.gray("      npx prisma generate"));
+    }
   } else {
     console.log(import_chalk.default.cyan("   2. Create database migration (if using TypeORM)"));
     console.log(import_chalk.default.gray("      npm run migration:generate -- src/migrations/CreateUserTable"));
@@ -1602,6 +1736,27 @@ async function run(cwd = process.cwd(), options = {}) {
     );
     process.exit(1);
   }
+  let prismaSchemaUpdated = false;
+  if (config.orm === "prisma") {
+    const prismaSpinner = createSpinner("Updating prisma/schema.prisma...").start();
+    try {
+      const { PrismaSchemaUpdater: PrismaSchemaUpdater2 } = await Promise.resolve().then(() => (init_installer(), installer_exports));
+      const schemaPath = path5.join(projectInfo.root, "prisma", "schema.prisma");
+      const prismaUpdater = new PrismaSchemaUpdater2(schemaPath);
+      const prismaResult = await prismaUpdater.update(config);
+      if (prismaResult.updated) {
+        await prismaUpdater.cleanupBackup();
+        prismaSpinner.succeed(prismaResult.message);
+        prismaSchemaUpdated = true;
+      } else {
+        prismaSpinner.warn(prismaResult.message);
+      }
+    } catch (error) {
+      prismaSpinner.warn(
+        `Could not update prisma/schema.prisma: ${error instanceof Error ? error.message : "Unknown error"}`
+      );
+    }
+  }
   const mainSpinner = createSpinner("Updating main.ts with global guards...").start();
   try {
     const { MainTsUpdater: MainTsUpdater2 } = await Promise.resolve().then(() => (init_installer(), installer_exports));
@@ -1651,7 +1806,8 @@ async function run(cwd = process.cwd(), options = {}) {
     orm: config.orm,
     swagger: config.features.swagger,
     emailVerification: config.features.emailVerification,
-    resetPassword: config.features.resetPassword
+    resetPassword: config.features.resetPassword,
+    prismaSchemaUpdated
   });
   console.log("\u{1F41B} Issues? https://github.com/Islamawad132/add-nest-auth/issues");
   console.log("\u2B50 Like it? https://www.npmjs.com/package/nest-authme");