@8btc/ppt-generator-mcp 0.0.1

package/dist/index.js

@@ -0,0 +1,161 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
+const stdio_js_1 = require("@modelcontextprotocol/sdk/server/stdio.js");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+const ppt_generator_js_1 = require("./ppt-generator.js");
+const node_path_1 = __importDefault(require("node:path"));
+const zod_1 = require("zod");
+const fs_extra_1 = __importDefault(require("fs-extra"));
+const package_json_1 = __importDefault(require("../package.json"));
+class PPTGeneratorMCPServer {
+    constructor() {
+        this.server = new mcp_js_1.McpServer({
+            name: "ppt-generator",
+            version: "0.1.0",
+        }, {
+            capabilities: {
+                tools: {},
+            },
+        });
+        this.pptGenerator = new ppt_generator_js_1.PPTGenerator();
+        this.setupToolHandlers();
+    }
+    setupToolHandlers() {
+        this.server.registerTool("get_ppt_template_list", {
+            title: "ppt模板列表",
+            description: "获取可使用的ppt模板",
+        }, async () => {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify([
+                            {
+                                name: "清新风格模板-图标比较多",
+                                path: node_path_1.default.join(__dirname, "tpls", "tpl-3.json"),
+                            },
+                            {
+                                name: "研究报告风格模板-紫色系",
+                                path: node_path_1.default.join(__dirname, "tpls", "tpl-4.json"),
+                            },
+                            {
+                                name: "简约风格模板-莫兰迪色系",
+                                path: node_path_1.default.join(__dirname, "tpls", "tpl-2.json"),
+                            },
+                            {
+                                name: "简单模板-蓝色调",
+                                path: node_path_1.default.join(__dirname, "tpls", "tpl-1.json"),
+                            },
+                        ]),
+                    },
+                ],
+            };
+        });
+        this.server.registerTool("generate_ppt", {
+            title: "ppt生成工具",
+            description: "根据大纲自动匹配模板生成PPT文件",
+            inputSchema: {
+                templateFilePath: zod_1.z.string().optional(),
+                outputPath: zod_1.z.string().optional(),
+                outline: zod_1.z.array(zod_1.z.discriminatedUnion("type", [
+                    zod_1.z.object({
+                        type: zod_1.z.literal("cover"),
+                        data: zod_1.z.object({
+                            title: zod_1.z.string(),
+                            text: zod_1.z.string(),
+                        }),
+                    }),
+                    zod_1.z.object({
+                        type: zod_1.z.literal("contents"),
+                        data: zod_1.z.object({
+                            items: zod_1.z.array(zod_1.z.string()),
+                        }),
+                        offset: zod_1.z.number().optional(),
+                    }),
+                    zod_1.z.object({
+                        type: zod_1.z.literal("transition"),
+                        data: zod_1.z.object({
+                            title: zod_1.z.string(),
+                            text: zod_1.z.string(),
+                        }),
+                    }),
+                    zod_1.z.object({
+                        type: zod_1.z.literal("content"),
+                        data: zod_1.z.object({
+                            title: zod_1.z.string(),
+                            items: zod_1.z.array(zod_1.z.object({
+                                title: zod_1.z.string(),
+                                text: zod_1.z.string(),
+                            })),
+                        }),
+                    }),
+                    zod_1.z.object({
+                        type: zod_1.z.literal("end"),
+                    }),
+                ])),
+            },
+        }, async ({ outline, templateFilePath, outputPath }) => {
+            if (!templateFilePath) {
+                templateFilePath = node_path_1.default.join(__dirname, "tpls", "tpl-2.json");
+            }
+            if (!outputPath) {
+                outputPath = node_path_1.default.join(__dirname, "..", "output");
+            }
+            try {
+                fs_extra_1.default.ensureDirSync(outputPath);
+                // 保存result.json
+                await fs_extra_1.default.writeJson(node_path_1.default.join(outputPath, "outline.json"), outline, {
+                    spaces: 2,
+                });
+                const result = await this.pptGenerator.generateResultJson({
+                    outline: outline,
+                    templateFilePath: templateFilePath,
+                    outputPath: node_path_1.default.join(outputPath, "result.json"),
+                });
+                if (!result) {
+                    console.error("generate_result_json 生成 result.json 失败");
+                    throw new types_js_1.McpError(types_js_1.ErrorCode.InternalError, "生成result.json失败");
+                }
+                const resp = await this.pptGenerator.generatePPTFromResult({
+                    result,
+                    outputPath: node_path_1.default.join(outputPath, "result.pptx"),
+                });
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: JSON.stringify(resp, null, 2),
+                        },
+                        {
+                            type: "resource_link",
+                            uri: `file:///${node_path_1.default.join(outputPath, "result.pptx")}`,
+                            name: "result.pptx",
+                            description: "生成的PPT",
+                            mimeType: "application/vnd.openxmlformats-officedocument.presentationml.presentation",
+                            annotations: {
+                                audience: ["assistant"],
+                                priority: 0.9,
+                            },
+                        },
+                    ],
+                };
+            }
+            catch (err) {
+                console.error(err);
+                throw new types_js_1.McpError(types_js_1.ErrorCode.InternalError, "生成PPT失败");
+            }
+        });
+    }
+    async run() {
+        const transport = new stdio_js_1.StdioServerTransport();
+        await this.server.connect(transport);
+        console.error("PPT Generator MCP server running on stdio");
+        console.error(`VERSION ${package_json_1.default.version}`);
+    }
+}
+const server = new PPTGeneratorMCPServer();
+server.run().catch(console.error);

package/dist/ppt-generator.d.ts

@@ -0,0 +1,39 @@
+import { Outline } from "./types/outline";
+import { Slide } from "./types/slide";
+interface IResult {
+    title: string;
+    width: number;
+    height: number;
+    theme: any;
+    slides: Slide[];
+}
+export declare class PPTGenerator {
+    /**
+     * 列出可用的模板
+     */
+    listTemplates(templateDir?: string): Promise<{
+        templates: string[];
+        defaultDir: string;
+    }>;
+    /**
+     * 两步生成PPT - 第一步：生成result.json
+     */
+    generateResultJson(params: {
+        outline: Outline[];
+        templateFilePath: string;
+        outputPath: string;
+        title?: string;
+    }): Promise<IResult | undefined>;
+    /**
+     * 两步生成PPT - 第二步：从result.json生成PPT
+     */
+    generatePPTFromResult(params: {
+        result: IResult;
+        outputPath: string;
+    }): Promise<{
+        success: boolean;
+        outputPath?: string;
+        message: string;
+    }>;
+}
+export {};

package/dist/ppt-generator.js

@@ -0,0 +1,110 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PPTGenerator = void 0;
+const fs_extra_1 = __importDefault(require("fs-extra"));
+const posix_1 = __importDefault(require("node:path/posix"));
+const export_pptx_1 = require("./export-pptx");
+const generate_ppt_slides_1 = require("./generate-ppt-slides");
+class PPTGenerator {
+    /**
+     * 列出可用的模板
+     */
+    async listTemplates(templateDir) {
+        const defaultDir = templateDir || posix_1.default.join(__dirname, "../../data");
+        const templates = [];
+        try {
+            if (await fs_extra_1.default.pathExists(defaultDir)) {
+                const files = await fs_extra_1.default.readdir(defaultDir);
+                const templateFiles = files.filter((file) => file.startsWith("tpl-") && file.endsWith(".json"));
+                templates.push(...templateFiles);
+            }
+        }
+        catch (error) {
+            console.warn(`无法读取模板目录 ${defaultDir}:`, error);
+        }
+        return {
+            templates,
+            defaultDir,
+        };
+    }
+    /**
+     * 两步生成PPT - 第一步：生成result.json
+     */
+    async generateResultJson(params) {
+        try {
+            const { outline, templateFilePath, outputPath } = params;
+            // 读取模板文件
+            const templateData = await fs_extra_1.default.readJson(templateFilePath);
+            // 使用AIPPT函数生成slides
+            const generatedSlides = (0, generate_ppt_slides_1.generatePPTSlides)(templateData.slides, outline);
+            // 构建完整的result对象，包含模板的其他属性
+            const resultData = {
+                title: templateData.title,
+                width: templateData.width,
+                height: templateData.height,
+                theme: templateData.theme,
+                slides: generatedSlides,
+            };
+            // 生成输出路径
+            const resultPath = outputPath;
+            // 保存result.json
+            await fs_extra_1.default.writeJson(resultPath, resultData, { spaces: 2 });
+            return resultData;
+        }
+        catch (error) {
+            console.error("generateResultJson catch: ", error);
+            throw error;
+        }
+    }
+    /**
+     * 两步生成PPT - 第二步：从result.json生成PPT
+     */
+    async generatePPTFromResult(params) {
+        try {
+            const { result: resultData, outputPath } = params;
+            // 构造导出选项
+            const options = {
+                title: resultData.title || "Generated Presentation",
+                author: "PPT Generator",
+                subject: "Auto-generated presentation",
+                keywords: "presentation, auto-generated",
+            };
+            // 构造完整的导出选项
+            const exportOptions = {
+                slides: resultData.slides,
+                theme: resultData.theme || {
+                    backgroundColor: "#ffffff",
+                    fontColor: "#333333",
+                    fontName: "Arial",
+                    themeColors: ["#1f4e79", "#0f2537"],
+                    outline: { width: 1, color: "#000000" },
+                    shadow: { h: 0, v: 0, blur: 0, color: "#000000" },
+                },
+                viewportRatio: (resultData.width || 1920) / (resultData.height || 1080),
+                title: resultData.title || "Generated Presentation",
+                viewportSize: resultData.width || 1920,
+            };
+            // 使用exportPPTX生成PPT
+            const pptx = (0, export_pptx_1.exportPPTX)(resultData.slides, false, exportOptions);
+            // 确定输出路径
+            const finalOutputPath = outputPath;
+            // 确保输出目录存在
+            await fs_extra_1.default.ensureDir(posix_1.default.dirname(finalOutputPath));
+            // 保存PPT文件
+            await pptx.writeFile({ fileName: finalOutputPath });
+            return {
+                success: true,
+                outputPath: finalOutputPath,
+                message: `PPT文件生成成功: ${finalOutputPath}`,
+            };
+        }
+        catch (error) {
+            console.error(`从result.json生成PPT失败: ${error.message}`);
+            throw error;
+        }
+    }
+}
+exports.PPTGenerator = PPTGenerator;