# -*- coding: utf-8 -*-
"""
Copyright (c) 2026 KenanZhu.
All rights reserved.

This software is provided "as is", without any warranty of any kind.
You may use, modify, and distribute this file under the terms of the MIT License.
See the LICENSE file for details.
"""
import base64
import queue

import ddddocr
from base.MsgBase import MsgBase
from pages.LoginPage import LoginPage


class CaptchaSolver(MsgBase):

    def __init__(
        self,
        input_queue: queue.Queue,
        output_queue: queue.Queue,
    ) -> None:

        super().__init__(input_queue, output_queue)
        self._ocr = ddddocr.DdddOcr()

    def _autoRecognize(
        self,
        login_page: LoginPage,
    ) -> str:

        try:
            img_src = login_page.getCaptchaImageSrc()
            if img_src is None:
                self._showTrace("验证码图片元素定位时发生错误 !", self.TraceLevel.ERROR)
                return ""
            base64_str = img_src.split(',', 1)[1]
            captcha_img = base64.b64decode(base64_str)
            captcha_text = self._ocr.classification(captcha_img)
            captcha_text = ''.join(filter(str.isalnum, captcha_text)).lower()
            self._showTrace(f"识别到验证码为 : '{captcha_text}'", 20, no_log=True)
            if len(captcha_text) != 4:
                self._showLog("识别到的验证码长度不等于 4 个字符 !", self.TraceLevel.WARNING)
                return ""
            return captcha_text
        except ValueError as e:
            self._showTrace(f"验证码识别失败 ! : {e}", self.TraceLevel.ERROR)
            return ""

    def _manualRecognize(
        self,
    ) -> str:

        self._showMsg("请输入验证码:")
        captcha_text = self._waitMsg(timeout=15)
        self._showTrace(f"输入的验证码为 : '{captcha_text}'", 20, no_log=True)
        if len(captcha_text) != 4:
            self._showLog("输入的验证码长度不等于 4 个字符 !", self.TraceLevel.WARNING)
            return ""
        return captcha_text

    def solveCaptcha(
        self,
        login_page: LoginPage,
        auto_captcha: bool = True,
    ) -> str:

        max_attempts = 3
        for _ in range(max_attempts):
            if auto_captcha:
                captcha_text = self._autoRecognize(login_page)
            else:
                self._showTrace("用户未配置自动识别验证码, 请手动输入验证码 !", 20, no_log=True)
                captcha_text = self._manualRecognize()
            if captcha_text:
                return captcha_text
            else:
                if not login_page.refreshCaptcha():
                    return ""
        self._showTrace(
            f"验证码识别失败 {max_attempts} 次, 达到最大尝试次数 !",
            self.TraceLevel.WARNING,
        )
        return ""