4bnode 3.5.3 → 4.0.0

package/add-crud.js

@@ -0,0 +1,226 @@
+#!/usr/bin/env node
+
+import path from "path";
+import { select, checkbox } from "@inquirer/prompts";
+import { toPascalCase } from "./lib/names.js";
+import { listRoutes, listModels, getModelSchema } from "./lib/project.js";
+import {
+  getRouteFilePath,
+  addImportToRoute,
+  insertCodeIntoRoute,
+  detectHttpMethod,
+  getRequestDataSource,
+} from "./lib/routeFile.js";
+import {
+  showHeader,
+  showTaskDone,
+  showError,
+  showFileAction,
+  installDeps,
+} from "./lib/ui.js";
+
+function buildInsertCode(pascalName, fields, dataSource) {
+  const assignments = fields
+    .map((f) => `    ${f}: ${dataSource}.${f}`)
+    .join(",\n");
+
+  let passwordLogic = "";
+  if (fields.includes("password")) {
+    passwordLogic = `
+  if (newData.password) {
+    const salt = await bcrypt.genSalt(10);
+    newData.password = await bcrypt.hash(newData.password, salt);
+  }`;
+  }
+
+  return `
+router.post('/create', async (req, res) => {
+  const newData = {
+${assignments}
+  };
+${passwordLogic}
+  try {
+    const newDocument = new ${pascalName}(newData);
+    await newDocument.save();
+    res.json({ message: 'Data inserted successfully', data: newDocument });
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ message: 'Server error' });
+  }
+});`;
+}
+
+function buildReadCode(pascalName) {
+  return `
+router.get('/', async (req, res) => {
+  try {
+    const documents = await ${pascalName}.find();
+    res.json({ data: documents });
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ message: 'Server error' });
+  }
+});
+
+router.get('/:id', async (req, res) => {
+  try {
+    const document = await ${pascalName}.findById(req.params.id);
+    if (!document) {
+      return res.status(404).json({ message: 'Document not found' });
+    }
+    res.json({ data: document });
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ message: 'Server error' });
+  }
+});`;
+}
+
+function buildUpdateCode(pascalName, fields, dataSource) {
+  const assignments = fields
+    .map((f) => `    ${f}: ${dataSource}.${f}`)
+    .join(",\n");
+
+  let passwordLogic = "";
+  if (fields.includes("password")) {
+    passwordLogic = `
+  if (updatedData.password) {
+    const salt = await bcrypt.genSalt(10);
+    updatedData.password = await bcrypt.hash(updatedData.password, salt);
+  }`;
+  }
+
+  return `
+router.put('/:id', async (req, res) => {
+  const updatedData = {
+${assignments}
+  };
+${passwordLogic}
+  try {
+    const updatedDocument = await ${pascalName}.findByIdAndUpdate(req.params.id, updatedData, { new: true });
+    if (!updatedDocument) {
+      return res.status(404).json({ message: 'Document not found' });
+    }
+    res.json({ message: 'Data updated successfully', data: updatedDocument });
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ message: 'Server error' });
+  }
+});`;
+}
+
+function buildDeleteCode(pascalName) {
+  return `
+router.delete('/:id', async (req, res) => {
+  try {
+    const deletedDocument = await ${pascalName}.findByIdAndDelete(req.params.id);
+    if (!deletedDocument) {
+      return res.status(404).json({ message: 'Document not found' });
+    }
+    res.json({ message: 'Data deleted successfully', data: deletedDocument });
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ message: 'Server error' });
+  }
+});`;
+}
+
+async function main() {
+  showHeader("add-crud", "Add CRUD operations to a route");
+
+  const routeFiles = listRoutes();
+  if (routeFiles.length === 0) {
+    showError("No route files found in src/routes.");
+    process.exit(1);
+  }
+
+  const modelFiles = listModels();
+  if (modelFiles.length === 0) {
+    showError("No model files found in src/models.");
+    process.exit(1);
+  }
+
+  const selectedRoute = await select({
+    message: "Select route file:",
+    choices: routeFiles.map((f) => ({ name: f, value: f })),
+  });
+
+  const selectedModelFile = await select({
+    message: "Select model file:",
+    choices: modelFiles.map((f) => ({ name: f, value: f })),
+  });
+
+  const operations = await checkbox({
+    message: "Select CRUD operations:",
+    choices: [
+      { name: "Create  (POST /create)", value: "create", checked: true },
+      { name: "Read    (GET / and GET /:id)", value: "read", checked: true },
+      { name: "Update  (PUT /:id)", value: "update", checked: true },
+      { name: "Delete  (DELETE /:id)", value: "delete", checked: true },
+    ],
+    validate: (v) => (v.length > 0 ? true : "Select at least one operation."),
+  });
+
+  const modelName = path.basename(selectedModelFile, ".js");
+  const pascalName = toPascalCase(modelName);
+  const fields = getModelSchema(modelName);
+
+  if (fields.length === 0) {
+    showError("No fields found in the selected model.");
+    process.exit(1);
+  }
+
+  console.log();
+
+  const routeFilePath = getRouteFilePath(selectedRoute);
+  const method = detectHttpMethod(routeFilePath);
+  const dataSource = getRequestDataSource(method);
+  const needsBcrypt = fields.includes("password") && (operations.includes("create") || operations.includes("update"));
+
+  // Build import
+  let importBlock = `import ${pascalName} from '../models/${modelName}.js';`;
+  if (needsBcrypt) {
+    importBlock = `import bcrypt from 'bcrypt';\n${importBlock}`;
+  }
+  addImportToRoute(routeFilePath, importBlock);
+
+  // Add selected operations
+  const added = [];
+
+  if (operations.includes("create")) {
+    insertCodeIntoRoute(routeFilePath, buildInsertCode(pascalName, fields, dataSource));
+    added.push("POST /create");
+  }
+
+  if (operations.includes("read")) {
+    insertCodeIntoRoute(routeFilePath, buildReadCode(pascalName));
+    added.push("GET / and GET /:id");
+  }
+
+  if (operations.includes("update")) {
+    insertCodeIntoRoute(routeFilePath, buildUpdateCode(pascalName, fields, dataSource));
+    added.push("PUT /:id");
+  }
+
+  if (operations.includes("delete")) {
+    insertCodeIntoRoute(routeFilePath, buildDeleteCode(pascalName));
+    added.push("DELETE /:id");
+  }
+
+  showFileAction("updated", `src/routes/${selectedRoute}`);
+
+  if (needsBcrypt) {
+    await installDeps("bcrypt");
+  }
+
+  showTaskDone(`CRUD operations added to ${selectedRoute}`, [
+    `Model: ${modelName}`,
+    ...added.map((a) => `Route: ${a}`),
+  ]);
+}
+
+main().catch((err) => {
+  if (err.name === "ExitPromptError") process.exit(0);
+  showError(err.message);
+  process.exit(1);
+});