import subprocess
import time
import re
from utils.logger_until import Logger

class SmsUtils:
    def __init__(self):
        self.logger = Logger(name=__name__).logger

    def get_sms_data_from_device(self):
        """從安卓設(shè)備讀取所有短信"""
        command = "adb shell content query --uri content://sms/inbox --projection _id,address,date,body"
        result = subprocess.run(command, shell=True, capture_output=True, text=True, encoding="utf-8")
        
        if result.returncode != 0:
            raise Exception("設(shè)備獲取短信失敗，請(qǐng)檢查ADB連接")
        
        # 解析短信數(shù)據(jù)
        parsed_sms = []
        for line in result.stdout.splitlines():
            sms_dict = {}
            for item in line.split(","):
                key_value = item.split("=")
                if len(key_value) == 2:
                    sms_dict[key_value[0].strip()] = key_value[1].strip()
            parsed_sms.append(sms_dict)
        
        return parsed_sms

    @staticmethod
    def filter_sms_by_keyword(sms_data, keyword):
        """根據(jù)關(guān)鍵字篩選短信"""
        return [sms for sms in sms_data if keyword in sms.get("body", "")]

    def get_latest_sms_by_timestamp(self, sms_data):
        """獲取最新時(shí)間戳的短信"""
        sms_data_with_date = [sms for sms in sms_data if 'date' in sms]
        sorted_sms = sorted(sms_data_with_date, key=lambda x: int(x['date']), reverse=True)
        
        if not sorted_sms:
            return None
            
        # 驗(yàn)證短信時(shí)效性（10分鐘內(nèi)）
        latest_sms = sorted_sms[0]
        sms_timestamp = int(latest_sms['date'])
        current_timestamp = int(time.time() * 1000)
        time_diff = current_timestamp - sms_timestamp
        
        if time_diff <= 600000:  # 10分鐘
            return latest_sms
        else:
            self.logger.error(f"驗(yàn)證碼已過期，時(shí)間差：{time_diff//60000}分鐘")
            return False

    def extract_verification_code(self, sms_body):
        """從短信內(nèi)容提取驗(yàn)證碼"""
        try:
            if isinstance(sms_body, dict):
                number_body = str(sms_body.get("body", ""))
                self.logger.info(f"原始短信內(nèi)容: {number_body}")
                match = re.search(r"\d{6}", number_body)
                if match:
                    return match.group(0)
            return None
        except Exception as e:
            self.logger.error(f"驗(yàn)證碼提取錯(cuò)誤: {e}")
            raise

    @staticmethod
    def get_latest_verification_code(keyword="魔意科技"):
        """獲取最新驗(yàn)證碼（整合方法）"""
        sms = SmsUtils()
        sms_data = sms.get_sms_data_from_device()
        filtered_sms = sms.filter_sms_by_keyword(sms_data, keyword)
        latest_sms = sms.get_latest_sms_by_timestamp(filtered_sms)
        
        if latest_sms:
            return sms.extract_verification_code(latest_sms)
        else:
            return False

import time
from functools import wraps
from utils.sms_util import SmsUtils

# 重試裝飾器
def retry(times=3, delay=2):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            last_exception = None
            for attempt in range(times):
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    last_exception = e
                    time.sleep(delay)
                    print(f"第{attempt+1}次重試失敗，剩余{times - attempt - 1}次重試")
            if last_exception:
                raise last_exception
        return wrapper
    return decorator

class LoginPage(BasePage):
    def __init__(self, device_manager):
        super().__init__(device_manager)
        self.logger = Logger(name=__name__).logger
    
    @retry(times=3, delay=15)
    def auto_fill_verification_code(self, keyword="魔意科技"):
        """自動(dòng)填充驗(yàn)證碼（帶重試機(jī)制）"""
        verification_code = SmsUtils.get_latest_verification_code(keyword)
        if verification_code:
            self.logger.info(f"驗(yàn)證碼提取成功: {verification_code}")
            self.input_text("phone_code_input", verification_code)
            self.logger.info(f"驗(yàn)證碼已自動(dòng)填充")
        else:
            self.logger.error("未找到有效驗(yàn)證碼")
            raise ValueError("驗(yàn)證碼提取失敗")
    
    def _perform_login_flow(self, account: str, locator_prefix: str):
        """執(zhí)行登錄流程"""
        # 確保在正確的登錄方式頁面
        if not self.assert_element_visible(f"{locator_prefix}_input"):
            self.click_element(f"switch_to_{locator_prefix}")
        
        # 輸入賬號(hào)并獲取驗(yàn)證碼
        self.input_text(f"{locator_prefix}_input", account)
        self.click_element("phone_code_get_captcha")
        
        # 等待并自動(dòng)填充驗(yàn)證碼
        time.sleep(10)  # 等待短信發(fā)送
        self.auto_fill_verification_code()
        
        # 后續(xù)登錄操作...

adb shell content query --uri content://sms/inbox --projection _id,address,date,body