@wenyan-md/mcp 1.0.10 → 2.0.0

package/README.md

@@ -1,5 +1,5 @@
 <div align="center">
-    <img alt = "logo" src="https://media.githubusercontent.com/media/caol64/wenyan-mcp/main/data/wenyan-mcp.png" />
+    <img alt="logo" src="https://media.githubusercontent.com/media/caol64/wenyan-mcp/main/data/wenyan-mcp.png" width="256" />
 </div>
 
 # 文颜 MCP Server
@@ -12,7 +12,7 @@
 
 ## 简介
 
-**文颜（Wenyan）** 是一款多平台 Markdown 排版与发布工具，支持将 Markdown 一键转换并发布至：
+**[文颜（Wenyan）](https://wenyan.yuzhi.tech)** 是一款多平台 Markdown 排版与发布工具，支持将 Markdown 一键转换并发布至：
 
 -   微信公众号
 -   知乎
@@ -21,8 +21,6 @@
 
 文颜的目标是：**让写作者专注内容，而不是排版和平台适配**。
 
-本仓库是 **文颜的 MCP Server 版本**，基于模型上下文协议（Model Context Protocol），旨在让 AI 助手（如 Claude Desktop）具备自动排版和发布公众号文章的能力。
-
 ## 文颜的不同版本
 
 文颜目前提供多种形态，覆盖不同使用场景：
@@ -31,57 +29,50 @@
 -   [跨平台桌面版](https://github.com/caol64/wenyan-pc) - Windows/Linux
 -   [CLI 版本](https://github.com/caol64/wenyan-cli) - 命令行 / CI 自动化发布
 -   👉 [MCP 版本](https://github.com/caol64/wenyan-mcp) - 本项目
--   [核心库](https://github.com/caol64/wenyan-core) - 嵌入 Node / Web 项目
-
-## 功能特性
-
--   列出并选择支持的文章主题
--   使用内置主题对 Markdown 内容排版
--   自动处理并上传图片（本地 / 网络）
--   一键发布文章到微信公众号草稿箱
--   **与 AI 深度集成**：[让 AI 帮你管理公众号的排版和发布](https://babyno.top/posts/2025/06/let-ai-help-you-manage-your-gzh-layout-and-publishing/)
 
-<video src="https://github.com/user-attachments/assets/2c355f76-f313-48a7-9c31-f0f69e5ec207"></video>
+## 文颜 MCP Server 是什么？
 
-> [!TIP]
->
-> 如果与 AI 集成遇到问题，可以参考 [test/list.js](./test/list.js) 和 [test/publish.js](./test/publish.js) 中的完整调用示例。
+简单来说，它打通了“AI 写作”与“公众号发文”的通道。
 
-## 主题效果预览
+基于 MCP 协议，Claude Desktop 等 AI 客户端现在可以直接调用文颜（Wenyan）的排版引擎。写完文章后，不需要再去第三方编辑器里来回复制粘贴，直接让 AI 帮你排版并塞进微信草稿箱。
 
-👉 [内置主题预览](https://yuzhi.tech/docs/wenyan/theme)
+**核心特性：**
 
-文颜内置并适配了多个优秀的 Typora 主题，在此感谢原作者：
+- **绕过排版工具**：AI 生成的 Markdown 直接转成微信富文本并上传，省去中间步骤。
+- **对话式排版**：直接打字跟 AI 说“换个橙色风格主题”，样式自动生效。
+- **不出窗口完成闭环**：在同一个聊天框里，顺滑搞定“想选题 -> 写文章 -> 调排版 -> 存草稿”的所有操作。
 
--   [Orange Heart](https://github.com/evgo2017/typora-theme-orange-heart)
--   [Rainbow](https://github.com/thezbm/typora-theme-rainbow)
--   [Lapis](https://github.com/YiNNx/typora-theme-lapis)
--   [Pie](https://github.com/kevinzhao2233/typora-theme-pie)
--   [Maize](https://github.com/BEATREE/typora-maize-theme)
--   [Purple](https://github.com/hliu202/typora-purple-theme)
--   [物理猫-薄荷](https://github.com/sumruler/typora-theme-phycat)
+**实战演示**：
+*   [让 AI 帮你管理公众号的排版和发布](https://babyno.top/posts/2025/06/let-ai-help-you-manage-your-gzh-layout-and-publishing/)
+*   [Moraya MCP 使用案例：微信公众号全托管](https://github.com/zouwei/moraya/wiki/Moraya-MCP-%E4%BD%BF%E7%94%A8%E6%A1%88%E4%BE%8B%EF%BC%9A%E5%BE%AE%E4%BF%A1%E5%85%AC%E4%BC%97%E5%8F%B7%E5%85%A8%E6%89%98%E7%AE%A1)
 
-## 安装与集成
+## 功能特性
 
-文颜 MCP Server 支持多种运行方式，请根据你的环境选择。
+- 一键发布 Markdown 到微信公众号草稿箱
+- 自动上传本地图片与封面
+- 支持远程 Server 发布（绕过 IP 白名单限制）
+- 内置多套精美排版主题
+- 支持自定义主题
+- 提供标准 MCP Tool 接口
+- 支持 AI 自动调用：
+  - 渲染 Markdown
+  - 主题管理
+  - 发布草稿
 
-### 方式一：npm 安装（推荐）
+## 快速开始
 
-直接安装到本地：
+**安装**
 
 ```bash
 npm install -g @wenyan-md/mcp
 ```
 
-**配置 MCP Client（如 Claude Desktop）：**
-
-在你的 MCP 配置文件中加入以下内容：
+**Claude Desktop 配置 (claude_desktop_config.json)：**：
 
 ```json
 {
   "mcpServers": {
     "wenyan-mcp": {
-      "name": "公众号助手",
       "command": "wenyan-mcp",
       "env": {
         "WECHAT_APP_ID": "your_app_id",
@@ -92,60 +83,103 @@ npm install -g @wenyan-md/mcp
 }
 ```
 
-### 方式二：Docker 运行（推荐）
+## 基本用法
 
-适合部署到服务器环境，或希望环境隔离的用户。
+### 列出主题
 
-**拉取镜像：**
+对AI说
+
+> 目前你可以使用哪些公众号主题？
+
+AI回复：
 
-```bash
-docker pull caol64/wenyan-mcp
 ```
+目前我支持的公众号主题包括以下这些，每个主题都有独特的风格和配色，适用于不同的内容调性：
 
-**配置 MCP Client：**
+1. **Default（默认）**  
+   简洁经典的排版，适合长文阅读，通用性强。
 
-```json
-{
-  "mcpServers": {
-    "wenyan-mcp": {
-      "name": "公众号助手",
-      "command": "docker",
-      "args": [
-        "run",
-        "--rm",
-        "-i",
-        "-v", "/your/host/file/path:/mnt/host-downloads",
-        "-e", "WECHAT_APP_ID=your_app_id",
-        "-e", "WECHAT_APP_SECRET=your_app_secret",
-        "-e", "HOST_FILE_PATH=/your/host/file/path",
-        "caol64/wenyan-mcp"
-      ]
-    }
-  }
-}
+2. **OrangeHeart（橙心）**  
+   暖橙色调，优雅而富有活力，适合情感、生活类内容。
+...
 ```
 
-> **Docker 配置特别说明：**
+### 注册主题
+
+对AI说
+
+> 帮我把这个css(https://wenyan.yuzhi.tech/manhua.css)注册为一个新的公众号主题，名称为：xiuluochang
+
+AI回复：
+
+```
+好的，已成功将 CSS 样式表注册为新的公众号主题，主题名称为：**xiuluochang**。
+
+您现在可以在公众号后台选择该主题进行应用，样式将按照 `https://wenyan.yuzhi.tech/manhua.css` 的定义进行渲染。
+
+如有需要进一步自定义或更换主题，随时告诉我！
+```
+
+### 删除主题
+
+对AI说
+
+> 把名为“xiuluochang”的公众号主题删除
+
+AI回复：
+
+```
+公众号主题 “xiuluochang” 已成功删除。
+```
+
+### 发布文章
+
+对AI说
+
+> 使用phycat主题将这篇文章发布到微信公众号：`./tests/publish.md`
+
+AI回复：
+
+```
+文章已成功发布至微信公众号草稿箱！🎉
+
+- **主题**：phycat  
+- **媒体ID**：xxx
+
+您可登录微信公众号后台，在「草稿箱」中查看并编辑文章，确认无误后即可一键发布。如需进一步排版优化、添加封面或设置摘要，我也可以协助您完成！
+
+是否需要我帮您生成一篇发布文案或封面建议？ 😊
+```
+
+## 概念
+
+### 环境变量配置
+
+> [!IMPORTANT]
 >
-> *   **挂载目录 (`-v`)**：必须将宿主机的文件/图片目录挂载到容器内的 `/mnt/host-downloads`。
-> *   **环境变量 (`HOST_FILE_PATH`)**：必须与宿主机挂载的文件/图片目录路径保持一致。
-> *   **原理**：你的 Markdown 文件/文章内所引用的本地图片应放置在该目录中，Docker 会自动将其映射，使容器能够读取并上传。
+> 请确保 MCP 启动时已配置如下环境变量，否则上传接口将调用失败。
 
-## 环境变量说明
+-   `WECHAT_APP_ID`
+-   `WECHAT_APP_SECRET`
 
-无论使用哪种方式，都需要配置以下环境变量以连接微信公众号平台：
+### 微信公众号 IP 白名单
 
--   `WECHAT_APP_ID`：微信公众号平台的 App ID
--   `WECHAT_APP_SECRET`：微信公众号平台的 App Secret
+> [!IMPORTANT]
+>
+> 请确保运行文颜的机器 IP 已加入微信公众号后台的 IP 白名单，否则上传接口将调用失败。
+
+配置说明文档：[https://yuzhi.tech/docs/wenyan/upload](https://yuzhi.tech/docs/wenyan/upload)
 
-## Markdown Frontmatter 说明（必读）
+### 文章格式
 
-为了正确上传文章，每篇 Markdown 顶部需要包含 frontmatter：
+为了正确上传文章，每篇 Markdown 顶部需要包含一段 `frontmatter`：
 
 ```md
 ---
 title: 在本地跑一个大语言模型(2) - 给模型提供外部知识库
 cover: /Users/xxx/image.jpg
+author: xxx
+source_url: http://
 ---
 ```
 
@@ -154,41 +188,100 @@ cover: /Users/xxx/image.jpg
 -   `title` 文章标题（必填）
 -   `cover` 文章封面
     -   本地路径或网络图片
-    -   如果正文有至少一张图片，可省略，此时将使用其中一张作为封面
-    -   如果正文无图片，则必须提供 cover
+    -   如果正文中已有图片，可省略
+-   `author` 文章作者
+-   `source_url` 原文地址
 
-## 关于图片自动上传
+**[示例文章](tests/publish.md)**
 
-支持以下图片来源：
+### 文内图片和文章封面
 
--   本地路径（如：`/Users/lei/Downloads/result_image.jpg`）
--   网络路径（如：`https://example.com/image.jpg`）
+把文章发布到公众号之前，文颜会按照微信要求自动处理文章内的所有图片，将其上传到公众号素材库。目前文颜对于以下图片都能很好的支持：
 
-## 微信公众号 IP 白名单
+- 本地硬盘绝对路径（如：`/Users/xxx/image.jpg`）
+- 网络路径（如：`https://example.com/image.jpg`）
+- 当前文章的相对路径（如：`./assets/image.png`）
 
-> [!IMPORTANT]
->
-> 请确保运行文颜 MCP Server 的机器 IP 已加入微信公众号后台的 IP 白名单，否则上传接口将调用失败。
+## Server 模式
 
-配置说明文档：[https://yuzhi.tech/docs/wenyan/upload](https://yuzhi.tech/docs/wenyan/upload)
+相较于纯本地运行的**本地模式（Stdio Mode）**，`wenyan-mcp`还提供了 **远程客户端模式（Client–Server Mode）**。两种模式运行效果完全一致，你可以根据运行环境和网络条件选择最合适的方式。
 
-## 示例文章格式
+在本地模式下，MCP 直接调用微信公众号 API 完成图片上传和草稿发布。
 
-```md
----
-title: 在本地跑一个大语言模型(2) - 给模型提供外部知识库
-cover: /Users/lei/Downloads/result_image.jpg
----
+```mermaid
+flowchart LR
+    MCP[Wenyan MCP] --> Wechat[公众号 API]
+```
+
+在远程客户端模式下，MCP 作为客户端，将发布请求发送到部署在云服务器上的 Wenyan Server，由 Server 完成微信公众号 API 调用。
+
+```mermaid
+flowchart LR
+    MCP[Wenyan MCP] --> Server[Wenyan Server] --> Wechat[公众号 API]
+```
+
+**适用于：**
+
+* 无本地固定 IP，需频繁添加IP 白名单的用户
+* 需团队协作的用户
+* 支持 CI/CD 自动发布
+* 支持 AI Agent 自动发布
+
+**[Server 模式部署](https://github.com/caol64/wenyan-cli/blob/main/docs/server.md)**
+
+**Claude Desktop 配置：**：
+
+```json
+{
+  "mcpServers": {
+    "wenyan-mcp": {
+      "command": "wenyan-mcp",
+      "args": ["--server", "https://api.example.com", "--api-key", "your-api-key"]
+      "env": {
+        "WECHAT_APP_ID": "your_app_id",
+        "WECHAT_APP_SECRET": "your_app_secret"
+      }
+    }
+  }
+}
+```
+
+## Docker 部署
 
-在[上一篇文章](https://babyno.top/posts/2024/02/running-a-large-language-model-locally/)中，我们展示了如何在本地运行大型语言模型。本篇将介绍如何让模型从外部知识库中检索定制数据，提升答题准确率，让它看起来更“智能”。
+适合不希望安装 Node.js 环境的用户。
 
-## 准备模型
+```bash
+docker pull caol64/wenyan-mcp:latest
+```
 
-访问 `Ollama` 的模型页面，搜索 `qwen`，我们使用支持中文语义的“[通义千问](https://ollama.com/library/qwen:7b)”模型进行实验。
+* **Claude Desktop 配置：**：
 
-![](https://mmbiz.qpic.cn/mmbiz_jpg/Jsq9IicjScDVUjkPc6O22ZMvmaZUzof5bLDjMyLg2HeAXd0icTvlqtL7oiarSlOicTtiaiacIxpVOV1EeMKl96PhRPPw/640?wx_fmt=jpeg)
+```json
+{
+  "mcpServers": {
+    "wenyan-mcp": {
+      "command": "docker",
+      "args": [
+        "run",
+        "--rm",
+        "-i",
+        "-v", "/your/host/file/path:/mnt/host-downloads",
+        "-e", "WECHAT_APP_ID=your_app_id",
+        "-e", "WECHAT_APP_SECRET=your_app_secret",
+        "-e", "HOST_FILE_PATH=/your/host/file/path",
+        "caol64/wenyan-mcp"
+      ]
+    }
+  }
+}
 ```
 
+> **Docker 配置特别说明：**
+>
+> *   **挂载目录 (`-v`)**：必须将宿主机的文件/图片目录挂载到容器内的 `/mnt/host-downloads`。
+> *   **环境变量 (`HOST_FILE_PATH`)**：必须与宿主机挂载的文件/图片目录路径保持一致。
+> *   **原理**：你的 Markdown 文件/文章内所引用的本地图片应放置在该目录中，Docker 会自动将其映射，使容器能够读取并上传。
+
 ## 如何调试
 
 推荐使用官方 Inspector 进行调试：

package/dist/index.js

@@ -1,145 +1,61 @@
 #!/usr/bin/env node
-/**
- * This is a template MCP server that implements a simple notes system.
- * It demonstrates core MCP concepts like resources and tools by allowing:
- * - Listing notes as resources
- * - Reading individual notes
- * - Creating new notes via a tool
- * - Summarizing all notes via a prompt
- */
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
-import { getGzhContent } from "@wenyan-md/core/wrapper";
-import { publishToDraft } from "@wenyan-md/core/publish";
-import { themes } from "@wenyan-md/core/theme";
-import { getNormalizeFilePath } from "./utils.js";
-import fs from "node:fs/promises";
-import path from "node:path";
+import { createServer } from "./mcpServer.js";
+import { globalStates } from "./utils.js";
 /**
- * Create an MCP server with capabilities for resources (to list/read notes),
- * tools (to create new notes), and prompts (to summarize notes).
- */
-const server = new Server({
-    name: "wenyan-mcp",
-    version: "0.1.0",
-}, {
-    capabilities: {
-        resources: {},
-        tools: {},
-        prompts: {},
-        // logging: {},
-    },
-});
-/**
- * Handler that lists available tools.
- * Exposes a single "publish_article" tool that lets clients publish new article.
+ * Start the server using stdio transport.
  */
-server.setRequestHandler(ListToolsRequestSchema, async () => {
-    return {
-        tools: [
-            {
-                name: "publish_article",
-                description: "Format a Markdown article using a selected theme and publish it to '微信公众号'.",
-                inputSchema: {
-                    type: "object",
-                    properties: {
-                        content: {
-                            type: "string",
-                            description: "The Markdown text to publish. REQUIRED if 'file' is not provided. Preserves frontmatter if present.",
-                        },
-                        file: {
-                            type: "string",
-                            description: "The path to the Markdown file (absolute or relative). REQUIRED if 'content' is not provided.",
-                        },
-                        theme_id: {
-                            type: "string",
-                            description: "ID of the theme to use (e.g., default, orangeheart, rainbow, lapis, pie, maize, purple, phycat).",
-                        },
-                    },
-                },
-            },
-            {
-                name: "list_themes",
-                description: "List the themes compatible with the 'publish_article' tool to publish an article to '微信公众号'.",
-                inputSchema: {
-                    type: "object",
-                    properties: {},
-                },
-            },
-        ],
-    };
-});
+async function mainStdio() {
+    const server = createServer();
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    console.error("[Init] Wenyan MCP server started successfully and listening for requests...");
+}
 /**
- * Handler for the publish_article tool.
- * Publish a new article with the provided title and content, and returns success message.
+ * Main entry point: parse command line arguments and start appropriate transport.
  */
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-    if (request.params.name === "publish_article") {
-        // server.sendLoggingMessage({
-        //     level: "debug",
-        //     data: JSON.stringify(request.params.arguments),
-        // });
-        const contentArg = request.params.arguments?.content;
-        const fileArg = request.params.arguments?.file;
-        if (!contentArg && !fileArg) {
-            throw new Error("You must provide either 'content' or 'file' to publish an article.");
-        }
-        let content = String(contentArg || "");
-        const file = String(fileArg || "");
-        const themeId = String(request.params.arguments?.theme_id || "");
-        let absoluteDirPath;
-        if (!content && file) {
-            const normalizePath = getNormalizeFilePath(file);
-            content = await fs.readFile(normalizePath, "utf-8");
-            if (!content) {
-                throw new Error("Can't read content from the specified file.");
-            }
-            absoluteDirPath = path.dirname(normalizePath);
-        }
-        const gzhContent = await getGzhContent(content, themeId, "solarized-light", true, true);
-        if (!gzhContent.title) {
-            throw new Error("Can't extract a valid title from the frontmatter.");
+async function main() {
+    const args = process.argv.slice(2);
+    const isClientMode = args.includes("--server");
+    if (isClientMode) {
+        globalStates.isClientMode = true;
+        console.error("[Init] Starting Wenyan MCP server in remote client mode...");
+        const serverIndex = args.indexOf("--server");
+        if (serverIndex !== -1 && args[serverIndex + 1] && !args[serverIndex + 1].startsWith("--")) {
+            globalStates.serverUrl = args[serverIndex + 1];
         }
-        if (!gzhContent.cover) {
-            throw new Error("Can't extract a valid cover from the frontmatter or article.");
+        const apiKeyIndex = args.indexOf("--api-key");
+        if (apiKeyIndex !== -1 && args[apiKeyIndex + 1] && !args[apiKeyIndex + 1].startsWith("--")) {
+            globalStates.apiKey = args[apiKeyIndex + 1];
         }
-        const response = await publishToDraft(gzhContent.title, gzhContent.content, gzhContent.cover, {
-            relativePath: absoluteDirPath,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Your article was successfully published to '公众号草稿箱'. The media ID is ${response.media_id}.`,
-                },
-            ],
-        };
     }
-    else if (request.params.name === "list_themes") {
-        const themeResources = Object.entries(themes).map(([id, theme]) => ({
-            type: "text",
-            text: JSON.stringify({
-                id: theme.id,
-                name: theme.name,
-                description: theme.description,
-            }),
-        }));
-        return {
-            content: themeResources,
-        };
+    else {
+        console.error("[Init] Starting Wenyan MCP server in local mode...");
     }
-    throw new Error("Unknown tool");
-});
-/**
- * Start the server using stdio transport.
- * This allows the server to communicate via standard input/output streams.
- */
-async function main() {
-    const transport = new StdioServerTransport();
-    await server.connect(transport);
+    await mainStdio();
 }
+// ==========================================
+// 全局异常与信号处理 (Graceful Shutdown)
+// ==========================================
 main().catch((error) => {
-    console.error("Server error:", error);
+    // 必须使用 console.error 输出到 stderr，防止污染 MCP 的 JSON-RPC stdout 通道
+    console.error("[Fatal Error] Server initialization failed:", error.message);
+    process.exit(1);
+});
+process.on("unhandledRejection", (reason) => {
+    console.error("[Unhandled Rejection] An unexpected error occurred:", reason?.message || reason);
+    // 注意：在 MCP 中通常不建议直接 exit，记录日志即可，让 Host 决定是否重启
+});
+process.on("uncaughtException", (error) => {
+    console.error("[Uncaught Exception] Critical error:", error.message);
     process.exit(1);
 });
+// 3. 优雅退出
+process.on("SIGINT", () => {
+    console.error("[Shutdown] Received SIGINT, exiting...");
+    process.exit(0);
+});
+process.on("SIGTERM", () => {
+    console.error("[Shutdown] Received SIGTERM, exiting...");
+    process.exit(0);
+});

package/dist/mcpServer.js

@@ -0,0 +1,93 @@
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { LIST_THEMES_SCHEMA, REGISTER_THEME_SCHEMA, REMOVE_THEME_SCHEMA } from "./theme.js";
+import { PUBLISH_ARTICLE_SCHEMA, publishArticle } from "./publish.js";
+import pkg from "../package.json" with { type: "json" };
+import { buildMcpResponse, globalStates } from "./utils.js";
+import { addTheme, listThemes, removeTheme } from "@wenyan-md/core/wrapper";
+/**
+ * Create and configure an MCP server instance.
+ */
+export function createServer() {
+    const server = new Server({
+        name: "wenyan-mcp",
+        version: pkg.version,
+    }, {
+        capabilities: {
+            resources: {},
+            tools: {},
+            prompts: {},
+            // logging: {},
+        },
+    });
+    /**
+     * Handler that lists available tools.
+     * Exposes a single "publish_article" tool that lets clients publish new article.
+     */
+    server.setRequestHandler(ListToolsRequestSchema, async () => {
+        return {
+            tools: [PUBLISH_ARTICLE_SCHEMA, LIST_THEMES_SCHEMA, REGISTER_THEME_SCHEMA, REMOVE_THEME_SCHEMA],
+        };
+    });
+    /**
+     * Handler for the publish_article tool.
+     * Publish a new article with the provided title and content, and returns success message.
+     */
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        try {
+            if (request.params.name === "publish_article") {
+                if (globalStates.isClientMode && !globalStates.serverUrl) {
+                    throw new Error("Missing server URL. Usage: --server <server_url>");
+                }
+                const args = request.params.arguments || {};
+                const content = String(args.content || "");
+                const contentUrl = String(args.content_url || "");
+                const file = String(args.file || "");
+                const themeId = String(args.theme_id || "");
+                return await publishArticle(contentUrl, file, content, themeId, pkg.version);
+            }
+            else if (request.params.name === "list_themes") {
+                const themes = await listThemes();
+                const builtinThemes = themes.filter((theme) => theme.isBuiltin);
+                const customThemes = themes.filter((theme) => !theme.isBuiltin);
+                return {
+                    content: [
+                        ...builtinThemes.map((theme) => ({
+                            type: "text",
+                            text: JSON.stringify({
+                                id: theme.id,
+                                name: theme.name,
+                                description: theme.description,
+                            }),
+                        })),
+                        ...customThemes.map((theme) => ({
+                            type: "text",
+                            text: JSON.stringify({
+                                id: theme.id,
+                                name: theme.name ?? theme.id,
+                                description: theme.description ?? "自定义主题，暂无描述。",
+                            }),
+                        })),
+                    ],
+                };
+            }
+            else if (request.params.name === "register_theme") {
+                const name = String(request.params.arguments?.name || "");
+                const path = String(request.params.arguments?.path || "");
+                await addTheme(name, path);
+                return buildMcpResponse(`Theme "${name}" has been added successfully.`);
+            }
+            else if (request.params.name === "remove_theme") {
+                const name = String(request.params.arguments?.name || "");
+                await removeTheme(name);
+                return buildMcpResponse(`Theme "${name}" has been removed successfully.`);
+            }
+            throw new Error("Unknown tool");
+        }
+        catch (error) {
+            console.error(`[MCP Tool Error] (${request.params.name}):`, error.message);
+            return buildMcpResponse(`执行工具失败: ${error.message}`);
+        }
+    });
+    return server;
+}

package/dist/publish.js

@@ -0,0 +1,48 @@
+import { renderAndPublish, renderAndPublishToServer } from "@wenyan-md/core/wrapper";
+import { buildMcpResponse, getInputContent, globalStates } from "./utils.js";
+export const PUBLISH_ARTICLE_SCHEMA = {
+    name: "publish_article",
+    description: "Format a Markdown article using a selected theme and publish it to '微信公众号'.",
+    inputSchema: {
+        type: "object",
+        properties: {
+            content: {
+                type: "string",
+                description: "The Markdown text to publish. REQUIRED if 'file' or 'content_url' is not provided. DO INCLUDE frontmatter if present.",
+            },
+            content_url: {
+                type: "string",
+                description: "A URL (e.g. GitHub raw link) to a Markdown file. Preferred over 'content' for large files to save tokens.",
+            },
+            file: {
+                type: "string",
+                description: "The local path (absolute or relative) to a Markdown file. Preferred over 'content' for large files to save tokens.",
+            },
+            theme_id: {
+                type: "string",
+                description: "ID of the theme to use (e.g., default, orangeheart, rainbow, lapis, pie, maize, purple, phycat).",
+            },
+        },
+    },
+};
+export async function publishArticle(contentUrl, file, content, themeId, clientVersion) {
+    let mediaId = "";
+    const publishOptions = {
+        file: file ? file : contentUrl,
+        theme: themeId,
+        highlight: "solarized-light",
+        macStyle: true,
+        footnote: true,
+        server: globalStates.serverUrl,
+        apiKey: globalStates.apiKey,
+        clientVersion,
+        disableStdin: true,
+    };
+    if (globalStates.isClientMode) {
+        mediaId = await renderAndPublishToServer(content, publishOptions, getInputContent);
+    }
+    else {
+        mediaId = await renderAndPublish(content, publishOptions, getInputContent);
+    }
+    return buildMcpResponse(`Your article was successfully published to '公众号草稿箱'. The media ID is ${mediaId}.`);
+}

package/dist/theme.js

@@ -0,0 +1,38 @@
+export const LIST_THEMES_SCHEMA = {
+    name: "list_themes",
+    description: "List the themes compatible with the 'publish_article' tool to publish an article to '微信公众号'.",
+    inputSchema: {
+        type: "object",
+        properties: {},
+    },
+};
+export const REGISTER_THEME_SCHEMA = {
+    name: "register_theme",
+    description: "Register a custom theme compatible with the 'publish_article' tool to publish an article to '微信公众号'.",
+    inputSchema: {
+        type: "object",
+        properties: {
+            name: {
+                type: "string",
+                description: "Name of the new custom theme.",
+            },
+            path: {
+                type: "string",
+                description: "Path to the new custom theme CSS file. It could be a path to a local file or a URL.",
+            },
+        },
+    },
+};
+export const REMOVE_THEME_SCHEMA = {
+    name: "remove_theme",
+    description: "Remove a custom theme compatible with the 'publish_article' tool to publish an article to '微信公众号'.",
+    inputSchema: {
+        type: "object",
+        properties: {
+            name: {
+                type: "string",
+                description: "Name of the custom theme to remove.",
+            },
+        },
+    },
+};

package/dist/types/index.d.ts

@@ -1,10 +1,2 @@
 #!/usr/bin/env node
-/**
- * This is a template MCP server that implements a simple notes system.
- * It demonstrates core MCP concepts like resources and tools by allowing:
- * - Listing notes as resources
- * - Reading individual notes
- * - Creating new notes via a tool
- * - Summarizing all notes via a prompt
- */
 export {};

package/dist/types/mcpServer.d.ts

@@ -0,0 +1,5 @@
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+/**
+ * Create and configure an MCP server instance.
+ */
+export declare function createServer(): Server;

package/dist/types/publish.d.ts

@@ -0,0 +1,31 @@
+export declare const PUBLISH_ARTICLE_SCHEMA: {
+    readonly name: "publish_article";
+    readonly description: "Format a Markdown article using a selected theme and publish it to '微信公众号'.";
+    readonly inputSchema: {
+        readonly type: "object";
+        readonly properties: {
+            readonly content: {
+                readonly type: "string";
+                readonly description: "The Markdown text to publish. REQUIRED if 'file' or 'content_url' is not provided. DO INCLUDE frontmatter if present.";
+            };
+            readonly content_url: {
+                readonly type: "string";
+                readonly description: "A URL (e.g. GitHub raw link) to a Markdown file. Preferred over 'content' for large files to save tokens.";
+            };
+            readonly file: {
+                readonly type: "string";
+                readonly description: "The local path (absolute or relative) to a Markdown file. Preferred over 'content' for large files to save tokens.";
+            };
+            readonly theme_id: {
+                readonly type: "string";
+                readonly description: "ID of the theme to use (e.g., default, orangeheart, rainbow, lapis, pie, maize, purple, phycat).";
+            };
+        };
+    };
+};
+export declare function publishArticle(contentUrl: string, file: string, content: string, themeId: string, clientVersion?: string): Promise<{
+    content: {
+        type: string;
+        text: string;
+    }[];
+}>;

package/dist/types/theme.d.ts

@@ -0,0 +1,38 @@
+export declare const LIST_THEMES_SCHEMA: {
+    readonly name: "list_themes";
+    readonly description: "List the themes compatible with the 'publish_article' tool to publish an article to '微信公众号'.";
+    readonly inputSchema: {
+        readonly type: "object";
+        readonly properties: {};
+    };
+};
+export declare const REGISTER_THEME_SCHEMA: {
+    readonly name: "register_theme";
+    readonly description: "Register a custom theme compatible with the 'publish_article' tool to publish an article to '微信公众号'.";
+    readonly inputSchema: {
+        readonly type: "object";
+        readonly properties: {
+            readonly name: {
+                readonly type: "string";
+                readonly description: "Name of the new custom theme.";
+            };
+            readonly path: {
+                readonly type: "string";
+                readonly description: "Path to the new custom theme CSS file. It could be a path to a local file or a URL.";
+            };
+        };
+    };
+};
+export declare const REMOVE_THEME_SCHEMA: {
+    readonly name: "remove_theme";
+    readonly description: "Remove a custom theme compatible with the 'publish_article' tool to publish an article to '微信公众号'.";
+    readonly inputSchema: {
+        readonly type: "object";
+        readonly properties: {
+            readonly name: {
+                readonly type: "string";
+                readonly description: "Name of the custom theme to remove.";
+            };
+        };
+    };
+};

package/dist/types/types.d.ts

@@ -0,0 +1,8 @@
+export type OutputType = "text";
+export interface OutputObject {
+    type: OutputType;
+    text: string;
+}
+export interface ResponseObject {
+    content: OutputObject[];
+}

package/dist/types/utils.d.ts

@@ -1 +1,23 @@
-export declare function getNormalizeFilePath(inputPath: string): string;
+export declare function buildMcpResponse(content: string): {
+    content: {
+        type: string;
+        text: string;
+    }[];
+};
+declare class GlobalStates {
+    private _isClientMode;
+    private _serverUrl?;
+    private _apiKey?;
+    get isClientMode(): boolean;
+    set isClientMode(value: boolean);
+    get serverUrl(): string | undefined;
+    set serverUrl(value: string | undefined);
+    get apiKey(): string | undefined;
+    set apiKey(value: string | undefined);
+}
+export declare const globalStates: GlobalStates;
+export declare function getInputContent(inputContent?: string, file?: string): Promise<{
+    content: string;
+    absoluteDirPath: string | undefined;
+}>;
+export {};

package/dist/types.js

@@ -0,0 +1 @@
+export {};

package/dist/utils.js

@@ -1,27 +1,51 @@
 import path from "node:path";
-/**
- * 路径标准化工具函数
- * 将 Windows 的反斜杠 \ 转换为正斜杠 /，并去除末尾斜杠
- * 目的：在 Linux 容器内也能正确处理 Windows 路径字符串
- */
-function normalizePath(p) {
-    return p.replace(/\\/g, "/").replace(/\/+$/, "");
+import fs from "node:fs/promises";
+import { getNormalizeFilePath } from "@wenyan-md/core/wrapper";
+export function buildMcpResponse(content) {
+    return {
+        content: [
+            {
+                type: "text",
+                text: content,
+            },
+        ],
+    };
 }
-export function getNormalizeFilePath(inputPath) {
-    const isContainer = !!process.env.CONTAINERIZED;
-    if (isContainer) {
-        const hostFilePath = normalizePath(process.env.HOST_FILE_PATH || "");
-        const containerFilePath = normalizePath(process.env.CONTAINER_FILE_PATH || "/mnt/host-downloads");
-        let relativePart = normalizePath(inputPath);
-        if (relativePart.startsWith(hostFilePath)) {
-            relativePart = relativePart.slice(hostFilePath.length);
-        }
-        if (!relativePart.startsWith("/")) {
-            relativePart = "/" + relativePart;
-        }
-        return containerFilePath + relativePart;
-    }
-    else {
-        return path.resolve(inputPath);
+class GlobalStates {
+    _isClientMode = false;
+    _serverUrl;
+    _apiKey;
+    get isClientMode() {
+        return this._isClientMode;
     }
+    set isClientMode(value) {
+        this._isClientMode = value;
+    }
+    get serverUrl() {
+        return this._serverUrl;
+    }
+    set serverUrl(value) {
+        this._serverUrl = value;
+    }
+    get apiKey() {
+        return this._apiKey;
+    }
+    set apiKey(value) {
+        this._apiKey = value;
+    }
+}
+export const globalStates = new GlobalStates();
+export async function getInputContent(inputContent, file) {
+    let absoluteDirPath = undefined;
+    // 2. 尝试从文件读取
+    if (!inputContent && file) {
+        const normalizePath = getNormalizeFilePath(file);
+        inputContent = await fs.readFile(normalizePath, "utf-8");
+        absoluteDirPath = path.dirname(normalizePath);
+    }
+    // 3. 校验输入
+    if (!inputContent) {
+        throw new Error("missing input-content (no argument, no stdin, and no file).");
+    }
+    return { content: inputContent, absoluteDirPath };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wenyan-md/mcp",
-  "version": "1.0.10",
+  "version": "2.0.0",
   "description": "MCP server for Wenyan, a Markdown formatting tool that allows AI assistants to apply elegant built-in themes and publish articles directly to 微信公众号.",
   "author": "Lei <caol64@gmail.com> (https://github.com/caol64)",
   "license": "Apache-2.0",
@@ -31,9 +31,11 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "0.6.0",
-    "@wenyan-md/core": "^1.0.17"
+    "@wenyan-md/core": "^2.0.8",
+    "zod": "^4.3.6"
   },
   "devDependencies": {
+    "@modelcontextprotocol/inspector": "^0.21.1",
     "@types/node": "^24.3.0",
     "dotenv-cli": "^10.0.0",
     "openai": "^6.16.0",
@@ -41,9 +43,11 @@
   },
   "scripts": {
     "build": "tsc",
-    "inspector": "pnpm build && dotenv -e .env.test -- node ./run-inspector.js",
-    "test:list": "pnpm build && dotenv -e .env.test -- node ./test/list.js",
-    "test:publish": "pnpm build && dotenv -e .env.test -- node ./test/publish.js",
+    "inspector": "pnpm build && node ./run-inspector.js",
+    "test:list": "pnpm build && dotenv -e .env.test -- node ./tests/list.js",
+    "test:publish": "pnpm build && dotenv -e .env.test -- node ./tests/publish.js",
+    "test:register": "pnpm build && dotenv -e .env.test -- node ./tests/registerTheme.js",
+    "test:remove": "pnpm build && dotenv -e .env.test -- node ./tests/removeTheme.js",
     "upgrade:core": "pnpm update @wenyan-md/core"
   }
 }