exguard-backend 1.0.26 → 1.0.28

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "exguard-backend",
-  "version": "1.0.26",
+  "version": "1.0.28",
   "private": false,
   "publishConfig": {
     "access": "public"

package/scripts/setup-nestjs.cjs

@@ -598,34 +598,45 @@ function updateAppModule() {
     return;
   }
 
-  // Add at the very top
-  appModuleContent = `import { ExGuardModule } from './exguard/exguard.module';\n${appModuleContent}`;
+  // Add import at the top
+  const importStatement = "import { ExGuardModule } from './exguard/exguard.module';\n";
+  appModuleContent = importStatement + appModuleContent;
+
+  // Check if ThrottlerModule.forRootAsync exists and add ExGuardModule to its imports
+  const throttlerAsyncMatch = appModuleContent.match(/ThrottlerModule\.forRootAsync\(\{[\s\S]*?imports:\s*\[([^\]]*)\]/);
+  let addedToThrottler = false;
+  if (throttlerAsyncMatch) {
+    const throttlerImports = throttlerAsyncMatch[1];
+    if (!throttlerImports.includes('ExGuardModule')) {
+      // Clean up trailing commas and add ExGuardModule
+      let cleanedImports = throttlerImports.trim().replace(/,\s*,/g, ',').replace(/,\s*\]/g, ']');
+      if (!cleanedImports.endsWith(']')) {
+        cleanedImports = cleanedImports.replace(/,\s*$/, '') + ',\n    ExGuardModule';
+        appModuleContent = appModuleContent.replace(
+          throttlerAsyncMatch[0],
+          throttlerAsyncMatch[0].replace(throttlerImports, cleanedImports)
+        );
+        addedToThrottler = true;
+        logSuccess('Added ExGuardModule to ThrottlerModule.forRootAsync imports');
+      }
+    }
+  }
 
-  // Find and update the @Module decorator
-  const moduleDecoratorMatch = appModuleContent.match(/@Module\s*\(\s*\{([^}]*)\}/s);
-  if (moduleDecoratorMatch) {
-    const moduleContent = moduleDecoratorMatch[1];
-    
-    // Check if imports array exists
-    const importsArrayMatch = moduleContent.match(/imports:\s*\[([^\]]*)\]/s);
-    if (importsArrayMatch) {
-      // Add to existing imports array
-      const importsContent = importsArrayMatch[1];
-      const newImportsContent = importsContent.trim() + ',\n    ExGuardModule';
-      appModuleContent = appModuleContent.replace(
-        importsArrayMatch[0], 
-        `imports: [${newImportsContent}]`
-      );
-    } else {
-      // Add imports array after @Module({
-      const moduleStartMatch = appModuleContent.match(/@Module\s*\(\s*\{/);
-      if (moduleStartMatch) {
-        const moduleStartIndex = appModuleContent.indexOf(moduleStartMatch[0]);
-        const afterModuleStart = appModuleContent.substring(moduleStartIndex + moduleStartMatch[0].length);
-        
-        appModuleContent = appModuleContent.substring(0, moduleStartIndex + moduleStartMatch[0].length) + 
-                         `\n  imports: [ExGuardModule],` + 
-                         afterModuleStart;
+  // Only add to main imports if not added to ThrottlerModule
+  if (!addedToThrottler) {
+    const moduleDecoratorMatch = appModuleContent.match(/@Module\s*\(\s*\{([\s\S]*?)\n\}\s*\)/);
+    if (moduleDecoratorMatch) {
+      const moduleContent = moduleDecoratorMatch[1];
+      const importsArrayMatch = moduleContent.match(/imports:\s*\[([\s\S]*?)\]/);
+      if (importsArrayMatch) {
+        let importsContent = importsArrayMatch[1].trim().replace(/,\s*,/g, ',').replace(/,\s*\]/g, ']');
+        if (!importsContent.endsWith(']')) {
+          importsContent = importsContent.replace(/,\s*$/, '') + ',\n    ExGuardModule';
+          appModuleContent = appModuleContent.replace(
+            importsArrayMatch[0], 
+            `imports: [${importsContent}]`
+          );
+        }
       }
     }
   }