cc-3-8-x-mcp/main.js

'use strict';

var fs = require('fs');
var path = require('path');
var { exec } = require('child_process');

var DEV_DIR = '.dev';

// 缓存预览地址
var _previewUrl = '';
// 定时刷新 dev-reload-info.json 的 interval handle
var _infoInterval = null;
// `.dev/refresh` 命令文件 watcher（唯一保留的 watcher）
var _refreshWatcher = null;

// dev-reload-info.json 输出路径（被 listWorktrees / getStatus / panel 读取，必须保留）
var INFO_FILE = '.dev/dev-reload-info.json';
// 信号文件：外部脚本写命令到此文件，插件读后执行（每行一条命令，读完清空）
var REFRESH_FILE = '.dev/refresh';
// 自定义按钮配置（用户可在 .dev/cc-mcp-panel.json 或旧名 .dev/dev-reload-panel.json 里维护）
var PANEL_CONFIG_FILE = '.dev/dev-reload-panel.json';

// 最近命令日志环形缓冲（供面板展示）
var _commandLog = [];
var COMMAND_LOG_MAX = 30;
function pushCommandLog(source, cmd) {
    _commandLog.push({ t: new Date().toISOString(), source: source, cmd: cmd });
    if (_commandLog.length > COMMAND_LOG_MAX) _commandLog.shift();
}

/**
 * 把当前预览状态写入 .dev/dev-reload-info.json。
 * 外部脚本（playwright/designer）通过此文件反查"本 worktree 对应哪个预览端口"。
 * 只在 previewUrl 已知时写入；previewUrl 为空则跳过，等待首次 getPreviewUrl 成功。
 * @param {string} previewUrl  已知的预览 URL（非空）
 */
function writeDevReloadInfo(previewUrl) {
    if (!previewUrl) return;
    var portMatch = previewUrl.match(/:(\d+)/);
    var previewPort = portMatch ? parseInt(portMatch[1], 10) : null;
    // 读取项目名（package.json name 字段）
    var projectName = '';
    try {
        var pkgPath = path.join(Editor.Project.path, 'package.json');
        if (fs.existsSync(pkgPath)) {
            var pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
            projectName = pkg.name || '';
        }
    } catch (e) { /* ignore */ }
    var info = {
        projectPath: Editor.Project.path,
        projectName: projectName,
        editorPid: process.pid,
        editorVersion: (Editor.App && Editor.App.version) ? Editor.App.version : '',
        previewUrl: previewUrl,
        previewPort: previewPort,
        updatedAt: new Date().toISOString(),
    };
    try {
        var devDir = path.join(Editor.Project.path, DEV_DIR);
        if (!fs.existsSync(devDir)) {
            fs.mkdirSync(devDir, { recursive: true });
        }
        fs.writeFileSync(
            path.join(Editor.Project.path, INFO_FILE),
            JSON.stringify(info, null, 2),
            'utf-8'
        );
        log('dev-reload-info.json updated — port:' + (previewPort || 'null'));
    } catch (e) {
        console.warn('[dev-reload] writeDevReloadInfo failed:', e.message || e);
    }
}

/** 启动 30s 定时刷新，保持 updatedAt 活跃供外部 stale 检测 */
function startInfoInterval() {
    if (_infoInterval) clearInterval(_infoInterval);
    _infoInterval = setInterval(function () {
        if (_previewUrl) writeDevReloadInfo(_previewUrl);
        // 顺便刷 registry，让 router 判活
        if (typeof writeRegistry === 'function') writeRegistry();
    }, 30000);
}

/** 停止定时刷新 */
function stopInfoInterval() {
    if (_infoInterval) { clearInterval(_infoInterval); _infoInterval = null; }
}

function log(msg) {
    console.log('[dev-reload] ' + msg);
}

function getFilePath(filename) {
    return path.join(Editor.Project.path, filename);
}

/**
 * 获取当前编辑器预览地址。
 * 每次都重新向 preview 扩展查询，避免首次预览未启动时缓存旧值。
 * 成功拿到 URL 后才更新 _previewUrl 缓存（供 writeDevReloadInfo 等用）。
 */
async function getPreviewUrl() {
    try {
        // CC3.8.x preview 扩展正式消息名：query-preview-url（见 builtin/preview/package.json contributions.messages）
        var url = await Editor.Message.request('preview', 'query-preview-url');
        if (url && typeof url === 'string' && url.startsWith('http')) {
            // host 规范化为 loopback：编辑器用 os.networkInterfaces() 挑的网卡 IP，在多网卡 /
            // 切换网络（家↔公司）时会指向当前不通的网卡。本机访问统一走 127.0.0.1，预览 server
            // 监听 0.0.0.0，loopback 永远通且与网卡/环境无关。写信号文件、open、返回值都用它。
            url = url.replace(/^(https?:\/\/)[^:\/]+/, '$1127.0.0.1');
            if (url !== _previewUrl) {
                log('preview url updated: ' + url);
            }
            _previewUrl = url;
            return _previewUrl;
        }
    } catch (e) {
        // Editor.Message.request 失败（消息不存在、preview 扩展未启动等）必须打印，不能静默
        console.error('[dev-reload] getPreviewUrl: Editor.Message.request failed —', e && (e.stack || e.message) || e);
    }

    // 降级：如果已有缓存（上次成功查到的），直接复用
    if (_previewUrl) return _previewUrl;

    // 无缓存、无法从编辑器获取，返回未知标记而非错误的硬编码端口
    log('preview url: unable to query from editor — preview may not be running');
    return 'http://localhost:unknown-port';
}

/**
 * 生成 AppleScript 匹配预览 tab 的条件：只按端口匹配、忽略 host。
 * 同一预览实例在多网卡 / 切换网络时 host(IP) 会变，但端口稳定（per-project 编辑器分配）。
 * 避免字面比完整 IP:port 在环境切换后失配（截图退化全屏、eval 报找不到 tab）。
 */
function tabMatchClause(url) {
    var m = url && url.match(/:(\d+)(?:\/|$)/);
    var port = m ? m[1] : '';
    return port ? ('URL of t contains ":' + port + '/"') : ('URL of t starts with "' + url + '"');
}

function sleep(ms) {
    return new Promise(function (resolve) { setTimeout(resolve, ms); });
}

async function doReimport(url) {
    log('reimporting: ' + url);
    await Editor.Message.request('asset-db', 'reimport-asset', url);
    log('reimported: ' + url);
}

async function doRefreshAssets() {
    log('refreshing assets...');
    await Editor.Message.request('asset-db', 'refresh-asset', 'db://assets/');
    log('assets refreshed.');
}

async function doReloadScene() {
    log('reloading scene...');
    await Editor.Message.request('scene', 'soft-reload');
    log('scene reloaded.');
}

/**
 * 在浏览器中打开预览
 */
async function doOpenPreview() {
    var url = await getPreviewUrl();
    // 先查有没有该端口的 tab,有就不重开——避免在已有(带 tid 的)预览 tab 之外再冒一个裸地址 tab
    var checkScript = [
        'tell application "Google Chrome"',
        '  repeat with w in windows',
        '    repeat with t in tabs of w',
        '      if ' + tabMatchClause(url) + ' then return "FOUND"',
        '    end repeat',
        '  end repeat',
        '  return "NONE"',
        'end tell'
    ].join('\n');
    return new Promise(function (resolve) {
        exec('osascript -e \'' + checkScript.replace(/'/g, "'\\''") + '\'', function (err, stdout) {
            if (!err && (stdout || '').trim() === 'FOUND') {
                log('preview tab already open (port matched), skip open');
                resolve();
            } else {
                log('opening preview: ' + url);
                exec('open "' + url + '"', function () { resolve(); });
            }
        });
    });
}

/**
 * 刷新已打开的预览浏览器页面
 */
async function doRefreshPreview() {
    var url = await getPreviewUrl();
    log('refreshing preview browser...');
    // 用 AppleScript 找到预览页面并刷新
    var script = [
        'tell application "Google Chrome"',
        '  set found to false',
        '  repeat with w in windows',
        '    repeat with t in tabs of w',
        '      if ' + tabMatchClause(url) + ' then',
        '        tell t to reload',
        '        set found to true',
        '      end if',
        '    end repeat',
        '  end repeat',
        '  if not found then',
        '    open location "' + url + '"',
        '  end if',
        'end tell'
    ].join('\n');

    return new Promise(function (resolve) {
        exec('osascript -e \'' + script.replace(/'/g, "'\\''") + '\'', function (err) {
            if (err) {
                log('Chrome refresh failed, trying Safari...');
                var safariScript = [
                    'tell application "Safari"',
                    '  set found to false',
                    '  repeat with w in windows',
                    '    repeat with t in tabs of w',
                    '      if ' + tabMatchClause(url) + ' then',
                    '        tell t to do JavaScript "location.reload()"',
                    '        set found to true',
                    '      end if',
                    '    end repeat',
                    '  end repeat',
                    '  if not found then',
                    '    open location "' + url + '"',
                    '  end if',
                    'end tell'
                ].join('\n');
                exec('osascript -e \'' + safariScript.replace(/'/g, "'\\''") + '\'', function () {
                    resolve();
                });
            } else {
                log('preview refreshed.');
                resolve();
            }
        });
    });
}

/**
 * 截取预览浏览器页面的截图
 */
async function doScreenshot(outputPath) {
    var url = await getPreviewUrl();
    log('taking screenshot → ' + outputPath);

    return new Promise(function (resolve) {
        // 等待渲染
        setTimeout(function () {
            // 优先用 Chrome DevTools Protocol 截图（更精准）
            var chromeScript = [
                'tell application "Google Chrome"',
                '  repeat with w in windows',
                '    repeat with t in tabs of w',
                '      if ' + tabMatchClause(url) + ' then',
                '        set index of w to 1',
                '        set active tab index of w to (index of t)',
                '        delay 0.5',
                '        return id of w',
                '      end if',
                '    end repeat',
                '  end repeat',
                '  return ""',
                'end tell'
            ].join('\n');

            exec('osascript -e \'' + chromeScript.replace(/'/g, "'\\''") + '\'', function (err, stdout) {
                var windowId = stdout ? stdout.trim() : '';
                if (windowId) {
                    // 截取 Chrome 窗口
                    exec('screencapture -o -l ' + windowId + ' "' + outputPath + '"', function (err2) {
                        if (err2) {
                            log('window capture failed, fallback to full screen');
                            exec('screencapture -o "' + outputPath + '"', function () { resolve(); });
                        } else {
                            log('screenshot saved (browser window).');
                            resolve();
                        }
                    });
                } else {
                    // 降级：截取整个屏幕
                    log('preview tab not found, capturing full screen');
                    exec('screencapture -o "' + outputPath + '"', function () { resolve(); });
                }
            });
        }, 1000);
    });
}

// ── 消息处理（支持从其他扩展或命令行调用） ──

exports.methods = {
    async refreshAssets() {
        await doRefreshAssets();
        await doReloadScene();
    },
    async screenshot() {
        var outputPath = path.join(Editor.Project.path, '.dev', 'screenshot.png');
        await doScreenshot(outputPath);
        return outputPath;
    },
    async queryPreviewUrl() {
        var url = await getPreviewUrl();
        // 顺便刷新 dev-reload-info.json，让外部脚本拿到最新端口
        writeDevReloadInfo(url);
        return url;
    },
    async openPanel() {
        await Editor.Panel.open('cc-3-8-x-mcp');
    },
    async restartServer() {
        await stopMcpServer();
        await startMcpServer();
        return { port: _mcpServer ? _mcpServer.port : null };
    },
    async getMcpConfig() {
        if (!_mcpServer) return { running: false };
        var url = 'http://' + _mcpServer.host + ':' + _mcpServer.port + '/mcp';
        return {
            running: _mcpServer.started,
            url: url,
            port: _mcpServer.port,
            host: _mcpServer.host,
            toolCount: _mcpServer.toolCount,
            resourceCount: _mcpServer.resourceCount,
            stats: _mcpServer.stats,
            // Claude Code 的 mcp add 命令
            cliAddCommand: 'claude mcp add cocos --transport http ' + url,
            // JSON 配置片段
            jsonConfig: {
                mcpServers: {
                    cocos: { transport: 'http', url: url },
                },
            },
        };
    },
    /** Panel 使用：刷新资源 + 重载场景 + 刷新预览 */
    async triggerRefresh() {
        await doRefreshAssets();
        await doReloadScene();
        await doRefreshPreview();
        return true;
    },
    /** Panel 使用：重新导入指定 assetUrl */
    async triggerReimport(url) {
        if (!url) return false;
        await doReimport(url);
        return true;
    },
    /** Panel 使用：打开 .dev 目录 */
    openDevDir() {
        var devDir = path.join(Editor.Project.path, DEV_DIR);
        if (!fs.existsSync(devDir)) fs.mkdirSync(devDir, { recursive: true });
        exec('open "' + devDir + '"');
        return devDir;
    },
    /** Panel 使用：只做场景软重载 */
    async softReloadScene() {
        pushCommandLog('panel', 'reload-scene');
        await doReloadScene();
        return true;
    },
    /** Panel 使用：在浏览器中打开预览 */
    async openPreview() {
        pushCommandLog('panel', 'open-preview');
        await doOpenPreview();
        return true;
    },
    /** Panel 使用：截图并把路径复制到剪贴板 */
    async screenshotCopy() {
        pushCommandLog('panel', 'screenshot');
        var outputPath = path.join(Editor.Project.path, DEV_DIR, 'screenshot.png');
        await doScreenshot(outputPath);
        return new Promise(function (resolve) {
            exec('printf %s "' + outputPath.replace(/"/g, '\\"') + '" | pbcopy', function () {
                resolve(outputPath);
            });
        });
    },
    /** Panel 使用：清理 .dev 临时产物（保留 dev-reload-info.json / dev-reload-panel.json / cc-mcp-panel.json） */
    cleanDevDir() {
        pushCommandLog('panel', 'clean-dev');
        var devDir = path.join(Editor.Project.path, DEV_DIR);
        var keep = { 'dev-reload-info.json': 1, 'dev-reload-panel.json': 1, 'cc-mcp-panel.json': 1 };
        var removed = [];
        try {
            var entries = fs.readdirSync(devDir);
            entries.forEach(function (name) {
                if (keep[name]) return;
                var full = path.join(devDir, name);
                try {
                    var st = fs.statSync(full);
                    if (st.isFile()) { fs.unlinkSync(full); removed.push(name); }
                } catch (e) { /* ignore */ }
            });
        } catch (e) { /* ignore */ }
        return removed;
    },
    /** Panel 使用：向预览 Chrome 页面注入 JS 代码并返回执行结果 */
    async evalInPreview(code) {
        if (!code) return { ok: false, error: 'empty code' };
        pushCommandLog('panel', 'eval:' + code.slice(0, 40));
        var url = await getPreviewUrl();
        // AppleScript 需要把 JS 代码里的双引号转义
        // 包一层 window.app 校验:连到的 tab 不是游戏(编辑器内嵌预览 / 错 tab)就明确报错,不默默 eval 错上下文
        var guarded = '(function(){ if(typeof window==="undefined"||!window.app){return "__NO_GAME_CONTEXT__";} return (' + code + '); })()';
        var escaped = guarded.replace(/\\/g, '\\\\').replace(/"/g, '\\"');
        var script = [
            'tell application "Google Chrome"',
            '  repeat with w in windows',
            '    repeat with t in tabs of w',
            '      if ' + tabMatchClause(url) + ' then',
            '        return (execute t javascript "' + escaped + '")',
            '      end if',
            '    end repeat',
            '  end repeat',
            '  return "__NO_PREVIEW_TAB__"',
            'end tell'
        ].join('\n');
        return new Promise(function (resolve) {
            exec('osascript -e \'' + script.replace(/'/g, "'\\''") + '\'', function (err, stdout, stderr) {
                if (err) {
                    resolve({ ok: false, error: (stderr || err.message || '').trim() });
                } else {
                    var out = (stdout || '').trim();
                    if (out === '__NO_PREVIEW_TAB__') {
                        resolve({ ok: false, error: '未找到预览标签页（先在 Chrome 打开 ' + url + '）' });
                    } else if (out === '__NO_GAME_CONTEXT__') {
                        resolve({ ok: false, error: '连到的 tab 没有游戏上下文（window.app undefined）——多半连的是编辑器内嵌预览或别的 tab。游戏要在浏览器跑且 app.ts 已加载；必要时用 playwright 直连游戏 tab' });
                    } else {
                        resolve({ ok: true, result: out });
                    }
                }
            });
        });
    },
    /** Panel 使用：读取用户自定义的 debug 按钮配置 */
    getDebugButtons() {
        var cfgPath = path.join(Editor.Project.path, PANEL_CONFIG_FILE);
        if (!fs.existsSync(cfgPath)) return [];
        try {
            var cfg = JSON.parse(fs.readFileSync(cfgPath, 'utf-8'));
            if (Array.isArray(cfg.buttons)) return cfg.buttons;
            return [];
        } catch (e) {
            return [];
        }
    },
    /** Panel 使用：扫同机其他 worktree 的 dev-reload-info.json */
    listWorktrees() {
        var results = [];
        // 扫当前项目同级目录里其它含 .dev/dev-reload-info.json 的项目实例（不假设目录命名）
        var cur = Editor.Project.path;
        var roots = [];
        try {
            var siblingDir = path.dirname(cur);
            fs.readdirSync(siblingDir).forEach(function (name) {
                var p = path.join(siblingDir, name);
                if (p !== cur && fs.existsSync(path.join(p, '.dev', 'dev-reload-info.json'))) {
                    roots.push(p);
                }
            });
        } catch (e) { /* ignore */ }
        // 再扫 worktree：`git worktree list` 输出的路径
        try {
            var wtOut = require('child_process').execSync('git -C "' + cur + '" worktree list --porcelain', { encoding: 'utf-8' });
            wtOut.split('\n').forEach(function (line) {
                if (line.indexOf('worktree ') === 0) {
                    var p = line.substring('worktree '.length).trim();
                    if (p && roots.indexOf(p) < 0) roots.push(p);
                }
            });
        } catch (e) { /* ignore */ }

        roots.forEach(function (root) {
            var candidates = [
                path.join(root, '.dev', 'dev-reload-info.json'),
            ];
            candidates.forEach(function (infoPath) {
                if (!fs.existsSync(infoPath)) return;
                try {
                    var info = JSON.parse(fs.readFileSync(infoPath, 'utf-8'));
                    var ageMs = Date.now() - new Date(info.updatedAt).getTime();
                    results.push({
                        projectPath: info.projectPath,
                        projectName: info.projectName,
                        previewPort: info.previewPort,
                        previewUrl: info.previewUrl,
                        editorPid: info.editorPid,
                        updatedAt: info.updatedAt,
                        staleSec: Math.floor(ageMs / 1000),
                        self: info.projectPath === Editor.Project.path,
                    });
                } catch (e) { /* ignore */ }
            });
        });
        return results;
    },
    /** Panel 使用：返回当前插件状态快照 */
    async getStatus() {
        var url = await getPreviewUrl();
        writeDevReloadInfo(url);
        var portMatch = url ? url.match(/:(\d+)/) : null;
        var infoPath = path.join(Editor.Project.path, INFO_FILE);
        var updatedAt = '';
        try {
            if (fs.existsSync(infoPath)) {
                var info = JSON.parse(fs.readFileSync(infoPath, 'utf-8'));
                updatedAt = info.updatedAt || '';
            }
        } catch (e) { /* ignore */ }
        // git 分支/commit
        var gitBranch = '', gitHead = '';
        try {
            var execSync = require('child_process').execSync;
            gitBranch = execSync('git -C "' + Editor.Project.path + '" rev-parse --abbrev-ref HEAD', { encoding: 'utf-8' }).trim();
            gitHead = execSync('git -C "' + Editor.Project.path + '" rev-parse --short HEAD', { encoding: 'utf-8' }).trim();
        } catch (e) { /* ignore */ }
        return {
            previewUrl: url,
            previewPort: portMatch ? parseInt(portMatch[1], 10) : null,
            editorPid: process.pid,
            editorVersion: (Editor.App && Editor.App.version) ? Editor.App.version : '',
            projectPath: Editor.Project.path,
            updatedAt: updatedAt,
            infoFile: INFO_FILE,
            watchers: {
                refresh: !!_refreshWatcher,
                infoInterval: !!_infoInterval,
            },
            gitBranch: gitBranch,
            gitHead: gitHead,
            commandLog: _commandLog.slice().reverse(),
            mcpServer: _mcpServer ? {
                running: _mcpServer.started,
                url: 'http://' + _mcpServer.host + ':' + _mcpServer.port + '/mcp',
                port: _mcpServer.port,
                toolCount: _mcpServer.toolCount,
                resourceCount: _mcpServer.resourceCount,
                stats: _mcpServer.stats,
            } : { running: false },
        };
    }
};

// ── MCP Server ──

var _mcpServer = null;
var MCP_DEFAULT_PORT = 7523;
var REGISTRY_DIR = path.join(require('os').homedir(), '.cocos-mcp', 'editors');
var SDK_PATH = path.join(__dirname, 'mcp-sdk', 'index.js');

/**
 * 计算项目短名（MCP 工具名前缀，需能区分不同项目）。
 * 禁 worktree（cocos 不支持同一项目多 worktree 同开），故不再为 worktree 做 parent 启发式；
 * base 是常见通用名（client/game/app/src）时取 parent 段区分不同项目，否则直接用 base。
 */
function getProjectShortName() {
    var p = Editor.Project.path || '';
    var parent = path.basename(path.dirname(p));
    var base = path.basename(p);
    if (base === 'client' || base === 'game' || base === 'app' || base === 'src') {
        return parent || base;
    }
    return base;
}

/**
 * 解析 Cocos 编辑器主进程可执行路径，写进注册文件供 router 的 editor_restart 拉起用。
 * 优先 process.argv[0]（Electron 主进程启动命令首段，即 .app 可执行），按 version 拼标准路径兜底。
 * 排除 Helper（渲染/GPU 子进程路径），要外层主可执行。解析不到返回空串，router 端还有 ps / version 两级 fallback。
 */
function getEditorExecPath() {
    var candidates = [];
    try { if (process.argv && process.argv[0]) candidates.push(process.argv[0]); } catch (e) { /* ignore */ }
    try { if (process.execPath) candidates.push(process.execPath); } catch (e) { /* ignore */ }
    var ver = (Editor.App && Editor.App.version) ? Editor.App.version : '';
    if (ver) candidates.push('/Applications/Cocos/Creator/' + ver + '/CocosCreator.app/Contents/MacOS/CocosCreator');
    for (var i = 0; i < candidates.length; i++) {
        var c = candidates[i];
        if (c && /CocosCreator/.test(c) && c.indexOf('Helper') < 0) {
            try { if (fs.existsSync(c)) return c; } catch (e) { /* ignore */ }
        }
    }
    return '';
}

function writeRegistry() {
    if (!_mcpServer || !_mcpServer.started) return;
    try {
        if (!fs.existsSync(REGISTRY_DIR)) fs.mkdirSync(REGISTRY_DIR, { recursive: true });
        var entry = {
            pid: process.pid,
            projectPath: Editor.Project.path,
            projectShortName: getProjectShortName(),
            host: _mcpServer.host,
            port: _mcpServer.port,
            url: 'http://' + _mcpServer.host + ':' + _mcpServer.port + '/mcp',
            editorVersion: (Editor.App && Editor.App.version) ? Editor.App.version : '',
            execPath: getEditorExecPath(),
            startedAt: _mcpServer.stats.startedAt,
            updatedAt: new Date().toISOString(),
        };
        fs.writeFileSync(path.join(REGISTRY_DIR, process.pid + '.json'), JSON.stringify(entry, null, 2), 'utf-8');
    } catch (e) {
        console.warn('[cc-mcp] writeRegistry failed:', e.message || e);
    }
}

function removeRegistry() {
    try {
        var f = path.join(REGISTRY_DIR, process.pid + '.json');
        if (fs.existsSync(f)) fs.unlinkSync(f);
    } catch (e) { /* ignore */ }
}

/** 分配可用端口：先试 DEFAULT，如果被占用递增 */
function findFreePort(startPort) {
    var net = require('net');
    return new Promise(function (resolve) {
        function tryPort(p) {
            var tester = net.createServer()
                .once('error', function () { tryPort(p + 1); })
                .once('listening', function () {
                    tester.close(function () { resolve(p); });
                })
                .listen(p, '127.0.0.1');
        }
        tryPort(startPort);
    });
}

async function startMcpServer() {
    if (_mcpServer && _mcpServer.started) return;
    var port = await findFreePort(MCP_DEFAULT_PORT);
    var sdk = require(SDK_PATH);

    // ── 使用 mcp-sdk ──────────────────────────────────────────────
    var tdef = require('./server/tools');
    var ctx = buildToolCtx();
    var toolDefs = tdef.defineTools(ctx);
    var resourceDefs = tdef.defineResources(ctx);

    var server = sdk.createServer({
        name: 'cc-3-8-x-mcp',
        version: '2.0.0',
        port: port,
        tools: toolDefs.map(function (t) {
            return {
                name: t.name,
                description: t.description,
                inputSchema: t.inputSchema,
                handler: t.handler,
            };
        }),
        resources: resourceDefs.map(function (r) {
            return {
                uri: r.uri,
                name: r.name,
                description: r.description,
                mimeType: r.mimeType,
                read: r.read,
            };
        }),
    });

    // 启动 HTTP（cc-3-8-x-mcp 只跑 HTTP，不跑 stdio）
    await server.start('http');
    _mcpServer = {
        started: true,
        host: '127.0.0.1',
        port: port,
        toolCount: toolDefs.length,
        resourceCount: resourceDefs.length,
        stats: { startedAt: new Date().toISOString(), requestCount: 0 },
        stop: function () { server.stop(); },
    };
    writeRegistry();
    log('MCP server up (SDK) — http://127.0.0.1:' + port + '/mcp  (tools:' + toolDefs.length + ') shortName=' + getProjectShortName());
}

/**
 * 构建 tool/resource 的 ctx（共享给 SDK 和 fallback）
 */
function buildToolCtx() {
    return {
        msg: function (target, name /*, ...args */) {
            var args = Array.prototype.slice.call(arguments, 2);
            return Editor.Message.request.apply(Editor.Message, [target, name].concat(args));
        },
        local: {
            getPreviewUrl: getPreviewUrl,
            doReimport: doReimport,
            doRefreshPreview: doRefreshPreview,
            doOpenPreview: doOpenPreview,
            doScreenshot: async function (outputPath) {
                var p = outputPath || path.join(Editor.Project.path, DEV_DIR, 'screenshot.png');
                await doScreenshot(p);
                return p;
            },
            doRefreshAssets: doRefreshAssets,
            doReloadScene: doReloadScene,
            evalInPreview: function (code) { return exports.methods.evalInPreview(code); },
            listWorktrees: function () { return exports.methods.listWorktrees(); },
            openDevDir: function () { return exports.methods.openDevDir(); },
            cleanDevDir: function () { return exports.methods.cleanDevDir(); },
            getStatus: function () { return exports.methods.getStatus(); },
        },
    };
}

async function stopMcpServer() {
    if (!_mcpServer) return;
    removeRegistry();
    if (_mcpServer.stop) {
        _mcpServer.stop();
    } else {
        try { await _mcpServer.stop(); } catch (e) { /* ignore */ }
    }
    _mcpServer = null;
}

// ── .dev/refresh 文件命令协议 ──

/**
 * 打开 prefab 到编辑器场景视图（等价于双击 prefab）。
 * @param {string} urlOrPath  db:// 路径 或 绝对路径
 */
async function doOpenPrefab(urlOrPath) {
    var dbUrl = urlOrPath;
    // 绝对路径 → 先通过 asset-db 反查 db:// url，再走统一路径
    if (!urlOrPath.startsWith('db://')) {
        var info = await Editor.Message.request('asset-db', 'query-asset-info', urlOrPath);
        if (!info || !info.url) throw new Error('open-prefab: cannot find db:// url for ' + urlOrPath);
        dbUrl = info.url;
    }
    var uuid = await Editor.Message.request('asset-db', 'query-uuid', dbUrl);
    if (!uuid) throw new Error('open-prefab: cannot resolve uuid for ' + dbUrl);
    await Editor.Message.request('asset-db', 'open-asset', uuid);
    log('open-prefab: opened ' + dbUrl + ' (uuid=' + uuid + ')');
}

/** 重启整个插件（disable → enable）让 main.js / tools.js / server/* 的代码改动生效。
 *  注意：本函数自身处于即将被卸载的 main.js 上下文，必须 fire-and-forget。
 *  Editor.Package.disable 是 host 进程 API，扩展沙箱卸载后仍然有效；
 *  enable 在 disable 完成后用 setTimeout 触发，给 unload 收尾留窗口。
 */
function doRestartSelf() {
    var name = 'cc-3-8-x-mcp';
    log('restart-package: scheduling disable → enable for ' + name);
    setImmediate(function () {
        Promise.resolve()
            .then(function () { return Editor.Package.disable(name, true); })
            // 200ms 给 unload 钩子（stopRefreshWatcher / stopMcpServer）跑完
            .then(function () { return new Promise(function (r) { setTimeout(r, 200); }); })
            .then(function () { return Editor.Package.enable(name, true); })
            .catch(function (err) {
                console.error('[restart-package] failed:', err && (err.stack || err.message) || err);
            });
    });
}

/**
 * 分发单条 refresh 命令。
 *
 * 协议精简：只支持 `restart-package`（disable→enable 整个扩展，让 JS 代码改动生效）。
 * 资源刷新 / 场景重载 / 预览刷新 / 截图等走 MCP tool（preview_refresh_and_reload /
 * asset_reimport / preview_screenshot 等）或面板按钮，不再通过文件协议触发。
 */
async function handleRefreshCommand(cmd) {
    if (!cmd) return;
    pushCommandLog('refresh', cmd);
    if (cmd === 'restart-package') {
        doRestartSelf();
        return;
    }
    log('refresh: unknown command — ' + cmd + '（仅支持 restart-package）');
}

/** 启动 .dev/refresh 文件 watcher（写入命令 → 读取 → 执行 → 清空） */
function startRefreshWatcher() {
    if (_refreshWatcher) return;
    var filePath = path.join(Editor.Project.path, REFRESH_FILE);
    // 确保文件存在，供 fs.watch 注册
    if (!fs.existsSync(filePath)) {
        try { fs.writeFileSync(filePath, '', 'utf-8'); } catch (e) { /* ignore */ }
    }
    var _debounceTimer = null;
    try {
        _refreshWatcher = fs.watch(filePath, function (event) {
            if (event !== 'change' && event !== 'rename') return;
            // debounce：macOS 下单次 write 可能触发多次事件
            clearTimeout(_debounceTimer);
            _debounceTimer = setTimeout(function () {
                var content = '';
                try { content = fs.readFileSync(filePath, 'utf-8').trim(); } catch (e) { return; }
                if (!content) return;
                // 清空信号文件，防止重复执行
                try { fs.writeFileSync(filePath, '', 'utf-8'); } catch (e) { /* ignore */ }
                var lines = content.split('\n');
                // 逐条串行执行（前一条完成再执行下一条）
                lines.reduce(function (chain, line) {
                    return chain.then(function () { return handleRefreshCommand(line.trim()); });
                }, Promise.resolve());
            }, 80);
        });
        log('refresh watcher started → ' + REFRESH_FILE);
    } catch (e) {
        console.error('[dev-reload] startRefreshWatcher failed:', e && (e.stack || e.message) || e);
    }
}

/** 停止 .dev/refresh 文件 watcher */
function stopRefreshWatcher() {
    if (_refreshWatcher) { try { _refreshWatcher.close(); } catch (e) { /* ignore */ } _refreshWatcher = null; }
}

// ── 插件生命周期 ──

exports.load = async function () {
    // 确保 .dev 目录存在
    var devDir = path.join(Editor.Project.path, DEV_DIR);
    if (!fs.existsSync(devDir)) {
        fs.mkdirSync(devDir, { recursive: true });
    }
    log('loaded');
    // 启动 .dev/refresh 文件 watcher
    startRefreshWatcher();
    // 异步拿预览地址，写 dev-reload-info.json，启动定时刷新
    getPreviewUrl().then(function(url) {
        if (url) writeDevReloadInfo(url);
        startInfoInterval();
    }).catch(function(e) {
        console.error('[dev-reload] load: getPreviewUrl failed —', e && (e.stack || e.message) || e);
        startInfoInterval();
    });
    // 启动 MCP server（失败打完整栈，不阻断扩展 load）
    startMcpServer().catch(function(e) {
        console.error('[cc-mcp] MCP server failed to start:', e && (e.stack || e.message) || e);
    });
};

exports.unload = async function () {
    stopRefreshWatcher();
    stopInfoInterval();
    await stopMcpServer();
    log('unloaded');
};