nodejs-quickstart-structure 1.18.1 → 1.19.1

package/lib/modules/database-setup.js

@@ -1,116 +1,116 @@
-import fs from 'fs-extra';
-import path from 'path';
-import ejs from 'ejs';
-
-export const setupDatabase = async (templatesDir, targetDir, config) => {
-    const { database, dbName, language, architecture } = config;
-    const langExt = language === 'TypeScript' ? 'ts' : 'js';
-
-    // 1. Migrations
-    if (database === 'MongoDB') {
-        // Copy migrate-mongo config
-        const migrateConfigTemplate = await fs.readFile(path.join(templatesDir, 'common', 'migrate-mongo-config.js.ejs'), 'utf-8');
-        const migrateConfigContent = ejs.render(migrateConfigTemplate, { ...config });
-        await fs.writeFile(path.join(targetDir, 'migrate-mongo-config.js'), migrateConfigContent);
-
-        // Setup migrations directory
-        await fs.ensureDir(path.join(targetDir, 'migrations'));
-        
-        // Create initial migration file with timestamp
-        const timestamp = new Date().toISOString().replace(/[-T:.Z]/g, '').slice(0, 14); // YYYYMMDDHHMMSS
-        const migrationTemplate = await fs.readFile(path.join(templatesDir, 'common', 'migrations', 'init.js.ejs'), 'utf-8');
-        await fs.writeFile(path.join(targetDir, 'migrations', `${timestamp}-initial-setup.js`), migrationTemplate);
-
-    } else if (database !== 'None') {
-        // Flyway for SQL
-        await fs.ensureDir(path.join(targetDir, 'flyway/sql'));
-        const dbType = database === 'PostgreSQL' ? 'postgres' : 'mysql';
-        const sourceDir = path.join(templatesDir, 'db', dbType);
-        
-        const files = await fs.readdir(sourceDir);
-        for (const file of files) {
-            if (file.endsWith('.ejs')) {
-                const template = await fs.readFile(path.join(sourceDir, file), 'utf-8');
-                const content = ejs.render(template, { ...config });
-                const targetFileName = file.replace('.ejs', '');
-                await fs.writeFile(path.join(targetDir, 'flyway/sql', targetFileName), content);
-            } else {
-                await fs.copy(path.join(sourceDir, file), path.join(targetDir, 'flyway/sql', file));
-            }
-        }
-    }
-
-    // 2. Database Config
-    if (database !== 'None') {
-        const dbConfigFileName = language === 'TypeScript' ? (database === 'MongoDB' ? 'mongoose.ts' : 'database.ts') : (database === 'MongoDB' ? 'mongoose.js' : 'database.js');
-        const dbConfigTemplateSource = path.join(templatesDir, 'common', 'database', langExt, `${dbConfigFileName}.ejs`);
-        
-        let dbConfigTarget;
-
-        if (architecture === 'MVC') {
-            await fs.ensureDir(path.join(targetDir, 'src/config'));
-            dbConfigTarget = path.join(targetDir, 'src/config', database === 'MongoDB' ? (language === 'TypeScript' ? 'database.ts' : 'database.js') : dbConfigFileName);
-        } else {
-            // Clean Architecture
-            await fs.ensureDir(path.join(targetDir, 'src/infrastructure/database'));
-            dbConfigTarget = path.join(targetDir, 'src/infrastructure/database', language === 'TypeScript' ? 'database.ts' : 'database.js');
-        }
-
-        if (await fs.pathExists(dbConfigTemplateSource)) {
-            const dbTemplate = await fs.readFile(dbConfigTemplateSource, 'utf-8');
-            const dbContent = ejs.render(dbTemplate, { ...config });
-            await fs.writeFile(dbConfigTarget, dbContent);
-
-            // Render Spec
-            const specTemplateName = dbConfigFileName.replace(`.${langExt}`, `.spec.${langExt}.ejs`);
-            const specTemplateSource = path.join(templatesDir, 'common', 'database', langExt, specTemplateName);
-            if (await fs.pathExists(specTemplateSource)) {
-                const specTemplate = await fs.readFile(specTemplateSource, 'utf-8');
-                const specContent = ejs.render(specTemplate, { ...config });
-                const specTarget = dbConfigTarget.replace(`${path.sep}src${path.sep}`, `${path.sep}tests${path.sep}unit${path.sep}`).replace(`.${langExt}`, `.spec.${langExt}`);
-                await fs.ensureDir(path.dirname(specTarget));
-                await fs.writeFile(specTarget, specContent);
-            }
-        }
-    } else if (architecture === 'MVC') {
-         // Even if DB is None, MVC needs src/config for other things (like swagger or general config)
-         await fs.ensureDir(path.join(targetDir, 'src/config'));
-    }
-
-    // 3. Models / Entities
-    await generateModels(templatesDir, targetDir, config);
-};
-
-export const generateModels = async (templatesDir, targetDir, config) => {
-    const { database, language, architecture } = config;
-    const langExt = language === 'TypeScript' ? 'ts' : 'js';
-    const modelFileName = language === 'TypeScript' ? 'User.ts' : 'User.js';
-
-    const sourceModelName = database === 'MongoDB' ? `${modelFileName}.mongoose.ejs` : `${modelFileName}.ejs`;
-    const modelTemplateSource = path.join(templatesDir, 'common', 'database', langExt, 'models', sourceModelName);
-    let modelTarget;
-
-    if (architecture === 'MVC') {
-        await fs.ensureDir(path.join(targetDir, 'src/models'));
-        modelTarget = path.join(targetDir, 'src/models', modelFileName);
-    } else {
-        await fs.ensureDir(path.join(targetDir, 'src/infrastructure/database/models'));
-        modelTarget = path.join(targetDir, 'src/infrastructure/database/models', modelFileName);
-    }
-
-    if (await fs.pathExists(modelTemplateSource)) {
-        const modelTemplate = await fs.readFile(modelTemplateSource, 'utf-8');
-        const modelContent = ejs.render(modelTemplate, { ...config });
-        await fs.writeFile(modelTarget, modelContent);
-
-        // Render Spec
-        const modelSpecTemplateSource = path.join(templatesDir, 'common', 'database', langExt, 'models', `User.spec.${langExt}.ejs`);
-        if (await fs.pathExists(modelSpecTemplateSource)) {
-            const modelSpecTemplate = await fs.readFile(modelSpecTemplateSource, 'utf-8');
-            const modelSpecContent = ejs.render(modelSpecTemplate, { ...config });
-            const modelSpecTarget = modelTarget.replace(`${path.sep}src${path.sep}`, `${path.sep}tests${path.sep}unit${path.sep}`).replace(`.${langExt}`, `.spec.${langExt}`);
-            await fs.ensureDir(path.dirname(modelSpecTarget));
-            await fs.writeFile(modelSpecTarget, modelSpecContent);
-        }
-    }
-};
+import fs from 'fs-extra';
+import path from 'path';
+import ejs from 'ejs';
+
+export const setupDatabase = async (templatesDir, targetDir, config) => {
+    const { database, dbName, language, architecture } = config;
+    const langExt = language === 'TypeScript' ? 'ts' : 'js';
+
+    // 1. Migrations
+    if (database === 'MongoDB') {
+        // Copy migrate-mongo config
+        const migrateConfigTemplate = await fs.readFile(path.join(templatesDir, 'common', 'migrate-mongo-config.js.ejs'), 'utf-8');
+        const migrateConfigContent = ejs.render(migrateConfigTemplate, { ...config });
+        await fs.writeFile(path.join(targetDir, 'migrate-mongo-config.js'), migrateConfigContent);
+
+        // Setup migrations directory
+        await fs.ensureDir(path.join(targetDir, 'migrations'));
+        
+        // Create initial migration file with timestamp
+        const timestamp = new Date().toISOString().replace(/[-T:.Z]/g, '').slice(0, 14); // YYYYMMDDHHMMSS
+        const migrationTemplate = await fs.readFile(path.join(templatesDir, 'common', 'migrations', 'init.js.ejs'), 'utf-8');
+        await fs.writeFile(path.join(targetDir, 'migrations', `${timestamp}-initial-setup.js`), migrationTemplate);
+
+    } else if (database !== 'None') {
+        // Flyway for SQL
+        await fs.ensureDir(path.join(targetDir, 'flyway/sql'));
+        const dbType = database === 'PostgreSQL' ? 'postgres' : 'mysql';
+        const sourceDir = path.join(templatesDir, 'db', dbType);
+        
+        const files = await fs.readdir(sourceDir);
+        for (const file of files) {
+            if (file.endsWith('.ejs')) {
+                const template = await fs.readFile(path.join(sourceDir, file), 'utf-8');
+                const content = ejs.render(template, { ...config });
+                const targetFileName = file.replace('.ejs', '');
+                await fs.writeFile(path.join(targetDir, 'flyway/sql', targetFileName), content);
+            } else {
+                await fs.copy(path.join(sourceDir, file), path.join(targetDir, 'flyway/sql', file));
+            }
+        }
+    }
+
+    // 2. Database Config
+    if (database !== 'None') {
+        const dbConfigFileName = language === 'TypeScript' ? (database === 'MongoDB' ? 'mongoose.ts' : 'database.ts') : (database === 'MongoDB' ? 'mongoose.js' : 'database.js');
+        const dbConfigTemplateSource = path.join(templatesDir, 'common', 'database', langExt, `${dbConfigFileName}.ejs`);
+        
+        let dbConfigTarget;
+
+        if (architecture === 'MVC') {
+            await fs.ensureDir(path.join(targetDir, 'src/config'));
+            dbConfigTarget = path.join(targetDir, 'src/config', database === 'MongoDB' ? (language === 'TypeScript' ? 'database.ts' : 'database.js') : dbConfigFileName);
+        } else {
+            // Clean Architecture
+            await fs.ensureDir(path.join(targetDir, 'src/infrastructure/database'));
+            dbConfigTarget = path.join(targetDir, 'src/infrastructure/database', language === 'TypeScript' ? 'database.ts' : 'database.js');
+        }
+
+        if (await fs.pathExists(dbConfigTemplateSource)) {
+            const dbTemplate = await fs.readFile(dbConfigTemplateSource, 'utf-8');
+            const dbContent = ejs.render(dbTemplate, { ...config });
+            await fs.writeFile(dbConfigTarget, dbContent);
+
+            // Render Spec
+            const specTemplateName = dbConfigFileName.replace(`.${langExt}`, `.spec.${langExt}.ejs`);
+            const specTemplateSource = path.join(templatesDir, 'common', 'database', langExt, specTemplateName);
+            if (await fs.pathExists(specTemplateSource)) {
+                const specTemplate = await fs.readFile(specTemplateSource, 'utf-8');
+                const specContent = ejs.render(specTemplate, { ...config });
+                const specTarget = dbConfigTarget.replace(`${path.sep}src${path.sep}`, `${path.sep}tests${path.sep}unit${path.sep}`).replace(`.${langExt}`, `.spec.${langExt}`);
+                await fs.ensureDir(path.dirname(specTarget));
+                await fs.writeFile(specTarget, specContent);
+            }
+        }
+    } else if (architecture === 'MVC') {
+         // Even if DB is None, MVC needs src/config for other things (like swagger or general config)
+         await fs.ensureDir(path.join(targetDir, 'src/config'));
+    }
+
+    // 3. Models / Entities
+    await generateModels(templatesDir, targetDir, config);
+};
+
+export const generateModels = async (templatesDir, targetDir, config) => {
+    const { database, language, architecture } = config;
+    const langExt = language === 'TypeScript' ? 'ts' : 'js';
+    const modelFileName = language === 'TypeScript' ? 'User.ts' : 'User.js';
+
+    const sourceModelName = database === 'MongoDB' ? `${modelFileName}.mongoose.ejs` : `${modelFileName}.ejs`;
+    const modelTemplateSource = path.join(templatesDir, 'common', 'database', langExt, 'models', sourceModelName);
+    let modelTarget;
+
+    if (architecture === 'MVC') {
+        await fs.ensureDir(path.join(targetDir, 'src/models'));
+        modelTarget = path.join(targetDir, 'src/models', modelFileName);
+    } else {
+        await fs.ensureDir(path.join(targetDir, 'src/infrastructure/database/models'));
+        modelTarget = path.join(targetDir, 'src/infrastructure/database/models', modelFileName);
+    }
+
+    if (await fs.pathExists(modelTemplateSource)) {
+        const modelTemplate = await fs.readFile(modelTemplateSource, 'utf-8');
+        const modelContent = ejs.render(modelTemplate, { ...config });
+        await fs.writeFile(modelTarget, modelContent);
+
+        // Render Spec
+        const modelSpecTemplateSource = path.join(templatesDir, 'common', 'database', langExt, 'models', `User.spec.${langExt}.ejs`);
+        if (await fs.pathExists(modelSpecTemplateSource)) {
+            const modelSpecTemplate = await fs.readFile(modelSpecTemplateSource, 'utf-8');
+            const modelSpecContent = ejs.render(modelSpecTemplate, { ...config });
+            const modelSpecTarget = modelTarget.replace(`${path.sep}src${path.sep}`, `${path.sep}tests${path.sep}unit${path.sep}`).replace(`.${langExt}`, `.spec.${langExt}`);
+            await fs.ensureDir(path.dirname(modelSpecTarget));
+            await fs.writeFile(modelSpecTarget, modelSpecContent);
+        }
+    }
+};