import sys, time, ctypes
import win32gui
from PIL import ImageGrab

sys.stdout.reconfigure(encoding='utf-8')
user32 = ctypes.windll.user32
hdesk = user32.OpenInputDesktop(0, False, 0x01FF)
if hdesk: user32.SetThreadDesktop(hdesk)

hwnd = 0x0063003A
print(f"Target HWND: 0x{hwnd:X}, Valid: {win32gui.IsWindow(hwnd)}")

def make_lparam(x, y):
    return ((y & 0xFFFF) << 16) | (x & 0xFFFF)

def click(x, y):
    lp = make_lparam(x, y)
    user32.PostMessageW(hwnd, 0x0200, 0, lp) # WM_MOUSEMOVE
    time.sleep(0.04)
    user32.PostMessageW(hwnd, 0x0201, 1, lp) # WM_LBUTTONDOWN
    time.sleep(0.08)
    user32.PostMessageW(hwnd, 0x0202, 0, lp) # WM_LBUTTONUP
    time.sleep(0.05)

def dblclick(x, y):
    lp = make_lparam(x, y)
    user32.PostMessageW(hwnd, 0x0200, 0, lp)
    time.sleep(0.03)
    user32.PostMessageW(hwnd, 0x0201, 1, lp)
    time.sleep(0.05)
    user32.PostMessageW(hwnd, 0x0202, 0, lp)
    time.sleep(0.06)
    user32.PostMessageW(hwnd, 0x0203, 1, lp)
    time.sleep(0.05)
    user32.PostMessageW(hwnd, 0x0202, 0, lp)
    time.sleep(0.05)

def send_key(vk):
    lp = (0x1C << 16) | 1
    user32.PostMessageW(hwnd, 0x0100, vk, lp) # WM_KEYDOWN
    time.sleep(0.05)
    user32.PostMessageW(hwnd, 0x0101, vk, lp | (3 << 30)) # WM_KEYUP
    time.sleep(0.05)

def send_char(c):
    lp = (ord(c) << 16) | 1
    user32.PostMessageW(hwnd, 0x0102, ord(c), lp) # WM_CHAR
    time.sleep(0.04)

def capture(name):
    if win32gui.IsWindow(hwnd):
        rect = win32gui.GetWindowRect(hwnd)
        img = ImageGrab.grab(bbox=rect, all_screens=True)
        path = rf"c:\Projects\JX\logs\{name}.png"
        img.save(path)
        print(f"Captured: {path}")

# Bring to foreground
user32.SetForegroundWindow(hwnd)
time.sleep(0.3)

# Step 1: Click "Bắt đầu trò chơi"
print("Step 1: Click 'Bắt đầu trò chơi'...")
click(400, 385)
time.sleep(0.1)
send_key(0x0D) # VK_RETURN
time.sleep(2.0)
capture("p2488_step1_start")

# Step 2: Confirm popup notice if present
print("Step 2: Notice popup...")
click(400, 440)
time.sleep(0.1)
send_key(0x0D)
time.sleep(1.5)
capture("p2488_step2_server_screen")

# Step 3: Server selection
print("Step 3: Select 'Võ Công Truyền Kỳ' -> 'Trọn Vẹn Chữ Tình'...")
send_key(0x26) # VK_UP
time.sleep(0.3)
click(208, 199)
time.sleep(0.4)
click(610, 209)
time.sleep(0.2)
dblclick(610, 209)
time.sleep(0.4)
click(260, 503) # Xác định
time.sleep(2.5)
capture("p2488_step3_login_box")

# Step 4: Enter username & password
print("Step 4: Enter credentials 'nguvansi1' / 'Ngu123@@'...")
click(400, 302)
time.sleep(0.1)
# Clear old username
send_key(0x23) # VK_END
for _ in range(25): send_key(0x08)
send_key(0x24) # VK_HOME
for _ in range(25): send_key(0x2E)
time.sleep(0.1)
for c in "nguvansi1":
    send_char(c)
time.sleep(0.2)

# Enter password
click(464, 342)
time.sleep(0.1)
send_key(0x23) # VK_END
for _ in range(25): send_key(0x08)
send_key(0x24) # VK_HOME
for _ in range(25): send_key(0x2E)
time.sleep(0.1)
for c in "Ngu123@@":
    send_char(c)
time.sleep(0.2)
capture("p2488_step4_typed")

# Submit credentials
print("Submitting login...")
click(300, 418)
time.sleep(0.1)
send_key(0x0D)
time.sleep(4.0)
capture("p2488_step4_char_select")

# Step 5: Select character Slot 1
print("Step 5: Select Slot 1 (ConCuaBoMe)...")
click(250, 350)
time.sleep(0.2)
dblclick(250, 350)
time.sleep(0.3)
click(92, 561) # Vào trò chơi
time.sleep(0.1)
send_key(0x0D)
time.sleep(5.0)

# Dismiss in-game dialog
click(300, 570)
time.sleep(0.2)
send_key(0x1B) # VK_ESCAPE
time.sleep(1.0)
capture("p2488_step5_in_game")
print("Done login sequence for nguvansi1!")