import time
import ctypes
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

from src.assistant_tool.screen_capturer import ScreenCapturer

capturer = ScreenCapturer(output_dir="captures")
hwnd = capturer.find_poe2_window()
if not hwnd:
    print("Could not find POE2 window")
    sys.exit(1)

capturer.bring_to_foreground(hwnd)
time.sleep(0.2)

user32 = ctypes.windll.user32
VK_SPACE = 0x20
VK_S = 0x53
VK_A = 0x41
KEYEVENTF_KEYUP = 0x0002

# Close inventory
user32.keybd_event(VK_SPACE, 0x39, 0, 0)
time.sleep(0.05)
user32.keybd_event(VK_SPACE, 0x39, KEYEVENTF_KEYUP, 0)
time.sleep(0.2)

# Walk South-West (Hold S + A for 1.2s)
print("Walking South-West towards portal entrance...")
user32.keybd_event(VK_S, 0x1F, 0, 0)
user32.keybd_event(VK_A, 0x1E, 0, 0)
time.sleep(1.2)
user32.keybd_event(VK_S, 0x1F, KEYEVENTF_KEYUP, 0)
user32.keybd_event(VK_A, 0x1E, KEYEVENTF_KEYUP, 0)
time.sleep(0.5)

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