新增代码

difyPlugin/app/api/workcase/QrCodeAPI.py

@@ -1,10 +1,14 @@
-"""二维码相关接口"""
-from fastapi import APIRouter
+"""二维码相关接口 - API层"""
+from fastapi import APIRouter, File, UploadFile
 
 from app.schemas import ResultDomain
+from app.services.workcase.qrcode import QrCodeService
 
 router = APIRouter()
 
+# 初始化服务
+qrcode_service = QrCodeService()
+
 
 @router.post(
     "/generate",
@@ -12,27 +16,135 @@ router = APIRouter()
     summary="生成二维码",
     description="根据内容生成二维码"
 )
-async def generate_qrcode(content: str) -> ResultDomain[dict]:
+async def generate_qrcode(
+    content: str,
+    size: int = 300,
+    error_correction: str = "H"
+) -> ResultDomain[dict]:
     """
     生成二维码
-    
+
     - **content**: 二维码内容
+    - **size**: 图片大小（像素，100-2000）
+    - **error_correction**: 纠错级别
+        - L: 7% 容错
+        - M: 15% 容错
+        - Q: 25% 容错
+        - H: 30% 容错 (推荐)
     """
-    # TODO: 实现二维码生成逻辑
-    return ResultDomain.success(message="生成成功", data={"content": content})
+    result = await qrcode_service.generate_qrcode(
+        content=content,
+        size=size,
+        error_correction=error_correction
+    )
+
+    if result["success"]:
+        return ResultDomain.success(message="生成成功", data=result)
+    else:
+        return ResultDomain.fail(message=result.get("error", "生成失败"))
 
 
 @router.post(
     "/parse",
     response_model=ResultDomain[dict],
     summary="解析二维码",
-    description="解析二维码图片内容"
+    description="解析二维码图片内容（支持URL、base64）"
 )
-async def parse_qrcode(image_url: str) -> ResultDomain[dict]:
+async def parse_qrcode(
+    image_source: str,
+    strategy: str = "auto"
+) -> ResultDomain[dict]:
     """
     解析二维码
-    
-    - **image_url**: 二维码图片URL
+
+    - **image_source**: 图片来源
+        - URL: http://... 或 https://...
+        - base64: data:image/...;base64,...
+        - 本地路径: /path/to/image.png
+    - **strategy**: 预处理策略
+        - basic: 基础模式，仅尝试原图和灰度图
+        - auto: 自动模式，尝试多种预处理方法 (推荐)
+        - enhanced: 增强模式，使用更多预处理技术
+        - all: 全部模式，尝试所有可能的预处理方法（包括多尺度）
     """
-    # TODO: 实现二维码解析逻辑
-    return ResultDomain.success(message="解析成功", data={"result": ""})
+    result = await qrcode_service.parse_qrcode(
+        image_source=image_source,
+        strategy=strategy
+    )
+
+    if result["success"]:
+        return ResultDomain.success(message="解析成功", data=result)
+    else:
+        return ResultDomain.fail(
+            message=result.get("error", "解析失败"),
+            data={"total_attempts": result.get("total_attempts", 0)}
+        )
+
+
+@router.post(
+    "/parse-file",
+    response_model=ResultDomain[dict],
+    summary="解析二维码文件",
+    description="通过文件上传解析二维码"
+)
+async def parse_qrcode_file(
+    file: UploadFile = File(...),
+    strategy: str = "auto"
+) -> ResultDomain[dict]:
+    """
+    解析二维码文件上传
+
+    - **file**: 二维码图片文件（支持 png/jpg/jpeg/bmp 等格式）
+    - **strategy**: 预处理策略 (basic/auto/enhanced/all)
+    """
+    # 读取文件内容
+    content = await file.read()
+
+    # 提取文件类型
+    if file.content_type:
+        file_type = file.content_type.split("/")[-1]
+    else:
+        # 从文件名提取扩展名
+        file_type = file.filename.split(".")[-1] if file.filename else "png"
+
+    # 调用服务
+    result = await qrcode_service.parse_qrcode_from_file(
+        file_content=content,
+        file_type=file_type,
+        strategy=strategy
+    )
+
+    if result["success"]:
+        return ResultDomain.success(message="解析成功", data=result)
+    else:
+        return ResultDomain.fail(
+            message=result.get("error", "解析失败"),
+            data={"total_attempts": result.get("total_attempts", 0)}
+        )
+
+
+@router.post(
+    "/validate",
+    response_model=ResultDomain[dict],
+    summary="验证二维码内容",
+    description="验证内容是否适合生成二维码"
+)
+async def validate_qrcode_content(
+    content: str,
+    max_length: int = 2953
+) -> ResultDomain[dict]:
+    """
+    验证二维码内容
+
+    - **content**: 要验证的内容
+    - **max_length**: 最大长度（字节）
+    """
+    result = qrcode_service.validate_qrcode_content(content, max_length)
+
+    if result["valid"]:
+        return ResultDomain.success(
+            message="内容有效",
+            data={"length": result["length"]}
+        )
+    else:
+        return ResultDomain.fail(message=result.get("error", "内容无效"))