cicy-desktop 1.0.8

package/src/tools/exec-js.js

@@ -0,0 +1,297 @@
+const { BrowserWindow } = require("electron");
+const { z } = require("zod");
+
+function registerTools(registerTool) {
+  registerTool(
+    "exec_js",
+    "在窗口中执行JS代码并返回结果,支持async/await。示例: document.title 或 return document.title 或 await fetch('/api').then(r=>r.json())",
+    z.object({
+      win_id: z.number().optional().default(1).describe("窗口 ID"),
+      code: z.string().describe("JS 代码"),
+    }),
+    async ({ win_id, code }) => {
+      try {
+        const win = BrowserWindow.fromId(win_id);
+        if (!win) throw new Error(`未找到窗口 ${win_id}`);
+
+        // 简化代码包装逻辑
+        let wrappedCode = code.trim();
+
+        // 如果代码不包含 return 且不是多语句，自动添加 return
+        if (
+          !wrappedCode.includes("return") &&
+          !wrappedCode.includes(";") &&
+          !wrappedCode.includes("\n")
+        ) {
+          wrappedCode = `return ${wrappedCode}`;
+        }
+
+        // 包装为异步函数
+        wrappedCode = `(async () => { ${wrappedCode} })()`;
+
+        const result = await win.webContents.executeJavaScript(wrappedCode);
+        const text =
+          result !== null && typeof result === "object"
+            ? JSON.stringify(result, null, 2)
+            : String(result);
+        return {
+          content: [
+            { type: "text", text },
+            { type: "text", text: `>> exec_js ok` },
+          ],
+        };
+      } catch (error) {
+        return { content: [{ type: "text", text: `Error: ${error.message}` }], isError: true };
+      }
+    },
+    { tag: "JavaScript" }
+  );
+
+  registerTool(
+    "exec_js_file",
+    "Execute JS file in browser window. Provide file path or content.",
+    z.object({
+      win_id: z.number().optional().default(1).describe("窗口 ID"),
+      file: z.string().optional().describe("本地 JS 文件路径"),
+      content: z.string().optional().describe("JS content (saved to temp and executed)"),
+    }),
+    async ({ win_id, file, content: jsContent }) => {
+      try {
+        const fs = require("fs");
+        const path = require("path");
+        const os = require("os");
+        let code;
+        if (jsContent) {
+          const tmp = path.join(os.homedir(), "tmp", "_exec_js_" + Date.now() + ".js");
+          fs.mkdirSync(path.dirname(tmp), { recursive: true });
+          fs.writeFileSync(tmp, jsContent, "utf-8");
+          code = jsContent;
+        } else if (file) {
+          const resolved = path.resolve(file);
+          if (!fs.existsSync(resolved)) throw new Error("文件不存在: " + resolved);
+          code = fs.readFileSync(resolved, "utf-8");
+        } else {
+          throw new Error("需要 file 或 content 参数");
+        }
+        const win = BrowserWindow.fromId(win_id);
+        if (!win) throw new Error("未找到窗口 " + win_id);
+        const wrappedCode = "(async () => { " + code + " })()";
+        const result = await win.webContents.executeJavaScript(wrappedCode);
+        const text = result !== null && typeof result === "object" ? JSON.stringify(result, null, 2) : String(result);
+        return { content: [{ type: "text", text }, { type: "text", text: ">> exec_js_file ok" }] };
+      } catch (error) {
+        return { content: [{ type: "text", text: "Error: " + error.message }], isError: true };
+      }
+    },
+    { tag: "JavaScript" }
+  );
+
+
+  registerTool(
+    "get_element_client_bound",
+    "获取页面元素的位置和尺寸信息(getBoundingClientRect)。返回元素的 x, y, width, height, top, left, right, bottom 坐标。selector 必须唯一且不能为空",
+    z.object({
+      win_id: z.number().optional().default(1).describe("窗口 ID"),
+      selector: z
+        .string()
+        .min(1, "CSS选择器不能为空")
+        .describe("CSS 选择器，如 '#id', '.class', 'button'"),
+    }),
+    async ({ win_id, selector }) => {
+      try {
+        const win = BrowserWindow.fromId(win_id);
+        if (!win) throw new Error(`未找到窗口 ${win_id}`);
+
+        const escapedSelector = selector.replace(/'/g, "\\'");
+        const result = await win.webContents.executeJavaScript(`
+          (function() {
+            const selector = '${escapedSelector}';
+
+            if (!selector || selector.trim() === '') {
+              return { error: 'CSS选择器不能为空' };
+            }
+
+            const allElements = document.querySelectorAll(selector);
+            if (allElements.length === 0) {
+              return { error: '未找到匹配的元素: ' + selector };
+            }
+            if (allElements.length > 1) {
+              return { error: '选择器匹配多个元素，请使用唯一选择器，匹配数量: ' + allElements.length };
+            }
+
+            const el = allElements[0];
+            const rect = el.getBoundingClientRect();
+            return {
+              selector: selector,
+              count: allElements.length,
+              x: Math.round(rect.x),
+              y: Math.round(rect.y),
+              width: Math.round(rect.width),
+              height: Math.round(rect.height),
+              top: Math.round(rect.top),
+              left: Math.round(rect.left),
+              right: Math.round(rect.right),
+              bottom: Math.round(rect.bottom)
+            };
+          })()
+        `);
+
+        if (result.error) {
+          return { content: [{ type: "text", text: result.error }], isError: true };
+        }
+
+        return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
+      } catch (error) {
+        return { content: [{ type: "text", text: `Error: ${error.message}` }], isError: true };
+      }
+    },
+    { tag: "JavaScript" }
+  );
+
+  registerTool(
+    "show_float_div",
+    "在页面中心显示一个可拖拽、可调整大小的浮动 div。支持自定义位置、尺寸和透明度。",
+    z.object({
+      win_id: z.number().optional().default(1).describe("窗口 ID"),
+      x: z.number().optional().describe("X 坐标 (默认居中)"),
+      y: z.number().optional().describe("Y 坐标 (默认居中)"),
+      w: z.number().optional().default(300).describe("宽度 (默认 300)"),
+      h: z.number().optional().default(200).describe("高度 (默认 200)"),
+      opacity: z.number().optional().default(0.7).describe("背景透明度 0-1 (默认 0.7)"),
+    }),
+    async ({ win_id, x, y, w, h, opacity }) => {
+      try {
+        const win = BrowserWindow.fromId(win_id);
+        if (!win) throw new Error(`未找到窗口 ${win_id}`);
+
+        const jsCode = `(function() {
+            try {
+            var existDiv = document.getElementById('__float_div__');
+            if (existDiv) { existDiv.remove(); }
+            
+            var div = document.createElement('div');
+            div.id = '__float_div__';
+            
+            var viewportW = window.innerWidth;
+            var viewportH = window.innerHeight;
+            var initW = ${w} || 300;
+            var initH = ${h} || 200;
+            var initX = ${x} || Math.round((viewportW - initW) / 2);
+            var initY = ${y} || Math.round((viewportH - initH) / 2);
+            var opacity = ${opacity} || 0.7;
+            
+            div.style.cssText = 'position:fixed;left:' + initX + 'px;top:' + initY + 'px;width:' + initW + 'px;height:' + initH + 'px;background:rgba(255,0,0,' + opacity + ');color:#fff;z-index:999999;border:2px solid #fff;';
+            
+            // 左上角坐标
+            var tl = document.createElement('div');
+            tl.textContent = initX + ',' + initY;
+            tl.style.cssText = 'position:absolute;top:4px;left:4px;font-family:monospace;font-size:14px;font-weight:bold;color:#fff;';
+            div.appendChild(tl);
+            
+            // 右上角坐标
+            var tr = document.createElement('div');
+            tr.textContent = (initX + initW) + ',' + initY;
+            tr.style.cssText = 'position:absolute;top:4px;right:4px;font-family:monospace;font-size:14px;font-weight:bold;color:#fff;';
+            div.appendChild(tr);
+            
+            // 左下角坐标
+            var bl = document.createElement('div');
+            bl.textContent = initX + ',' + (initY + initH);
+            bl.style.cssText = 'position:absolute;bottom:4px;left:4px;font-family:monospace;font-size:14px;font-weight:bold;color:#fff;';
+            div.appendChild(bl);
+            
+            // 右下角坐标
+            var br = document.createElement('div');
+            br.textContent = (initX + initW) + ',' + (initY + initH);
+            br.style.cssText = 'position:absolute;bottom:4px;right:4px;font-family:monospace;font-size:14px;font-weight:bold;color:#fff;';
+            div.appendChild(br);
+            
+            // 关闭按钮
+            var closeBtn = document.createElement('div');
+            closeBtn.textContent = 'X';
+            closeBtn.style.cssText = 'position:absolute;top:50%;right:4px;transform:translateY(-50%);width:16px;height:16px;cursor:pointer;font-size:12px;font-weight:bold;background:#fff;color:#f00;text-align:center;line-height:16px;';
+            closeBtn.onclick = function(e) { e.stopPropagation(); div.remove(); };
+            div.appendChild(closeBtn);
+            
+            // resize 把手
+            var resizeHandle = document.createElement('div');
+            resizeHandle.style.cssText = 'position:absolute;right:0;bottom:0;width:14px;height:14px;background:#fff;cursor:se-resize;';
+            div.appendChild(resizeHandle);
+            
+            var dragX = initX, dragY = initY, curW = initW, curH = initH;
+            var isDrag = false, isResize = false, startX, startY;
+            
+            div.addEventListener('mousedown', function(e) {
+              if (e.target === closeBtn || e.target === resizeHandle) return;
+              isDrag = true;
+              startX = e.clientX;
+              startY = e.clientY;
+              e.preventDefault();
+              e.stopPropagation();
+            });
+            
+            resizeHandle.addEventListener('mousedown', function(e) {
+              isResize = true;
+              startX = e.clientX;
+              startY = e.clientY;
+              e.preventDefault();
+              e.stopPropagation();
+            });
+            
+            document.addEventListener('mousemove', function(e) {
+              if (isDrag) {
+                dragX += e.clientX - startX;
+                dragY += e.clientY - startY;
+                div.style.left = dragX + 'px';
+                div.style.top = dragY + 'px';
+                startX = e.clientX;
+                startY = e.clientY;
+                tl.textContent = Math.round(dragX) + ',' + Math.round(dragY);
+                tr.textContent = Math.round(dragX + curW) + ',' + Math.round(dragY);
+                bl.textContent = Math.round(dragX) + ',' + Math.round(dragY + curH);
+                br.textContent = Math.round(dragX + curW) + ',' + Math.round(dragY + curH);
+              } else if (isResize) {
+                curW = Math.max(100, curW + e.clientX - startX);
+                curH = Math.max(60, curH + e.clientY - startY);
+                div.style.width = curW + 'px';
+                div.style.height = curH + 'px';
+                startX = e.clientX;
+                startY = e.clientY;
+                tr.textContent = Math.round(dragX + curW) + ',' + Math.round(dragY);
+                bl.textContent = Math.round(dragX) + ',' + Math.round(dragY + curH);
+                br.textContent = Math.round(dragX + curW) + ',' + Math.round(dragY + curH);
+              }
+            });
+            
+            document.addEventListener('mouseup', function() {
+              isDrag = false;
+              isResize = false;
+            });
+            
+            document.body.appendChild(div);
+            return 'OK: div at ' + Math.round(initX) + ',' + Math.round(initY) + ' size ' + initW + 'x' + initH;
+            } catch(e) { return 'ERROR: ' + e.message; }
+          })()`;
+
+        const result = await win.webContents.debugger.sendCommand("Runtime.evaluate", {
+          expression: jsCode,
+        });
+        console.log("[show_float_div] result:", result);
+
+        return {
+          content: [
+            {
+              type: "text",
+              text: result.result?.value || result.result?.description || JSON.stringify(result),
+            },
+          ],
+        };
+      } catch (error) {
+        return { content: [{ type: "text", text: `Error: ${error.message}` }], isError: true };
+      }
+    },
+    { tag: "JavaScript" }
+  );
+}
+
+module.exports = registerTools;

package/src/tools/exec-tools.js

@@ -0,0 +1,139 @@
+const { z } = require("zod");
+const { exec } = require("child_process");
+const util = require("util");
+const execPromise = util.promisify(exec);
+const fs = require("fs");
+const path = require("path");
+const os = require("os");
+
+const TMP = path.join(os.homedir(), "tmp");
+if (!fs.existsSync(TMP)) fs.mkdirSync(TMP, { recursive: true });
+
+function writeTemp(name, content) {
+  const p = path.join(TMP, name);
+  fs.writeFileSync(p, content, "utf-8");
+  return p;
+}
+
+function resolveFile(file, content, ext) {
+  if (content) return writeTemp(`_exec_${Date.now()}${ext}`, content);
+  const resolved = path.resolve(file);
+  if (!fs.existsSync(resolved)) throw new Error("File not found: " + resolved);
+  return resolved;
+}
+
+function result(stdout, stderr) {
+  return { content: [{ type: "text", text: JSON.stringify({ stdout: stdout || "", stderr: stderr || "", exitCode: 0 }, null, 2) }] };
+}
+function errorResult(error) {
+  return { content: [{ type: "text", text: JSON.stringify({ stdout: error.stdout || "", stderr: error.stderr || error.message, exitCode: error.code || 1 }, null, 2) }], isError: true };
+}
+
+function registerTools(registerTool) {
+  registerTool(
+    "exec_shell",
+    "Execute shell command",
+    z.object({
+      command: z.string().describe("Shell command to execute"),
+      cwd: z.string().optional().describe("Working directory"),
+    }),
+    async ({ command, cwd }) => {
+      try {
+        const { stdout, stderr } = await execPromise(command, { cwd: cwd || process.cwd(), maxBuffer: 1024 * 1024 * 10 });
+        return result(stdout, stderr);
+      } catch (e) { return errorResult(e); }
+    },
+    { tag: "Exec" }
+  );
+
+  registerTool(
+    "exec_python",
+    "Execute Python code",
+    z.object({
+      code: z.string().describe("Python code to execute"),
+      cwd: z.string().optional().describe("Working directory"),
+    }),
+    async ({ code, cwd }) => {
+      try {
+        const py = process.platform === "win32" ? "python" : "python3";
+        const { stdout, stderr } = await execPromise(`${py} -c ${JSON.stringify(code)}`, { cwd: cwd || process.cwd(), maxBuffer: 1024 * 1024 * 10 });
+        return result(stdout, stderr);
+      } catch (e) { return errorResult(e); }
+    },
+    { tag: "Exec" }
+  );
+
+  registerTool(
+    "exec_node",
+    "Execute Node.js code",
+    z.object({
+      code: z.string().describe("Node.js code to execute"),
+      cwd: z.string().optional().describe("Working directory"),
+    }),
+    async ({ code, cwd }) => {
+      try {
+        const { stdout, stderr } = await execPromise(`node -e ${JSON.stringify(code)}`, { cwd: cwd || process.cwd(), maxBuffer: 1024 * 1024 * 10 });
+        return result(stdout, stderr);
+      } catch (e) { return errorResult(e); }
+    },
+    { tag: "Exec" }
+  );
+
+  registerTool(
+    "exec_shell_file",
+    "Execute shell script. Provide file path or content (content will be saved to temp file and executed).",
+    z.object({
+      file: z.string().optional().describe("Path to shell script file"),
+      content: z.string().optional().describe("Shell script content (uploaded and executed)"),
+      cwd: z.string().optional().describe("Working directory"),
+    }),
+    async ({ file, content, cwd }) => {
+      try {
+        const resolved = resolveFile(file, content, ".bat");
+        const cmd = process.platform === "win32" ? `"${resolved}"` : `bash "${resolved}"`;
+        const { stdout, stderr } = await execPromise(cmd, { cwd: cwd || process.cwd(), maxBuffer: 1024 * 1024 * 10 });
+        return result(stdout, stderr);
+      } catch (e) { return errorResult(e); }
+    },
+    { tag: "Exec" }
+  );
+
+  registerTool(
+    "exec_python_file",
+    "Execute Python script. Provide file path or content (content will be saved to temp file and executed).",
+    z.object({
+      file: z.string().optional().describe("Path to Python script file"),
+      content: z.string().optional().describe("Python script content (uploaded and executed)"),
+      cwd: z.string().optional().describe("Working directory"),
+    }),
+    async ({ file, content, cwd }) => {
+      try {
+        const resolved = resolveFile(file, content, ".py");
+        const py = process.platform === "win32" ? "python" : "python3";
+        const { stdout, stderr } = await execPromise(`${py} "${resolved}"`, { cwd: cwd || process.cwd(), maxBuffer: 1024 * 1024 * 10 });
+        return result(stdout, stderr);
+      } catch (e) { return errorResult(e); }
+    },
+    { tag: "Exec" }
+  );
+
+  registerTool(
+    "exec_node_file",
+    "Execute Node.js script. Provide file path or content (content will be saved to temp file and executed).",
+    z.object({
+      file: z.string().optional().describe("Path to Node.js script file"),
+      content: z.string().optional().describe("Node.js script content (uploaded and executed)"),
+      cwd: z.string().optional().describe("Working directory"),
+    }),
+    async ({ file, content, cwd }) => {
+      try {
+        const resolved = resolveFile(file, content, ".js");
+        const { stdout, stderr } = await execPromise(`node "${resolved}"`, { cwd: cwd || process.cwd(), maxBuffer: 1024 * 1024 * 10 });
+        return result(stdout, stderr);
+      } catch (e) { return errorResult(e); }
+    },
+    { tag: "Exec" }
+  );
+}
+
+module.exports = registerTools;

package/src/tools/file-tools.js

@@ -0,0 +1,212 @@
+const { z } = require("zod");
+const fs = require("fs");
+const path = require("path");
+const crypto = require("crypto");
+
+const FILES_DIR = path.join(require("os").homedir(), "electron-mcp-files");
+
+function registerTools(registerTool) {
+  // 文本文件工具（默认 UTF-8）
+  registerTool(
+    "file_read",
+    "读取文本文件内容",
+    z.object({
+      path: z.string().describe("文件路径"),
+    }),
+    async ({ path: filePath }) => {
+      try {
+        if (!fs.existsSync(filePath)) {
+          throw new Error(`File not found: ${filePath}`);
+        }
+
+        const content = fs.readFileSync(filePath, "utf8");
+        const stats = fs.statSync(filePath);
+
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  path: filePath,
+                  size: stats.size,
+                  content,
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      } catch (error) {
+        return {
+          content: [{ type: "text", text: `Error: ${error.message}` }],
+          isError: true,
+        };
+      }
+    },
+    { tag: "File" }
+  );
+
+  registerTool(
+    "file_write",
+    "写入文本文件内容",
+    z.object({
+      path: z.string().describe("文件路径"),
+      content: z.string().describe("文件内容"),
+    }),
+    async ({ path: filePath, content }) => {
+      try {
+        const dir = path.dirname(filePath);
+        if (!fs.existsSync(dir)) {
+          fs.mkdirSync(dir, { recursive: true });
+        }
+
+        fs.writeFileSync(filePath, content, "utf8");
+        const stats = fs.statSync(filePath);
+
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  success: true,
+                  path: filePath,
+                  size: stats.size,
+                  message: `File written: ${stats.size} bytes`,
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      } catch (error) {
+        return {
+          content: [{ type: "text", text: `Error: ${error.message}` }],
+          isError: true,
+        };
+      }
+    },
+    { tag: "File" }
+  );
+
+  // 文件上传下载（通过 HTTP）
+  registerTool(
+    "file_upload",
+    "上传文件到服务器（base64编码），返回可访问的 URL",
+    z.object({
+      filename: z.string().describe("文件名"),
+      content: z.string().describe("文件内容（base64编码）"),
+    }),
+    async ({ filename, content }) => {
+      try {
+        if (!fs.existsSync(FILES_DIR)) {
+          fs.mkdirSync(FILES_DIR, { recursive: true });
+        }
+
+        const hash = crypto.createHash("md5").update(content).digest("hex").slice(0, 8);
+        const ext = path.extname(filename);
+        const basename = path.basename(filename, ext);
+        const uniqueFilename = `${basename}-${hash}${ext}`;
+        const filePath = path.join(FILES_DIR, uniqueFilename);
+
+        const buffer = Buffer.from(content, "base64");
+        fs.writeFileSync(filePath, buffer);
+        const stats = fs.statSync(filePath);
+
+        const baseUrl =
+          process.env.ELECTRON_MCP_BASE_URL || `http://localhost:${process.env.PORT || 8101}`;
+        const url = `${baseUrl}/files/${uniqueFilename}`;
+
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  success: true,
+                  filename: uniqueFilename,
+                  path: filePath,
+                  url,
+                  size: stats.size,
+                  message: `File uploaded: ${stats.size} bytes`,
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      } catch (error) {
+        return {
+          content: [{ type: "text", text: `Error: ${error.message}` }],
+          isError: true,
+        };
+      }
+    },
+    { tag: "File" }
+  );
+
+  registerTool(
+    "file_download",
+    "从本地路径下载文件，复制到服务器并返回 URL",
+    z.object({
+      path: z.string().describe("本地文件路径"),
+    }),
+    async ({ path: filePath }) => {
+      try {
+        if (!fs.existsSync(filePath)) {
+          throw new Error(`File not found: ${filePath}`);
+        }
+
+        if (!fs.existsSync(FILES_DIR)) {
+          fs.mkdirSync(FILES_DIR, { recursive: true });
+        }
+
+        const stats = fs.statSync(filePath);
+        const buffer = fs.readFileSync(filePath);
+        const hash = crypto.createHash("md5").update(buffer).digest("hex").slice(0, 8);
+        const ext = path.extname(filePath);
+        const basename = path.basename(filePath, ext);
+        const uniqueFilename = `${basename}-${hash}${ext}`;
+        const destPath = path.join(FILES_DIR, uniqueFilename);
+
+        fs.copyFileSync(filePath, destPath);
+
+        const baseUrl =
+          process.env.ELECTRON_MCP_BASE_URL || `http://localhost:${process.env.PORT || 8101}`;
+        const url = `${baseUrl}/files/${uniqueFilename}`;
+
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  success: true,
+                  filename: uniqueFilename,
+                  path: destPath,
+                  url,
+                  size: stats.size,
+                  message: `File copied: ${stats.size} bytes`,
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      } catch (error) {
+        return {
+          content: [{ type: "text", text: `Error: ${error.message}` }],
+          isError: true,
+        };
+      }
+    },
+    { tag: "File" }
+  );
+}
+
+module.exports = registerTools;