4brains-node-generator
Version:
A CLI tool to generate a Node.js app by 4Brains Technologies
153 lines (133 loc) • 4.29 kB
JavaScript
import fs from "fs";
import path from "path";
import readline from "readline";
import chalk from "chalk";
import { execSync } from "child_process";
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout,
});
function listRoutes() {
const projectRoot = process.cwd();
const routesDir = path.join(projectRoot, "src", "routes");
if (!fs.existsSync(routesDir)) {
console.error(chalk.red("No routes directory found in src/routes"));
return null;
}
const routeFiles = fs
.readdirSync(routesDir)
.filter((file) => file.endsWith(".js"));
if (routeFiles.length === 0) {
console.error(chalk.red("No route files found in src/routes"));
return null;
}
return routeFiles;
}
function listModels() {
const projectRoot = process.cwd();
const modelsDir = path.join(projectRoot, "src", "models");
if (!fs.existsSync(modelsDir)) {
console.error(chalk.red("No models directory found in src/models"));
return null;
}
const modelFiles = fs
.readdirSync(modelsDir)
.filter((file) => file.endsWith(".js"));
if (modelFiles.length === 0) {
console.error(chalk.red("No model files found in src/models"));
return null;
}
return modelFiles;
}
function addReadTemplateToRoute(routeFile, modelName) {
const projectRoot = process.cwd();
const routeFilePath = path.join(projectRoot, "src", "routes", routeFile);
const pascalModelName =
modelName.charAt(0).toUpperCase() + modelName.slice(1);
const readTemplateImport = `
import ${pascalModelName} from '../models/${modelName}.js';
`;
const readTemplateFunction = `
// Data read logic
try {
const document = await ${pascalModelName}.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" });
}
`;
if (!fs.existsSync(routeFilePath)) {
console.error(
chalk.red(`The specified route file does not exist: ${routeFilePath}`)
);
return;
}
let routeContent = fs.readFileSync(routeFilePath, "utf8");
if (!routeContent.includes(`import ${pascalModelName}`)) {
routeContent = readTemplateImport + routeContent;
}
const lastCloseIndex = routeContent.lastIndexOf("});");
if (lastCloseIndex !== -1) {
const updatedContent = [
routeContent.slice(0, lastCloseIndex),
readTemplateFunction,
routeContent.slice(lastCloseIndex),
].join("\n");
routeContent = updatedContent;
} else {
routeContent += `\n${readTemplateFunction}\n`;
}
fs.writeFileSync(routeFilePath, routeContent, "utf8");
console.log(chalk.green(`Read template added to ${routeFile}`));
}
const routeFiles = listRoutes();
const modelFiles = listModels();
if (!routeFiles || !modelFiles) {
rl.close();
process.exit(0);
}
console.log(chalk.blue("Select the route file to modify:"));
routeFiles.forEach((file, index) => {
console.log(chalk.yellow(`${index + 1}. ${file}`));
});
rl.question(
chalk.cyan("Enter the number of the route file: "),
(routeAnswer) => {
const routeIndex = parseInt(routeAnswer, 10) - 1;
if (routeIndex >= 0 && routeIndex < routeFiles.length) {
const selectedRoute = routeFiles[routeIndex];
console.log(chalk.blue("Select the model file to use:"));
modelFiles.forEach((file, index) => {
console.log(chalk.yellow(`${index + 1}. ${file}`));
});
rl.question(
chalk.cyan("Enter the number of the model file: "),
(modelAnswer) => {
const modelIndex = parseInt(modelAnswer, 10) - 1;
if (modelIndex >= 0 && modelIndex < modelFiles.length) {
const selectedModel = path.basename(modelFiles[modelIndex], ".js");
addReadTemplateToRoute(selectedRoute, selectedModel);
rl.close();
} else {
console.error(
chalk.red("Invalid selection. Please choose a valid number.")
);
rl.close();
process.exit(0);
}
}
);
} else {
console.error(
chalk.red("Invalid selection. Please choose a valid number.")
);
rl.close();
process.exit(0);
}
}
);