@svton/cli 2.0.0 → 2.1.0

package/dist/index.js

@@ -860,11 +860,16 @@ function collectEnvVars(features, config) {
   }
   return envVars;
 }
-async function generateEnvExample(features, config, targetPath) {
+async function generateEnvExample(features, config, targetPath, database = "mysql") {
   const envVars = collectEnvVars(features, config);
   if (envVars.length === 0) {
     return;
   }
+  const databaseUrls = {
+    mysql: "mysql://root:root123456@localhost:3306/{{PROJECT_NAME}}",
+    postgresql: "postgresql://postgres:postgres@localhost:5432/{{PROJECT_NAME}}",
+    sqlite: "file:./dev.db"
+  };
   const content = [
     "# ========================================",
     "# Environment Variables",
@@ -884,7 +889,7 @@ async function generateEnvExample(features, config, targetPath) {
     "# ========================================",
     "# Database Configuration",
     "# ========================================",
-    "DATABASE_URL=mysql://root:root123456@localhost:3306/{{PROJECT_NAME}}",
+    `DATABASE_URL=${databaseUrls[database] || databaseUrls.mysql}`,
     ""
   ];
   const featureGroups = {};
@@ -1011,7 +1016,7 @@ ${featuresList}
   await import_fs_extra5.default.writeFile(indexPath, content);
   logger.info("Generated capabilities index");
 }
-async function copyPrismaTemplates(templateDir, targetPath) {
+async function copyPrismaTemplates(templateDir, targetPath, database = "mysql") {
   const prismaTemplatesDir = import_path3.default.join(templateDir, "apps/backend/prisma");
   const prismaDestDir = import_path3.default.join(targetPath, "apps/backend/prisma");
   if (await import_fs_extra5.default.pathExists(prismaTemplatesDir)) {
@@ -1022,10 +1027,13 @@ async function copyPrismaTemplates(templateDir, targetPath) {
       if (file.endsWith(".tpl")) {
         const filePath = import_path3.default.join(prismaDestDir, file);
         const newPath = filePath.replace(/\.tpl$/, "");
-        await import_fs_extra5.default.rename(filePath, newPath);
+        let content = await import_fs_extra5.default.readFile(filePath, "utf-8");
+        content = content.replace(/provider\s*=\s*"postgresql"/, `provider = "${database}"`);
+        await import_fs_extra5.default.writeFile(newPath, content);
+        await import_fs_extra5.default.remove(filePath);
       }
     }
-    logger.info("Copied Prisma templates");
+    logger.info(`Copied Prisma templates with ${database} database`);
   } else {
     logger.warn(`Prisma templates not found: ${prismaTemplatesDir}`);
   }
@@ -1217,6 +1225,8 @@ async function createProject(projectName, options = {}) {
       answers = {
         org: options.org || projectName,
         template: options.template || "full-stack",
+        database: "mysql",
+        // 默认使用 MySQL
         features: [],
         // 默认不选择额外功能
         packageManager: options.packageManager || "pnpm",
@@ -1247,6 +1257,18 @@ async function createProject(projectName, options = {}) {
           ],
           default: options.template || "full-stack"
         },
+        {
+          type: "list",
+          name: "database",
+          message: "Choose a database:",
+          choices: [
+            { name: "MySQL", value: "mysql" },
+            { name: "PostgreSQL", value: "postgresql" },
+            { name: "SQLite", value: "sqlite" }
+          ],
+          default: "mysql",
+          when: (answers2) => answers2.template === "backend-only" || answers2.template === "full-stack"
+        },
         {
           type: "checkbox",
           name: "features",
@@ -1279,6 +1301,7 @@ async function createProject(projectName, options = {}) {
       projectName,
       orgName: answers.org.startsWith("@") ? answers.org : `@${answers.org}`,
       template: answers.template,
+      database: answers.database || "mysql",
       features: answers.features || [],
       packageManager: answers.packageManager,
       installDeps: answers.installDeps,
@@ -1290,6 +1313,9 @@ async function createProject(projectName, options = {}) {
     logger.info(`  Project Name: ${import_chalk2.default.white(config.projectName)}`);
     logger.info(`  Organization: ${import_chalk2.default.white(config.orgName)}`);
     logger.info(`  Template: ${import_chalk2.default.white(config.template)}`);
+    if (config.template === "backend-only" || config.template === "full-stack") {
+      logger.info(`  Database: ${import_chalk2.default.white(config.database)}`);
+    }
     if (config.features.length > 0) {
       logger.info(`  Features: ${import_chalk2.default.white(config.features.join(", "))}`);
     }
@@ -1374,9 +1400,9 @@ async function createProjectFromTemplate(config) {
       await copyExampleFiles(config.features, featuresConfig, templateDir, config.projectPath);
       await copySkillFiles(config.features, featuresConfig, templateDir, config.projectPath);
       if (config.template === "backend-only" || config.template === "full-stack") {
-        await copyPrismaTemplates(templateDir, config.projectPath);
+        await copyPrismaTemplates(templateDir, config.projectPath, config.database);
       }
-      await generateEnvExample(config.features, featuresConfig, config.projectPath);
+      await generateEnvExample(config.features, featuresConfig, config.projectPath, config.database);
       if (config.template === "backend-only" || config.template === "full-stack") {
         await updateAppModule(config.features, featuresConfig, config.projectPath);
       }
@@ -1384,6 +1410,19 @@ async function createProjectFromTemplate(config) {
     if (config.installDeps) {
       spinner.text = "Installing dependencies...";
       await installDependencies(config.packageManager);
+      if (config.template === "backend-only" || config.template === "full-stack") {
+        spinner.text = "Generating Prisma client...";
+        try {
+          const { execSync: execSync4 } = require("child_process");
+          execSync4(`${config.packageManager} --filter=backend prisma:generate`, {
+            cwd: config.projectPath,
+            stdio: "inherit"
+          });
+          logger.info("Prisma client generated successfully");
+        } catch (error) {
+          logger.warn('Failed to generate Prisma client. Please run "pnpm --filter=backend prisma:generate" manually.');
+        }
+      }
     }
     if (config.initGit) {
       spinner.text = "Initializing Git repository...";
@@ -1401,7 +1440,7 @@ async function createProjectFromTemplate(config) {
 }
 
 // package.json
-var version = "2.0.0";
+var version = "2.1.0";
 
 // src/index.ts
 async function cli() {

package/dist/index.mjs

@@ -697,11 +697,16 @@ function collectEnvVars(features, config) {
   }
   return envVars;
 }
-async function generateEnvExample(features, config, targetPath) {
+async function generateEnvExample(features, config, targetPath, database = "mysql") {
   const envVars = collectEnvVars(features, config);
   if (envVars.length === 0) {
     return;
   }
+  const databaseUrls = {
+    mysql: "mysql://root:root123456@localhost:3306/{{PROJECT_NAME}}",
+    postgresql: "postgresql://postgres:postgres@localhost:5432/{{PROJECT_NAME}}",
+    sqlite: "file:./dev.db"
+  };
   const content = [
     "# ========================================",
     "# Environment Variables",
@@ -721,7 +726,7 @@ async function generateEnvExample(features, config, targetPath) {
     "# ========================================",
     "# Database Configuration",
     "# ========================================",
-    "DATABASE_URL=mysql://root:root123456@localhost:3306/{{PROJECT_NAME}}",
+    `DATABASE_URL=${databaseUrls[database] || databaseUrls.mysql}`,
     ""
   ];
   const featureGroups = {};
@@ -848,7 +853,7 @@ ${featuresList}
   await fs4.writeFile(indexPath, content);
   logger.info("Generated capabilities index");
 }
-async function copyPrismaTemplates(templateDir, targetPath) {
+async function copyPrismaTemplates(templateDir, targetPath, database = "mysql") {
   const prismaTemplatesDir = path3.join(templateDir, "apps/backend/prisma");
   const prismaDestDir = path3.join(targetPath, "apps/backend/prisma");
   if (await fs4.pathExists(prismaTemplatesDir)) {
@@ -859,10 +864,13 @@ async function copyPrismaTemplates(templateDir, targetPath) {
       if (file.endsWith(".tpl")) {
         const filePath = path3.join(prismaDestDir, file);
         const newPath = filePath.replace(/\.tpl$/, "");
-        await fs4.rename(filePath, newPath);
+        let content = await fs4.readFile(filePath, "utf-8");
+        content = content.replace(/provider\s*=\s*"postgresql"/, `provider = "${database}"`);
+        await fs4.writeFile(newPath, content);
+        await fs4.remove(filePath);
       }
     }
-    logger.info("Copied Prisma templates");
+    logger.info(`Copied Prisma templates with ${database} database`);
   } else {
     logger.warn(`Prisma templates not found: ${prismaTemplatesDir}`);
   }
@@ -1054,6 +1062,8 @@ async function createProject(projectName, options = {}) {
       answers = {
         org: options.org || projectName,
         template: options.template || "full-stack",
+        database: "mysql",
+        // 默认使用 MySQL
         features: [],
         // 默认不选择额外功能
         packageManager: options.packageManager || "pnpm",
@@ -1084,6 +1094,18 @@ async function createProject(projectName, options = {}) {
           ],
           default: options.template || "full-stack"
         },
+        {
+          type: "list",
+          name: "database",
+          message: "Choose a database:",
+          choices: [
+            { name: "MySQL", value: "mysql" },
+            { name: "PostgreSQL", value: "postgresql" },
+            { name: "SQLite", value: "sqlite" }
+          ],
+          default: "mysql",
+          when: (answers2) => answers2.template === "backend-only" || answers2.template === "full-stack"
+        },
         {
           type: "checkbox",
           name: "features",
@@ -1116,6 +1138,7 @@ async function createProject(projectName, options = {}) {
       projectName,
       orgName: answers.org.startsWith("@") ? answers.org : `@${answers.org}`,
       template: answers.template,
+      database: answers.database || "mysql",
       features: answers.features || [],
       packageManager: answers.packageManager,
       installDeps: answers.installDeps,
@@ -1127,6 +1150,9 @@ async function createProject(projectName, options = {}) {
     logger.info(`  Project Name: ${chalk2.white(config.projectName)}`);
     logger.info(`  Organization: ${chalk2.white(config.orgName)}`);
     logger.info(`  Template: ${chalk2.white(config.template)}`);
+    if (config.template === "backend-only" || config.template === "full-stack") {
+      logger.info(`  Database: ${chalk2.white(config.database)}`);
+    }
     if (config.features.length > 0) {
       logger.info(`  Features: ${chalk2.white(config.features.join(", "))}`);
     }
@@ -1211,9 +1237,9 @@ async function createProjectFromTemplate(config) {
       await copyExampleFiles(config.features, featuresConfig, templateDir, config.projectPath);
       await copySkillFiles(config.features, featuresConfig, templateDir, config.projectPath);
       if (config.template === "backend-only" || config.template === "full-stack") {
-        await copyPrismaTemplates(templateDir, config.projectPath);
+        await copyPrismaTemplates(templateDir, config.projectPath, config.database);
       }
-      await generateEnvExample(config.features, featuresConfig, config.projectPath);
+      await generateEnvExample(config.features, featuresConfig, config.projectPath, config.database);
       if (config.template === "backend-only" || config.template === "full-stack") {
         await updateAppModule(config.features, featuresConfig, config.projectPath);
       }
@@ -1221,6 +1247,19 @@ async function createProjectFromTemplate(config) {
     if (config.installDeps) {
       spinner.text = "Installing dependencies...";
       await installDependencies(config.packageManager);
+      if (config.template === "backend-only" || config.template === "full-stack") {
+        spinner.text = "Generating Prisma client...";
+        try {
+          const { execSync: execSync4 } = __require("child_process");
+          execSync4(`${config.packageManager} --filter=backend prisma:generate`, {
+            cwd: config.projectPath,
+            stdio: "inherit"
+          });
+          logger.info("Prisma client generated successfully");
+        } catch (error) {
+          logger.warn('Failed to generate Prisma client. Please run "pnpm --filter=backend prisma:generate" manually.');
+        }
+      }
     }
     if (config.initGit) {
       spinner.text = "Initializing Git repository...";
@@ -1238,7 +1277,7 @@ async function createProjectFromTemplate(config) {
 }
 
 // package.json
-var version = "2.0.0";
+var version = "2.1.0";
 
 // src/index.ts
 async function cli() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@svton/cli",
-  "version": "2.0.0",
+  "version": "2.1.0",
   "description": "Svton CLI - Create full-stack applications with NestJS, Next.js, and Taro",
   "keywords": [
     "cli",