lightpdf-aipdf-mcp 0.1.31__tar.gz

lightpdf_aipdf_mcp-0.1.31/.gitignore

@@ -0,0 +1,20 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv/
+node_modules/
+
+# macOS system files
+.DS_Store
+
+mcp_server/demo/
+mcp_server/docs/
+homepage/backend/uploads
+.version
+lightpdf-mcp.zip

lightpdf_aipdf_mcp-0.1.31/PKG-INFO

@@ -0,0 +1,208 @@
+Metadata-Version: 2.4
+Name: lightpdf-aipdf-mcp
+Version: 0.1.31
+Summary: MCP Server for LightPDF AI-PDF
+Author: LightPDF Team
+License: Proprietary
+Requires-Python: >=3.12
+Requires-Dist: httpx-sse>=0.4.0
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: mcp>=1.6.0
+Requires-Dist: pydantic-settings>=2.8.1
+Requires-Dist: pydantic>=2.11.2
+Requires-Dist: python-dotenv>=1.1.0
+Requires-Dist: uvicorn>=0.34.0
+Description-Content-Type: text/markdown
+
+# LightPDF AI助手 MCP Server
+
+LightPDF AI助手 MCP Server 是一个基于 MCP (Model Control Protocol) 的文档格式转换工具，支持多种文档格式之间的相互转换以及PDF的各种处理功能。该工具需要在支持 MCP 功能的软件（如 Cursor）中配置使用。
+
+## 配置说明
+
+在 Cursor 中配置 LightPDF AI助手 MCP Server 工具需要修改 `~/.cursor/mcp.json` 文件。添加以下配置：
+
+```json
+{
+  "mcpServers": {
+    "lightpdf": {
+      "command": "uvx",
+      "args": [
+        "lightpdf-aipdf-mcp@latest"
+      ],
+      "env": {
+        "API_KEY": "your_api_key_here"
+      }
+    }
+  }
+}
+```
+
+配置说明：
+- `command`: 使用 `uvx` 命令运行工具
+- `args`: 命令行参数
+  - 使用最新版的 `lightpdf-aipdf-mcp` 包
+- `env`: 环境变量配置
+  - `API_KEY`: 你的API密钥
+
+配置完成后，重启 Cursor 使配置生效。
+
+## 功能特性
+
+- 支持多种文档格式的转换
+- 支持本地文件和网络URL
+- 支持批量转换
+- 支持PDF文件的高级处理功能（水印去除/添加、添加页码、压缩、加密/解密等）
+- 并发处理，提高转换效率
+- 详细的转换进度和结果报告
+
+## 支持的格式
+
+### PDF转换为其他格式
+- Word (DOCX)
+- Excel (XLSX)
+- PowerPoint (PPTX)
+- 图片 (JPG, JPEG, PNG, BMP, GIF)
+- HTML
+- 文本 (TXT)
+
+### 其他格式转换为PDF
+- Word (DOCX)
+- Excel (XLSX)
+- PowerPoint (PPTX)
+- 图片 (JPG, JPEG, PNG, BMP, GIF)
+- CAD (DWG)
+- CAJ
+- OFD
+
+## PDF处理功能
+
+### PDF水印处理
+- 支持去除PDF文档中的水印
+- 支持添加自定义文本水印
+- 可设置水印位置、透明度、角度和应用页面范围
+
+### PDF页码添加
+- 支持在PDF文档中添加页码
+- 可自定义起始页码
+- 可选择页码位置（左上、上中、右上、左下、下中、右下）
+- 可设置页码边距
+
+### PDF编辑功能
+- **拆分PDF**: 支持按页码范围拆分，或每页生成一个新文件
+- **合并PDF**: 将多个PDF文件合并为一个
+- **旋转PDF**: 支持90°、180°、270°旋转，可指定页面范围
+- **压缩PDF**: 优化PDF文件大小，可调整图像质量
+
+### PDF安全功能
+- **加密PDF**: 添加密码保护
+- **解密PDF**: 移除密码保护（需提供原密码）
+
+## 使用方法
+
+1. 确保你使用的软件支持 MCP 功能（如 Cursor）
+2. 在软件中完成 LightPDF AI助手 MCP Server 工具的配置
+3. 调用相应的工具，指定输入文件和所需参数
+4. 处理完成后，系统会提供结果文件的在线下载链接
+
+### 格式转换
+使用 `convert_document` 工具，指定输入文件和目标格式。
+
+### PDF处理功能
+- 去除水印: `remove_watermark`
+- 添加水印: `add_watermark`
+- 添加页码: `add_page_numbers`
+- 拆分PDF: `split_pdf`
+- 合并PDF: `merge_pdfs`
+- 旋转PDF: `rotate_pdf`
+- 压缩PDF: `compress_pdf`
+- 加密PDF: `protect_pdf`
+- 解密PDF: `unlock_pdf`
+
+## 参数说明
+
+### 格式转换工具 (convert_document)
+- `files`: 要转换的文件对象列表
+  - `path`: 文件路径或URL
+  - `password`: 文档密码（如有）
+- `format`: 目标格式，支持：
+  - `pdf`: 转换为PDF格式
+  - `docx`: 转换为Word格式
+  - `xlsx`: 转换为Excel格式
+  - `pptx`: 转换为PowerPoint格式
+  - `jpg`/`jpeg`: 转换为JPG格式
+  - `png`: 转换为PNG格式
+  - `bmp`: 转换为BMP格式
+  - `gif`: 转换为GIF格式
+  - `html`: 转换为HTML格式
+  - `txt`: 转换为文本格式
+
+### 去除水印工具 (remove_watermark)
+- `files`: 要处理的PDF文件对象列表
+  - `path`: 文件路径或URL
+  - `password`: 文档密码（如有）
+
+### 添加水印工具 (add_watermark)
+- `files`: 要处理的PDF文件对象列表
+- `text`: 水印文本内容
+- `position`: 水印位置，可选值包括"center"、"topleft"等
+- `opacity`: 透明度（0.0-1.0）
+- `deg`: 旋转角度
+- `range`: 页面范围，如 "1,3,5-7"
+- `layout`: 水印显示位置，"on"（在内容上）或"under"（在内容下）
+- 可选的字体设置: `font_family`、`font_size`、`font_color`
+
+### 添加页码工具 (add_page_numbers)
+- `files`: 要处理的PDF文件对象列表
+- `start_num`: 起始页码，默认为1
+- `position`: 页码位置（1-6），对应左上/上中/右上/左下/下中/右下
+- `margin`: 页码边距，可选值为10/30/60，默认为30
+
+### 拆分PDF工具 (split_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `split_type`: 拆分类型
+  - `every`: 每页拆分为一个文件
+  - `page`: 按页面规则拆分（默认）
+- `pages`: 拆分页面规则，如 "1,3,5-7"
+- `merge_all`: 是否合并拆分后的文件（0=否，1=是）
+
+### 合并PDF工具 (merge_pdfs)
+- `files`: 要合并的PDF文件对象列表（至少两个文件）
+
+### 旋转PDF工具 (rotate_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `angle`: 旋转角度，可选值为90、180、270
+- `pages`: 页面范围，如 "1,3,5-7"
+
+### 压缩PDF工具 (compress_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `image_quantity`: 图像质量（1-100），值越低压缩率越高，默认为60
+
+### 加密PDF工具 (protect_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `password`: 要设置的新密码
+
+### 解密PDF工具 (unlock_pdf)
+- `files`: 要处理的PDF文件对象列表
+  - `path`: 文件路径或URL
+  - `password`: 文档的当前密码（必需）
+
+## 注意事项
+
+1. 确保有足够的磁盘空间用于文件转换
+2. 对于网络URL，确保URL可访问且文件大小在合理范围内
+3. 批量转换时会自动控制并发数，避免过度占用系统资源
+4. 某些格式转换可能需要较长时间，请耐心等待
+5. 水印去除功能仅适用于PDF格式文件
+6. 复杂水印可能无法完全去除，效果取决于水印类型
+7. 所有PDF编辑功能只支持PDF格式的输入文件
+
+## 错误处理
+
+工具会提供详细的错误信息，常见的错误包括：
+
+- API密钥未配置或无效
+- 文件格式不支持
+- 文件不存在或无法访问
+- 网络连接问题
+- PDF密码不正确或文件受保护

lightpdf_aipdf_mcp-0.1.31/README.md

@@ -0,0 +1,192 @@
+# LightPDF AI助手 MCP Server
+
+LightPDF AI助手 MCP Server 是一个基于 MCP (Model Control Protocol) 的文档格式转换工具，支持多种文档格式之间的相互转换以及PDF的各种处理功能。该工具需要在支持 MCP 功能的软件（如 Cursor）中配置使用。
+
+## 配置说明
+
+在 Cursor 中配置 LightPDF AI助手 MCP Server 工具需要修改 `~/.cursor/mcp.json` 文件。添加以下配置：
+
+```json
+{
+  "mcpServers": {
+    "lightpdf": {
+      "command": "uvx",
+      "args": [
+        "lightpdf-aipdf-mcp@latest"
+      ],
+      "env": {
+        "API_KEY": "your_api_key_here"
+      }
+    }
+  }
+}
+```
+
+配置说明：
+- `command`: 使用 `uvx` 命令运行工具
+- `args`: 命令行参数
+  - 使用最新版的 `lightpdf-aipdf-mcp` 包
+- `env`: 环境变量配置
+  - `API_KEY`: 你的API密钥
+
+配置完成后，重启 Cursor 使配置生效。
+
+## 功能特性
+
+- 支持多种文档格式的转换
+- 支持本地文件和网络URL
+- 支持批量转换
+- 支持PDF文件的高级处理功能（水印去除/添加、添加页码、压缩、加密/解密等）
+- 并发处理，提高转换效率
+- 详细的转换进度和结果报告
+
+## 支持的格式
+
+### PDF转换为其他格式
+- Word (DOCX)
+- Excel (XLSX)
+- PowerPoint (PPTX)
+- 图片 (JPG, JPEG, PNG, BMP, GIF)
+- HTML
+- 文本 (TXT)
+
+### 其他格式转换为PDF
+- Word (DOCX)
+- Excel (XLSX)
+- PowerPoint (PPTX)
+- 图片 (JPG, JPEG, PNG, BMP, GIF)
+- CAD (DWG)
+- CAJ
+- OFD
+
+## PDF处理功能
+
+### PDF水印处理
+- 支持去除PDF文档中的水印
+- 支持添加自定义文本水印
+- 可设置水印位置、透明度、角度和应用页面范围
+
+### PDF页码添加
+- 支持在PDF文档中添加页码
+- 可自定义起始页码
+- 可选择页码位置（左上、上中、右上、左下、下中、右下）
+- 可设置页码边距
+
+### PDF编辑功能
+- **拆分PDF**: 支持按页码范围拆分，或每页生成一个新文件
+- **合并PDF**: 将多个PDF文件合并为一个
+- **旋转PDF**: 支持90°、180°、270°旋转，可指定页面范围
+- **压缩PDF**: 优化PDF文件大小，可调整图像质量
+
+### PDF安全功能
+- **加密PDF**: 添加密码保护
+- **解密PDF**: 移除密码保护（需提供原密码）
+
+## 使用方法
+
+1. 确保你使用的软件支持 MCP 功能（如 Cursor）
+2. 在软件中完成 LightPDF AI助手 MCP Server 工具的配置
+3. 调用相应的工具，指定输入文件和所需参数
+4. 处理完成后，系统会提供结果文件的在线下载链接
+
+### 格式转换
+使用 `convert_document` 工具，指定输入文件和目标格式。
+
+### PDF处理功能
+- 去除水印: `remove_watermark`
+- 添加水印: `add_watermark`
+- 添加页码: `add_page_numbers`
+- 拆分PDF: `split_pdf`
+- 合并PDF: `merge_pdfs`
+- 旋转PDF: `rotate_pdf`
+- 压缩PDF: `compress_pdf`
+- 加密PDF: `protect_pdf`
+- 解密PDF: `unlock_pdf`
+
+## 参数说明
+
+### 格式转换工具 (convert_document)
+- `files`: 要转换的文件对象列表
+  - `path`: 文件路径或URL
+  - `password`: 文档密码（如有）
+- `format`: 目标格式，支持：
+  - `pdf`: 转换为PDF格式
+  - `docx`: 转换为Word格式
+  - `xlsx`: 转换为Excel格式
+  - `pptx`: 转换为PowerPoint格式
+  - `jpg`/`jpeg`: 转换为JPG格式
+  - `png`: 转换为PNG格式
+  - `bmp`: 转换为BMP格式
+  - `gif`: 转换为GIF格式
+  - `html`: 转换为HTML格式
+  - `txt`: 转换为文本格式
+
+### 去除水印工具 (remove_watermark)
+- `files`: 要处理的PDF文件对象列表
+  - `path`: 文件路径或URL
+  - `password`: 文档密码（如有）
+
+### 添加水印工具 (add_watermark)
+- `files`: 要处理的PDF文件对象列表
+- `text`: 水印文本内容
+- `position`: 水印位置，可选值包括"center"、"topleft"等
+- `opacity`: 透明度（0.0-1.0）
+- `deg`: 旋转角度
+- `range`: 页面范围，如 "1,3,5-7"
+- `layout`: 水印显示位置，"on"（在内容上）或"under"（在内容下）
+- 可选的字体设置: `font_family`、`font_size`、`font_color`
+
+### 添加页码工具 (add_page_numbers)
+- `files`: 要处理的PDF文件对象列表
+- `start_num`: 起始页码，默认为1
+- `position`: 页码位置（1-6），对应左上/上中/右上/左下/下中/右下
+- `margin`: 页码边距，可选值为10/30/60，默认为30
+
+### 拆分PDF工具 (split_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `split_type`: 拆分类型
+  - `every`: 每页拆分为一个文件
+  - `page`: 按页面规则拆分（默认）
+- `pages`: 拆分页面规则，如 "1,3,5-7"
+- `merge_all`: 是否合并拆分后的文件（0=否，1=是）
+
+### 合并PDF工具 (merge_pdfs)
+- `files`: 要合并的PDF文件对象列表（至少两个文件）
+
+### 旋转PDF工具 (rotate_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `angle`: 旋转角度，可选值为90、180、270
+- `pages`: 页面范围，如 "1,3,5-7"
+
+### 压缩PDF工具 (compress_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `image_quantity`: 图像质量（1-100），值越低压缩率越高，默认为60
+
+### 加密PDF工具 (protect_pdf)
+- `files`: 要处理的PDF文件对象列表
+- `password`: 要设置的新密码
+
+### 解密PDF工具 (unlock_pdf)
+- `files`: 要处理的PDF文件对象列表
+  - `path`: 文件路径或URL
+  - `password`: 文档的当前密码（必需）
+
+## 注意事项
+
+1. 确保有足够的磁盘空间用于文件转换
+2. 对于网络URL，确保URL可访问且文件大小在合理范围内
+3. 批量转换时会自动控制并发数，避免过度占用系统资源
+4. 某些格式转换可能需要较长时间，请耐心等待
+5. 水印去除功能仅适用于PDF格式文件
+6. 复杂水印可能无法完全去除，效果取决于水印类型
+7. 所有PDF编辑功能只支持PDF格式的输入文件
+
+## 错误处理
+
+工具会提供详细的错误信息，常见的错误包括：
+
+- API密钥未配置或无效
+- 文件格式不支持
+- 文件不存在或无法访问
+- 网络连接问题
+- PDF密码不正确或文件受保护

lightpdf_aipdf_mcp-0.1.31/pyproject.toml

@@ -0,0 +1,32 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "lightpdf-aipdf-mcp"
+version = "0.1.31"
+description = "MCP Server for LightPDF AI-PDF"
+readme = "README.md"
+requires-python = ">=3.12"
+license = { text = "Proprietary" }
+authors = [
+    {name = "LightPDF Team"}
+]
+dependencies = [
+    "mcp>=1.6.0",
+    "httpx>=0.28.1",
+    "httpx-sse>=0.4.0",
+    "python-dotenv>=1.1.0",
+    "uvicorn>=0.34.0",
+    "pydantic>=2.11.2",
+    "pydantic-settings>=2.8.1",
+]
+
+[project.scripts]
+lightpdf-aipdf-mcp = "lightpdf_aipdf_mcp:main"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/lightpdf_aipdf_mcp"]
+
+[tool.hatch.build]
+exclude = ["docs/", "__pycache__/", "publish.sh", "run.py", ".env", ".gitignore"]

lightpdf_aipdf_mcp-0.1.31/src/lightpdf_aipdf_mcp/__init__.py

@@ -0,0 +1,8 @@
+from . import server
+
+def main():
+    """Main entry point for the package."""
+    server.cli_main()
+
+# 只导出main函数
+__all__ = ['main']

lightpdf_aipdf_mcp-0.1.31/src/lightpdf_aipdf_mcp/common.py

@@ -0,0 +1,181 @@
+"""通用工具模块"""
+import asyncio
+import json
+import os
+import tempfile
+import time
+import urllib.parse
+from dataclasses import dataclass, field
+from typing import List, Optional, Dict, Any, Tuple
+
+import httpx
+from mcp.types import TextContent, LoggingMessageNotification, LoggingMessageNotificationParams, LoggingLevel
+
+@dataclass
+class BaseResult:
+    """基础结果数据类"""
+    success: bool
+    file_path: str
+    error_message: Optional[str] = None
+    download_url: Optional[str] = None
+
+class Logger:
+    """日志记录器类"""
+    def __init__(self, context, collect_info: bool = True):
+        self.context = context
+        self.collect_info = collect_info
+        self._info_log = []
+
+    async def log(self, level: str, message: str, add_to_result: bool = True):
+        """记录日志消息"""
+        if self.collect_info and add_to_result:
+            self._info_log.append(message)
+            
+        level_map = {
+            "debug": LoggingLevel.Debug,
+            "info": LoggingLevel.Info,
+            "warning": LoggingLevel.Warning,
+            "error": LoggingLevel.Error
+        }
+        
+        mcp_level = level_map.get(level.lower(), LoggingLevel.Info)
+        
+        # 使用更简洁的send_log_message方法
+        await self.context.session.send_log_message(mcp_level, message)
+    
+    async def error(self, message: str, error_class=RuntimeError):
+        """记录错误并引发异常"""
+        await self.log("error", message)
+        raise error_class(message)
+        
+    def get_result_info(self) -> List[str]:
+        """获取收集的信息日志"""
+        return self._info_log
+
+class FileHandler:
+    """文件处理工具类"""
+    def __init__(self, logger: Logger):
+        self.logger = logger
+
+    @staticmethod
+    def is_url(path: str) -> bool:
+        """检查路径是否为URL"""
+        return path.startswith(("http://", "https://"))
+
+    @staticmethod
+    def get_file_extension(file_path: str) -> str:
+        """获取文件扩展名（小写）"""
+        if "?" in file_path:  # 处理URL中的查询参数
+            file_path = file_path.split("?")[0]
+        return os.path.splitext(file_path)[1].lower()
+        
+    @staticmethod
+    def get_input_format(file_path: str):
+        """根据文件路径获取输入格式
+        
+        此方法需要导入InputFormat和INPUT_EXTENSIONS，
+        但为避免循环导入，由调用者提供转换逻辑
+        """
+        ext = FileHandler.get_file_extension(file_path)
+        return ext
+        
+    @staticmethod
+    def get_available_output_formats(input_format):
+        """获取指定输入格式支持的输出格式
+        
+        此方法需要导入FORMAT_CONVERSION_MAP，
+        但为避免循环导入，由调用者提供转换逻辑
+        """
+        # 实际实现在converter.py
+        return {}
+
+    async def validate_file_exists(self, file_path: str) -> Tuple[bool, bool]:
+        """验证文件是否存在
+        
+        Args:
+            file_path: 文件路径
+            
+        Returns:
+            tuple: (文件是否存在, 是否为URL)
+        """
+        is_url = self.is_url(file_path)
+        if not is_url and not os.path.exists(file_path):
+            await self.logger.error(f"文件不存在：{file_path}", FileNotFoundError)
+            return False, is_url
+        return True, is_url
+
+class BaseApiClient:
+    """API客户端基类"""
+    def __init__(self, logger: Logger, file_handler: FileHandler):
+        self.logger = logger
+        self.file_handler = file_handler
+        self.api_key = os.getenv("API_KEY")
+        # 子类必须设置api_base_url
+        self.api_base_url = None
+
+    async def _wait_for_task(self, client: httpx.AsyncClient, task_id: str, operation_type: str = "处理") -> str:
+        """等待任务完成并返回下载链接
+        
+        Args:
+            client: HTTP客户端
+            task_id: 任务ID
+            operation_type: 操作类型描述，用于日志，默认为"处理"
+            
+        Returns:
+            str: 下载链接
+            
+        Raises:
+            RuntimeError: 如果任务失败或超时
+        """
+        headers = {"X-API-KEY": self.api_key}
+        MAX_ATTEMPTS = 100
+        
+        for attempt in range(MAX_ATTEMPTS):
+            await asyncio.sleep(3)
+            
+            status_response = await client.get(
+                f"{self.api_base_url}/{task_id}",
+                headers=headers
+            )
+            
+            if status_response.status_code != 200:
+                await self.logger.log("warning", f"获取任务状态失败。状态码: {status_response.status_code}")
+                continue
+            
+            status_result = status_response.json()
+            state = status_result.get("data", {}).get("state")
+            progress = status_result.get("data", {}).get("progress", 0)
+            
+            if state == 1:  # 完成
+                download_url = status_result.get("data", {}).get("file")
+                if not download_url:
+                    await self.logger.error(f"任务完成但未找到下载链接。任务状态：{json.dumps(status_result, ensure_ascii=False)}")
+                return download_url
+            elif state < 0:  # 失败
+                await self.logger.error(f"任务失败: {json.dumps(status_result, ensure_ascii=False)}")
+            else:  # 进行中
+                await self.logger.log("debug", f"{operation_type}进度: {progress}%", add_to_result=False)
+        
+        await self.logger.error(f"超过最大尝试次数（{MAX_ATTEMPTS}），任务未完成")
+
+    async def _handle_api_response(self, response: httpx.Response, error_prefix: str) -> str:
+        """处理API响应并提取任务ID
+        
+        Args:
+            response: API响应
+            error_prefix: 错误消息前缀
+            
+        Returns:
+            str: 任务ID
+            
+        Raises:
+            RuntimeError: 如果响应无效或任务ID缺失
+        """
+        if response.status_code != 200:
+            await self.logger.error(f"{error_prefix}失败。状态码: {response.status_code}\n响应: {response.text}")
+        
+        result = response.json()
+        if "data" not in result or "task_id" not in result["data"]:
+            await self.logger.error(f"无法获取任务ID。API响应：{json.dumps(result, ensure_ascii=False)}")
+        
+        return result["data"]["task_id"]