@birthday8/doc-mcp 1.0.0

package/python/requirements.txt

@@ -0,0 +1,4 @@
+mcp>=1.0.0
+python-docx>=0.8.11
+beautifulsoup4>=4.9.3
+lxml>=4.6.3

package/python/server.py

@@ -0,0 +1,320 @@
+#!/usr/bin/env python3
+"""
+Doc Creator MCP Server
+文档生成器 - 提供HTML转Word文档的工具服务
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# MCP imports
+from mcp.server import Server
+from mcp.types import TextContent, Tool
+
+# Import the conversion module
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+from docx_converter import convert_html_to_docx as docx_convert
+
+# Create MCP server
+app = Server("doc-creator")
+
+
+@app.list_tools()
+async def list_tools() -> list[Tool]:
+    """列出可用的工具"""
+    return [
+        Tool(
+            name="convert_document",
+            description="将HTML文件转换为Word文档(.docx)",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "html_path": {
+                        "type": "string",
+                        "description": "HTML文件的完整路径"
+                    },
+                    "output_path": {
+                        "type": "string",
+                        "description": "输出DOCX文件的路径（可选，默认与HTML同目录）"
+                    }
+                },
+                "required": ["html_path"]
+            }
+        ),
+        Tool(
+            name="generate_document",
+            description="根据HTML内容生成Word文档",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "content": {
+                        "type": "string",
+                        "description": "HTML内容（不需要完整的HTML结构，只需要body内的内容）"
+                    },
+                    "title": {
+                        "type": "string",
+                        "description": "文档标题",
+                        "default": "文档"
+                    },
+                    "output_dir": {
+                        "type": "string",
+                        "description": "输出目录（可选，默认使用workspace/documents/）"
+                    }
+                },
+                "required": ["content"]
+            }
+        ),
+        Tool(
+            name="get_html_template",
+            description="获取完整的HTML模板，包含CSS样式",
+            inputSchema={
+                "type": "object",
+                "properties": {}
+            }
+        )
+    ]
+
+
+@app.call_tool()
+async def call_tool(name: str, arguments: dict) -> list[TextContent]:
+    """调用工具"""
+    
+    if name == "convert_document":
+        html_path = arguments.get("html_path")
+        output_path = arguments.get("output_path")
+        
+        if not html_path:
+            return [TextContent(type="text", text="错误：必须提供html_path参数")]
+        
+        if not os.path.exists(html_path):
+            return [TextContent(type="text", text=f"错误：HTML文件不存在: {html_path}")]
+        
+        # 如果未指定输出路径，使用相同目录和文件名
+        if not output_path:
+            base_name = os.path.splitext(html_path)[0]
+            output_path = base_name + ".docx"
+        
+        try:
+            docx_convert(html_path, output_path)
+            return [TextContent(
+                type="text", 
+                text=f"✅ 转换成功！\n📄 HTML文件: {html_path}\n📝 Word文件: {output_path}"
+            )]
+        except Exception as e:
+            return [TextContent(type="text", text=f"❌ 转换失败: {str(e)}")]
+    
+    elif name == "generate_document":
+        content = arguments.get("content", "")
+        title = arguments.get("title", "文档")
+        output_dir = arguments.get("output_dir")
+        
+        if not content:
+            return [TextContent(type="text", text="错误：必须提供content参数")]
+        
+        # 构建输出目录
+        if not output_dir:
+            workspace_dir = Path(__file__).parent.parent / "workspace" / "documents"
+            from datetime import datetime
+            timestamp = datetime.now().strftime("%Y%m%d")
+            output_dir = workspace_dir / f"{timestamp}_{title.replace(' ', '_')}"
+        
+        output_dir = Path(output_dir)
+        output_dir.mkdir(parents=True, exist_ok=True)
+        
+        # 生成HTML文件
+        html_path = output_dir / "document.html"
+        docx_path = output_dir / "document.docx"
+        
+        # 构建完整HTML
+        html_template = f"""<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <title>{title}</title>
+    <style>
+        :root {{
+            --default-font: '微软雅黑';
+            --default-size: 12pt;
+            --default-color: #333;
+            --line-height: 1.8;
+        }}
+        body {{
+            font-family: var(--default-font);
+            font-size: var(--default-size);
+            color: var(--default-color);
+            line-height: var(--line-height);
+            padding: 20px;
+            max-width: 800px;
+            margin: 0 auto;
+        }}
+        h1 {{
+            font-family: '微软雅黑';
+            font-size: 18pt;
+            color: #4a3f6b;
+            text-align: center;
+            margin-bottom: 20pt;
+        }}
+        h2 {{
+            font-family: '微软雅黑';
+            font-size: 16pt;
+            color: #5b4e8c;
+            border-bottom: 2px solid #667eea;
+            padding-bottom: 10px;
+            margin-bottom: 15pt;
+        }}
+        h3 {{
+            font-family: '微软雅黑';
+            font-size: 14pt;
+            color: #6b5b7a;
+            margin-bottom: 10pt;
+        }}
+        p {{
+            text-indent: 2em;
+            margin-bottom: 10pt;
+        }}
+        .center {{ text-align: center; }}
+        .red {{ color: red; }}
+        .blue {{ color: blue; }}
+        .green {{ color: green; }}
+        .highlight {{ background-color: yellow; }}
+        .info {{
+            background-color: #e3f2fd;
+            padding: 10px;
+            border-left: 4px solid #2196F3;
+            margin: 10pt 0;
+        }}
+        .warning {{
+            background-color: #fff3cd;
+            padding: 10px;
+            border-left: 4px solid #ffc107;
+            margin: 10pt 0;
+        }}
+        .success {{
+            background-color: #d4edda;
+            padding: 10px;
+            border-left: 4px solid #28a745;
+            margin: 10pt 0;
+        }}
+        table {{
+            width: 100%;
+            border-collapse: collapse;
+            margin: 20px 0;
+        }}
+        th, td {{
+            border: 1px solid #ddd;
+            padding: 12px;
+            text-align: center;
+        }}
+        th {{
+            background-color: #667eea;
+            color: white;
+        }}
+        tr:nth-child(even) {{
+            background-color: #f9f9f9;
+        }}
+    </style>
+</head>
+<body>
+    <h1>{title}</h1>
+    {content}
+</body>
+</html>"""
+        
+        # 保存HTML
+        with open(html_path, 'w', encoding='utf-8') as f:
+            f.write(html_template)
+        
+        # 转换为DOCX
+        try:
+            docx_convert(str(html_path), str(docx_path))
+            return [TextContent(
+                type="text",
+                text=f"✅ 文档生成成功！\n📁 输出目录: {output_dir}\n📄 HTML文件: {html_path}\n📝 Word文件: {docx_path}"
+            )]
+        except Exception as e:
+            return [TextContent(type="text", text=f"❌ 转换失败: {str(e)}")]
+    
+    elif name == "get_html_template":
+        template = """<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <title>文档标题</title>
+    <style>
+        /* 全局配置 */
+        :root {
+            --default-font: '微软雅黑';
+            --default-size: 12pt;
+            --default-color: #333;
+            --line-height: 1.8;
+        }
+        
+        body {
+            font-family: var(--default-font);
+            font-size: var(--default-size);
+            color: var(--default-color);
+            line-height: var(--line-height);
+            padding: 20px;
+            max-width: 800px;
+            margin: 0 auto;
+        }
+        
+        /* 标题样式 */
+        h1 { font-size: 18pt; color: #4a3f6b; text-align: center; }
+        h2 { font-size: 16pt; color: #5b4e8c; border-bottom: 2px solid #667eea; }
+        h3 { font-size: 14pt; color: #6b5b7a; }
+        
+        /* 段落 */
+        p { text-indent: 2em; margin-bottom: 10pt; }
+        
+        /* 文本格式 */
+        .red { color: red; }
+        .blue { color: blue; }
+        .green { color: green; }
+        .highlight { background-color: yellow; }
+        
+        /* 提示框 */
+        .info { background-color: #e3f2fd; padding: 10px; border-left: 4px solid #2196F3; }
+        .warning { background-color: #fff3cd; padding: 10px; border-left: 4px solid #ffc107; }
+        .success { background-color: #d4edda; padding: 10px; border-left: 4px solid #28a745; }
+        
+        /* 表格 */
+        table { width: 100%; border-collapse: collapse; margin: 20px 0; }
+        th, td { border: 1px solid #ddd; padding: 12px; text-align: center; }
+        th { background-color: #667eea; color: white; }
+        tr:nth-child(even) { background-color: #f9f9f9; }
+    </style>
+</head>
+<body>
+    <h1>文档标题</h1>
+    
+    <h2>一、章节标题</h2>
+    <p>这是正文段落，<strong>支持加粗</strong>、<em>斜体</em>、<span class="red">彩色文字</span>等格式。</p>
+    
+    <div class="info">
+        <strong>提示：</strong> 这是信息提示框
+    </div>
+</body>
+</html>"""
+        return [TextContent(type="text", text=template)]
+    
+    else:
+        return [TextContent(type="text", text=f"未知工具: {name}")]
+
+
+async def main():
+    """主函数"""
+    from mcp.server.stdio import stdio_server
+    
+    async with stdio_server() as (read_stream, write_stream):
+        await app.run(
+            read_stream,
+            write_stream,
+            app.create_initialization_options()
+        )
+
+
+if __name__ == "__main__":
+    import asyncio
+    asyncio.run(main())