cicy-desktop 1.0.8

package/src/utils/auth.js

@@ -0,0 +1,81 @@
+const fs = require("fs");
+const path = require("path");
+const crypto = require("crypto");
+const os = require("os");
+const log = require("electron-log");
+
+/**
+ * 认证模块 - 处理令牌生成、验证和管理
+ */
+class AuthManager {
+  constructor() {
+    this.authToken = this.getOrGenerateToken();
+    log.info("[MCP] Auth token enabled");
+    log.info("[MCP] Token saved to ~/data/electron/token.txt");
+  }
+
+  /**
+   * 获取或生成认证令牌
+   * @returns {string} 认证令牌
+   */
+  getOrGenerateToken() {
+    const tokenPath = path.join(os.homedir(), "data", "electron", "token.txt");
+
+    try {
+      // 检查是否已存在令牌
+      if (fs.existsSync(tokenPath)) {
+        const token = fs.readFileSync(tokenPath, "utf8").trim();
+        if (token) {
+          log.info("[MCP] Using existing token from", tokenPath);
+          return token;
+        }
+      }
+
+      // 生成新令牌
+      const newToken = crypto.randomBytes(32).toString("hex");
+      fs.mkdirSync(path.dirname(tokenPath), { recursive: true });
+      fs.writeFileSync(tokenPath, newToken);
+      log.info("[MCP] Generated new token and saved to", tokenPath);
+      return newToken;
+    } catch (error) {
+      log.error("[MCP] Token management error:", error);
+      return crypto.randomBytes(32).toString("hex"); // fallback
+    }
+  }
+
+  /**
+   * 验证认证令牌（支持 Bearer 和 Basic Auth）
+   * @param {Object} req - HTTP 请求对象
+   * @returns {boolean} 验证结果
+   */
+  validateAuth(req) {
+    const authHeader = req.headers.authorization;
+    if (!authHeader) return false;
+
+    // Bearer token
+    if (authHeader.startsWith("Bearer ")) {
+      const token = authHeader.replace("Bearer ", "");
+      return token === this.authToken;
+    }
+
+    // Basic Auth (username:password where password is token)
+    if (authHeader.startsWith("Basic ")) {
+      const base64Credentials = authHeader.replace("Basic ", "");
+      const credentials = Buffer.from(base64Credentials, "base64").toString("utf8");
+      const [, password] = credentials.split(":");
+      return password === this.authToken;
+    }
+
+    return false;
+  }
+
+  /**
+   * 获取当前认证令牌
+   * @returns {string} 当前令牌
+   */
+  getToken() {
+    return this.authToken;
+  }
+}
+
+module.exports = { AuthManager };

package/src/utils/cdp-utils.js

@@ -0,0 +1,8 @@
+async function sendCDP(webContents, method, params = {}) {
+  if (!webContents.debugger.isAttached()) {
+    webContents.debugger.attach("1.3");
+  }
+  return await webContents.debugger.sendCommand(method, params);
+}
+
+module.exports = { sendCDP };

package/src/utils/download-manager.js

@@ -0,0 +1,41 @@
+// 全局下载管理器
+const downloads = new Map();
+let downloadIdCounter = 0;
+
+function addDownload(downloadInfo) {
+  const id = ++downloadIdCounter;
+  downloads.set(id, {
+    id,
+    ...downloadInfo,
+    createdAt: new Date().toISOString(),
+  });
+  return id;
+}
+
+function updateDownload(id, updates) {
+  const download = downloads.get(id);
+  if (download) {
+    Object.assign(download, updates, { updatedAt: new Date().toISOString() });
+  }
+}
+
+function getDownload(id) {
+  return downloads.get(id);
+}
+
+function getAllDownloads() {
+  return Array.from(downloads.values());
+}
+
+function clearDownloads() {
+  downloads.clear();
+  downloadIdCounter = 0;
+}
+
+module.exports = {
+  addDownload,
+  updateDownload,
+  getDownload,
+  getAllDownloads,
+  clearDownloads,
+};

package/src/utils/process-utils.js

@@ -0,0 +1,185 @@
+const { exec } = require("child_process");
+const net = require("net");
+const { promisify } = require("util");
+
+const execAsync = promisify(exec);
+
+/**
+ * 检查端口是否开放（被占用）
+ * @param {number} port - 端口号
+ * @param {string} [host='localhost'] - 主机地址
+ * @param {number} [timeout=1000] - 超时时间（毫秒）
+ * @returns {Promise<boolean>} true=端口开放/被占用, false=端口关闭/可用
+ */
+async function isPortOpen(port, host = "localhost", timeout = 1000) {
+  // 验证端口范围
+  if (typeof port !== "number" || port < 0 || port >= 65536) {
+    return false;
+  }
+
+  return new Promise((resolve) => {
+    const socket = new net.Socket();
+    let isResolved = false;
+
+    const onError = () => {
+      if (!isResolved) {
+        isResolved = true;
+        socket.destroy();
+        resolve(false);
+      }
+    };
+
+    const onTimeout = () => {
+      if (!isResolved) {
+        isResolved = true;
+        socket.destroy();
+        resolve(false);
+      }
+    };
+
+    const onConnect = () => {
+      if (!isResolved) {
+        isResolved = true;
+        socket.destroy();
+        resolve(true);
+      }
+    };
+
+    socket.setTimeout(timeout);
+    socket.once("error", onError);
+    socket.once("timeout", onTimeout);
+    socket.once("connect", onConnect);
+
+    socket.connect(port, host);
+  });
+}
+
+/**
+ * 杀死占用指定端口的进程
+ * @param {number} port - 端口号
+ * @returns {Promise<{success: boolean, message: string, pid?: number}>}
+ */
+async function killPort(port) {
+  const platform = process.platform;
+
+  try {
+    if (platform === "win32") {
+      return await killPortWindows(port);
+    } else {
+      return await killPortUnix(port);
+    }
+  } catch (error) {
+    return {
+      success: false,
+      message: `Failed to kill process on port ${port}: ${error.message}`,
+    };
+  }
+}
+
+/**
+ * Windows 平台杀死端口进程
+ * @private
+ */
+async function killPortWindows(port) {
+  try {
+    // 查找占用端口的进程
+    const { stdout } = await execAsync(`netstat -ano | findstr :${port}`);
+
+    if (!stdout.trim()) {
+      return {
+        success: false,
+        message: `No process found on port ${port}`,
+      };
+    }
+
+    // 提取 PID（最后一列）
+    const lines = stdout.trim().split("\n");
+    const pids = new Set();
+
+    for (const line of lines) {
+      const parts = line.trim().split(/\s+/);
+      const pid = parts[parts.length - 1];
+      if (pid && pid !== "0") {
+        pids.add(pid);
+      }
+    }
+
+    if (pids.size === 0) {
+      return {
+        success: false,
+        message: `No valid PID found for port ${port}`,
+      };
+    }
+
+    // 杀死所有相关进程
+    for (const pid of pids) {
+      try {
+        await execAsync(`taskkill /F /PID ${pid}`);
+      } catch (err) {
+        // 进程可能已经结束，忽略错误
+      }
+    }
+
+    return {
+      success: true,
+      message: `Killed process(es) on port ${port}`,
+      pid: parseInt(Array.from(pids)[0]),
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `Failed to kill process on port ${port}: ${error.message}`,
+    };
+  }
+}
+
+/**
+ * Unix 平台（Linux/macOS）杀死端口进程
+ * @private
+ */
+async function killPortUnix(port) {
+  try {
+    // 使用 fuser 查找占用端口的进程（不需要 root 权限）
+    const { stdout } = await execAsync(`fuser ${port}/tcp 2>/dev/null || true`);
+
+    if (!stdout.trim()) {
+      return {
+        success: false,
+        message: `No process found on port ${port}`,
+      };
+    }
+
+    // 提取 PID
+    const pids = stdout
+      .trim()
+      .split(/\s+/)
+      .filter(Boolean)
+      .map((pid) => parseInt(pid));
+
+    if (pids.length === 0) {
+      return {
+        success: false,
+        message: `No valid PID found for port ${port}`,
+      };
+    }
+
+    // 杀死进程
+    await execAsync(`kill -9 ${pids.join(" ")}`);
+
+    return {
+      success: true,
+      message: `Killed process(es) on port ${port}`,
+      pid: pids[0],
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `Failed to kill process on port ${port}: ${error.message}`,
+    };
+  }
+}
+
+module.exports = {
+  isPortOpen,
+  killPort,
+};

package/src/utils/snapshot-utils.js

@@ -0,0 +1,56 @@
+// snapshot-utils.js
+// Snapshot utilities for Electron MCP server
+// Provides page snapshot capture with screenshots and element references
+const { clipboard, nativeImage } = require("electron");
+
+async function captureSnapshot(webContents, options = {}) {
+  const { win_id = 1 } = options;
+
+  const image = await webContents.capturePage();
+  let size = image.getSize(); // 逻辑尺寸
+
+  // 检测是否为 macOS
+  if (process.platform === "darwin") {
+    // 将图片缩小到逻辑尺寸（即原始物理像素的一半）
+    // resize 会返回一个新的 NativeImage 对象
+    const resizedImage = image.resize({
+      width: size.width,
+      height: size.height,
+      quality: "better", // 可选: 'good', 'better', 'best'
+    });
+
+    // 更新数据
+    var finalImage = resizedImage;
+  } else {
+    var finalImage = image;
+  }
+  const pngBuffer = finalImage.toPNG();
+  // 2. 使用 Buffer 重新创建 NativeImage (确保数据完整)
+  const imageToPaste = nativeImage.createFromBuffer(pngBuffer);
+
+  // 3. 写入剪贴板
+  clipboard.writeImage(imageToPaste);
+
+  const screenshotData = finalImage.toPNG().toString("base64");
+  return {
+    content: [
+      {
+        type: "text",
+        text: `Size: ${size.width}x${size.height} (macOS auto-scaled)`,
+      },
+      {
+        type: "text",
+        text: `Image has write to clipboard`,
+      },
+      {
+        type: "image",
+        data: screenshotData,
+        mimeType: "image/png",
+      },
+    ],
+  };
+}
+
+module.exports = {
+  captureSnapshot,
+};