"""用 session_name 模块跑一遍真实窗口：标题和列表各行的昵称、md5、耗时。"""
import os
import sys
import time

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

import numpy as np  # noqa: E402
import session_name  # noqa: E402
import wechat_bot as bot_module  # noqa: E402


def main() -> None:
    bot = bot_module.WeChatBot()
    bot.safe_window_mode = True
    bot.auto_activate_window = False
    if not bot.connect(activate=False, wait_if_missing=False):
        print("[-] 未挂载到企业微信窗口")
        return

    reader = session_name.NameReader()
    if not reader.available:
        print("[-] OCR 引擎不可用")
        return

    scale = max(0.75, float(bot.scale or 1.0))
    full = bot._capture_full_window()
    x1 = int(bot._list_x + bot._list_w)

    print("=" * 78)
    print("当前打开的会话（聊天标题）")
    print("=" * 78)
    panel = np.ascontiguousarray(
        full[int(20 * scale):int(78 * scale),
             x1 + int(8 * scale):x1 + int(600 * scale), :3]
    )
    start = time.perf_counter()
    name = reader.read(panel)
    cost = (time.perf_counter() - start) * 1000
    print(f"  昵称   {name!r}")
    print(f"  会话ID {session_name.session_id_for(name)}")
    print(f"  耗时   {cost:.0f} ms")

    print()
    print("=" * 78)
    print("会话列表")
    print("=" * 78)
    listing = bot.capture_session_list()
    item_h = max(1, int(bot.session_item_h))
    rows = max(1, listing.shape[0] // item_h)
    total = 0.0
    for index in range(min(rows, 10)):
        y0 = index * item_h
        panel = np.ascontiguousarray(
            listing[y0 + int(item_h * 0.14):y0 + int(item_h * 0.50),
                    int(72 * scale):int(310 * scale), :3]
        )
        start = time.perf_counter()
        name = reader.read(panel)
        cost = (time.perf_counter() - start) * 1000
        total += cost
        sid = session_name.session_id_for(name)
        print(f"  第 {index} 行  {cost:5.0f} ms  {name!r:<24} {sid[:16] if sid else '(读不到)'}")
    print(f"  整列共 {total:.0f} ms")

    print()
    print(f"已认识的昵称: {reader.known_names()}")


if __name__ == "__main__":
    main()