urbanLifeline/difyPlugin/app/services/workcase/qrcode/QrCodeService.py

"""二维码服务层 - 提供统一的业务逻辑接口"""
import base64
from typing import Optional

from .QrCode import QRCodeProcessor


class QrCodeService:
    """二维码服务 - 业务逻辑层"""

    def __init__(self):
        """初始化服务"""
        self.processor = QRCodeProcessor()

    async def generate_qrcode(
        self,
        content: str,
        size: int = 300,
        error_correction: str = "H"
    ) -> dict:
        """
        生成二维码

        Args:
            content: 二维码内容
            size: 图片大小（像素）
            error_correction: 纠错级别 (L/M/Q/H)

        Returns:
            {
                "success": bool,
                "image": str,  # base64编码的图片
                "content": str,
                "size": int,
                "error_correction": str,
                "error": str  # 仅失败时有
            }
        """
        try:
            # 验证参数
            if not content:
                return {
                    "success": False,
                    "error": "内容不能为空"
                }

            if size < 100 or size > 2000:
                return {
                    "success": False,
                    "error": "尺寸必须在100-2000之间"
                }

            if error_correction not in ["L", "M", "Q", "H"]:
                return {
                    "success": False,
                    "error": "纠错级别必须是 L/M/Q/H 之一"
                }

            # 生成二维码
            img_base64 = self.processor.generate(
                content=content,
                size=size,
                error_correction=error_correction
            )

            return {
                "success": True,
                "image": img_base64,
                "content": content,
                "size": size,
                "error_correction": error_correction
            }

        except Exception as e:
            return {
                "success": False,
                "error": f"生成失败: {str(e)}"
            }

    async def parse_qrcode(
        self,
        image_source: str,
        strategy: str = "auto"
    ) -> dict:
        """
        解析二维码

        Args:
            image_source: 图片来源（URL/base64/本地路径）
            strategy: 预处理策略 (basic/auto/enhanced/all)

        Returns:
            {
                "success": bool,
                "content": str or None,
                "strategy_used": str,
                "total_attempts": int,
                "error": str  # 仅失败时有
            }
        """
        try:
            # 验证参数
            if not image_source:
                return {
                    "success": False,
                    "error": "图片来源不能为空"
                }

            if strategy not in ["basic", "auto", "enhanced", "all"]:
                return {
                    "success": False,
                    "error": "策略必须是 basic/auto/enhanced/all 之一"
                }

            # 解析二维码
            result = await self.processor.parse(image_source, strategy)

            if result["success"]:
                return result
            else:
                return {
                    "success": False,
                    "content": None,
                    "error": result.get("message", "解析失败"),
                    "total_attempts": result.get("total_attempts", 0)
                }

        except Exception as e:
            return {
                "success": False,
                "content": None,
                "error": f"解析失败: {str(e)}"
            }

    async def parse_qrcode_from_file(
        self,
        file_content: bytes,
        file_type: str = "png",
        strategy: str = "auto"
    ) -> dict:
        """
        从文件内容解析二维码

        Args:
            file_content: 文件二进制内容
            file_type: 文件类型 (png/jpg/jpeg等)
            strategy: 预处理策略

        Returns:
            解析结果（格式同parse_qrcode）
        """
        try:
            # 转换为base64
            img_base64 = base64.b64encode(file_content).decode()
            image_source = f"data:image/{file_type};base64,{img_base64}"

            # 调用解析方法
            return await self.parse_qrcode(image_source, strategy)

        except Exception as e:
            return {
                "success": False,
                "content": None,
                "error": f"文件解析失败: {str(e)}"
            }

    def validate_qrcode_content(self, content: str, max_length: int = 2953) -> dict:
        """
        验证二维码内容是否合法

        Args:
            content: 要验证的内容
            max_length: 最大长度（默认2953字节，version 40 with L级别）

        Returns:
            {
                "valid": bool,
                "length": int,
                "error": str  # 仅无效时有
            }
        """
        if not content:
            return {
                "valid": False,
                "error": "内容不能为空"
            }

        content_bytes = content.encode("utf-8")
        length = len(content_bytes)

        if length > max_length:
            return {
                "valid": False,
                "length": length,
                "error": f"内容过长，当前{length}字节，最大支持{max_length}字节"
            }

        return {
            "valid": True,
            "length": length
        }