"""
图标捕获工具
用于捕获微信、小程序图标、一见星球等图标
"""
import pyautogui
import time
from pathlib import Path


def capture_icon(icon_name: str, delay: int = 5):
    """
    捕获图标
    1. 运行此函数
    2. 在延迟时间内将鼠标移动到目标图标上
    3. 程序会自动截图保存
    """
    template_dir = Path(__file__).parent / "images"
    template_dir.mkdir(parents=True, exist_ok=True)
    
    print(f"请在 {delay} 秒内将鼠标移动到要捕获的图标上...")
    time.sleep(delay)
    
    x, y = pyautogui.position()
    print(f"当前鼠标位置：({x}, {y})")
    
    screenshot = pyautogui.screenshot(region=(x-30, y-30, 60, 60))
    filepath = template_dir / f"{icon_name}.png"
    screenshot.save(str(filepath))
    print(f"图标已保存到：{filepath}")


if __name__ == "__main__":
    print("=== 微信图标捕获工具 ===\n")
    
    while True:
        print("\n请选择要捕获的图标:")
        print("1. 桌面微信图标 (wechat_icon)")
        print("2. 小程序图标 (miniprogram_icon)")
        print("3. 一见星球小程序 (yijian_planet_icon)")
        print("4. 手动指定名称")
        print("0. 退出")
        
        choice = input("\n选择：")
        
        if choice == "0":
            break
        elif choice == "1":
            capture_icon("wechat_icon", delay=5)
        elif choice == "2":
            capture_icon("miniprogram_icon", delay=5)
        elif choice == "3":
            capture_icon("yijian_planet_icon", delay=5)
        elif choice == "4":
            name = input("输入图标名称（不带扩展名）: ")
            capture_icon(name, delay=5)
        else:
            print("无效选择，请重试")