import pyautogui
import time
import subprocess
import sys
from pathlib import Path
from typing import Optional, Tuple

BASE_DIR = Path(__file__).parent.parent.parent
sys.path.insert(0, str(BASE_DIR))

from wechat_auto.config import settings
from wechat_auto.utils.logger import logger


class DesktopAutomation:
    """桌面自动化操作 - 点击微信、进入小程序、打开一见星球"""
    
    def __init__(self):
        pyautogui.FAILSAFE = settings.failsafe
        pyautogui.PAUSE = 0.5
        self.screenshot_dir = Path(settings.screenshot_dir)
        self.screenshot_dir.mkdir(parents=True, exist_ok=True)
        
        # 微信窗口已知位置（从xwininfo获取）
        self.wechat_window = {
            'x': 877,
            'y': 207,
            'width': 980,
            'height': 710
        }
    
    def click_at(self, x: int, y: int):
        """在指定位置点击"""
        pyautogui.click(x, y)
        logger.info(f"点击坐标：({x}, {y})")
    
    def screenshot(self, name: str):
        """截图保存"""
        filepath = self.screenshot_dir / f"{name}_{time.strftime('%Y%m%d_%H%M%S')}.png"
        try:
            subprocess.run(['scrot', str(filepath)], capture_output=True, timeout=5)
        except FileNotFoundError:
            try:
                subprocess.run(['gnome-screenshot', '-f', str(filepath)], capture_output=True, timeout=5)
            except:
                pass
        logger.info(f"截图：{filepath}")
    
    def get_screen_size(self) -> Tuple[int, int]:
        """获取屏幕尺寸"""
        return pyautogui.size()
    
    def open_wechat_and_miniprogram(self) -> bool:
        """
        打开微信并进入一见星球小程序
        流程：
        1. 点击桌面微信图标
        2. 等待微信窗口
        3. 点击左侧小程序图标
        4. 点击一见星球
        """
        screen_width, screen_height = self.get_screen_size()
        logger.info(f"屏幕尺寸：{screen_width}x{screen_height}")
        
        self.screenshot("step0_start")
        
        # 步骤1：点击桌面微信图标
        logger.info("步骤1：点击桌面微信图标")
        self.click_at(int(screen_width * 0.15), int(screen_height * 0.15))
        time.sleep(4)
        self.screenshot("step1_click_wechat")
        
        # 步骤2：点击左侧小程序图标
        # 微信窗口内相对位置
        wx = self.wechat_window['x']
        wy = self.wechat_window['y']
        ww = self.wechat_window['width']
        wh = self.wechat_window['height']
        
        logger.info("步骤2：点击左侧小程序图标")
        # 小程序图标在左侧边栏，约为窗口宽度的4%，高度的22%
        mini_x = wx + int(ww * 0.04)
        mini_y = wy + int(wh * 0.22)
        self.click_at(mini_x, mini_y)
        time.sleep(2)
        self.screenshot("step2_miniprogram_panel")
        
        # 步骤3：点击一见星球小程序
        # 一见星球在主面板中，约为窗口宽度的35%，高度的25%
        logger.info("步骤3：点击一见星球小程序")
        planet_x = wx + int(ww * 0.35)
        planet_y = wy + int(wh * 0.25)
        self.click_at(planet_x, planet_y)
        time.sleep(3)
        self.screenshot("step3_yijian_planet")
        
        logger.info("✅ 已成功打开一见星球小程序！")
        return True


if __name__ == "__main__":
    automation = DesktopAutomation()
    
    print("=" * 50)
    print("开始执行：打开微信 -> 进入小程序 -> 一见星球")
    print("=" * 50)
    
    result = automation.open_wechat_and_miniprogram()
    
    if result:
        print("\n✅ 成功完成！")
    else:
        print("\n❌ 执行失败，请检查日志")