wanzi-mcp 0.2.0__tar.gz

wanzi_mcp-0.2.0/.gitignore

@@ -0,0 +1,11 @@
+__pycache__/
+*.py[cod]
+*.egg-info/
+dist/
+build/
+.eggs/
+*.egg
+.venv/
+venv/
+.env
+*.db

wanzi_mcp-0.2.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: wanzi-mcp
+Version: 0.2.0
+Summary: 丸子MCP - Cursor AI 聊天桥接工具
+Requires-Python: >=3.10
+Requires-Dist: click>=8.0.0
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: pywebview>=5.0.0
+Requires-Dist: starlette>=0.37.0
+Requires-Dist: uvicorn>=0.30.0

wanzi_mcp-0.2.0/pyproject.toml

@@ -0,0 +1,19 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "wanzi-mcp"
+version = "0.2.0"
+description = "丸子MCP - Cursor AI 聊天桥接工具"
+requires-python = ">=3.10"
+dependencies = [
+    "mcp>=1.0.0",
+    "uvicorn>=0.30.0",
+    "starlette>=0.37.0",
+    "click>=8.0.0",
+    "pywebview>=5.0.0",
+]
+
+[project.scripts]
+wanzi = "wanzi_mcp.__main__:cli"

wanzi_mcp-0.2.0/src/wanzi_mcp/__init__.py

@@ -0,0 +1,3 @@
+"""丸子MCP - Cursor AI 聊天桥接工具"""
+
+__version__ = "0.1.0"

wanzi_mcp-0.2.0/src/wanzi_mcp/__main__.py

@@ -0,0 +1,86 @@
+"""CLI 入口：wanzi / wanzi serve / wanzi gui / wanzi install"""
+
+from __future__ import annotations
+
+import threading
+import time
+
+import click
+
+
+@click.group(invoke_without_command=True)
+@click.option("--port", default=8765, help="服务端口", show_default=True)
+@click.pass_context
+def cli(ctx, port: int):
+    """丸子 MCP — Cursor AI 聊天桥接工具"""
+    ctx.ensure_object(dict)
+    ctx.obj["port"] = port
+    if ctx.invoked_subcommand is None:
+        _start_all(port)
+
+
+@cli.command()
+@click.pass_context
+def serve(ctx):
+    """只启动后端服务"""
+    _start_server(ctx.obj["port"])
+
+
+@cli.command()
+@click.pass_context
+def gui(ctx):
+    """只启动 GUI 窗口（需后端已运行）"""
+    from .gui.app import start_gui
+
+    start_gui(ctx.obj["port"])
+
+
+@cli.command()
+@click.pass_context
+def install(ctx):
+    """写入 ~/.cursor/mcp.json 注册 MCP"""
+    from .installer import install_to_cursor
+
+    install_to_cursor(ctx.obj["port"])
+
+
+def _free_port(port: int):
+    from .server_ctl import kill_port_occupants
+
+    killed = kill_port_occupants(port)
+    if killed:
+        click.echo(f"已释放端口 {port}（终止 PID: {killed}）")
+        time.sleep(0.5)
+
+
+def _start_server(port: int):
+    import uvicorn
+    from .server import app
+
+    _free_port(port)
+    click.echo(f"丸子 MCP 后端启动: http://localhost:{port}")
+    uvicorn.run(app, host="0.0.0.0", port=port, log_level="info")
+
+
+def _run_uvicorn(port: int):
+    import uvicorn
+    from .server import app
+
+    uvicorn.run(app, host="0.0.0.0", port=port, log_level="info")
+
+
+def _start_all(port: int):
+    _free_port(port)
+    click.echo(f"丸子 MCP 后端启动: http://localhost:{port}")
+    t = threading.Thread(target=_run_uvicorn, args=(port,), daemon=True)
+    t.start()
+    time.sleep(1)
+
+    from .gui.app import start_gui
+
+    click.echo("正在打开 GUI 窗口…")
+    start_gui(port)
+
+
+if __name__ == "__main__":
+    cli()

wanzi_mcp-0.2.0/src/wanzi_mcp/gui/app.py

@@ -0,0 +1,17 @@
+"""pywebview GUI 窗口：包裹本地 Web 页面。"""
+
+from __future__ import annotations
+
+import webview
+
+
+def start_gui(port: int = 8765):
+    url = f"http://localhost:{port}/"
+    webview.create_window(
+        "丸子 MCP",
+        url,
+        width=920,
+        height=660,
+        min_size=(640, 400),
+    )
+    webview.start()

wanzi_mcp-0.2.0/src/wanzi_mcp/gui/static/app.js

@@ -0,0 +1,530 @@
+(() => {
+  "use strict";
+
+  // ── State ──────────────────────────────────
+  let ws = null;
+  let activeSessionId = null;
+  let sessions = [];              // [{id, title, ...}]
+  let messageCache = {};          // sessionId -> [msg]
+  let ctxTargetId = null;
+  let pendingImages = [];           // [{url, file}] waiting to send
+
+  const $ = (sel) => document.querySelector(sel);
+  const $messages   = $("#messages");
+  const $input      = $("#msg-input");
+  const $btnSend    = $("#btn-send");
+  const $sessionList = $("#session-list");
+  const $chatTitle  = $("#chat-title");
+  const $ctxMenu    = $("#ctx-menu");
+  const $previewBar = $("#preview-bar");
+  const $fileInput  = $("#file-input");
+
+  // ── Markdown Setup ─────────────────────────
+  marked.setOptions({
+    highlight: (code, lang) => {
+      if (lang && hljs.getLanguage(lang))
+        return hljs.highlight(code, { language: lang }).value;
+      return hljs.highlightAuto(code).value;
+    },
+    breaks: true,
+  });
+
+  // ── WebSocket ──────────────────────────────
+  function connect() {
+    const proto = location.protocol === "https:" ? "wss:" : "ws:";
+    ws = new WebSocket(`${proto}//${location.host}/ws`);
+
+    ws.onopen = () => {
+      console.log("[ws] connected");
+      ws.send(JSON.stringify({ type: "list_sessions" }));
+    };
+
+    ws.onmessage = (e) => {
+      const data = JSON.parse(e.data);
+      handleMessage(data);
+    };
+
+    ws.onclose = () => {
+      console.log("[ws] disconnected, reconnecting in 2s…");
+      setTimeout(connect, 2000);
+    };
+  }
+
+  function send(obj) {
+    if (ws && ws.readyState === WebSocket.OPEN)
+      ws.send(JSON.stringify(obj));
+  }
+
+  // ── Message Router ─────────────────────────
+  function handleMessage(data) {
+    switch (data.type) {
+      case "session_list":
+        sessions = data.sessions;
+        renderSidebar();
+        break;
+
+      case "session_detail":
+        messageCache[data.session.id] = data.session.messages;
+        if (data.session.id === activeSessionId) renderMessages();
+        break;
+
+      case "session_created": {
+        if (!sessions.find((s) => s.id === data.session.id)) {
+          sessions.unshift(data.session);
+          renderSidebar();
+        }
+        const createdEl = $sessionList.querySelector(`[data-id="${data.session.id}"]`);
+        if (createdEl) createdEl.classList.add("session-flash");
+        if (!activeSessionId) selectSession(data.session.id);
+        break;
+      }
+
+      case "user_message":
+      case "assistant_message": {
+        const sid = data.session_id;
+        if (!messageCache[sid]) messageCache[sid] = [];
+        messageCache[sid].push(data.message);
+        if (sid === activeSessionId) {
+          appendBubble(data.message);
+          scrollToBottom();
+        }
+        updateSidebarPreview(sid, data.message);
+        if (data.type === "assistant_message") {
+          notifyNewMessage(data.message.content);
+        }
+        break;
+      }
+
+      case "session_waiting":
+        selectSession(data.session_id);
+        break;
+
+      case "session_renamed":
+        updateSessionTitle(data.session_id, data.title);
+        break;
+
+      case "message_deleted": {
+        const sid2 = data.session_id;
+        const mid = data.message_id;
+        if (messageCache[sid2]) {
+          messageCache[sid2] = messageCache[sid2].filter((m) => m.id !== mid);
+        }
+        if (sid2 === activeSessionId) {
+          const el = $messages.querySelector(`[data-msg-id="${mid}"]`);
+          if (el) el.remove();
+          if ((messageCache[sid2] || []).length === 0) {
+            $messages.innerHTML = '<div class="empty-state">发送第一条消息开始聊天</div>';
+          }
+        }
+        updateSidebarPreview(sid2, (messageCache[sid2] || []).at(-1) || { content: "暂无消息" });
+        break;
+      }
+
+      case "session_deleted":
+        sessions = sessions.filter((s) => s.id !== data.session_id);
+        delete messageCache[data.session_id];
+        renderSidebar();
+        if (activeSessionId === data.session_id) {
+          activeSessionId = null;
+          $chatTitle.textContent = "选择或新建一个会话";
+          $messages.innerHTML = '<div class="empty-state">暂无会话</div>';
+        }
+        break;
+    }
+  }
+
+  // ── Sidebar ────────────────────────────────
+  function renderSidebar() {
+    $sessionList.innerHTML = "";
+    if (sessions.length === 0) {
+      $sessionList.innerHTML = '<div class="empty-state" style="padding:40px 0;font-size:13px">暂无会话</div>';
+      return;
+    }
+    sessions.forEach((s) => {
+      const el = document.createElement("div");
+      el.className = "session-item" + (s.id === activeSessionId ? " active" : "");
+      el.dataset.id = s.id;
+
+      const preview = s.last_message
+        ? truncate(s.last_message.content, 30)
+        : "暂无消息";
+
+      el.innerHTML = `
+        <div class="s-title">${esc(s.title || s.id)}</div>
+        <div class="s-preview">${esc(preview)}</div>
+        <button class="s-del" title="删除会话">×</button>`;
+
+      el.addEventListener("click", (e) => {
+        if (e.target.closest(".s-del")) return;
+        selectSession(s.id);
+      });
+      el.addEventListener("dblclick", (e) => {
+        e.stopPropagation();
+        startInlineRename(el, s.id);
+      });
+      el.addEventListener("contextmenu", (e) => showCtxMenu(e, s.id));
+      $sessionList.appendChild(el);
+    });
+  }
+
+  $sessionList.addEventListener("click", (e) => {
+    const btn = e.target.closest(".s-del");
+    if (!btn) return;
+    const item = btn.closest(".session-item");
+    const sid = item?.dataset.id;
+    if (sid) {
+      send({ type: "delete_session", session_id: sid });
+    }
+  });
+
+  function selectSession(sid) {
+    activeSessionId = sid;
+    const s = sessions.find((x) => x.id === sid);
+    $chatTitle.textContent = s ? s.title || sid : sid;
+
+    document.querySelectorAll(".session-item").forEach((el) => {
+      el.classList.toggle("active", el.dataset.id === sid);
+      if (el.dataset.id === sid) el.classList.remove("session-flash");
+    });
+
+    if (messageCache[sid]) {
+      renderMessages();
+    } else {
+      send({ type: "get_session", session_id: sid });
+    }
+    $input.focus();
+  }
+
+  function updateSidebarPreview(sid, msg) {
+    const s = sessions.find((x) => x.id === sid);
+    if (s) s.last_message = msg;
+    const el = $sessionList.querySelector(`[data-id="${sid}"] .s-preview`);
+    if (el) el.textContent = truncate(msg.content, 30);
+  }
+
+  function updateSessionTitle(sid, title) {
+    const s = sessions.find((x) => x.id === sid);
+    if (s) s.title = title;
+    const el = $sessionList.querySelector(`[data-id="${sid}"] .s-title`);
+    if (el) el.textContent = title;
+    if (sid === activeSessionId) $chatTitle.textContent = title;
+  }
+
+  // ── Messages ───────────────────────────────
+  function renderMessages() {
+    $messages.innerHTML = "";
+    const msgs = messageCache[activeSessionId] || [];
+    if (msgs.length === 0) {
+      $messages.innerHTML = '<div class="empty-state">发送第一条消息开始聊天</div>';
+      return;
+    }
+    msgs.forEach((m) => appendBubble(m));
+    scrollToBottom();
+  }
+
+  function appendBubble(msg) {
+    const empty = $messages.querySelector(".empty-state");
+    if (empty) empty.remove();
+
+    const div = document.createElement("div");
+    div.className = `msg ${msg.role}`;
+    div.dataset.msgId = msg.id;
+
+    const avatarText = msg.role === "user" ? "你" : "AI";
+    let rendered = msg.role === "assistant"
+      ? marked.parse(msg.content)
+      : esc(msg.content);
+
+    const imgs = (msg.images || [])
+      .map((u) => `<img class="msg-img" src="${esc(u)}" onclick="window.open('${esc(u)}')">`)
+      .join("");
+
+    const time = formatTime(msg.timestamp);
+
+    div.innerHTML = `
+      <div class="avatar">${avatarText}</div>
+      <div class="bubble">${rendered}${imgs}</div>
+      <span class="msg-time">${time}</span>
+      <button class="msg-del" title="删除">×</button>`;
+
+    $messages.appendChild(div);
+  }
+
+  function scrollToBottom() {
+    requestAnimationFrame(() => {
+      $messages.scrollTop = $messages.scrollHeight;
+    });
+  }
+
+  // ── Delete Message (event delegation) ───
+  $messages.addEventListener("click", (e) => {
+    const btn = e.target.closest(".msg-del");
+    if (!btn) return;
+    const msgEl = btn.closest(".msg");
+    const msgId = msgEl?.dataset.msgId;
+    if (msgId && activeSessionId) {
+      send({ type: "delete_message", session_id: activeSessionId, message_id: msgId });
+    }
+  });
+
+  // ── Image Upload ────────────────────────────
+  async function uploadFile(file) {
+    const fd = new FormData();
+    fd.append("file", file);
+    const res = await fetch("/api/upload", { method: "POST", body: fd });
+    const data = await res.json();
+    if (data.ok) return data.url;
+    throw new Error(data.message);
+  }
+
+  function addPendingImage(file) {
+    const url = URL.createObjectURL(file);
+    pendingImages.push({ url, file });
+    renderPreviews();
+  }
+
+  function removePendingImage(idx) {
+    URL.revokeObjectURL(pendingImages[idx].url);
+    pendingImages.splice(idx, 1);
+    renderPreviews();
+  }
+
+  function renderPreviews() {
+    if (pendingImages.length === 0) {
+      $previewBar.classList.add("hidden");
+      $previewBar.innerHTML = "";
+      return;
+    }
+    $previewBar.classList.remove("hidden");
+    $previewBar.innerHTML = pendingImages
+      .map((p, i) => `<div class="preview-item">
+        <img src="${p.url}">
+        <button class="remove-btn" data-idx="${i}">✕</button>
+      </div>`)
+      .join("");
+    $previewBar.querySelectorAll(".remove-btn").forEach((btn) =>
+      btn.addEventListener("click", () => removePendingImage(+btn.dataset.idx))
+    );
+  }
+
+  $fileInput.addEventListener("change", () => {
+    for (const f of $fileInput.files) addPendingImage(f);
+    $fileInput.value = "";
+  });
+
+  $input.addEventListener("paste", (e) => {
+    const items = e.clipboardData?.items;
+    if (!items) return;
+    for (const item of items) {
+      if (item.type.startsWith("image/")) {
+        e.preventDefault();
+        addPendingImage(item.getAsFile());
+      }
+    }
+  });
+
+  // ── Send ───────────────────────────────────
+  async function sendUserMessage() {
+    const text = $input.value.trim();
+    if ((!text && pendingImages.length === 0) || !activeSessionId) return;
+
+    let imageUrls = [];
+    if (pendingImages.length > 0) {
+      try {
+        imageUrls = await Promise.all(pendingImages.map((p) => uploadFile(p.file)));
+      } catch (err) {
+        alert("图片上传失败：" + err.message);
+        return;
+      }
+      pendingImages.forEach((p) => URL.revokeObjectURL(p.url));
+      pendingImages = [];
+      renderPreviews();
+    }
+
+    send({
+      type: "send_message",
+      session_id: activeSessionId,
+      content: text,
+      images: imageUrls,
+    });
+    $input.value = "";
+    autoResize();
+  }
+
+  $btnSend.addEventListener("click", sendUserMessage);
+  $input.addEventListener("keydown", (e) => {
+    if (e.key === "Enter") {
+      if (e.shiftKey) return;
+      if (e.ctrlKey) {
+        e.preventDefault();
+        const start = $input.selectionStart;
+        const end = $input.selectionEnd;
+        $input.value = $input.value.substring(0, start) + "\n" + $input.value.substring(end);
+        $input.selectionStart = $input.selectionEnd = start + 1;
+        autoResize();
+        return;
+      }
+      e.preventDefault();
+      sendUserMessage();
+    }
+  });
+
+  // auto-resize textarea
+  function autoResize() {
+    $input.style.height = "auto";
+    $input.style.height = Math.min($input.scrollHeight, 120) + "px";
+  }
+  $input.addEventListener("input", autoResize);
+
+  // ── Install to Cursor ──────────────────────
+  $("#btn-install").addEventListener("click", async () => {
+    const port = location.port || "8765";
+    try {
+      const res = await fetch(`/api/install?port=${port}`, { method: "POST" });
+      const data = await res.json();
+      alert(data.ok ? "✓ 已自动写入 ~/.cursor/mcp.json" : "✗ " + data.message);
+    } catch (e) {
+      alert("写入失败：" + e.message);
+    }
+  });
+
+  // ── Inline Rename ─────────────────────────
+  function startInlineRename(el, sid) {
+    const titleEl = el.querySelector(".s-title");
+    if (!titleEl) return;
+    const oldTitle = titleEl.textContent;
+
+    const input = document.createElement("input");
+    input.type = "text";
+    input.value = oldTitle;
+    input.className = "rename-input";
+    titleEl.replaceWith(input);
+    input.focus();
+    input.select();
+
+    let committed = false;
+    function finish() {
+      if (committed) return;
+      committed = true;
+      const newTitle = input.value.trim() || oldTitle;
+      const span = document.createElement("div");
+      span.className = "s-title";
+      span.textContent = newTitle;
+      input.replaceWith(span);
+      if (newTitle !== oldTitle) {
+        send({ type: "rename_session", session_id: sid, title: newTitle });
+      }
+    }
+
+    input.addEventListener("blur", finish);
+    input.addEventListener("keydown", (ev) => {
+      if (ev.key === "Enter") { ev.preventDefault(); input.blur(); }
+      if (ev.key === "Escape") { input.value = oldTitle; input.blur(); }
+    });
+  }
+
+  // ── Context Menu ───────────────────────────
+  function showCtxMenu(e, sid) {
+    e.preventDefault();
+    ctxTargetId = sid;
+    $ctxMenu.style.left = e.clientX + "px";
+    $ctxMenu.style.top = e.clientY + "px";
+    $ctxMenu.classList.remove("hidden");
+  }
+
+  document.addEventListener("click", () => $ctxMenu.classList.add("hidden"));
+
+  $ctxMenu.querySelectorAll(".ctx-item").forEach((el) => {
+    el.addEventListener("click", () => {
+      const action = el.dataset.action;
+      if (!ctxTargetId) return;
+
+      if (action === "rename") {
+        const title = prompt("新标题：");
+        if (title) send({ type: "rename_session", session_id: ctxTargetId, title });
+      } else if (action === "delete") {
+        if (confirm("确定删除此会话？"))
+          send({ type: "delete_session", session_id: ctxTargetId });
+      }
+      ctxTargetId = null;
+    });
+  });
+
+  // ── Notifications ──────────────────────────
+  let notifGranted = false;
+  if ("Notification" in window) {
+    Notification.requestPermission().then((p) => { notifGranted = p === "granted"; });
+  }
+
+  let titleTimer = null;
+  const origTitle = document.title;
+
+  function notifyNewMessage(text) {
+    const preview = truncate(text || "新消息", 50);
+
+    if (notifGranted && document.hidden) {
+      new Notification("丸子 MCP", { body: preview, icon: "" });
+    }
+
+    if (document.hidden) {
+      clearInterval(titleTimer);
+      let flash = true;
+      titleTimer = setInterval(() => {
+        document.title = flash ? `💬 ${preview}` : origTitle;
+        flash = !flash;
+      }, 800);
+    }
+  }
+
+  document.addEventListener("visibilitychange", () => {
+    if (!document.hidden && titleTimer) {
+      clearInterval(titleTimer);
+      titleTimer = null;
+      document.title = origTitle;
+    }
+  });
+
+  // ── Helpers ────────────────────────────────
+  function esc(str) {
+    const d = document.createElement("div");
+    d.textContent = str;
+    return d.innerHTML;
+  }
+
+  function truncate(s, n) {
+    return s.length > n ? s.slice(0, n) + "…" : s;
+  }
+
+  function formatTime(iso) {
+    try {
+      const d = new Date(iso);
+      return d.toLocaleTimeString("zh-CN", { hour: "2-digit", minute: "2-digit" });
+    } catch {
+      return "";
+    }
+  }
+
+  // ── Status Polling ─────────────────────────
+  const $statusDot  = $("#status-dot");
+  const $statusText = $("#status-text");
+
+  async function pollStatus() {
+    try {
+      const res = await fetch("/api/status");
+      const data = await res.json();
+      $statusDot.className = "dot online";
+      const parts = [`${data.sse_connections} 个 Cursor 连接`];
+      if (data.sessions > 0) parts.push(`${data.sessions} 个会话`);
+      $statusText.textContent = parts.join(" · ");
+    } catch {
+      $statusDot.className = "dot offline";
+      $statusText.textContent = "服务已断开";
+    }
+  }
+
+  setInterval(pollStatus, 3000);
+  pollStatus();
+
+  // ── Boot ───────────────────────────────────
+  connect();
+})();