@szc-ft/mcp-szcd-client 0.11.0

package/scripts/lib/trae-cli.js

@@ -0,0 +1,337 @@
+/**
+ * trae-cli.js — Trae CLI 兼容逻辑
+ *
+ * 负责：
+ * - ~/.trae/trae_cli.yaml 的读写同步
+ * - trae-cli mcp add-json / remove 命令调用
+ * - .traecli/agents/ 目录的 agent 复制
+ *
+ * 导出 setupTraeCli(deps) 供 trae.js 入口调用。
+ * deps 需包含: { getMcpServerUrl, getMcpServerName, safeExecSync, isCommandAvailable, ensureDirectory, copyFile, PROJECT_ROOT, PACKAGE_ROOT }
+ */
+
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+import { execSync } from "node:child_process";
+
+// ==================== 路径工具 ====================
+
+function getHomeDir() {
+    if (os.platform() === "win32") {
+        return process.env.USERPROFILE || process.env.HOME || os.homedir();
+    }
+    return process.env.HOME || os.homedir();
+}
+
+function getTraeCliYamlPath() {
+    return path.join(getHomeDir(), ".traecli", "trae_cli.yaml");
+}
+
+// ==================== YAML 解析 ====================
+
+function parseYamlMcpServers(content) {
+    const servers = [];
+    const lines = content.split("\n");
+    let inMcpServers = false;
+    let currentServer = null;
+
+    for (const line of lines) {
+        if (/^mcp_servers\s*:/.test(line)) {
+            inMcpServers = true;
+            continue;
+        }
+        if (inMcpServers && /^\s*$/.test(line)) {
+            inMcpServers = false;
+            if (currentServer) servers.push(currentServer);
+            currentServer = null;
+            continue;
+        }
+        if (inMcpServers && /^[a-zA-Z]/.test(line)) {
+            inMcpServers = false;
+            if (currentServer) servers.push(currentServer);
+            currentServer = null;
+            continue;
+        }
+        if (inMcpServers) {
+            const nameMatch = line.match(/^\s+-\s+name\s*:\s*(.+)/);
+            if (nameMatch) {
+                if (currentServer) servers.push(currentServer);
+                currentServer = { name: nameMatch[1].trim(), raw: [line] };
+                continue;
+            }
+            if (currentServer) {
+                currentServer.raw.push(line);
+                const urlMatch = line.match(/^\s+url\s*:\s*(.+)/);
+                if (urlMatch) currentServer.url = urlMatch[1].trim();
+                const typeMatch = line.match(/^\s+type\s*:\s*(.+)/);
+                if (typeMatch) currentServer.type = typeMatch[1].trim();
+            }
+        }
+    }
+    if (currentServer) servers.push(currentServer);
+    return { servers, lines };
+}
+
+// ==================== YAML 文件同步 ====================
+
+export function syncTraeCliYaml(deps) {
+    const yamlPath = getTraeCliYamlPath();
+    const sseUrl = `${deps.getMcpServerUrl()}/sse`;
+    const serverName = deps.getMcpServerName();
+
+    if (!fs.existsSync(yamlPath)) {
+        const dir = path.dirname(yamlPath);
+        if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+        const content = `mcp_servers:\n    - name: ${serverName}\n      url: ${sseUrl}\n      type: sse\n`;
+        fs.writeFileSync(yamlPath, content);
+        console.log(`✓ Created Trae CLI YAML: ${yamlPath}`);
+        return;
+    }
+
+    let content = fs.readFileSync(yamlPath, "utf8");
+    const { servers } = parseYamlMcpServers(content);
+
+    const existing = servers.find(s => s.name === serverName);
+    if (existing && existing.url === sseUrl) {
+        console.log(`✓ Trae CLI YAML already up-to-date: ${sseUrl}`);
+        return;
+    }
+
+    if (existing) {
+        const newRaw = existing.raw.map(l =>
+            l.replace(/^(\s+url\s*:\s*).*/, `$1${sseUrl}`)
+        );
+        const oldBlock = existing.raw.join("\n");
+        const newBlock = newRaw.join("\n");
+        content = content.replace(oldBlock, newBlock);
+        console.log(`✓ Updated Trae CLI YAML: ${existing.url} → ${sseUrl}`);
+    } else {
+        const entry = `    - name: ${serverName}\n      url: ${sseUrl}\n      type: sse\n`;
+        if (/^mcp_servers\s*:\s*\[\]\s*$/m.test(content)) {
+            content = content.replace(/^mcp_servers\s*:\s*\[\]\s*$/m, `mcp_servers:\n${entry}`);
+        } else if (/^mcp_servers\s*:\s*$/m.test(content)) {
+            content = content.replace(/^mcp_servers\s*:\s*$/m, `mcp_servers:\n${entry}`);
+        } else if (/^mcp_servers\s*:/m.test(content)) {
+            content = content.replace(/^(mcp_servers\s*:)/m, `$1\n${entry}`);
+        } else {
+            content = content.trimEnd() + `\n\nmcp_servers:\n${entry}`;
+        }
+        console.log(`✓ Added to Trae CLI YAML: ${sseUrl}`);
+    }
+
+    fs.writeFileSync(yamlPath, content);
+}
+
+// ==================== CLI 命令同步 ====================
+
+export function createTraeCliConfig(deps) {
+    if (deps.isCommandAvailable("trae-cli")) {
+        const sseUrl = `${deps.getMcpServerUrl()}/sse`;
+        const serverName = deps.getMcpServerName();
+        const jsonConfig = `{"type":"sse","url":"${sseUrl}"}`;
+        deps.safeExecSync(`trae-cli mcp remove ${serverName} 2>/dev/null`);
+        const result = deps.safeExecSync(`trae-cli mcp add-json ${serverName} '${jsonConfig}'`);
+        if (result === true || result === "already_exists") {
+            console.log(`✓ Trae CLI command synced: ${sseUrl}`);
+        }
+    }
+    syncTraeCliYaml(deps);
+}
+
+// ==================== Skill 复制（Trae CLI） ====================
+
+export function getTraeCliSkillsDirectory() {
+    return path.join(getHomeDir(), ".traecli", "skills");
+}
+
+export function copySkillToTraeCli(deps, isProjectLevel = false) {
+    const skillName = deps.getMcpServerName();
+    let traeCliSkillsDir;
+
+    if (isProjectLevel) {
+        traeCliSkillsDir = path.join(deps.PROJECT_ROOT, ".traecli", "skills");
+    } else {
+        traeCliSkillsDir = getTraeCliSkillsDirectory();
+    }
+
+    const skillDir = path.join(traeCliSkillsDir, skillName);
+
+    try {
+        deps.ensureDirectory(skillDir);
+        const skillDest = path.join(skillDir, "SKILL.md");
+        deps.copyFile(deps.SKILL_SOURCE, skillDest);
+        console.log(`✓ Copied skill to Trae CLI ${isProjectLevel ? 'project' : 'user'} directory: ${skillDir}`);
+        return true;
+    } catch (error) {
+        console.log(`⚠️  Failed to copy skill to Trae CLI ${isProjectLevel ? 'project' : 'user'} directory: ${error.message}`);
+        return false;
+    }
+}
+
+// ==================== Slash Command 安装（Trae CLI） ====================
+
+export function getTraeCliCommandsDirectory() {
+    return path.join(getHomeDir(), ".traecli", "commands");
+}
+
+export function installTraeCliSlashCommand(deps) {
+    const commandSource = path.join(deps.PACKAGE_ROOT, "commands", "szcd-mcp-url.md");
+    if (!fs.existsSync(commandSource)) {
+        console.log("⚠️  Slash command source not found, skipping");
+        return;
+    }
+
+    const traeCliCommandsDir = getTraeCliCommandsDirectory();
+    try {
+        deps.ensureDirectory(traeCliCommandsDir);
+        deps.copyFile(commandSource, path.join(traeCliCommandsDir, "szcd-mcp-url.md"));
+        console.log(`✓ Installed Trae CLI slash command: /szcd-mcp-url`);
+    } catch (error) {
+        console.log(`⚠️  Failed to install Trae CLI slash command: ${error.message}`);
+    }
+}
+
+// ==================== MCP URL 同步（供 update-mcp-url.js 调用） ====================
+
+/**
+ * 同步 Trae CLI YAML 和 CLI 命令中的 MCP 服务器 URL
+ * @param {string} targetUrl - 新的 MCP 服务器基础 URL（如 http://localhost:3456）
+ * @param {string} serverName - MCP 服务器名称
+ */
+export function syncMcpUrl(targetUrl, serverName) {
+    const sseUrl = `${targetUrl}/sse`;
+
+    // 1. CLI 命令同步
+    if (isCommandAvailable("trae-cli")) {
+        const jsonConfig = `{"type":"sse","url":"${sseUrl}"}`;
+        safeExecSync(`trae-cli mcp remove ${serverName} 2>/dev/null`);
+        const result = safeExecSync(`trae-cli mcp add-json ${serverName} '${jsonConfig}'`);
+        if (result === true || result === "already_exists") {
+            console.log(`✓ Trae CLI command synced: ${sseUrl}`);
+        } else {
+            console.warn(`⚠️  trae-cli command sync failed (YAML file will be updated directly)`);
+        }
+    } else {
+        console.log("⏭️  Skipping trae-cli command sync: trae-cli not found");
+    }
+
+    // 2. 直接文件操作同步 YAML（确保最终状态正确）
+    syncTraeCliYamlDirect(targetUrl, serverName);
+}
+
+/**
+ * 不依赖 deps 的 CLI 可用性检测
+ */
+function isCommandAvailable(cmd) {
+    try {
+        execSync(`which ${cmd} 2>/dev/null`, { stdio: "pipe", timeout: 3000 });
+        return true;
+    } catch {
+        return false;
+    }
+}
+
+/**
+ * 不依赖 deps 的安全命令执行
+ */
+function safeExecSync(command) {
+    try {
+        execSync(command, { stdio: "pipe", timeout: 10000 });
+        return true;
+    } catch (e) {
+        const stderr = e.stderr?.toString() || "";
+        if (stderr.includes("already exists")) return "already_exists";
+        return false;
+    }
+}
+
+/**
+ * 直接文件操作同步 YAML（独立于 deps，供 syncMcpUrl 使用）
+ */
+function syncTraeCliYamlDirect(targetUrl, serverName) {
+    const yamlPath = getTraeCliYamlPath();
+    const sseUrl = `${targetUrl}/sse`;
+
+    if (!fs.existsSync(yamlPath)) {
+        const dir = path.dirname(yamlPath);
+        if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+        const content = `mcp_servers:\n    - name: ${serverName}\n      url: ${sseUrl}\n      type: sse\n`;
+        fs.writeFileSync(yamlPath, content);
+        console.log(`✓ Created Trae CLI YAML: ${yamlPath}`);
+        return;
+    }
+
+    let content = fs.readFileSync(yamlPath, "utf8");
+    const { servers } = parseYamlMcpServers(content);
+
+    const existing = servers.find(s => s.name === serverName);
+    if (existing && existing.url === sseUrl) {
+        console.log(`⏭️  Trae CLI YAML already up-to-date: ${sseUrl}`);
+        return;
+    }
+
+    if (existing) {
+        const newRaw = existing.raw.map(l =>
+            l.replace(/^(\s+url\s*:\s*).*/, `$1${sseUrl}`)
+        );
+        const oldBlock = existing.raw.join("\n");
+        const newBlock = newRaw.join("\n");
+        content = content.replace(oldBlock, newBlock);
+        console.log(`✓ Updated Trae CLI YAML: ${existing.url} → ${sseUrl}`);
+    } else {
+        const entry = `    - name: ${serverName}\n      url: ${sseUrl}\n      type: sse\n`;
+        if (/^mcp_servers\s*:\s*\[\]\s*$/m.test(content)) {
+            content = content.replace(/^mcp_servers\s*:\s*\[\]\s*$/m, `mcp_servers:\n${entry}`);
+        } else if (/^mcp_servers\s*:\s*$/m.test(content)) {
+            content = content.replace(/^mcp_servers\s*:\s*$/m, `mcp_servers:\n${entry}`);
+        } else if (/^mcp_servers\s*:/m.test(content)) {
+            content = content.replace(/^(mcp_servers\s*:)/m, `$1\n${entry}`);
+        } else {
+            content = content.trimEnd() + `\n\nmcp_servers:\n${entry}`;
+        }
+        console.log(`✓ Added to Trae CLI YAML: ${sseUrl}`);
+    }
+
+    fs.writeFileSync(yamlPath, content);
+}
+
+// ==================== Agent 复制（Trae CLI） ====================
+
+export function copyTraeCliAgent(deps, isProjectLevel = false) {
+    const agentsSourceDir = path.join(deps.PACKAGE_ROOT, "agents");
+    if (!fs.existsSync(agentsSourceDir)) {
+        console.log("⏭️  Skipping agent install for Trae CLI: agents source directory not found");
+        return false;
+    }
+
+    let traeCliAgentsDir;
+    if (isProjectLevel) {
+        traeCliAgentsDir = path.join(deps.PROJECT_ROOT, ".traecli", "agents");
+    } else {
+        traeCliAgentsDir = path.join(getHomeDir(), ".traecli", "agents");
+    }
+
+    // Trae CLI 只复制 .trae.md 文件，并重命名为 .md（适配 YAML frontmatter 格式）
+    const agentFiles = fs.readdirSync(agentsSourceDir).filter(f => f.endsWith(".trae.md"));
+    let copied = 0;
+
+    for (const agentFile of agentFiles) {
+        const sourcePath = path.join(agentsSourceDir, agentFile);
+        const destName = agentFile.replace(".trae.md", ".md");
+
+        try {
+            deps.ensureDirectory(traeCliAgentsDir);
+            deps.copyFile(sourcePath, path.join(traeCliAgentsDir, destName));
+            copied++;
+        } catch (error) {
+            console.log(`⚠️  Failed to copy agent ${agentFile} to Trae CLI: ${error.message}`);
+        }
+    }
+
+    if (copied > 0) {
+        console.log(`✓ Copied ${copied} agent(s) to Trae CLI ${isProjectLevel ? 'project' : 'user'} directory: ${traeCliAgentsDir}`);
+    }
+
+    return copied > 0;
+}

package/scripts/lib/trae-ide.js

@@ -0,0 +1,198 @@
+/**
+ * trae-ide.js — Trae IDE (Trae CN) 兼容逻辑
+ *
+ * 负责：
+ * - mcp.json 的读写同步（支持 Windows %APPDATA%\Trae CN\User\mcp.json）
+ * - skills 复制到 ~/.trae-cn/skills/
+ * - agents 复制到 ~/.trae-cn/agents/ 与项目级 .trae/agents/
+ *
+ * 导出 setupTraeIde(deps) 供 trae.js 入口调用。
+ * deps 需包含: { getMcpServerUrl, getMcpServerName, ensureDirectory, copyFile, writeFile, PROJECT_ROOT, PACKAGE_ROOT }
+ */
+
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+
+// ==================== 路径工具 ====================
+
+function getHomeDir() {
+    if (os.platform() === "win32") {
+        return process.env.USERPROFILE || process.env.HOME || os.homedir();
+    }
+    return process.env.HOME || os.homedir();
+}
+
+export function getTraeUserDirectory() {
+    const homeDir = getHomeDir();
+    if (!homeDir) throw new Error("Home directory not found");
+    return path.join(homeDir, ".trae-cn");
+}
+
+/**
+ * Trae IDE 的 mcp.json 路径。
+ * Windows: %APPDATA%\Trae CN\User\mcp.json
+ * macOS/Linux: ~/.trae-cn/mcp.json
+ */
+export function getTraeIdeMcpJsonPath() {
+    if (os.platform() === "win32") {
+        const appData = process.env.APPDATA;
+        if (appData) {
+            return path.join(appData, "Trae CN", "User", "mcp.json");
+        }
+    }
+    return path.join(getTraeUserDirectory(), "mcp.json");
+}
+
+export function getTraeSkillsDirectory() {
+    return path.join(getTraeUserDirectory(), "skills");
+}
+
+export function getTraeAgentsDirectory() {
+    return path.join(getTraeUserDirectory(), "agents");
+}
+
+function getTraeProjectDirectory(deps) {
+    return path.join(deps.PROJECT_ROOT, ".trae");
+}
+
+// ==================== mcp.json 文件同步 ====================
+
+export function syncTraeCnMcpJson(deps) {
+    const mcpJsonPath = getTraeIdeMcpJsonPath();
+    const sseUrl = `${deps.getMcpServerUrl()}/sse`;
+    const serverName = deps.getMcpServerName();
+
+    let config = {};
+    if (fs.existsSync(mcpJsonPath)) {
+        try {
+            config = JSON.parse(fs.readFileSync(mcpJsonPath, "utf8"));
+        } catch { /* 忽略 */ }
+    }
+
+    if (!config.mcpServers) config.mcpServers = {};
+
+    const current = config.mcpServers[serverName];
+    if (current && current.url === sseUrl && current.type === "sse") {
+        console.log(`✓ Trae CN mcp.json already up-to-date: ${sseUrl}`);
+        return;
+    }
+
+    config.mcpServers[serverName] = {
+        type: "sse",
+        url: sseUrl,
+    };
+
+    const dir = path.dirname(mcpJsonPath);
+    if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+    fs.writeFileSync(mcpJsonPath, JSON.stringify(config, null, 2));
+    console.log(`✓ Updated Trae CN mcp.json: ${sseUrl}`);
+}
+
+export function createTraeIdeConfig(deps) {
+    syncTraeCnMcpJson(deps);
+}
+
+// ==================== MCP URL 同步（供 update-mcp-url.js 调用） ====================
+
+/**
+ * 同步 Trae IDE (Trae CN) mcp.json 中的 MCP 服务器 URL
+ * @param {string} targetUrl - 新的 MCP 服务器基础 URL（如 http://localhost:3456）
+ * @param {string} serverName - MCP 服务器名称
+ */
+export function syncMcpUrl(targetUrl, serverName) {
+    const mcpJsonPath = getTraeIdeMcpJsonPath();
+    const sseUrl = `${targetUrl}/sse`;
+
+    let config = {};
+    if (fs.existsSync(mcpJsonPath)) {
+        try {
+            config = JSON.parse(fs.readFileSync(mcpJsonPath, "utf8"));
+        } catch { /* 忽略 */ }
+    }
+
+    if (!config.mcpServers) config.mcpServers = {};
+
+    const current = config.mcpServers[serverName];
+    if (current && current.url === sseUrl && current.type === "sse") {
+        console.log(`⏭️  Trae CN mcp.json already up-to-date: ${sseUrl}`);
+        return;
+    }
+
+    config.mcpServers[serverName] = {
+        type: "sse",
+        url: sseUrl,
+    };
+
+    const dir = path.dirname(mcpJsonPath);
+    if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+    fs.writeFileSync(mcpJsonPath, JSON.stringify(config, null, 2));
+    console.log(`✓ Updated Trae CN mcp.json: ${sseUrl}`);
+}
+
+// ==================== Skill 复制 ====================
+
+export function copySkillToTrae(deps, isProjectLevel = false) {
+    const skillName = deps.getMcpServerName();
+    let traeSkillsDir;
+
+    if (isProjectLevel) {
+        traeSkillsDir = path.join(deps.PROJECT_ROOT, ".trae", "skills");
+    } else {
+        traeSkillsDir = getTraeSkillsDirectory();
+    }
+
+    const skillDir = path.join(traeSkillsDir, skillName);
+
+    try {
+        deps.ensureDirectory(skillDir);
+        const skillDest = path.join(skillDir, "SKILL.md");
+        deps.copyFile(deps.SKILL_SOURCE, skillDest);
+        console.log(`✓ Copied standard skill to Trae ${isProjectLevel ? 'project' : 'user'} directory: ${skillDir}`);
+        return true;
+    } catch (error) {
+        console.log(`⚠️  Failed to copy skill to Trae ${isProjectLevel ? 'project' : 'user'} directory: ${error.message}`);
+        return false;
+    }
+}
+
+// ==================== Agent 复制（Trae IDE） ====================
+
+export function copyTraeIdeAgent(deps, isProjectLevel = false) {
+    const agentsSourceDir = path.join(deps.PACKAGE_ROOT, "agents");
+    if (!fs.existsSync(agentsSourceDir)) {
+        console.log("⏭️  Skipping agent install for Trae IDE: agents source directory not found");
+        return false;
+    }
+
+    let traeIdeAgentsDir;
+    if (isProjectLevel) {
+        traeIdeAgentsDir = path.join(deps.PROJECT_ROOT, ".trae", "agents");
+    } else {
+        traeIdeAgentsDir = getTraeAgentsDirectory();
+    }
+
+    // Trae IDE 只复制 .trae.md 文件，并重命名为 .md（适配 YAML frontmatter 格式）
+    const agentFiles = fs.readdirSync(agentsSourceDir).filter(f => f.endsWith(".trae.md"));
+    let copied = 0;
+
+    for (const agentFile of agentFiles) {
+        const sourcePath = path.join(agentsSourceDir, agentFile);
+        const destName = agentFile.replace(".trae.md", ".md");
+
+        try {
+            deps.ensureDirectory(traeIdeAgentsDir);
+            deps.copyFile(sourcePath, path.join(traeIdeAgentsDir, destName));
+            copied++;
+        } catch (error) {
+            console.log(`⚠️  Failed to copy agent ${agentFile} to Trae IDE: ${error.message}`);
+        }
+    }
+
+    if (copied > 0) {
+        console.log(`✓ Copied ${copied} agent(s) to Trae IDE ${isProjectLevel ? 'project' : 'user'} directory: ${traeIdeAgentsDir}`);
+    }
+
+    return copied > 0;
+}
+

package/scripts/lib/trae.js

@@ -0,0 +1,65 @@
+/**
+ * trae.js — Trae CLI / Trae IDE 兼容逻辑入口
+ *
+ * 负责编排 Trae CLI 与 Trae IDE 的初始化流程。
+ * 具体逻辑已按方向拆离到：
+ *   - trae-cli.js   Trae CLI 专属逻辑（YAML、trae-cli 命令、.traecli/skills/agents/commands）
+ *   - trae-ide.js   Trae IDE 专属逻辑（mcp.json、.trae-cn/skills、.trae-cn/agents）
+ *
+ * 导出 setupTrae(deps) 供 postinstall.js 调用。
+ * deps 需包含: { getMcpServerUrl, getMcpServerName, safeExecSync, isCommandAvailable, ensureDirectory, copyFile, writeFile, SKILL_SOURCE, PROJECT_ROOT, PACKAGE_ROOT }
+ */
+
+import {
+    createTraeCliConfig,
+    copyTraeCliAgent,
+    copySkillToTraeCli,
+    installTraeCliSlashCommand,
+    getTraeCliSkillsDirectory,
+} from "./trae-cli.js";
+
+import {
+    createTraeIdeConfig,
+    copySkillToTrae,
+    copyTraeIdeAgent,
+    getTraeAgentsDirectory,
+} from "./trae-ide.js";
+
+// ==================== 导出 ====================
+
+export function setupTrae(deps) {
+    // ---- Trae IDE ----
+    createTraeIdeConfig(deps);
+    copySkillToTrae(deps, false);
+    // IDE 暂不支持 agent，跳过
+    // copyTraeIdeAgent(deps, false);
+
+    // ---- Trae CLI ----
+    createTraeCliConfig(deps);
+    copySkillToTraeCli(deps, false);
+    copyTraeCliAgent(deps, false);
+    installTraeCliSlashCommand(deps);
+
+    // 项目级 Trae 配置暂不启用，用户级已足够
+    // let traeProjectInstalled = false;
+    // if (deps.PROJECT_ROOT && fs.existsSync(deps.PROJECT_ROOT)) {
+    //     try {
+    //         traeProjectInstalled = createTraeProjectConfig(deps);
+    //         if (traeProjectInstalled) {
+    //             copySkillToTrae(deps, true);
+    //             // IDE 暂不支持 agent，跳过
+    //             // copyTraeIdeAgent(deps, true);
+    //             copySkillToTraeCli(deps, true);
+    //             copyTraeCliAgent(deps, true);
+    //         }
+    //     } catch (e) {
+    //         console.log(`⚠️  Failed to install Trae project config: ${e.message}`);
+    //     }
+    // }
+
+    return {
+        traeProjectInstalled: false,
+        skillsDirectory: getTraeCliSkillsDirectory(),
+        agentsDirectory: getTraeAgentsDirectory(),
+    };
+}