# coding=utf-8
import os
import sys
import time

import cv2
import uiautomator2 as u2

# 添加项目根目录到 sys.path
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if project_root not in sys.path:
    sys.path.append(project_root)

from WeiXin.WxUtil import find_all_template_matches


def run_cv_debug():
    # 1. 拍照 (获取当前设备屏幕)
    print("📸 正在连接设备并截取屏幕...")
    try:
        d = u2.connect()
        screenshot_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Screenshots")
        if not os.path.exists(screenshot_dir):
            os.makedirs(screenshot_dir)
            
        image_path = os.path.join(screenshot_dir, "t6_live_shot.jpg")
        output_path = os.path.join(screenshot_dir, "T6_debug_view.jpg")
        
        d.screenshot(image_path)
        print(f"✅ 截图已保存: {image_path}")
    except Exception as e:
        print(f"❌ 拍照失败: {e}")
        return

    print(f"🔍 正在分析实时图片...")
    
    # 模板路径
    audio_template = r"d:\dsWork\aiData\WeiXin\Templates\audio.jpg"
    red_point_template = r"d:\dsWork\aiData\WeiXin\Templates\red_point.jpg"
    
    if not os.path.exists(audio_template) or not os.path.exists(red_point_template):
        print("错误: 模板文件不存在")
        return

    # 2. 识别逻辑
    audio_matches = find_all_template_matches(image_path, audio_template, threshold=0.8)
    red_points = find_all_template_matches(image_path, red_point_template, threshold=0.8)
    
    print(f"发现语音图标数量: {len(audio_matches)}")
    print(f"发现红点数量: {len(red_points)}")

    # 3. 读取图片并绘制
    img = cv2.imread(image_path)
    if img is None:
        print("错误: 无法读取图片")
        return

    for ax, ay in audio_matches:
        # 排除顶部标题栏和底部输入区 (假设 300-1800 为有效区)
        if ay < 300 or ay > 1800:
            continue
            
        sender = "对方" if ax < 500 else "我"
        is_unread = False
        
        # 1. 绘制语音图标绿框 (根据模板大小，假设 60x60)
        cv2.rectangle(img, (int(ax-30), int(ay-30)), (int(ax+30), int(ay+30)), (0, 255, 0), 3)
        
        # 2. 绘制模拟点击红点 (用户要求用红点标出模拟点击位置)
        cv2.circle(img, (int(ax), int(ay)), 15, (0, 0, 255), -1)
        
        # 3. 检查并标记未读红点 (如果存在)
        for rx, ry in red_points:
            if abs(ry - ay) < 50 and rx > ax:
                is_unread = True
                # 绘制未读红点
                cv2.circle(img, (int(rx), int(ry)), 12, (0, 0, 255), -1)
                break
        
        print(f"标注语音消息: ({ax}, {ay}), 发送者: {sender}, 未读: {is_unread}")

    # 保存结果
    cv2.imwrite(output_path, img)
    print(f"✅ 调试图片已保存至: {output_path}")

if __name__ == "__main__":
    run_cv_debug()