@wenyan-md/mcp 1.0.9 → 1.0.11

package/README.md

@@ -21,8 +21,6 @@
 
 文颜的目标是：**让写作者专注内容，而不是排版和平台适配**。
 
-本仓库是 **文颜的 MCP Server 版本**，基于模型上下文协议（Model Context Protocol），旨在让 AI 助手（如 Claude Desktop）具备自动排版和发布公众号文章的能力。
-
 ## 文颜的不同版本
 
 文颜目前提供多种形态，覆盖不同使用场景：
@@ -33,12 +31,8 @@
 -   👉 [MCP 版本](https://github.com/caol64/wenyan-mcp) - 本项目
 -   [核心库](https://github.com/caol64/wenyan-core) - 嵌入 Node / Web 项目
 
-## 功能特性
+本仓库是 **文颜的 MCP Server 版本**，基于模型上下文协议（Model Context Protocol），旨在让 AI 助手（如 Claude Desktop）具备自动排版和发布公众号文章的能力。
 
--   列出并选择支持的文章主题
--   使用内置主题对 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>
@@ -47,20 +41,6 @@
 >
 > 如果与 AI 集成遇到问题，可以参考 [test/list.js](./test/list.js) 和 [test/publish.js](./test/publish.js) 中的完整调用示例。
 
-## 主题效果预览
-
-👉 [内置主题预览](https://yuzhi.tech/docs/wenyan/theme)
-
-文颜内置并适配了多个优秀的 Typora 主题，在此感谢原作者：
-
--   [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)
-
 ## 安装与集成
 
 文颜 MCP Server 支持多种运行方式，请根据你的环境选择。
@@ -131,6 +111,81 @@ docker pull caol64/wenyan-mcp
 > *   **环境变量 (`HOST_FILE_PATH`)**：必须与宿主机挂载的文件/图片目录路径保持一致。
 > *   **原理**：你的 Markdown 文件/文章内所引用的本地图片应放置在该目录中，Docker 会自动将其映射，使容器能够读取并上传。
 
+## 基本用法
+
+### 列出主题
+
+对AI说
+
+> 目前你可以使用哪些公众号主题？
+
+AI回复：
+
+```
+目前我支持的公众号主题包括以下这些，每个主题都有独特的风格和配色，适用于不同的内容调性：
+
+1. **Default（默认）**  
+   简洁经典的排版，适合长文阅读，通用性强。
+
+2. **OrangeHeart（橙心）**  
+   暖橙色调，优雅而富有活力，适合情感、生活类内容。
+...
+```
+
+### 注册主题
+
+对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
+
+您可登录微信公众号后台，在「草稿箱」中查看并编辑文章，确认无误后即可一键发布。如需进一步排版优化、添加封面或设置摘要，我也可以协助您完成！
+
+是否需要我帮您生成一篇发布文案或封面建议？ 😊
+```
+
+## 关于图片自动上传
+
+支持以下图片来源：
+
+-   本地路径（如：`/Users/lei/Downloads/result_image.jpg`）
+-   网络路径（如：`https://example.com/image.jpg`）
+
 ## 环境变量说明
 
 无论使用哪种方式，都需要配置以下环境变量以连接微信公众号平台：
@@ -157,13 +212,6 @@ cover: /Users/xxx/image.jpg
     -   如果正文有至少一张图片，可省略，此时将使用其中一张作为封面
     -   如果正文无图片，则必须提供 cover
 
-## 关于图片自动上传
-
-支持以下图片来源：
-
--   本地路径（如：`/Users/lei/Downloads/result_image.jpg`）
--   网络路径（如：`https://example.com/image.jpg`）
-
 ## 微信公众号 IP 白名单
 
 > [!IMPORTANT]

package/dist/index.js

@@ -10,12 +10,12 @@
 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 { configStore, renderStyledContent } 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 { listThemes, REGISTER_THEME_SCHEMA, registerTheme, REMOVE_THEME_SCHEMA, removeTheme } from "./theme.js";
 /**
  * Create an MCP server with capabilities for resources (to list/read notes),
  * tools (to create new notes), and prompts (to summarize notes).
@@ -57,7 +57,6 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
                             description: "ID of the theme to use (e.g., default, orangeheart, rainbow, lapis, pie, maize, purple, phycat).",
                         },
                     },
-                    anyOf: [{ required: ["content"] }, { required: ["file"] }],
                 },
             },
             {
@@ -68,6 +67,8 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
                     properties: {},
                 },
             },
+            REGISTER_THEME_SCHEMA,
+            REMOVE_THEME_SCHEMA,
         ],
     };
 });
@@ -81,24 +82,37 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         //     level: "debug",
         //     data: JSON.stringify(request.params.arguments),
         // });
-        let content = String(request.params.arguments?.content || "");
-        const file = String(request.params.arguments?.file || "");
+        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 = undefined;
+        // 先尝试从已注册的主题中获取主题
+        const customTheme = configStore.getThemeById(themeId);
+        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);
         }
-        if (!content) {
-            throw new Error("Missing content to publish");
-        }
-        const gzhContent = await getGzhContent(content, themeId, "solarized-light", true, true);
+        const gzhContent = await renderStyledContent(content, {
+            themeId,
+            hlThemeId: "solarized-light",
+            isMacStyle: true,
+            isAddFootnote: true,
+            themeCss: customTheme,
+        });
         if (!gzhContent.title) {
-            throw new Error("未能找到文章标题");
+            throw new Error("Can't extract a valid title from the frontmatter.");
         }
         if (!gzhContent.cover) {
-            throw new Error("未能找到文章封面");
+            throw new Error("Can't extract a valid cover from the frontmatter or article.");
         }
         const response = await publishToDraft(gzhContent.title, gzhContent.content, gzhContent.cover, {
             relativePath: absoluteDirPath,
@@ -113,17 +127,13 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         };
     }
     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,
-        };
+        return listThemes();
+    }
+    else if (request.params.name === "register_theme") {
+        return await registerTheme(String(request.params.arguments?.name || ""), String(request.params.arguments?.path || ""));
+    }
+    else if (request.params.name === "remove_theme") {
+        return removeTheme(String(request.params.arguments?.name || ""));
     }
     throw new Error("Unknown tool");
 });

package/dist/theme.js

@@ -0,0 +1,112 @@
+import { getAllGzhThemes } from "@wenyan-md/core";
+import { configStore } from "@wenyan-md/core/wrapper";
+import fs from "node:fs/promises";
+import { getNormalizeFilePath } from "./utils.js";
+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.",
+            },
+        },
+    },
+};
+export function listThemes() {
+    const themes = getAllGzhThemes();
+    const customThemes = configStore.getThemes();
+    return {
+        content: [
+            ...themes.map((theme) => ({
+                type: "text",
+                text: JSON.stringify({
+                    id: theme.meta.id,
+                    name: theme.meta.name,
+                    description: theme.meta.description,
+                }),
+            })),
+            ...customThemes.map((theme) => ({
+                type: "text",
+                text: JSON.stringify({
+                    id: theme.id,
+                    name: theme.name ?? theme.id,
+                    description: theme.description ?? "自定义主题，暂无描述。",
+                }),
+            })),
+        ],
+    };
+}
+export async function registerTheme(name, path) {
+    if (!name || !path) {
+        throw new Error("When adding a theme, you must provide a name and a path.");
+    }
+    if (checkThemeExists(name) || checkCustomThemeExists(name)) {
+        throw new Error("A theme with the given name already exists.");
+    }
+    if (path.startsWith("http")) {
+        const response = await fetch(path);
+        if (!response.ok) {
+            throw new Error(`Failed to retrieve theme from url: ${response.statusText}`);
+        }
+        const content = await response.text();
+        configStore.addThemeToConfig(name, content);
+    }
+    else {
+        const normalizePath = getNormalizeFilePath(path);
+        const content = await fs.readFile(normalizePath, "utf-8");
+        configStore.addThemeToConfig(name, content);
+    }
+    return {
+        content: [
+            {
+                type: "text",
+                text: `Theme "${name}" has been added successfully.`,
+            },
+        ],
+    };
+}
+export function removeTheme(name) {
+    if (checkThemeExists(name)) {
+        throw new Error(`Can't remove builtin theme "${name}"`);
+    }
+    if (!checkCustomThemeExists(name)) {
+        throw new Error(`Custom theme "${name}" does not exist`);
+    }
+    configStore.deleteThemeFromConfig(name);
+    return {
+        content: [
+            {
+                type: "text",
+                text: `Theme "${name}" has been removed successfully.`,
+            },
+        ],
+    };
+}
+function checkThemeExists(themeId) {
+    const themes = getAllGzhThemes();
+    return themes.some((theme) => theme.meta.id === themeId);
+}
+function checkCustomThemeExists(themeId) {
+    const customThemes = configStore.getThemes();
+    return customThemes.some((theme) => theme.id === themeId);
+}

package/dist/type.js

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

package/dist/types/theme.d.ts

@@ -0,0 +1,48 @@
+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.";
+            };
+        };
+    };
+};
+export declare function listThemes(): {
+    content: {
+        type: "text";
+        text: string;
+    }[];
+};
+export declare function registerTheme(name: string, path: string): Promise<{
+    content: {
+        type: string;
+        text: string;
+    }[];
+}>;
+export declare function removeTheme(name: string): {
+    content: {
+        type: string;
+        text: string;
+    }[];
+};

package/dist/types/type.d.ts

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wenyan-md/mcp",
-  "version": "1.0.9",
+  "version": "1.0.11",
   "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,7 +31,10 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "0.6.0",
-    "@wenyan-md/core": "^1.0.17"
+    "@wenyan-md/core": "^2.0.2",
+    "form-data-encoder": "^4.1.0",
+    "formdata-node": "^6.0.3",
+    "jsdom": "^27.4.0"
   },
   "devDependencies": {
     "@types/node": "^24.3.0",
@@ -42,8 +45,10 @@
   "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",
+    "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"
   }
 }