sd-ai-rules/dist/index.js

AI 规则模板生成 CLI

#!/usr/bin/env node
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
import chalk from "chalk";
import prompts from "prompts";
import fs from "node:fs";
import path from "node:path";
import { fileURLToPath } from "node:url";
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
function init() {
    return __awaiter(this, void 0, void 0, function* () {
        let result;
        try {
            result = yield prompts([
                {
                    type: 'select',
                    name: 'ruleType',
                    message: chalk.reset(`你想要创建哪种编辑器的规则？`),
                    choices: [
                        { title: chalk.cyan('Cursor'), value: 'cursor' },
                        { title: chalk.green('Trae'), value: 'trae' }
                    ],
                    initial: 0
                }
            ], {
                onCancel: () => {
                    throw new Error(chalk.red('✖') + ' 操作已取消');
                }
            });
        }
        catch (error) {
            console.log(error);
            return;
        }
        const { ruleType } = result;
        const templateDir = path.join(__dirname, '..', 'rules', ruleType);
        console.log(templateDir);
        const rootDir = path.join(process.cwd(), `.${ruleType}`, `rules`);
        console.log(rootDir);
        if (!fs.existsSync(templateDir)) {
            console.log(chalk.red('✖') + ` 未找到模板目录：${templateDir}`);
            return;
        }
        if (!fs.existsSync(rootDir)) {
            fs.mkdirSync(rootDir, { recursive: true });
        }
        const files = fs.readdirSync(templateDir);
        for (const file of files) {
            fs.copyFileSync(path.join(templateDir, file), path.join(rootDir, file));
        }
        console.log(chalk.green('✔') + `${ruleType === 'cursor' ? ' Cursor' : ' Trae'}编辑器规则已创建！`);
        console.log(chalk.green(`请前往 ${rootDir} 目录查看/编辑规则文件。`));
    });
}
init().catch(e => {
    console.log(e);
});