classworks.js

#!/usr/bin/env node
import {execSync} from "child_process";
import dotenv from "dotenv";

dotenv.config();


// 🔄 执行数据库迁移函数
function runDatabaseMigration() {
    try {
        console.log("🔄 执行数据库迁移...");
        execSync("npx prisma migrate deploy", {stdio: "inherit"});
        console.log("✅ 数据库迁移完成");
    } catch (error) {
        console.error("❌ 数据库迁移失败:", error.message);
        process.exit(1);
    }
}

// 🧱 数据库初始化函数
function setupDatabase() {
    try {
        // 执行数据库迁移
        runDatabaseMigration();
    } catch (error) {
        console.error("❌ 数据库初始化失败:", error.message);
        process.exit(1);
    }
}

// 🔨 本地构建函数
function buildLocal() {
    try {
        // 确保数据库迁移已执行
        runDatabaseMigration();
        execSync("npm install", {stdio: "inherit"}); // 安装依赖
        execSync("npx prisma generate", {stdio: "inherit"}); // 生成 Prisma 客户端
        console.log("✅ 构建完成");
    } catch (error) {
        console.error("❌ 构建失败:", error.message);
        process.exit(1);
    }
}

// 🚀 启动服务函数
function startServer() {
    try {
        execSync("npm run start", {stdio: "inherit"}); // 启动项目
    } catch (error) {
        console.error("❌ 服务启动失败:", error.message);
        process.exit(1);
    }
}

// ▶️ 执行 Prisma CLI 命令函数
function runPrismaCommand(args) {
    try {
        const command = `npx prisma ${args.join(" ")}`;
        execSync(command, {stdio: "inherit"});
    } catch (error) {
        console.error("❌ Prisma 命令执行失败:", error.message);
        process.exit(1);
    }
}

// 🧠 主函数，根据命令行参数判断执行哪种流程
async function main() {
    const args = process.argv.slice(2); // 获取命令行参数
    if (args[0] === "prisma") {
        // 如果输入的是 prisma 命令，则执行 prisma 子命令
        runPrismaCommand(args.slice(1));
    } else {
        // 否则按默认流程：初始化 → 构建 → 启动服务
        setupDatabase();
        buildLocal();
        startServer();
    }
}

// 🚨 捕捉主函数异常
main().catch((error) => {
    console.error("❌ 脚本执行失败:", error);
    process.exit(1);
});