import ctypes
import ctypes.wintypes as w
import time
import win32gui
import win32con
from PIL import ImageGrab

u32 = ctypes.windll.user32

hdesk = u32.OpenDesktopW("Default", 0, False, 0x10000000)
if not hdesk:
    hdesk = u32.OpenInputDesktop(0, False, 0x10000000)
if hdesk:
    u32.SetThreadDesktop(hdesk)

hwnd = win32gui.FindWindow("POEWindowClass", "Path of Exile 2")
print(f"Found POE Window: {hwnd:#x}")
if hwnd:
    win32gui.ShowWindow(hwnd, win32con.SW_RESTORE)
    win32gui.SetForegroundWindow(hwnd)
    time.sleep(0.5)

img = ImageGrab.grab(all_screens=True)
img.save("captures/game_window_focused.png")
print("Saved captures/game_window_focused.png")
