@seip/blue-bird 0.7.5 → 0.7.6

package/core/cli/init.js

@@ -12,220 +12,318 @@ import { execSync } from "node:child_process";
  * Initializes a new Blue Bird project by copying the base structure.
  */
 class ProjectInit {
-    constructor() {
-        this.appDir = process.cwd();
-        this.sourceDir = path.resolve(import.meta.dirname, "../../");
+  constructor() {
+    this.appDir = process.cwd();
+    this.sourceDir = path.resolve(import.meta.dirname, "../../");
+  }
+
+  /**
+   * Runs the project initialization process.
+   */
+  async run() {
+    console.log(chalk.cyan("Starting Blue Bird project initialization..."));
+
+    const rl = readline.createInterface({ input, output });
+
+    let title = "Blue-Bird";
+    let port = 3000;
+    let appUrl = "http://localhost:3000";
+    let dbType = "none";
+    let dbName = "blue_bird";
+    let dbUser = "root";
+    let dbPassword = "root";
+    let dbPort = 3306;
+
+    try {
+      const ask = async (query, defaultValue) => {
+        const formattedQuery =
+          defaultValue !== undefined
+            ? `${query} [${defaultValue}]: `
+            : `${query}: `;
+        const answer = await rl.question(formattedQuery);
+        return answer.trim() || defaultValue;
+      };
+
+      title = await ask("Project Title", title);
+      const portInput = await ask("Server Port", port);
+      port = parseInt(portInput, 10);
+      if (Number.isNaN(port)) {
+        port = 3000;
+      }
+
+      const defaultAppUrl = `http://localhost:${port}`;
+      appUrl = await ask("Application URL", defaultAppUrl);
+
+      const dbTypeAns = await ask(
+        "Which database do you want to configure? (none / mysql / postgres)",
+        "none",
+      );
+      const cleanDbTypeAns = dbTypeAns.toLowerCase().trim();
+      if (
+        cleanDbTypeAns === "postgres" ||
+        cleanDbTypeAns === "postgresql" ||
+        cleanDbTypeAns === "pg" ||
+        cleanDbTypeAns === "postgr"
+      ) {
+        dbType = "postgres";
+      } else if (cleanDbTypeAns === "mysql") {
+        dbType = "mysql";
+      } else {
+        dbType = "none";
+      }
+
+      if (dbType !== "none") {
+        dbName = await ask("Database Name", dbName);
+        dbUser = await ask(
+          "Database User",
+          dbType === "postgres" ? "postgres" : "root",
+        );
+        dbPassword = await ask("Database Password", dbPassword);
+        const defaultDbPort = dbType === "postgres" ? 5432 : 3306;
+        const dbPortInput = await ask("Database Port", defaultDbPort);
+        dbPort = parseInt(dbPortInput, 10);
+        if (Number.isNaN(dbPort)) {
+          dbPort = defaultDbPort;
+        }
+      }
+    } catch (error) {
+      console.error(
+        chalk.red("[ERROR] Error reading configuration input:"),
+        error.message,
+      );
+      rl.close();
+      return;
+    } finally {
+      rl.close();
     }
 
-    /**
-     * Runs the project initialization process.
-     */
-    async run() {
-        console.log(chalk.cyan("Starting Blue Bird project initialization..."));
-
-        const rl = readline.createInterface({ input, output });
-
-        let title = "Blue-Bird";
-        let port = 3000;
-        let appUrl = "http://localhost:3000";
-        let useMysql = false;
-        let dbName = "blue_bird";
-        let dbUser = "root";
-        let dbPassword = "root";
-        let dbPort = 3306;
-
-        try {
-            const ask = async (query, defaultValue) => {
-                const formattedQuery = defaultValue !== undefined ? `${query} [${defaultValue}]: ` : `${query}: `;
-                const answer = await rl.question(formattedQuery);
-                return answer.trim() || defaultValue;
-            };
-
-            title = await ask("Project Title", title);
-            const portInput = await ask("Server Port", port);
-            port = parseInt(portInput, 10);
-            if (Number.isNaN(port)) {
-                port = 3000;
-            }
-
-            const defaultAppUrl = `http://localhost:${port}`;
-            appUrl = await ask("Application URL", defaultAppUrl);
-
-            const mysqlAns = await ask("Do you want to configure MySQL? (y/n)", "n");
-            useMysql = mysqlAns.toLowerCase() === "y" || mysqlAns.toLowerCase() === "yes";
-
-            if (useMysql) {
-                dbName = await ask("Database Name", dbName);
-                dbUser = await ask("Database User", dbUser);
-                dbPassword = await ask("Database Password", dbPassword);
-                const dbPortInput = await ask("Database Port", dbPort);
-                dbPort = parseInt(dbPortInput, 10);
-                if (Number.isNaN(dbPort)) {
-                    dbPort = 3306;
-                }
-            }
-        } catch (error) {
-            console.error(chalk.red("[ERROR] Error reading configuration input:"), error.message);
-            rl.close();
-            return;
-        } finally {
-            rl.close();
+    const itemsToCopy = [
+      "backend",
+      "frontend",
+      "docker",
+      ".env_example",
+      "AGENTS.md",
+      "index.js",
+    ];
+
+    try {
+      itemsToCopy.forEach((item) => {
+        const src = path.join(this.sourceDir, item);
+        const dest = path.join(this.appDir, item);
+
+        if (fs.existsSync(src)) {
+          if (!fs.existsSync(dest)) {
+            this.copyRecursive(src, dest);
+            console.log(chalk.green(`[OK] Copied ${item} to root.`));
+          } else {
+            console.log(
+              chalk.yellow(`[SKIP] ${item} already exists, skipping.`),
+            );
+          }
+        } else {
+          console.warn(
+            chalk.red(`[ERROR] Source ${item} not found in ${this.sourceDir}`),
+          );
+        }
+      });
+
+      const composeTemplateName =
+        dbType === "postgres"
+          ? "docker-compose.postgres.yml"
+          : dbType === "mysql"
+            ? "docker-compose.mysql.yml"
+            : "docker-compose.none.yml";
+      const composeSrc = path.join(
+        this.sourceDir,
+        "docker",
+        composeTemplateName,
+      );
+      const composeDest = path.join(this.appDir, "docker-compose.yml");
+      if (fs.existsSync(composeSrc)) {
+        if (!fs.existsSync(composeDest)) {
+          fs.copyFileSync(composeSrc, composeDest);
+          console.log(
+            chalk.green(`[OK] Created docker-compose.yml (${dbType} mode).`),
+          );
+        } else {
+          console.log(
+            chalk.yellow(`[SKIP] docker-compose.yml already exists, skipping.`),
+          );
+        }
+      } else {
+        const fallbackSrc = path.join(this.sourceDir, "docker-compose.yml");
+        if (fs.existsSync(fallbackSrc) && !fs.existsSync(composeDest)) {
+          fs.copyFileSync(fallbackSrc, composeDest);
+          console.log(chalk.green(`[OK] Created docker-compose.yml.`));
+        }
+      }
+
+      const envPath = path.join(this.appDir, ".env");
+      const envExamplePath = path.join(this.appDir, ".env_example");
+
+      if (fs.existsSync(envExamplePath)) {
+        let envContent = fs.readFileSync(envExamplePath, "utf-8");
+
+        const jwtSecret = crypto.randomBytes(32).toString("hex");
+
+        const updates = {
+          TITLE: title,
+          PORT: port,
+          APP_URL: appUrl,
+          JWT_SECRET: jwtSecret,
+          DB_TYPE: dbType,
+        };
+
+        if (dbType === "mysql") {
+          updates.DB_NAME = dbName;
+          updates.DB_USER = dbUser;
+          updates.DB_PASSWORD = dbPassword;
+          updates.DB_PORT = dbPort;
+          updates.DATABASE_URL = `mysql://${dbUser}:${dbPassword}@localhost:${dbPort}/${dbName}`;
+        } else if (dbType === "postgres") {
+          updates.DB_NAME = dbName;
+          updates.DB_USER = dbUser;
+          updates.DB_PASSWORD = dbPassword;
+          updates.DB_PORT = dbPort;
+          updates.DATABASE_URL = `postgresql://${dbUser}:${dbPassword}@localhost:${dbPort}/${dbName}?schema=public`;
         }
 
-        const itemsToCopy = [
-            "backend",
-            "frontend",
-            "docker",
-            "docker-compose.yml",
-            ".env_example",
-            "AGENTS.md",
-            "index.js"
-        ];
-
-        try {
-            itemsToCopy.forEach(item => {
-                const src = path.join(this.sourceDir, item);
-                const dest = path.join(this.appDir, item);
-
-                if (fs.existsSync(src)) {
-                    if (!fs.existsSync(dest)) {
-                        this.copyRecursive(src, dest);
-                        console.log(chalk.green(`[OK] Copied ${item} to root.`));
-                    } else {
-                        console.log(chalk.yellow(`[SKIP] ${item} already exists, skipping.`));
-                    }
-                } else {
-                    console.warn(chalk.red(`[ERROR] Source ${item} not found in ${this.sourceDir}`));
-                }
-            });
-
-            const envPath = path.join(this.appDir, ".env");
-            const envExamplePath = path.join(this.appDir, ".env_example");
-
-            if (fs.existsSync(envExamplePath)) {
-                let envContent = fs.readFileSync(envExamplePath, "utf-8");
-
-                const jwtSecret = crypto.randomBytes(32).toString("hex");
-
-                const updates = {
-                    TITLE: title,
-                    PORT: port,
-                    APP_URL: appUrl,
-                    JWT_SECRET: jwtSecret,
-                };
-
-                if (useMysql) {
-                    updates.DB_NAME = dbName;
-                    updates.DB_USER = dbUser;
-                    updates.DB_PASSWORD = dbPassword;
-                    updates.DB_PORT = dbPort;
-                    updates.DATABASE_URL = `mysql://${dbUser}:${dbPassword}@localhost:${dbPort}/${dbName}`;
-                }
-
-                const lines = envContent.split(/\r?\n/);
-                const updatedLines = lines.map(line => {
-                    const match = line.match(/^([A-Z_]+)=(.+)/);
-                    if (match) {
-                        const key = match[1];
-                        if (updates[key] !== undefined) {
-                            const value = updates[key];
-                            if (typeof value === "string" && !value.startsWith('"')) {
-                                return `${key}="${value}"`;
-                            }
-                            return `${key}=${value}`;
-                        }
-                    }
-                    return line;
-                });
-                envContent = updatedLines.join("\n");
-
-                fs.writeFileSync(envPath, envContent, "utf-8");
-                console.log(chalk.green("[OK] Created and configured .env file."));
+        const lines = envContent.split(/\r?\n/);
+        const updatedLines = lines.map((line) => {
+          const match = line.match(/^([A-Z_]+)=(.+)/);
+          if (match) {
+            const key = match[1];
+            if (updates[key] !== undefined) {
+              const value = updates[key];
+              if (typeof value === "string" && !value.startsWith('"')) {
+                return `${key}="${value}"`;
+              }
+              return `${key}=${value}`;
             }
+          }
+          return line;
+        });
+        envContent = updatedLines.join("\n");
 
-            this.updatePackageJson();
-
-            if (useMysql) {
-                console.log(chalk.cyan("[INFO] Installing mysql2 and redis packages..."));
-                try {
-                    execSync("npm install mysql2 redis", { stdio: "inherit", cwd: this.appDir });
-                    console.log(chalk.green("[OK] Successfully installed mysql2 and redis."));
-                } catch (error) {
-                    console.warn(chalk.yellow("[ERROR] Automatic package installation failed. Please run 'npm install mysql2 redis' manually."));
-                }
-            }
+        fs.writeFileSync(envPath, envContent, "utf-8");
+        console.log(chalk.green("[OK] Created and configured .env file."));
+      }
 
-            console.log(chalk.blue("\nBlue Bird initialization completed!"));
-            console.log(chalk.white("Next steps:"));
-            console.log(chalk.bold("  npm install"));
-            console.log(chalk.bold("  npm run dev"));
+      this.updatePackageJson();
 
+      if (dbType !== "none") {
+        console.log(
+          chalk.cyan(`[INFO] Installing ${dbType} and redis packages...`),
+        );
+        try {
+          let packagesToInstall = ["redis"];
+          if (dbType === "postgres") {
+            packagesToInstall.push("pg");
+          } else {
+            packagesToInstall.push("mysql2");
+          }
+          execSync(`npm install ${packagesToInstall.join(" ")}`, {
+            stdio: "inherit",
+            cwd: this.appDir,
+          });
+          console.log(
+            chalk.green(
+              `[OK] Successfully installed ${packagesToInstall.join(" ")}.`,
+            ),
+          );
         } catch (error) {
-            console.error(chalk.red("[ERROR] Error during initialization:"), error.message);
+          console.warn(
+            chalk.yellow(
+              "[ERROR] Automatic package installation failed. Please run 'npm install' manually.",
+            ),
+          );
         }
+      }
+
+      console.log(chalk.blue("\nBlue Bird initialization completed!"));
+      console.log(chalk.white("Next steps:"));
+      console.log(chalk.bold("  npm install"));
+      console.log(chalk.bold("  npm run dev"));
+    } catch (error) {
+      console.error(
+        chalk.red("[ERROR] Error during initialization:"),
+        error.message,
+      );
     }
+  }
+
+  /**
+   * Updates package.json with needed scripts and module type if not set.
+   */
+  updatePackageJson() {
+    const pkgPath = path.join(this.appDir, "package.json");
+    if (fs.existsSync(pkgPath)) {
+      const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
+      pkg.scripts = pkg.scripts || {};
+
+      const scriptsToAdd = {
+        dev: "node --watch --env-file=.env index.js",
+        "dev:astro": "astro dev --root frontend",
+        "dev:api": "node --watch --env-file=.env index.js",
+        start: "node --env-file=.env index.js",
+        build: "astro build --root frontend",
+        init: "blue-bird",
+        route: "blue-bird route",
+        "swagger-install": "blue-bird swagger-install",
+        docker: "blue-bird docker",
+      };
+
+      let updated = false;
+      for (const [key, value] of Object.entries(scriptsToAdd)) {
+        if (!pkg.scripts[key]) {
+          pkg.scripts[key] = value;
+          updated = true;
+        }
+      }
 
-    /**
-     * Updates the user's package.json with Blue Bird scripts.
-     */
-    updatePackageJson() {
-        const pkgPath = path.join(this.appDir, "package.json");
-        if (fs.existsSync(pkgPath)) {
-            const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
-            pkg.scripts = pkg.scripts || {};
-
-            const scriptsToAdd = {
-                "dev": "node --watch --env-file=.env index.js",
-                "dev:astro": "astro dev --root frontend",
-                "dev:api": "node --watch --env-file=.env index.js",
-                "start": "node --env-file=.env index.js",
-                "build": "astro build --root frontend",
-                "init": "blue-bird",
-                "route": "blue-bird route",
-                "swagger-install": "blue-bird swagger-install",
-                "docker": "blue-bird docker"
-            };
-
-            let updated = false;
-            for (const [key, value] of Object.entries(scriptsToAdd)) {
-                if (!pkg.scripts[key]) {
-                    pkg.scripts[key] = value;
-                    updated = true;
-                }
-            }
-
-            if (pkg.type !== "module") {
-                pkg.type = "module";
-                updated = true;
-            }
+      if (pkg.type !== "module") {
+        pkg.type = "module";
+        updated = true;
+      }
 
-            if (updated) {
-                fs.writeFileSync(pkgPath, JSON.stringify(pkg, null, 2));
-                console.log(chalk.green("[OK] Updated package.json configuration."));
-            }
-        }
+      if (updated) {
+        fs.writeFileSync(pkgPath, JSON.stringify(pkg, null, 2));
+        console.log(chalk.green("[OK] Updated package.json configuration."));
+      }
     }
-
-    /**
-     * Copies a file or directory recursively.
-     * @param {string} src - Source path.
-     * @param {string} dest - Destination path.
-     */
-    copyRecursive(src, dest) {
-        const stats = fs.statSync(src);
-        const isDirectory = stats.isDirectory();
-
-        if (isDirectory) {
-            if (!fs.existsSync(dest)) {
-                fs.mkdirSync(dest, { recursive: true });
-            }
-            fs.readdirSync(src).forEach(childItemName => {
-                this.copyRecursive(path.join(src, childItemName), path.join(dest, childItemName));
-            });
-        } else {
-            fs.copyFileSync(src, dest);
+  }
+
+  /**
+   * Copies a file or directory recursively.
+   * @param {string} src - Source path.
+   * @param {string} dest - Destination path.
+   */
+  copyRecursive(src, dest) {
+    const stats = fs.statSync(src);
+    const isDirectory = stats.isDirectory();
+
+    if (isDirectory) {
+      if (!fs.existsSync(dest)) {
+        fs.mkdirSync(dest, { recursive: true });
+      }
+      fs.readdirSync(src).forEach((childItemName) => {
+        if (
+          childItemName === "node_modules" ||
+          childItemName === ".astro" ||
+          childItemName === "dist" ||
+          childItemName === ".git"
+        ) {
+          return;
         }
+        this.copyRecursive(
+          path.join(src, childItemName),
+          path.join(dest, childItemName),
+        );
+      });
+    } else {
+      fs.copyFileSync(src, dest);
     }
+  }
 }
 
 const initializer = new ProjectInit();