db: seeder

src/db/seeders/warrantySeeders.ts

@@ -1,46 +1,74 @@
-import { faker } from "@faker-js/faker"; // Use faker for generating random data
-
+import { cleanMongoData, extractDumpData, getCollectionData } from "./utils/extractDumpData";
 import { StatusEnum } from "../../common/enums/status.enum";
 import { Logger } from "../../core/logging/logger";
 import { WarrantyModel } from "../../modules/warranty/models/warranty.model";
 
-const warrantyNames = [
-  "Iran Warranty Co.",
-  // "Pars Guarantee",
-  // "Tehran Warranty Services",
-  // "Safir Warranty",
-  // "Farda Warranty Solutions",
-  // "Aryan Protection",
-  // "Mehr Warranty",
-  // "Saman Warranty Group",
-  // "Persian Warranty Solutions",
-  // "Shahab After Sales",
-  // "Payam Warranty",
-  // "Kavir Protection",
-  // "Tajrish Warranty",
-  // "Sina Warranty",
-  // "Atlas Warranty Services",
-  // "Sepahan Warranty",
-  // "Kish Warranty Group",
-  // "Navid Warranty Solutions",
-  // "Alborz Protection Services",
-  // "Raha Warranty Solutions",
-];
-
 export const seedWarranty = async (logger: Logger) => {
   try {
     logger.info("Warranty seeding started");
 
-    for (const name of warrantyNames) {
+    // Extract data from dump
+    const dumpData = extractDumpData();
+    const warrantiesData = getCollectionData(dumpData, "warranties");
+
+    if (warrantiesData.length === 0) {
+      logger.warn("No warranty data found in dump, using fallback data");
       await WarrantyModel.create({
-        name,
+        name: "Iran Warranty Co.",
         status: StatusEnum.Approved,
-        duration: faker.datatype.number({ min: 1, max: 36 }) + " months",
-        logoUrl: faker.image.business(),
+        duration: "12 months",
+        logoUrl: "",
       });
+      logger.info("Warranty seeding complete with fallback data!");
+      return;
     }
 
-    logger.info("Warranty seeding complete!");
+    // Clean and prepare warranty data
+    const cleanedWarranties = warrantiesData
+      .map((warranty) => {
+        const cleaned = cleanMongoData(warranty);
+        // Only include non-deleted warranties
+        if (cleaned.deleted) {
+          return null;
+        }
+        return {
+          name: cleaned.name,
+          status: cleaned.status || StatusEnum.Pending,
+          duration: cleaned.duration || "0",
+          logoUrl: cleaned.logoUrl || "",
+        };
+      })
+      .filter((w) => w !== null);
+
+    if (cleanedWarranties.length > 0) {
+      // Check for existing warranties and only create new ones to avoid duplicate key errors
+      let createdCount = 0;
+      let skippedCount = 0;
+      
+      for (const warranty of cleanedWarranties) {
+        const existing = await WarrantyModel.findOne({ name: warranty.name });
+        if (!existing) {
+          try {
+            await WarrantyModel.create(warranty);
+            createdCount++;
+          } catch (error: any) {
+            // Handle duplicate key errors gracefully
+            if (error.code === 11000) {
+              logger.warn(`Skipping duplicate warranty: ${warranty.name}`);
+              skippedCount++;
+            } else {
+              throw error;
+            }
+          }
+        } else {
+          skippedCount++;
+        }
+      }
+      
+      logger.info(`Warranty seeding complete! (${createdCount} created, ${skippedCount} skipped)`);
+    } else {
+      logger.warn("No valid warranties found after filtering");
+    }
   } catch (error) {
     logger.error("Error seeding warranties:", error);
   }