feat: add MongoDB setup and installation options in orchestrateSetup

Author: talhabilal-dev

src/orchestrator.js

@@ -8,6 +8,7 @@ import { addMulter } from "./tasks/addMulter.js";
 import { configureScripts } from "./tasks/configureScripts.js";
 import { configurePackageJson } from "./tasks/configurePackageJson.js";
 import { setupESLintConfig } from "./tasks/setupESLint.js";
+import { setupDb } from "./tasks/setupDb.js";
 
 export async function orchestrateSetup() {
   console.log("🚀 Welcome to create-node-backend!");
@@ -16,15 +17,17 @@ export async function orchestrateSetup() {
   const useAuth = await askQuestion("Include auth? (y/n): ");
   const useMulter = await askQuestion("Include Multer (file uploads)? (y/n): ");
   const useLint = await askQuestion("Use ESLint? (y/n): ");
+  const useDb = await askQuestion("Use MongoDB? (y/n): ");
 
   await setupNpm(projectName);
-  await installDependencies(useAuth, useMulter, useLint);
+  await installDependencies(useAuth, useMulter, useLint, useDb);
   await setupFolderStructure(projectName);
   await setupEnv(projectName);
 
   if (useAuth.toLowerCase() === "y") await addAuth(projectName);
   if (useMulter.toLowerCase() === "y") await addMulter(projectName);
 
+  if (useDb.toLowerCase() === "y") await setupDb(projectName);
   await configureScripts(projectName);
   await configurePackageJson(projectName);
   if (useLint.toLowerCase() === "y") await setupESLintConfig(projectName);

src/tasks/configureScripts.js

@@ -5,8 +5,10 @@ export async function configureScripts(projectName) {
   const indexContent = `
 import express from 'express';
 import dotenv from 'dotenv';
+import connectDB from './config/db.config.js';
 import routes from './routes/userRoutes.js';
 
+
 dotenv.config();
 
 const app = express();

src/tasks/installDependencies.js

@@ -1,6 +1,11 @@
 import { execSync } from "child_process";
 
-export async function installDependencies(useAuth, useMulter, useLint) {
+export async function installDependencies(
+  useAuth,
+  useMulter,
+  useLint = "n",
+  useDb
+) {
   console.log("📥 Installing core dependencies...");
   execSync("npm install express dotenv", { stdio: "inherit" });
 
@@ -14,14 +19,22 @@ export async function installDependencies(useAuth, useMulter, useLint) {
     execSync("npm install multer", { stdio: "inherit" });
   }
 
+  if (useDb.toLowerCase() === "y") {
+    console.log("🗄️ Installing MongoDB dependencies...");
+    execSync("npm install mongoose", { stdio: "inherit" });
+  }
+
   console.log("📦 Installing dev dependencies...");
 
   execSync("npm install --save-dev nodemon", { stdio: "inherit" });
   if (useLint.toLowerCase() === "y") {
     console.log("🔍 Installing ESLint...");
-    execSync("npm install --save-dev eslint eslint-plugin-n eslint-plugin-promise", {
-      stdio: "inherit",
-    });
+    execSync(
+      "npm install --save-dev eslint eslint-plugin-n eslint-plugin-promise",
+      {
+        stdio: "inherit",
+      }
+    );
   }
 
   console.log("⚙️ Installing dev dependencies...");

src/tasks/setupDb.js

@@ -0,0 +1,26 @@
+import fs from "fs";
+import path from "path";
+
+export async function setupDb(projectName) {
+  const authMiddleware = `
+import mongoose from "mongoose";
+import dotenv from "dotenv";
+
+const connectDB = async () => {
+  try {
+    await mongoose.connect(process.env.MONGO_URI);
+    console.log("MongoDB connected successfully");
+  } catch (err) {
+    console.error("Error connecting to MongoDB:", err);
+    process.exit(1);
+  }
+};
+
+export default connectDB;
+`;
+
+  fs.writeFileSync(
+    path.join(process.cwd(), "src", "config", "db.config.js"),
+    authMiddleware.trim()
+  );
+}

src/tasks/setupEnv.js

@@ -5,6 +5,8 @@ export async function setupEnv(projectName) {
   const envContent = `
 PORT=3000
 JWT_SECRET=your_jwt_secret
+MONGO_URI=mongodb://localhost:27017/${projectName}
+
 `;
   fs.writeFileSync(path.join(process.cwd(), ".env"), envContent.trim());
 }