feat: seeder

package.json

@@ -7,7 +7,9 @@
     "start": "build && node dist/index.js",
     "start:dev": "ts-node-dev --files -r tsconfig-paths/register src/index.ts",
     "build": "rimraf dist && tsc && tsc-alias",
-    "lint": "eslint . --ext .ts"
+    "lint": "eslint . --ext .ts",
+    "seed": "ts-node-dev --files -r tsconfig-paths/register src/seeder/seed.ts",
+    "seed:reset": "ts-node --files -r tsconfig-paths/register src/seeder/seed.ts --reset"
   },
   "author": "",
   "license": "ISC",

src/configs/database.ts

@@ -2,7 +2,8 @@ import mongoose from "mongoose";
 import { config } from "./config";
 
 export const connectDatabase = async () => {
-  return mongoose.connect(config.db.uri).then(() => {
+  return mongoose.connect(config.db.uri).then((o) => {
     console.log(`Connected to MongoDB database successfully!`);
+    return o;
   });
 };

src/lib/decorators/swagger-controller.decorator.ts

@@ -1,8 +0,0 @@
-import { swaggerRegistry } from "@lib/swagger/swagger";
-
-export const SwaggerController = (path: string) => {
-  return (target: any) => {
-    target = target.prototype.constructor.name;
-    swaggerRegistry.setControllerPrefix(target, path);
-  };
-};

src/seeder/helpers/db-store.ts

@@ -0,0 +1,3 @@
+export const dbStore = {
+  dbConnected: false,
+};

src/seeder/helpers/seeder-wrapper.ts

@@ -0,0 +1,27 @@
+import { connectDatabase } from "@configs/database";
+import { dbStore } from "./db-store";
+
+export const seederWrapper =
+  (model, seederFunction: () => Promise<void>) => async () => {
+    // get cli arguments
+    const args = process.argv.slice(2);
+
+    // check if reset flag is passed
+    const resetFlag = args.includes("--reset");
+
+    // connect to database
+    if (!dbStore.dbConnected) {
+      await connectDatabase();
+      dbStore.dbConnected = true;
+    }
+
+    // clear the collection
+    if (resetFlag) {
+      console.log(`Clearing collection: ${model.collection.collectionName}`);
+      await model.deleteMany({});
+      console.log(`Collection ${model.collection.collectionName} cleared!`);
+    }
+
+    // run the seeder
+    return seederFunction();
+  };

src/seeder/seed.ts

@@ -0,0 +1,55 @@
+// mongoose seeder script
+// takes an optional cli argument "--reset" to reset the database
+//
+// Usage:
+// npm run seed
+// npm run seed:reset
+// node src/seeder/seed.ts
+// node src/seeder/seed.ts --reset
+
+import * as glob from "glob";
+import path from "path";
+
+const main = async () => {
+  // get cli arguments
+  const args = process.argv.slice(2);
+
+  // get cli names that don't start with "--"
+  const seederNames = args.filter((arg) => !arg.startsWith("--"));
+
+  // list all files under ./seeders
+  const seedersPath = path
+    .relative(process.cwd(), path.join(__dirname, "**/*.seeder.{ts,js}"))
+    .replace(/\\/g, "/");
+
+  const seedersFiles = glob
+    .sync(seedersPath, {})
+    .map((file) => {
+      return path.resolve(file);
+    })
+    .filter((file) => {
+      if (seederNames.length === 0) {
+        return true;
+      }
+
+      return seederNames.includes(path.basename(file));
+    });
+
+  // run all seeders
+  let count = 0;
+  for await (const file of seedersFiles) {
+    const baseName = path.basename(file);
+    console.log(`Running ${baseName} ...`);
+    const seeder = await import(file);
+    await seeder.default();
+    console.log(`${baseName} completed! (${++count}/${seedersFiles.length})`);
+  }
+
+  console.log("All seeders completed!");
+  process.exit(0);
+};
+
+main().catch((err) => {
+  console.error(err);
+  process.exit(1);
+});

src/seeder/seeders/admins.seeder.ts

@@ -0,0 +1,15 @@
+import { Role } from "@common/enums/role.enum";
+import { Admin } from "modules/console/common/models/admin.model";
+import { seederWrapper } from "seeder/helpers/seeder-wrapper";
+
+export default seederWrapper(Admin, async () => {
+  // create super admin
+  await Admin.create({
+    email: "[email protected]",
+    password: "super",
+    role: Role.SUPER_ADMIN,
+    gender: "M1 Abrams Tank",
+    name: "Super Admin",
+    dob: new Date(),
+  });
+});