import base64

from loguru import logger
from PyQt6.QtCore import Qt, QThread, pyqtSignal
from PyQt6.QtGui import (QColor, QPainter, QPainterPath, QPen, QPixmap)

from ..api import miaoStarsBasicApi
from ...core import cfg, qconfig, userConfig


class CaptchaThread(QThread):
    captchaReady = pyqtSignal(QPixmap)
    captchaFailed = pyqtSignal(str)

    def __init__(self):
        super().__init__()

    @staticmethod
    def _createRoundedPixmap(pixmap, radius=10):
        """创建圆角图片"""
        try:
            # 获取原始图片尺寸
            if pixmap.isNull():
                logger.error("原始图片为空，无法创建圆角图片")
                return pixmap
            size = pixmap.size()
            # 创建透明背景的图片
            rounded_pixmap = QPixmap(size)
            rounded_pixmap.fill(Qt.GlobalColor.transparent)
            # 创建对象
            painter = QPainter(rounded_pixmap)
            painter.setRenderHints(
                QPainter.RenderHint.Antialiasing
                | QPainter.RenderHint.SmoothPixmapTransform
            )
            # 创建圆角矩形路径
            path = QPainterPath()
            path.addRoundedRect(0, 0, size.width(), size.height(), radius, radius)
            # 设置裁剪区域
            painter.setClipPath(path)
            # 绘制原始图片
            painter.drawPixmap(0, 0, pixmap)
            # 绘制边框
            pen = QPen(QColor(200, 200, 200))  # 浅灰色边框
            pen.setWidth(1)
            painter.setPen(pen)
            painter.drawRoundedRect(
                0, 0, size.width() - 1, size.height() - 1, radius, radius
            )

            painter.end()
            return rounded_pixmap
        except Exception as e:
            logger.error(f"创建圆角图片失败：{e}")
            return pixmap  # 如果出错，返回原始图片

    def run(self):
        try:
            logger.debug("开始获取验证码")
            response = miaoStarsBasicApi.getCaptcha()
            logger.debug(f"验证码API返回响应: {response}")
            
            if response["code"] == 0:
                # 确保data字段存在且为字符串
                if "data" in response and isinstance(response["data"], str):
                    # 分割base64前缀和实际数据
                    try:
                        captchaImageData = response["data"].split(",")[1]
                        logger.debug(f"成功提取base64数据，长度: {len(captchaImageData)}")
                        
                        # 解码base64数据
                        captchaImage = base64.b64decode(captchaImageData)
                        logger.debug(f"成功解码base64数据，长度: {len(captchaImage)} bytes")
                        
                        # 加载图片
                        pixmap = QPixmap()
                        load_success = pixmap.loadFromData(captchaImage)
                        
                        if load_success:
                            logger.debug(f"成功加载图片，尺寸: {pixmap.width()}x{pixmap.height()}")
                            # 创建圆角图片
                            pixmap = self._createRoundedPixmap(pixmap, radius=10)
                            self.captchaReady.emit(pixmap)
                        else:
                            logger.error("图片加载失败")
                            self.captchaFailed.emit("验证码图片加载失败")
                            self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
                    except (IndexError, ValueError, TypeError) as e:
                        logger.error(f"验证码数据格式错误: {e}")
                        self.captchaFailed.emit(f"验证码数据格式错误: {str(e)}")
                        self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
                else:
                    logger.error("验证码响应中缺少有效的data字段")
                    self.captchaFailed.emit("验证码数据无效")
                    self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
            else:
                error_msg = response.get("msg", "获取验证码失败")
                logger.error(f"获取验证码失败: {error_msg}")
                self.captchaFailed.emit(error_msg)
                self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
        except Exception as e:
            logger.exception(f"获取验证码过程中发生异常: {e}")
            self.captchaFailed.emit(str(e))
            self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))


class LoginThread(QThread):
    successLogin = pyqtSignal()
    errorLogin = pyqtSignal(str)

    def __init__(self, email: str, password: str, captchaCode: str):
        super().__init__()
        logger.debug(f"初始化许可证服务线程 - 邮箱: {email}")
        self.email = email
        self.password = password
        self.captchaCode = captchaCode

    def run(self):
        logger.info(f"开始验证用户登录 - 邮箱: {self.email}")
        try:
            loginResponse = miaoStarsBasicApi.login(
                self.email, self.password, self.captchaCode
            )
            if loginResponse["code"] == 0:
                self.successLogin.emit()
                qconfig.set(cfg.email, self.email)
                qconfig.set(cfg.activationCode, self.password)
                userConfig.userData = loginResponse
                # 从登录响应中提取token并保存
                # 在basicApi的login方法中已经处理了token的设置，这里保存到userConfig中以便程序启动时恢复
                token = miaoStarsBasicApi.token
                if token:
                    userConfig.setToken(token)
            else:
                self.errorLogin.emit(loginResponse["msg"])

        except Exception as e:
            logger.error(f"登录验证过程中发生异常: {e}")
            self.errorLogin.emit("系统错误，请稍后重试")


class RegisterThread(QThread):
    successRegister = pyqtSignal()
    errorRegister = pyqtSignal(str)

    def __init__(self, email: str, password: str, captchaCode: str):
        super().__init__()
        logger.debug(f"初始化许可证服务线程 - 邮箱: {email}")
        self.email = email
        self.password = password
        self.captchaCode = captchaCode

    def run(self):
        logger.info(f"开始验证用户注册 - 邮箱: {self.email}")
        try:
            registerRespond = miaoStarsBasicApi.register(
                self.email, self.password, self.captchaCode
            )
            if registerRespond["code"] == 203:
                self.successRegister.emit()
            else:
                logger.error(f"注册失败: {registerRespond['msg']}")
                self.errorRegister.emit(registerRespond["msg"])
        except Exception as e:
            logger.error(f"登录验证过程中发生异常: {e}")
            self.errorRegister.emit("系统错误，请稍后重试")