"""网易云音乐二维码登录模块 (已修改为API)

提供网易云音乐二维码登录功能，被 main.py 调用。
- 二维码生成 (key 和 base64)
- 登录状态检查 (返回 code 和 cookie)
"""

import json
import logging
import base64
import io
import re                     # [!! 已添加 !!]
from random import randrange # [!! 关键修复 !!] 导入 randrange
from typing import Optional, Dict, Any, Tuple

try:
    # qrcode 库在 music_api.py 中被导入，这里我们也需要它
    import qrcode
except ImportError:
    print("错误：缺少 'qrcode' 库。请运行 `pip install qrcode`")
    qrcode = None

try:
    # 我们需要从 music_api 借用很多底层工具
    from music_api import (
        QRLoginManager, APIException, HTTPClient, 
        CryptoUtils, APIConstants
    )
    from cookie_manager import CookieManager, CookieException
except ImportError as e:
    print(f"导入模块失败: {e}")
    print("请确保 music_api.py 和 cookie_manager.py 文件存在且可用")


def api_generate_qr_key() -> Dict[str, Any]:
    """
    [新API] 生成二维码Key和Base64图像
    
    Returns:
        {'success': bool, 'qr_key': str, 'qr_img_b64': str, 'message': str}
    """
    if not qrcode:
        return {'success': False, 'message': "qrcode 库未安装"}
        
    try:
        qr_manager = QRLoginManager()
        unikey = qr_manager.generate_qr_key()
        
        if not unikey:
            raise APIException("生成二维码key失败")
            
        # 生成二维码数据
        qr_data = f'https://music.163.com/login?codekey={unikey}'
        
        # 在内存中生成二维码图片
        qr = qrcode.QRCode()
        qr.add_data(qr_data)
        qr.make(fit=True)
        img = qr.make_image(fill_color="black", back_color="white")
        
        # 将图片保存到 BytesIO
        buffered = io.BytesIO()
        img.save(buffered, format="PNG")
        
        # 转换为 Base64 字符串
        img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
        qr_img_b64 = f"data:image/png;base64,{img_str}"
        
        return {
            'success': True,
            'qr_key': unikey,
            'qr_img_b64': qr_img_b64
        }
        
    except Exception as e:
        logging.error(f"生成二维码失败: {e}")
        return {'success': False, 'message': str(e)}


def api_check_qr_status(unikey: str) -> Dict[str, Any]:
    """
    [新API] 检查二维码登录状态
    
    此函数复制并*修复*了 music_api.py 中 check_qr_login 的逻辑，
    以便返回完整的 Cookie 字符串，而不只是 MUSIC_U。
    
    Args:
        unikey: 二维码key
        
    Returns:
        {'code': int, 'cookie': Optional[str], 'message': str}
        code: 800=过期, 801=等待, 802=已扫码, 803=成功
    """
    try:
        http_client = HTTPClient()
        crypto_utils = CryptoUtils()
        config = APIConstants.DEFAULT_CONFIG.copy()
        
        # [!! 关键修复 !!] 现在 randrange 已经被导入，这行代码可以正常工作
        config["requestId"] = str(randrange(20000000, 30000000))
        
        payload = {
            'key': unikey,
            'type': 1,
            'header': json.dumps(config)
        }
        
        params = crypto_utils.encrypt_params(APIConstants.QR_LOGIN_API, payload)
        
        # 我们需要完整的 response 对象来获取 headers
        response = http_client.post_request_full(
            APIConstants.QR_LOGIN_API, params, {}
        )
        
        result = json.loads(response.text)
        code = result.get('code', -1)
        
        cookie_string = None
        
        if code == 803:
            # 登录成功，提取cookie
            raw_cookies = response.headers.get('Set-Cookie', '')
            
            # 使用 re.sub 来处理 'path=/,' 这种导致错误分割的情况
            raw_cookies = re.sub(r"path=/,", "path=/", raw_cookies)
            cookie_list = [c.strip() for c in raw_cookies.split(',') if c.strip()]
            
            final_cookie_parts = []
            for item in cookie_list:
                if not item:
                    continue
                part = item.split(';')[0]
                if part.strip():
                    final_cookie_parts.append(part.strip())
            
            cookie_string = '; '.join(final_cookie_parts)
            
            if "MUSIC_U" not in cookie_string:
                return {'code': -1, 'cookie': None, 'message': '登录成功但未获取到 MUSIC_U'}

        return {
            'code': code,
            'cookie': cookie_string,
            'message': result.get('message', '')
        }
        
    except Exception as e:
        # 在 `api_check_qr_status` 捕获异常时，记录详细的 traceback
        logging.error(f"检查登录状态响应失败: {e}", exc_info=True)
        return {'code': -1, 'cookie': None, 'message': f"检查失败: {e}"}

# --- 删除了所有旧的交互式代码 ---