import os
import sys
import time
import ctypes
from pathlib import Path

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

from src.assistant_tool.screen_capturer import ScreenCapturer
from src.common.win32_window import find_poe2_window, ensure_window_focus

capturer = ScreenCapturer(output_dir="captures")
hwnd = find_poe2_window()
ensure_window_focus(hwnd)
time.sleep(0.3)

user32 = ctypes.windll.user32
VK_U = 0x55
KEYEVENTF_KEYUP = 0x0002

print("Pressing 'U' to open World Map...")
user32.keybd_event(VK_U, 0x16, 0, 0)
time.sleep(0.08)
user32.keybd_event(VK_U, 0x16, KEYEVENTF_KEYUP, 0)
time.sleep(1.0)

p = capturer.capture_screenshot(event_name="TEST_AFTER_PRESS_U", capture_ram=False)
print(f"Captured: {p}")