j^ TUdZddlZddlZddlZddlZddlmZmZmZddl m Z dZ ee e d<dee fdZiZde defd Zdd eeede fd Zde dee e e e ffd Zde defd Zdde de defdZdde dedefdZdde de de defdZddee defdZdS)uaThư viện dùng chung quản lý cửa sổ Win32 cho Path of Exile 2. Cung cấp các hàm tương tác cửa sổ cấp thấp, chuẩn hóa tọa độ Client Rectangle, tính toán hệ số DPI Scale và gắn luồng vào Desktop tương tác của người dùng. Tuân thủ nghiêm ngặt chuẩn tài liệu Google Docstrings theo Rule 16. N)ListOptionalTuple) is_poe2_pid_global_default_deskreturnctjdkrdS tjj}|ddd}|r|||ddd}|r$||t|S| dddd}|r$||t|Sn#t$rYnwxYwdS)uGắn luồng hiện tại vào Desktop 'Default' tương tác của phiên người dùng. Hàm này mở trạm làm việc 'WinSta0' và desktop 'Default', sau đó gọi SetThreadDesktop để cho phép tiến trình chạy ngầm hoặc service có thể tìm thấy và chụp cửa sổ Win32 của game Path of Exile 2 mà không bị lỗi Access Denied. Preconditions: Hệ điều hành Windows (nt). Luồng gọi có quyền truy cập desktop tương tác. Postconditions: Nếu thành công, luồng hiện tại được liên kết với 'Default' desktop. Returns: Optional[int]: Handle của desktop nếu thành công, hoặc None nếu thất bại / không chạy trên Windows. Thread-safety: Thread-safe. Có thể gọi an toàn từ bất kỳ worker thread nào. Example: >>> hdesk = attach_to_default_desktop() >>> if hdesk: ... print("Đã gắn luồng vào Default Desktop thành công.") ntNWinSta0FrDefault) osnamectypeswindlluser32OpenWindowStationWSetProcessWindowStationOpenInputDesktopSetThreadDesktopint OpenDesktopW Exception)rhwinstah_input h_defaults src/common/win32_window.pyattach_to_default_desktoprs0 w$t %++IujII  4  * *7 3 3 3))!UJ??   # #G , , ,w<< '' 1eZHH  "  # #I . . .y>> ! "      4sA;C=C CChwndc tjdks|sdStjj}||rdSt |}|r,||r||rdS | |d}|rM| |}|r6||r!||r |t |<dSn#t$rYnwxYw | }|rtj tjjtjtjj}g fd}||||d D]h}||ddd} | rL|| r"||r | t |<dS|| in#t$rYnwxYwdS)uGắn luồng hiện tại vào Desktop chứa cửa sổ HWND chỉ định. Tự động truy vấn Thread ID sở hữu HWND qua GetWindowThreadProcessId, lấy handle Desktop qua GetThreadDesktop hoặc duyệt EnumDesktopsW với quyền MAXIMUM_ALLOWED. Đảm bảo 100% tương thích bất kể cửa sổ nằm trên Default desktop hay sandbox desktop (exebox). r FTNcP|r"t|dSNTappendstr)dlpdesktopss r_d_cbz'attach_to_window_desktop.._d_cbks(,OOCFF+++trr )rrrrrIsWindow_cached_window_desktopsgetrGetWindowThreadProcessIdGetThreadDesktoprGetProcessWindowStation WINFUNCTYPEwintypesBOOL c_wchar_pLPARAM EnumDesktopsWr CloseDesktop) rrcachedtidhdeskrDESKTOPENUMPROCWr)r&hdr(s @rattach_to_window_desktopr=HsI w$du ] !F tt $ ( ( . .F &))&11 ??4  4  --dD99  ++C00E 0077 ??4(( 49+D14       0022  ,%1&/2FHXZ`ZiZpqq H       *:*:5*A*A1 E E E , ,((AujAA,..r22(!??400(<>3D9#'44''+++      5s,A#C-- C:9C:>CG%G%% G21G2title_candidatesc tjdkrdSt|ddg}d|Dddgtjjdd t d tffd d D]6}|d}|r |rt |cS7|D]6}d|}|r |rt |cS7t}tj tj j tj j tj j }hd  fd}||} |r||dn|dn#t $rYnwxYws= tj tj j tjtj j }gfd} } | rЉ| || dD]} | dkr| ddd} | rx | |dr-|  | n2 | #| wxYwn#t $rYnwxYwS)u]Tìm handle (HWND) của cửa sổ game Path of Exile 2. Thực hiện dò tìm thông qua lớp cửa sổ đã biết (POEWindowClass, POEC2_MainWindow) và danh sách tiêu đề ứng viên (title candidates). Tự động duyệt qua toàn bộ các cửa sổ trên Default Desktop nếu EnumDesktopWindows khả dụng. Args: title_candidates: Danh sách chuỗi tiêu đề cửa sổ cần khớp (không phân biệt hoa thường). Nếu None, mặc định sử dụng ["path of exile 2", "path of exile"]. Returns: int: HWND của cửa sổ game nếu tìm thấy, hoặc 0 nếu không tìm thấy. Thread-safety: Thread-safe. Example: >>> hwnd = find_poe2_window() >>> if hwnd != 0: ... print(f"Tìm thấy cửa sổ game: HWND={hex(hwnd)}") r rNzpath of exile 2 path of exilec6g|]}|S)lower).0ts r z$find_poe2_window..s ===q===r*poewindowclasspoec2_mainwindowhrc|r*|r|sdStj}|tj||jdkrdSt|jS)NFr)r+IsWindowVisiblerc_ulongr.byrefvaluer)rIw_pidrs r_is_valid_poe2_hwndz-find_poe2_window.._is_valid_poe2_hwnds ** &2H2H2K2K 5  ''6<+>+>??? ;!  55;'''r*)POEWindowClassPOEC2_MainWindow> window_class cabinetwclasschrome_widgetwin_1mozilla_window_classapplicationframewindowcP |r |sdStjd} ||d|j}| vrdStjd} ||d|j}| vr|rt|dS D]#}||vr|rt|dS$dS)NTF) r+rKrcreate_unicode_buffer GetClassNameWrNstriprCGetWindowTextWr) rlparamcbufcls_nametbuf win_title target_titlerP found_hwndignored_classesnormalized_titlestarget_classesrs r _enum_procz$find_poe2_window.._enum_procsFt$$ F,B,B4,H,H 4+C00T4---:##%%++--  & &4+C00dD#...J$$&&,,..  ~ % %""4((  YY u. ! !Ly((&&t,,!!$TJ 55tr*cP|r"t|dSr"r#)dnamer' desktops_lists rr)zfind_poe2_window.._d_cbs*5!((U444tr*defaultFr )rrrrrrrbool FindWindowWr1r2r3HWNDr5EnumDesktopWindows EnumWindowsrr4r0r6rCrrr7)r>clsrtitler: WNDENUMPROCrhcbr;r)rr&r<rPrkrdrerfrgrs @@@@@@@rfind_poe2_windowrv~s, w$q-?==,<===&(:;N ] !FJ(s(t((((((6!!#t,,  ''-- t99   "!!$..  ''-- t99    & ' 'E$V_%96?;OQWQ`QghhKO > Z B  &  % %eR 3 3 3 3   r1 % % %        %1&/2FHXZ`ZiZpqq M     4466G 4$$W.>.>u.E.EqIII& 4 4AwwyyI-- ,,Q5*EEB44"55b"a@@@)& & 7 7 ; ; ; %"//33333 &#//3333F//33334    D  sC0E33 F?FB2J3:.J)J3J3J..J33 K?Kctjdks|sdSt|tjj}||sdStj}| |tj |sdStj dd}| |tj |sdS|j |jz }|j|jz }|j|j||fS)uqLấy tọa độ vùng vẽ thực tế (Client Rectangle) của cửa sổ game theo pixel màn hình. Tự động bù trừ thanh tiêu đề (title bar) và viền cửa sổ thông qua Win32 API GetClientRect và ClientToScreen, trả về tọa độ tuyệt đối chuẩn xác cho việc tính toán giao diện, quét mẫu pixel và bắt sự kiện chuột. Args: hwnd: Handle cửa sổ Win32 của tiến trình Path of Exile 2. Returns: Tuple[int, int, int, int]: Bộ tứ (left, top, width, height) theo tọa độ màn hình. Trả về (0, 0, 0, 0) nếu HWND không hợp lệ hoặc gọi Win32 API thất bại. Thread-safety: Thread-safe. Example: >>> left, top, width, height = get_client_rect(hwnd) >>> print(f"Client Area: {width}x{height} tại ({left}, {top})") r )rrrrr)rrr=rrrr+r2RECT GetClientRectrMPOINTClientToScreenrightleftbottomtopxy)rrrectpointwidthheights rget_client_rectrs* w$d|T""" ] !F ??4 | ?   ! !D   fl4&8&8 9 9| O ! !!Q ' 'E  v|E':': ; ;| J "E [48 #F GUWeV ,,r*cVtjdks|sdStjj}t |dr? ||}|dkrt|dz Sn#t$rYnwxYw tjj }| |}|rz d}| ||}|dkr(t|dz | ||S | ||n#| ||wxYwn#t$rYnwxYwdS)uTính toán hệ số phóng đại DPI (DPI Scale Factor) của cửa sổ chỉ định. Sử dụng GetDpiForWindow (Windows 10 v1607+ / Windows 11) với cơ chế fallback về Device Context GetDeviceCaps(LOGPIXELSX) để đảm bảo độ co giãn giao diện và tọa độ chuột luôn chính xác trên mọi thiết lập màn hình (HiDPI / 4K / Ultrawide). Args: hwnd: Handle cửa sổ Win32 cần đo lường. Returns: float: Hệ số tỉ lệ DPI (ví dụ 1.0 cho 100% 96 DPI, 1.25 cho 125% 120 DPI, 1.5 cho 150%). Trả về 1.0 nếu không xác định được. Thread-safety: Thread-safe. Example: >>> scale = get_dpi_scale(hwnd) >>> print(f"DPI Scale: {scale * 100:.0f}%") r g?GetDpiForWindowrgX@X) rrrrrhasattrrfloatrgdi32GetDC GetDeviceCaps ReleaseDC)rrdpirhdc LOGPIXELSXs r get_dpi_scaler1sg* w$ds ] !Fv()) ((..CQwwSzzD((    D    #ll4    , , ))#z::77 ::,  s++++  s++++  s++++       3sA,A%% A21A26(D/C=D&D=DD D&%D& timeout_msc  tjdks|sdStjj}tjj}t |||sdS ddlm }||r,| |r| |sdSn#t$rYnwxYw| |krt|rdS | |r||dn||d||dddddd|}| }|r||d nd}||d }d} d} |r*||kr$t'|||d} |r*||kr$t'|||d} t+|d r|d d } d } || dd| n#t$rYnwxYw t+|d r|dt+|dr||d|dddd|||||||||ddd d| r|r|||d| r|r|||dn=#| r|r|||d| r|r|||dwwwxYwt?j|dz z} t?j| kr| }||krdStj }||tj!|tj }||tj!||j"dkr|j"|j"krdSt|rdSt?j#dt?j| kt|S#t$rE}dd l$}tKd|d|&t|cYd }~Sd }~wwxYw)u6Tự động chuyển tab và đưa cửa sổ chỉ định lên foreground với quyền tương tác an toàn. Sử dụng kỹ thuật AttachThreadInput kết hợp SetForegroundWindow và BringWindowToTop để đảm bảo quyền kích hoạt cửa sổ ngay cả khi tiến trình gọi đang chạy nền hoặc không có quyền input focus trực tiếp. Tự động khôi phục nếu cửa sổ đang bị thu nhỏ (minimized). Args: hwnd: Handle cửa sổ Win32 cần đưa lên tiêu điểm. timeout_ms: Thời gian chờ tối đa tính bằng mili-giây để xác nhận focus (mặc định 500ms). Returns: bool: True nếu cửa sổ đã nằm ở foreground thành công, False nếu thất bại hoặc không hợp lệ. Thread-safety: Thread-safe. Có thể gọi từ bất kỳ worker thread nào. Example: >>> hwnd = find_poe2_window() >>> if hwnd and ensure_window_focus(hwnd): ... print("Cửa sổ POE2 đã sẵn sàng nhận phím/chuột.") r Fris_window_on_virtual_displayT CNLockSetForegroundWindowi AllowSetForegroundWindowSwitchToThisWindowg@@g{Gz?z [ensure_window_focus Exception] )flush)'rrrrrkernel32r=r+"src.common.virtual_display_managerrrKIsIconicrGetForegroundWindowis_poe2_window_focused ShowWindow SetWindowPosGetCurrentThreadIdr.rmAttachThreadInputrrSystemParametersInfoWrr keybd_eventBringWindowToTopSetForegroundWindowSetActiveWindowSetFocustimerLrMrNsleep tracebackprint print_exc)rrrrr cur_threadfg_hwnd fg_thread target_thread attached_fgattached_targetSPI_SETFOREGROUNDLOCKTIMEOUTSPIF_SENDCHANGEdeadlinefg target_pidfg_piders rensure_window_focusres, w$du ] !F}%HT""" ??4 u SSSSSS ' ' - - %%d++ FOOD4I4I t      !!##t++0Ft0L0L+tF, ??4  '   dA & & & &   dA & & &D"aAq2JKKK0022 ,,..FMTF33GTBBBST 77dCC    Vy00v77 ItTTUUK  ^Z=88"6#;#;J W[#\#\]]O v899 2..q111+1 ($O  ( ()Eq!_ ] ] ] ]    D  Gv9:: 4//333v344 6))$555   tQ1 - - -  # #D ) ) )  & &t , , ,  " "4 ( ( ( OOD ! ! !   tQ1 - - - K= K((]EJJJ Gy G((YFFF K= K((]EJJJ Gy G((YFFFF G G9;;*v"56ikkH$$++--BTzzt))J  + +D&,z2J2J K K K^%%F  + +B V0D0D E E E!## 8H(H(Ht%d++ t Jt   ikkH$$&d+++ ,,, 444DAAAA%d++++++++ ,s;B B)(B)DR((AH*)R(* H74R(6H77R(;CM 7R(:M;;A R( BR(R(.9R(( S72:S2,S72S7r|sidec ztjdks|sdStjj}||sdStj}d}||dtj |dsBd|_ d|_ | d|_ | d|_|j |j z }|j|j z }|dz}|dkr |j |zn|j }|j } |} |} d} d } t||d|| | | | | zS) uITự động đặt cửa sổ game chiếm nửa màn hình (trái hoặc phải) để người dùng làm việc bên còn lại. Args: hwnd: Handle cửa sổ Win32 của game. side: 'right' (nửa phải) hoặc 'left' (nửa trái). Returns: bool: True nếu thành công. r F0rrr|@)rrrrrr+r2rxrrMr}rGetSystemMetricsr|r~rmr)rrrrSPI_GETWORKAREAscreen_wscreen_hhalf_wnew_xnew_ynew_wnew_h SWP_NOZORDERSWP_SHOWWINDOWs rsnap_window_to_halfrs; w$du ] !F ??4 u ?   ! !DO  ' 'FLrs;  ((((((((((000000&*hsm***/8C=////d33343333lFFxS ':FcFFFFR'-#'-%S#s(:";'-'-'-'-T111111hv,v,cv,sv,Tv,v,v,v,r#i#ic#i#i4#i#i#i#iLcS#W[&RR#R$RRRRRRr*