@szc-ft/mcp-szcd-client 0.11.0

package/scripts/lib/qwen-code.js

@@ -0,0 +1,408 @@
+/**
+ * qwen-code.js — Qwen Code 兼容逻辑
+ *
+ * 导出 setupQwenCode(deps) 供 postinstall.js 调用。
+ * deps 需包含: { getMcpServerUrl, getMcpServerName, safeExecSync, isCommandAvailable, ensureDirectory, copyFile, writeFile, SKILL_SOURCE, PROJECT_ROOT, PACKAGE_ROOT }
+ *
+ * Qwen Code 配置方式（v0.9.0+）：
+ *   - 优先使用 Extension 扩展机制（qwen extensions install）
+ *   - 扩展目录: ~/.qwen/extensions/szcd-component-helper/
+ *   - 扩展配置: qwen-extension.json（声明 MCP 服务器、skills、agents、commands）
+ *   - 回退方案: 直接复制扩展目录到 ~/.qwen/extensions/
+ *   - 清理旧配置: 移除 settings.json 中的 mcpServers 条目、旧的 skills/ 目录
+ *
+ * 扩展优势（vs 手动改 settings.json）：
+ *   - 声明式配置，Qwen Code 自动加载 MCP 服务器
+ *   - 原生支持 skills/agents/commands 目录
+ *   - 支持 qwen extensions install/uninstall/disable/enable 完整生命周期
+ *   - 支持 settings（如 MCP_SERVER_URL 环境变量）
+ *   - contextFileName 自动注入 QWEN.md 上下文
+ */
+
+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 getQwenCodeUserDirectory() {
+    const homeDir = getHomeDir();
+    if (!homeDir) throw new Error("Home directory not found");
+    return path.join(homeDir, ".qwen");
+}
+
+function getQwenCodeExtensionsDirectory() {
+    return path.join(getQwenCodeUserDirectory(), "extensions");
+}
+
+function getQwenCodeSettingsPath() {
+    return path.join(getHomeDir(), ".qwen", "settings.json");
+}
+
+function getQwenCodeProjectSettingsPath(projectRoot) {
+    return path.join(projectRoot, ".qwen", "settings.json");
+}
+
+function getExtensionSourcePath(deps) {
+    return path.join(deps.PACKAGE_ROOT, "qwen-extension");
+}
+
+// ==================== JSON 读写 ====================
+
+function readJsonFile(filePath) {
+    if (!fs.existsSync(filePath)) return {};
+    try {
+        return JSON.parse(fs.readFileSync(filePath, "utf8"));
+    } catch {
+        return {};
+    }
+}
+
+function writeJsonFile(filePath, data) {
+    const dir = path.dirname(filePath);
+    if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+    }
+    fs.writeFileSync(filePath, JSON.stringify(data, null, 2));
+}
+
+// ==================== 旧配置清理 ====================
+
+/**
+ * 清理 settings.json 中旧的 mcpServers 条目
+ * 扩展机制下 MCP 由 qwen-extension.json 声明，settings.json 中不再需要
+ */
+function cleanupLegacyMcpConfig(deps) {
+    const settingsPath = getQwenCodeSettingsPath();
+    const serverName = deps.getMcpServerName();
+
+    if (!fs.existsSync(settingsPath)) return;
+
+    const settings = readJsonFile(settingsPath);
+    if (!settings.mcpServers || !settings.mcpServers[serverName]) return;
+
+    delete settings.mcpServers[serverName];
+    if (Object.keys(settings.mcpServers).length === 0) {
+        delete settings.mcpServers;
+    }
+
+    writeJsonFile(settingsPath, settings);
+    console.log(`✓ Cleaned up legacy mcpServers entry in ~/.qwen/settings.json`);
+}
+
+/**
+ * 清理项目级 .qwen/settings.json 中的旧配置
+ * 将 mcpServers 迁移到 _mcpServers_disabled（如果还有的话）
+ */
+function cleanupLegacyProjectConfig(deps) {
+    const settingsPath = getQwenCodeProjectSettingsPath(deps.PROJECT_ROOT);
+    const serverName = deps.getMcpServerName();
+
+    if (!fs.existsSync(settingsPath)) return;
+
+    const settings = readJsonFile(settingsPath);
+    const disabledKey = "_mcpServers_disabled";
+
+    if (settings.mcpServers && settings.mcpServers[serverName]) {
+        settings[disabledKey] = settings[disabledKey] || {};
+        settings[disabledKey][serverName] = settings.mcpServers[serverName];
+        delete settings.mcpServers[serverName];
+        if (Object.keys(settings.mcpServers).length === 0) {
+            delete settings.mcpServers;
+        }
+        settings._comment =
+            "项目级 MCP 已禁用，扩展机制已覆盖。如需恢复，将 _mcpServers_disabled 改回 mcpServers";
+        writeJsonFile(settingsPath, settings);
+        console.log(`✓ Cleaned up project .qwen/settings.json: migrated to _mcpServers_disabled`);
+    }
+}
+
+/**
+ * 清理旧的用户级 skills/ 目录中的手动安装文件
+ * 扩展机制下 skills 由扩展目录提供，不再需要手动复制
+ */
+function cleanupLegacySkills(deps) {
+    const skillsDir = path.join(getQwenCodeUserDirectory(), "skills", deps.getMcpServerName());
+    if (fs.existsSync(skillsDir)) {
+        try {
+            fs.rmSync(skillsDir, { recursive: true, force: true });
+            console.log(`✓ Cleaned up legacy skills directory: ${skillsDir}`);
+        } catch (e) {
+            console.log(`⚠️  Failed to clean up legacy skills: ${e.message}`);
+        }
+    }
+}
+
+/**
+ * 清理旧的用户级 agents/ 目录中的手动安装文件
+ */
+function cleanupLegacyAgents(deps) {
+    const agentsDir = path.join(getQwenCodeUserDirectory(), "agents");
+    const serverName = deps.getMcpServerName();
+
+    if (!fs.existsSync(agentsDir)) return;
+
+    const agentFiles = fs.readdirSync(agentsDir).filter(
+        (f) => f.includes(serverName) && f.endsWith(".md"),
+    );
+
+    for (const f of agentFiles) {
+        try {
+            fs.unlinkSync(path.join(agentsDir, f));
+            console.log(`✓ Cleaned up legacy agent: ${f}`);
+        } catch (e) {
+            console.log(`⚠️  Failed to clean up agent ${f}: ${e.message}`);
+        }
+    }
+}
+
+// ==================== 扩展安装 ====================
+
+/**
+ * 方式1: 通过 CLI 安装扩展（推荐）
+ */
+function installExtensionViaCLI(deps) {
+    if (!deps.isCommandAvailable("qwen")) {
+        console.log("⏭️  Skipping Qwen Code CLI extension install: qwen command not found");
+        return false;
+    }
+
+    const extensionPath = getExtensionSourcePath(deps);
+    const result = deps.safeExecSync(`qwen extensions install "${extensionPath}"`);
+
+    if (result === true) {
+        console.log(`✓ Qwen Code extension installed via CLI: ${deps.getMcpServerName()}`);
+        return true;
+    }
+
+    console.warn(`⚠️  Failed to install Qwen Code extension via CLI, falling back to direct copy`);
+    return false;
+}
+
+/**
+ * 方式2: 直接复制扩展目录到 ~/.qwen/extensions/
+ */
+function installExtensionViaCopy(deps) {
+    const sourcePath = getExtensionSourcePath(deps);
+    const extName = deps.getMcpServerName();
+    const destPath = path.join(getQwenCodeExtensionsDirectory(), extName);
+
+    if (!fs.existsSync(sourcePath)) {
+        console.log(`⚠️  Extension source not found: ${sourcePath}`);
+        return false;
+    }
+
+    try {
+        // 递归复制目录
+        copyDirRecursive(sourcePath, destPath);
+        console.log(`✓ Qwen Code extension installed via direct copy: ${destPath}`);
+        return true;
+    } catch (error) {
+        console.log(`⚠️  Failed to copy extension: ${error.message}`);
+        return false;
+    }
+}
+
+/**
+ * 递归复制目录
+ */
+function copyDirRecursive(src, dest) {
+    if (!fs.existsSync(dest)) {
+        fs.mkdirSync(dest, { recursive: true });
+    }
+
+    const entries = fs.readdirSync(src, { withFileTypes: true });
+    for (const entry of entries) {
+        const srcPath = path.join(src, entry.name);
+        const destPath = path.join(dest, entry.name);
+
+        if (entry.isDirectory()) {
+            copyDirRecursive(srcPath, destPath);
+        } else {
+            fs.copyFileSync(srcPath, destPath);
+        }
+    }
+}
+
+// ==================== MCP URL 同步（供 update-mcp-url.js 调用） ====================
+
+/**
+ * 不依赖 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;
+    }
+}
+
+/**
+ * 同步 Qwen Code 配置中的 MCP 服务器 URL
+ * 
+ * 优先更新 Extension 扩展目录中的 qwen-extension.json，
+ * 同时更新 settings.json（兜底，给非扩展模式用户使用）。
+ * 
+ * @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("qwen")) {
+        safeExecSync(`qwen mcp remove --scope user ${serverName} 2>/dev/null`);
+        const result = safeExecSync(
+            `qwen mcp add --transport sse --scope user ${serverName} ${sseUrl}`,
+        );
+        if (result === true || result === "already_exists") {
+            console.log(`✓ Qwen Code CLI synced: ${sseUrl}`);
+        } else {
+            console.warn(`⚠️  Qwen Code CLI sync failed (settings.json will be updated directly)`);
+        }
+    } else {
+        console.log("⏭️  Skipping Qwen Code CLI sync: qwen not found");
+    }
+
+    // 2. 更新 Extension 扩展目录中的 qwen-extension.json
+    syncExtensionConfig(targetUrl, serverName);
+
+    // 3. 直接文件操作同步 ~/.qwen/settings.json（兜底）
+    syncQwenCodeSettingsDirect(targetUrl, serverName);
+}
+
+/**
+ * 更新已安装扩展目录中的 qwen-extension.json
+ */
+function syncExtensionConfig(targetUrl, serverName) {
+    const extDir = path.join(getQwenCodeExtensionsDirectory(), serverName);
+    const manifestPath = path.join(extDir, "qwen-extension.json");
+
+    if (!fs.existsSync(manifestPath)) {
+        console.log("⏭️  Skipping extension manifest update: extension not installed");
+        return;
+    }
+
+    const manifest = readJsonFile(manifestPath);
+    if (!manifest.mcpServers || !manifest.mcpServers[serverName]) {
+        console.log("⏭️  Skipping extension manifest update: mcpServers entry not found");
+        return;
+    }
+
+    const sseUrl = `${targetUrl}/sse`;
+    if (manifest.mcpServers[serverName].url === sseUrl) {
+        console.log(`⏭️  Extension qwen-extension.json already up-to-date: ${sseUrl}`);
+        return;
+    }
+
+    manifest.mcpServers[serverName].url = sseUrl;
+    writeJsonFile(manifestPath, manifest);
+    console.log(`✓ Updated extension qwen-extension.json: ${sseUrl}`);
+}
+
+/**
+ * 直接文件操作同步 ~/.qwen/settings.json（独立于 deps）
+ */
+function syncQwenCodeSettingsDirect(targetUrl, serverName) {
+    const settingsPath = getQwenCodeSettingsPath();
+    const sseUrl = `${targetUrl}/sse`;
+
+    let settings = {};
+    if (fs.existsSync(settingsPath)) {
+        try {
+            settings = JSON.parse(fs.readFileSync(settingsPath, "utf8"));
+        } catch { /* 忽略 */ }
+    }
+
+    if (!settings.mcpServers) settings.mcpServers = {};
+
+    const current = settings.mcpServers[serverName];
+    if (current && current.type === "sse" && current.url === sseUrl) {
+        console.log(`⏭️  Qwen Code ~/.qwen/settings.json already up-to-date: ${sseUrl}`);
+        return;
+    }
+
+    settings.mcpServers[serverName] = {
+        type: "sse",
+        url: sseUrl,
+        timeout: 30000,
+    };
+
+    const dir = path.dirname(settingsPath);
+    if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+    fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
+    console.log(`✓ Updated Qwen Code ~/.qwen/settings.json: ${sseUrl}`);
+}
+
+// ==================== 导出 ====================
+
+export function setupQwenCode(deps) {
+    console.log("\n📦 Setting up Qwen Code extension...\n");
+
+    // ---- 清理旧配置 ----
+    cleanupLegacyMcpConfig(deps);
+    cleanupLegacySkills(deps);
+    cleanupLegacyAgents(deps);
+
+    // ---- 项目级清理 ----
+    if (deps.PROJECT_ROOT && fs.existsSync(deps.PROJECT_ROOT)) {
+        try {
+            cleanupLegacyProjectConfig(deps);
+        } catch (e) {
+            console.log(`⚠️  Failed to clean up project config: ${e.message}`);
+        }
+    }
+
+    // ---- 安装扩展 ----
+    let extensionInstalled = false;
+
+    // 优先使用 CLI 安装
+    extensionInstalled = installExtensionViaCLI(deps);
+
+    // CLI 失败则直接复制
+    if (!extensionInstalled) {
+        extensionInstalled = installExtensionViaCopy(deps);
+    }
+
+    // ---- 结果 ----
+    if (extensionInstalled) {
+        console.log(`\n✅ Qwen Code extension installed successfully!`);
+        console.log(`   Extension: ~/.qwen/extensions/${deps.getMcpServerName()}/`);
+        console.log(`   Restart Qwen Code to activate the extension.`);
+    } else {
+        console.log(`\n⚠️  Qwen Code extension installation failed.`);
+        console.log(`   You can manually install: qwen extensions install ${getExtensionSourcePath(deps)}`);
+    }
+
+    return {
+        qwenProjectInstalled: extensionInstalled,
+        extensionInstalled,
+        extensionsDirectory: getQwenCodeExtensionsDirectory(),
+        // 保留旧字段兼容 postinstall.js
+        skillsDirectory: path.join(getQwenCodeExtensionsDirectory(), deps.getMcpServerName(), "skills"),
+        agentsDirectory: path.join(getQwenCodeExtensionsDirectory(), deps.getMcpServerName(), "agents"),
+    };
+}