# -*- coding: utf-8 -*-
"""
继续截取微信通讯录 - 从当前位置继续滚动到底
"""
import uiautomation as auto
import time
import os
from PIL import Image


def find_wechat_window():
    """查找微信窗口"""
    wechat_window = auto.WindowControl(searchDepth=1, Name='微信')
    if wechat_window.Exists(3, 1):
        return wechat_window
    return None


def capture_contact_region(wechat_window, index, save_dir):
    """截取通讯录区域"""
    rect = wechat_window.BoundingRectangle
    
    x_offset = 70
    y_offset = 130
    width = 280
    height = rect.height() - 160
    
    screenshot_path = os.path.join(save_dir, f"region_{index:03d}.png")
    
    try:
        bitmap = wechat_window.ToBitmap(x=x_offset, y=y_offset, width=width, height=height)
        bitmap.ToFile(screenshot_path)
        return screenshot_path
    except Exception as e:
        print(f"截图失败: {e}")
        return None


def get_image_hash(image_path):
    """计算图片哈希"""
    try:
        img = Image.open(image_path)
        img = img.resize((16, 16), Image.Resampling.LANCZOS)
        img = img.convert('L')
        pixels = list(img.get_flattened_data())
        avg = sum(pixels) / len(pixels)
        return ''.join(['1' if p > avg else '0' for p in pixels])
    except:
        return None


def images_similarity(hash1, hash2):
    """计算相似度"""
    if not hash1 or not hash2:
        return 0
    diff = sum(c1 != c2 for c1, c2 in zip(hash1, hash2))
    return 1 - diff / len(hash1)


def scroll_down(wechat_window):
    """向下滚动一页"""
    try:
        rect = wechat_window.BoundingRectangle
        center_x = rect.left + 200
        center_y = rect.top + 400
        
        auto.SetCursorPos(center_x, center_y)
        auto.Click(center_x, center_y)
        time.sleep(0.2)
        auto.WheelDown(wheelTimes=3)
        time.sleep(0.4)
        return True
    except:
        return False


def main():
    print("=" * 60)
    print("继续截取微信通讯录")
    print("=" * 60)
    
    # 使用已有的截图目录
    save_dir = r"D:\夏骥\微信研究\scroll_full"
    
    # 获取已有截图数量
    existing_files = [f for f in os.listdir(save_dir) if f.endswith('.png')]
    start_index = len(existing_files)
    
    print(f"已有 {start_index} 张截图")
    print(f"将从第 {start_index + 1} 张开始继续...")
    
    # 查找微信窗口
    print("\n查找微信窗口...")
    wechat_window = find_wechat_window()
    if not wechat_window:
        print("未找到微信窗口！")
        return
    
    print(f"找到微信窗口: {wechat_window.Name}")
    
    # 提示用户
    print("\n请确保微信通讯录界面已打开并滚动到之前的位置...")
    time.sleep(3)
    
    # 开始截图
    print("\n开始继续截图...")
    screenshots = []
    last_hash = None
    no_change = 0
    max_screenshots = 100
    
    for i in range(max_screenshots):
        actual_index = start_index + i
        path = capture_contact_region(wechat_window, actual_index, save_dir)
        if path:
            screenshots.append(path)
            print(f"  截图 {actual_index + 1}")
            
            # 检测是否到底
            current_hash = get_image_hash(path)
            if last_hash:
                sim = images_similarity(last_hash, current_hash)
                if sim > 0.95:
                    no_change += 1
                    if no_change >= 2:
                        print(f"\n检测到到底，共新增截图 {len(screenshots)} 张")
                        break
                else:
                    no_change = 0
            last_hash = current_hash
        
        scroll_down(wechat_window)
    
    print(f"\n完成！新增 {len(screenshots)} 张截图")
    print(f"总截图数: {start_index + len(screenshots)}")
    print(f"截图保存目录: {save_dir}")
    
    # 提示运行OCR
    print("\n接下来请运行 batch_ocr_full.py 进行OCR识别")


if __name__ == '__main__':
    main()