import json
import os
import random
import sys
import time
import math
import ctypes
import cv2
import numpy as np
import win32gui
import win32con

# 开启 DPI 意识
try:
    ctypes.windll.shcore.SetProcessDpiAwareness(1)
except Exception:
    ctypes.windll.user32.SetProcessDPIAware()

from hardware_control import hw_ctrl
from game_state import parse_game_state, load_layout_config
from stuck_handler import StuckHandler

# 定义按键常量
KEY_TAB = '3'
KEY_LOOT = '4'      # 假设你在游戏里设置了互动按键为 F
KEY_ATTACK = '2'    # 假设你的主攻击技能是 1
WIN_TITLE = "魔兽世界"

def _config_base():
    if getattr(sys, 'frozen', False):
        return os.path.dirname(sys.executable)
    return os.path.dirname(os.path.abspath(__file__))


def get_config_path(filename):
    base = _config_base()
    p = os.path.join(base, filename)
    if os.path.exists(p):
        return p
    if getattr(sys, 'frozen', False):
        meipass = getattr(sys, '_MEIPASS', '')
        if meipass:
            p2 = os.path.join(meipass, filename)
            if os.path.exists(p2):
                return p2
    return p


def move_cursor_hw(x, y, settle_sec=0.02):
    hw_ctrl.move_to(int(x), int(y))
    if settle_sec > 0:
        time.sleep(settle_sec)


def get_wow_client_rect(hwnd):
    client_left, client_top, client_right, client_bottom = win32gui.GetClientRect(hwnd)
    screen_left, screen_top = win32gui.ClientToScreen(hwnd, (client_left, client_top))
    screen_right, screen_bottom = win32gui.ClientToScreen(hwnd, (client_right, client_bottom))
    return screen_left, screen_top, screen_right, screen_bottom

class CursorManager:
    """通过图像识别判断鼠标图标类型"""
    def __init__(self):
        self.templates = {}
        self.handle_cache = {}
        self.log_path = get_config_path('cursor_recognition.log')
        self._load_templates()

    def _load_templates(self):
        # 强制使用纯相对路径，由 Python 自动处理 CWD，完美避开中文路径编码问题
        cursor_dir = get_config_path(os.path.join('images', 'cursor'))
        files = {'Point': 'Point.PNG', 'Attack': 'Attack.PNG', 'LootAll': 'LootAll.PNG', 'Skin': 'Skin.PNG'}
        for name, fname in files.items():
            path = os.path.join(cursor_dir, fname)
            if os.path.exists(path):
                try:
                    # 使用 numpy 读取二进制流再解码
                    img_array = np.fromfile(path, dtype=np.uint8)
                    img = cv2.imdecode(img_array, cv2.IMREAD_UNCHANGED)
                    if img is not None:
                        if img.shape[2] == 3:
                            img = cv2.cvtColor(img, cv2.COLOR_BGR2BGRA)
                        self.templates[name] = img
                except Exception as e:
                    print(f">>> [CursorMgr] 加载 {fname} 失败: {e}")
            else:
                print(f">>> [CursorMgr] 找不到文件: {path}")

    def get_type(self, hcursor):
        if not hcursor or hcursor == 0: return 'Other', 0.0
        if hcursor in self.handle_cache: return self.handle_cache[hcursor]
        res = self._identify(hcursor)
        self.handle_cache[hcursor] = res
        return res

    def _debug_log(self, message):
        try:
            with open(self.log_path, 'a', encoding='utf-8') as f:
                f.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} {message}\n")
        except Exception:
            pass

    def _normalize_for_match(self, img):
        if img is None:
            return None
        if len(img.shape) == 2:
            gray = img
        elif img.shape[2] == 4:
            alpha = img[:, :, 3].astype(np.float32) / 255.0
            bgr = img[:, :, :3].astype(np.float32)
            bg = np.full_like(bgr, 255.0)
            composed = (bgr * alpha[..., None]) + (bg * (1.0 - alpha[..., None]))
            gray = cv2.cvtColor(composed.astype(np.uint8), cv2.COLOR_BGR2GRAY)
        else:
            gray = cv2.cvtColor(img[:, :, :3], cv2.COLOR_BGR2GRAY)
        return cv2.GaussianBlur(gray, (3, 3), 0)

    def _edge_map(self, gray):
        if gray is None:
            return None
        return cv2.Canny(gray, 32, 96)

    def _identify(self, hcursor):
        import win32ui
        try:
            width, height = 48, 48
            hdc = win32gui.GetDC(0)
            dc = win32ui.CreateDCFromHandle(hdc)
            memdc = dc.CreateCompatibleDC()
            bitmap = win32ui.CreateBitmap()
            bitmap.CreateCompatibleBitmap(dc, width, height)
            memdc.SelectObject(bitmap)
            win32gui.DrawIconEx(memdc.GetSafeHdc(), 0, 0, hcursor, width, height, 0, 0, win32con.DI_NORMAL)
            bits = bitmap.GetBitmapBits(True)
            target = np.frombuffer(bits, dtype='uint8').reshape(height, width, 4)
            memdc.DeleteDC()
            win32gui.ReleaseDC(0, hdc)

            best_name = 'Other'
            max_score = 0.0
            
            best_scale = 1.0
            target_gray = self._normalize_for_match(target)
            target_edge = self._edge_map(target_gray)
            target_has_edge = target_edge is not None and np.count_nonzero(target_edge) > 0
            scales = [0.55, 0.65, 0.75, 0.85, 0.95, 1.0, 1.1, 1.2, 1.35, 1.5]
            for name, temp in self.templates.items():
                t_h, t_w = temp.shape[:2]
                for s in scales:
                    s_w, s_h = int(t_w * s), int(t_h * s)
                    if s_w > width or s_h > height:
                        continue
                    interp = cv2.INTER_AREA if s < 1.0 else cv2.INTER_CUBIC
                    res_temp = cv2.resize(temp, (s_w, s_h), interpolation=interp)
                    temp_gray = self._normalize_for_match(res_temp)
                    if temp_gray is None:
                        continue
                    gray_res = cv2.matchTemplate(target_gray, temp_gray, cv2.TM_CCOEFF_NORMED)
                    _, gray_score, _, _ = cv2.minMaxLoc(gray_res)
                    score = gray_score
                    temp_edge = self._edge_map(temp_gray)
                    if target_has_edge and temp_edge is not None and np.count_nonzero(temp_edge) > 0:
                        edge_res = cv2.matchTemplate(target_edge, temp_edge, cv2.TM_CCOEFF_NORMED)
                        _, edge_score, _, _ = cv2.minMaxLoc(edge_res)
                        score = max(score, edge_score)
                    if score > max_score:
                        max_score = score
                        best_name = name
                        best_scale = s
            
            if max_score > 0.2:
                self._debug_log(f"best={best_name} score={max_score:.3f} scale={best_scale:.2f}")
                print(f">>>> [雷达识别] 目标: {best_name} | 最高分: {max_score:.3f}")
            return best_name, max_score
        except Exception:
            return 'Other', 0.0

def load_attack_loop(path):
    """从 JSON 加载攻击循环配置。支持 hp_below_percent/hp_key、mp_below_percent/mp_key（0=不启用）"""
    if not path or not os.path.exists(path):
        return None
    try:
        with open(path, 'r', encoding='utf-8') as f:
            cfg = json.load(f)
        steps = cfg.get('steps') or []
        trigger = float(cfg.get('trigger_chance', 0.3))
        hp_below = int(cfg.get('hp_below_percent', 0)) or 0
        mp_below = int(cfg.get('mp_below_percent', 0)) or 0
        return {
            'trigger_chance': trigger,
            'steps': steps,
            'hp_below_percent': hp_below,
            'hp_key': (cfg.get('hp_key') or '').strip() or None,
            'mp_below_percent': mp_below,
            'mp_key': (cfg.get('mp_key') or '').strip() or None,
        }
    except Exception:
        return None


class AutoBot:
    def __init__(
        self,
        waypoints=None,
        attack_loop_path=None,
        skinning_wait_sec=None,
        enable_mouse_loot=True,
        turn_error_key=None,
        turn_error_hold_sec=None,
        distance_interact_pause_sec=None,
    ):
        self.last_tab_time = 0
        self.last_interaction_time = 0
        self.last_target_hp = 0
        self._has_braked_for_target = False
        self._was_in_combat_or_target = False
        self.is_running = True
        self.attack_loop_config = load_attack_loop(attack_loop_path)
        self.tab_no_target_count = 0  # 连续按 Tab 仍无目标的次数，满 5 次则转向
        self.stuck_handler = StuckHandler()
        self.target_acquired_time = None  # 本次获得目标的时间（目前仅用于调试/保留接口）
        self.last_turn_end_time = 0  # 最近一次结束 A/D 转向的时间，供卡死检测排除
        self.skinning_wait_sec = float(skinning_wait_sec) if skinning_wait_sec is not None else 1.5
        self.enable_mouse_loot = enable_mouse_loot
        self.cursor_mgr = CursorManager()
        self.last_turn_signal_time = 0.0
        self.turn_error_key = (turn_error_key or 's').strip().lower() or 's'
        self.combat_signal_retry_sec = 1.0
        self.turn_error_hold_sec = float(turn_error_hold_sec) if turn_error_hold_sec is not None else 0.8
        self.distance_interact_pause_until = 0.0
        self.distance_interact_brake_pending = False
        self.distance_interact_pause_sec = (
            float(distance_interact_pause_sec) if distance_interact_pause_sec is not None else 1.0
        )
        self.distance_interact_brake_sec = 0.08
        self._last_mouse_path_scale_signature = None

    def execute_disengage_loot(self):
        """从有战斗/目标切换到完全脱战的瞬间，执行拾取 + 剥皮。"""
        try:
            # 1. 拟人化鼠标扫雷补漏拾取
            if self.enable_mouse_loot:
                self.mouse_sweep_loot()
            else:
                hw_ctrl.press(KEY_LOOT)
                time.sleep(0.5)
                # 关闭扫雷时，保留原有的补按交互和固定等待
                hw_ctrl.press(KEY_LOOT)
                time.sleep(self.skinning_wait_sec + 0.5)
        except Exception:
            pass

    def _load_mouse_path_points(self, client_width, client_height):
        path_points = []
        layout = load_layout_config()
        base_width = int(layout.get('mouse_path_base_window_width', 2560) or 2560)
        base_height = int(layout.get('mouse_path_base_window_height', 1600) or 1600)
        path_file = get_config_path("loot_path.json")
        if os.path.exists(path_file):
            with open(path_file, 'r', encoding='utf-8') as f:
                raw_path = json.load(f)

            if isinstance(raw_path, dict):
                path_points = raw_path.get('points') or raw_path.get('path_points') or raw_path.get('offsets') or []
                base_width = int(
                    raw_path.get('base_window_width')
                    or raw_path.get('window_width')
                    or raw_path.get('base_client_width')
                    or base_width
                )
                base_height = int(
                    raw_path.get('base_window_height')
                    or raw_path.get('window_height')
                    or raw_path.get('base_client_height')
                    or base_height
                )
            else:
                path_points = raw_path

            if path_points:
                scale_x = client_width / max(base_width, 1)
                scale_y = client_height / max(base_height, 1)
                signature = (client_width, client_height, base_width, base_height)
                if signature != self._last_mouse_path_scale_signature:
                    print(
                        f">>> [扫雷路径] 当前窗口 {client_width}x{client_height}，"
                        f"基准 {base_width}x{base_height}，缩放 x={scale_x:.3f} y={scale_y:.3f}"
                    )
                    self._last_mouse_path_scale_signature = signature

                scaled_points = []
                for point in path_points:
                    if not isinstance(point, (list, tuple)) or len(point) < 2:
                        continue
                    dx = int(round(float(point[0]) * scale_x))
                    dy = int(round(float(point[1]) * scale_y))
                    scaled_points.append((dx, dy))
                if scaled_points:
                    return scaled_points

        x_scale, y_scale = 1.8, 0.8
        for r in range(50, client_height // 2, 40):
            angles = range(180, 360, 5) if (r // 40) % 2 == 0 else range(360, 180, -5)
            for a in angles:
                rad = math.radians(a)
                path_points.append((int(r * math.cos(rad) * x_scale), int(r * math.sin(rad) * y_scale)))
        return path_points

    def mouse_sweep_scan(
        self,
        target_cursor_types,
        click_wait_map=None,
        max_scan_sec=15.0,
        score_threshold=0.7,
        return_on_first_click=False,
    ):
        hwnd = win32gui.FindWindow(None, WIN_TITLE)
        if not hwnd:
            return False

        click_wait_map = click_wait_map or {}
        target_cursor_types = set(target_cursor_types or [])

        try:
            left, top, right, bottom = get_wow_client_rect(hwnd)
            client_width = max(right - left, 1)
            client_height = max(bottom - top, 1)
            center_x = left + (right - left) // 2
            center_y = top + (bottom - top) // 2

            move_cursor_hw(left + 50, top + 50, settle_sec=0.2)
            _, default_hcursor, _ = win32gui.GetCursorInfo()

            path_points = self._load_mouse_path_points(client_width, client_height)
            start_time = time.time()
            clicked_positions = []

            for dx, dy in path_points:
                if time.time() - start_time > max_scan_sec:
                    break

                target_x = center_x + dx + random.randint(-5, 5)
                target_y = center_y + dy + random.randint(-5, 5)

                if not (left + 10 < target_x < right - 10 and top + 10 < target_y < bottom - 10):
                    continue
                if any(math.dist((target_x, target_y), pos) < 30 for pos in clicked_positions):
                    continue

                move_cursor_hw(target_x, target_y, settle_sec=0.02)

                _, hcursor, _ = win32gui.GetCursorInfo()
                if hcursor == 0 or hcursor == default_hcursor:
                    continue

                ctype_name, score = self.cursor_mgr.get_type(hcursor)
                if score > score_threshold and ctype_name in target_cursor_types:
                    print(f">>> [扫雷] 识别成功: {ctype_name} (得分: {score:.3f}), 执行右键点击")
                    hw_ctrl.right_click()
                    clicked_positions.append((target_x, target_y))

                    wait_sec = float(click_wait_map.get(ctype_name, 0.3))
                    if wait_sec > 0:
                        time.sleep(wait_sec)

                    if return_on_first_click:
                        move_cursor_hw(center_x, center_y, settle_sec=0.02)
                        return True

                    wait_start = time.time()
                    while time.time() - wait_start < 0.8:
                        _, curr_h, _ = win32gui.GetCursorInfo()
                        check_name, _ = self.cursor_mgr.get_type(curr_h)
                        if check_name == 'Point':
                            break
                        time.sleep(0.1)
                    time.sleep(random.uniform(0.1, 0.2))
                elif score > 0.4:
                    print(f">>> [扫雷] 疑似图标: {ctype_name} (得分: {score:.3f} < 阈值 {score_threshold})")

            move_cursor_hw(center_x, center_y, settle_sec=0.02)
            return bool(clicked_positions)
        except Exception as e:
            print(f">>> [扫雷扫描] 出错: {e}")
            return False

    def mouse_scan_attack_target(self):
        return self.mouse_sweep_scan(
            ['Attack'],
            click_wait_map={'Attack': 0.3},
            max_scan_sec=4.0,
            score_threshold=0.6,
            return_on_first_click=True,
        )

    def mouse_sweep_loot(self):
        """支持图标识别的高精度扫雷拾取。"""
        return self.mouse_sweep_scan(
            ['LootAll', 'Skin'],
            click_wait_map={'LootAll': 1.3, 'Skin': self.skinning_wait_sec},
            max_scan_sec=15.0,
            score_threshold=0.7,
            return_on_first_click=False,
        )
        hwnd = win32gui.FindWindow(None, WIN_TITLE)
        if not hwnd: return False
        
        try:
            rect = win32gui.GetWindowRect(hwnd)
            left, top, right, bottom = rect
            center_x = left + (right - left) // 2
            center_y = top + (bottom - top) // 2
            
            # 1. 强制“角落校准”采样
            move_cursor_hw(left + 50, top + 50, settle_sec=0.2)
            _, default_hcursor, _ = win32gui.GetCursorInfo()
            
            # 2. 获取扫瞄路径点位
            path_points = []
            path_file = get_config_path("loot_path.json")
            if os.path.exists(path_file):
                with open(path_file, 'r') as f:
                    path_points = json.load(f)
            else:
                # 降级方案：生成拟人化半椭圆
                x_scale, y_scale = 1.8, 0.8
                for r in range(50, (bottom-top)//2, 40):
                    angles = range(180, 360, 5) if (r//40)%2==0 else range(360, 180, -5)
                    for a in angles:
                        rad = math.radians(a)
                        path_points.append((int(r * math.cos(rad) * x_scale), int(r * math.sin(rad) * y_scale)))

            # 3. 开始沿路径扫瞄
            start_time = time.time()
            looted_positions = []
            
            for dx, dy in path_points:
                if time.time() - start_time > 15.0: break
                
                target_x = center_x + dx + random.randint(-5, 5)
                target_y = center_y + dy + random.randint(-5, 5)
                
                if not (left+10 < target_x < right-10 and top+10 < target_y < bottom-10): continue
                if any(math.dist((target_x, target_y), pos) < 30 for pos in looted_positions): continue

                move_cursor_hw(target_x, target_y, settle_sec=0.02)
                
                _, hcursor, _ = win32gui.GetCursorInfo()
                if hcursor != 0 and hcursor != default_hcursor:
                    # 识别图标
                    ctype_name, score = self.cursor_mgr.get_type(hcursor)
                    if score > 0.7 and ctype_name in ['LootAll', 'Skin']:
                        print(f">>> [扫雷] 识别成功: {ctype_name} (得分: {score:.3f}), 执行右键点击")
                        hw_ctrl.right_click()
                        looted_positions.append((target_x, target_y))
                        
                        # 根据类型等待
                        ws = 1.3 if ctype_name == 'LootAll' else self.skinning_wait_sec
                        time.sleep(ws)
                        
                        # 等待指针恢复
                        w_start = time.time()
                        while time.time() - w_start < 0.8:
                            _, ch, _ = win32gui.GetCursorInfo()
                            check_name, _ = self.cursor_mgr.get_type(ch)
                            if check_name == 'Point': break
                            time.sleep(0.1)
                        time.sleep(random.uniform(0.1, 0.2))
                    elif score > 0.4:
                        print(f">>> [扫雷] 疑似图标: {ctype_name} (得分: {score:.3f} < 门槛 0.7)")
                    
            move_cursor_hw(center_x, center_y, settle_sec=0.02)
            return True
        except Exception as e:
            print(f">>> [扫雷拾取] 出错: {e}")
            return False
        
    def execute_combat_logic(self, state):
        if self.attack_loop_config:
            cfg = self.attack_loop_config
            if random.random() >= cfg['trigger_chance']:
                return
            hp = state.get('hp')
            if hp is not None and cfg.get('hp_below_percent') and cfg.get('hp_key') and hp < cfg['hp_below_percent']:
                hw_ctrl.press(cfg['hp_key'])
                time.sleep(0.2)
            mp = state.get('mp')
            if mp is not None and cfg.get('mp_below_percent') and cfg.get('mp_key') and mp < cfg['mp_below_percent']:
                hw_ctrl.press(cfg['mp_key'])
                time.sleep(0.2)

            # 每次攻击前选中目标
            # hw_ctrl.press(KEY_LOOT)
            for step in cfg['steps']:
                key = step.get('key') or KEY_ATTACK
                delay = float(step.get('delay', 0.5))
                hw_ctrl.press(key)
                time.sleep(delay)
            return
        # 默认：模拟手动按键的节奏
        if random.random() < 0.3:
            hw_ctrl.press(KEY_ATTACK)
            time.sleep(0.5)

    def execute_logic(self, state):
        current_time = time.time()
        target_hp = state.get('target_hp', 0)
        combat_error_signal = int(state.get('combat_error_signal', 255) or 255)
        
        effective_target = bool(state['target'] and target_hp > 0)
        in_combat_or_target = bool(state['combat'] or effective_target)

        if in_combat_or_target:
            self.target_acquired_time = (self.target_acquired_time or current_time)
            self.tab_no_target_count = 0 
            
            if effective_target:
                is_new_target = (target_hp > self.last_target_hp + 5)
                if is_new_target:
                    self._has_braked_for_target = False
                    self.distance_interact_brake_pending = False
                    self.distance_interact_pause_until = 0.0

                # 1. 刹车逻辑
                if not self._has_braked_for_target and 0 < target_hp < self.last_target_hp:
                    hw_ctrl.keyDown('s')
                    time.sleep(0.05)
                    hw_ctrl.keyUp('s')
                    self._has_braked_for_target = True

                if self.distance_interact_brake_pending:
                    if current_time < self.distance_interact_pause_until:
                        self.last_target_hp = target_hp
                        self._was_in_combat_or_target = True
                        return
                    hw_ctrl.keyDown('s')
                    time.sleep(self.distance_interact_brake_sec)
                    hw_ctrl.keyUp('s')
                    self.distance_interact_brake_pending = False
                    self.distance_interact_pause_until = 0.0
                    self.last_target_hp = target_hp
                    self._was_in_combat_or_target = True
                    return

                # 2. 根据战斗信号执行纠偏动作
                if (
                    combat_error_signal == 100
                    and (current_time - self.last_interaction_time) >= self.combat_signal_retry_sec
                ):
                    hw_ctrl.press(KEY_LOOT)
                    self.last_interaction_time = current_time
                    self.distance_interact_pause_until = current_time + self.distance_interact_pause_sec
                    self.distance_interact_brake_pending = True
                    self.last_target_hp = target_hp
                    self._was_in_combat_or_target = True
                    return

                if (
                    combat_error_signal == 50
                    and (current_time - self.last_turn_signal_time) >= self.combat_signal_retry_sec
                ):
                    hw_ctrl.keyDown(self.turn_error_key)
                    time.sleep(self.turn_error_hold_sec)
                    hw_ctrl.keyUp(self.turn_error_key)
                    self.last_turn_signal_time = current_time
                self.last_target_hp = target_hp
                if state['combat']:
                    self.execute_combat_logic(state)
            
            self._was_in_combat_or_target = True
            return
        else:
            if self._was_in_combat_or_target:
                self.execute_disengage_loot()
                self._was_in_combat_or_target = False
                self.last_tab_time = current_time + 1.0 
                self.last_target_hp = 0
                self.last_turn_signal_time = 0.0
                self.last_interaction_time = 0.0
                self.distance_interact_brake_pending = False
                self.distance_interact_pause_until = 0.0
                self._has_braked_for_target = False
                return 

            self._was_in_combat_or_target = False
            self.last_target_hp = 0
            self.last_turn_signal_time = 0.0
            self.last_interaction_time = 0.0
            self.distance_interact_brake_pending = False
            self.distance_interact_pause_until = 0.0
            self._has_braked_for_target = False

            self.tab_no_target_count = min(self.tab_no_target_count, 5)
            if self.tab_no_target_count >= 5:
                turn_key = random.choice(["a", "d"])
                hw_ctrl.keyDown(turn_key)
                time.sleep(random.uniform(0.3, 0.6))
                hw_ctrl.keyUp(turn_key)
                self.tab_no_target_count = 0
                self.last_tab_time = current_time
                self.last_turn_end_time = current_time
            elif current_time - self.last_tab_time > random.uniform(0.8, 1.5):
                hw_ctrl.press(KEY_TAB)
                self.last_tab_time = current_time
                self.tab_no_target_count += 1

# 在 main 循环中使用：从 game_state 获取 state
if __name__ == "__main__":
    bot = AutoBot()
    try:
        while True:
            state = parse_game_state()
            if state:
                target_hp = state.get('target_hp')
                if target_hp is not None:
                    hp_part = f"血量:{state['hp']}% 目标血:{target_hp}% | 法力:{state['mp']}% | "
                else:
                    hp_part = f"血量:{state['hp']}% | 法力:{state['mp']}% | "
                print(
                    f"\r[状态] {hp_part}"
                    f"战斗:{'YES' if state['combat'] else 'NO '} | "
                    f"目标:{'YES' if state['target'] else 'NO '}",
                    end=""
                )
                bot.execute_logic(state)
            time.sleep(0.1)
    except KeyboardInterrupt:
        print("\n已停止。")
