Merge pull request #4 from sudoriaa/main

feat: 添加管理员权限检查与分辨率检测，并输出警告日志

Author: 1bananachicken

agent/custom/action/auto_make_coffee.py

@@ -67,7 +67,6 @@ def __init__(self):
     def run(self, context: Context, argv: CustomAction.RunArg) -> CustomAction.RunResult:
         print("=== Auto Make Coffee Action Started ===")
         controller = context.tasker.controller
-        
         make_count = 10
         check_freq = 0.5
         if argv.custom_action_param:
@@ -92,7 +91,7 @@ def run(self, context: Context, argv: CustomAction.RunArg) -> CustomAction.RunRe
             if context.tasker.stopping:  
                 return CustomAction.RunResult(success=False)
             print(f"=== Making Coffee {count + 1}/{make_count} ===")
-            
+
             # Step 1: 选择关卡
             print("Tapping on select level...")
             click_rect(controller, select_level_target)

agent/main.py

@@ -393,6 +393,59 @@ def check_and_install_dependencies():
         logger.info("Pip 依赖安装已禁用，跳过依赖安装")
 
 
+# -----
+# region 环境检测
+# -----
+
+
+def _check_chinese_path():
+    """检测当前程序是否运行在含中文字符的目录下"""
+    import re
+
+    cwd = os.getcwd()
+    if re.search(r"[一-鿿]", cwd):
+        logger.warning(
+            f"当前运行目录含中文字符: {cwd}\n请将程序移动至纯英文路径下运行。"
+        )
+
+
+def _check_admin_privilege():
+    """检查是否以管理员权限运行，若否则输出警告"""
+    import ctypes
+
+    if ctypes.windll.shell32.IsUserAnAdmin():
+        return
+
+    logger.warning(
+        "未以管理员权限运行，部分输入功能可能无法正常使用。"
+        "请右键 MaaNTE.exe → 以管理员身份运行。"
+    )
+
+
+def _check_game_resolution():
+    """连接控制器后检测游戏窗口分辨率"""
+    from utils.win32_process import find_window_by_process, get_client_size
+
+    hwnd = find_window_by_process("HTGame.exe")
+    if hwnd is None:
+        logger.warning("分辨率检测: 未找到游戏窗口 (HTGame.exe)")
+        return
+
+    size = get_client_size(hwnd)
+    if size is None:
+        logger.warning("分辨率检测: 无法获取窗口尺寸")
+        return
+
+    w, h = size
+    if (w, h) == (1280, 720):
+        logger.info(f"当前窗口分辨率: {w}x{h} [正常]")
+    else:
+        logger.warning(
+            f"当前窗口分辨率: {w}x{h}，请使用 1280x720 分辨率。"
+            "请将游戏设置为 1280x720 窗口化模式，否则部分功能可能异常。"
+        )
+
+
 # -----
 # region 核心业务
 # -----
@@ -442,6 +495,7 @@ def agent(is_dev_mode=False):
         log_pi_environment()
         AgentServer.start_up(socket_id)
         logger.info("AgentServer启动")
+        _check_game_resolution()
         AgentServer.join()
         AgentServer.shut_down()
         logger.info("AgentServer关闭")
@@ -463,6 +517,9 @@ def main():
     current_version = read_interface_version()
     is_dev_mode = current_version == "DEBUG"
 
+    if sys.platform.startswith("win"):
+        _check_admin_privilege()
+
     # 如果是Linux系统或开发模式，启动虚拟环境
     if sys.platform.startswith("linux") or is_dev_mode:
         ensure_venv_and_relaunch_if_needed()

agent/utils/win32_process.py

@@ -0,0 +1,72 @@
+import ctypes
+from ctypes import wintypes
+
+user32 = ctypes.windll.user32
+kernel32 = ctypes.windll.kernel32
+
+# 使进程感知 DPI，避免 GetClientRect 返回缩放后的虚拟坐标
+# 150% 缩放时未设置此项会导致返回值只有实际分辨率的 2/3
+user32.SetProcessDPIAware()
+
+WNDENUMPROC = ctypes.WINFUNCTYPE(wintypes.BOOL, wintypes.HWND, wintypes.LPARAM)
+
+TH32CS_SNAPPROCESS = 0x00000002
+
+
+class PROCESSENTRY32W(ctypes.Structure):
+    _fields_ = [
+        ("dwSize", wintypes.DWORD),
+        ("cntUsage", wintypes.DWORD),
+        ("th32ProcessID", wintypes.DWORD),
+        ("th32DefaultHeapID", ctypes.POINTER(wintypes.ULONG)),
+        ("th32ModuleID", wintypes.DWORD),
+        ("cntThreads", wintypes.DWORD),
+        ("th32ParentProcessID", wintypes.DWORD),
+        ("pcPriClassBase", wintypes.LONG),
+        ("dwFlags", wintypes.DWORD),
+        ("szExeFile", ctypes.c_wchar * 260),
+    ]
+
+
+def get_pids_by_name(process_name):
+    snapshot = kernel32.CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0)
+    if snapshot == -1:
+        return []
+    entry = PROCESSENTRY32W()
+    entry.dwSize = ctypes.sizeof(PROCESSENTRY32W)
+    pids = []
+    if kernel32.Process32FirstW(snapshot, ctypes.byref(entry)):
+        while True:
+            if entry.szExeFile.lower() == process_name.lower():
+                pids.append(entry.th32ProcessID)
+            if not kernel32.Process32NextW(snapshot, ctypes.byref(entry)):
+                break
+    kernel32.CloseHandle(snapshot)
+    return pids
+
+
+def find_window_by_process(process_name):
+    pids = get_pids_by_name(process_name)
+    if not pids:
+        return None
+    pid_set = set(pids)
+    results = []
+
+    def callback(hwnd, _lparam):
+        if not user32.IsWindowVisible(hwnd):
+            return True
+        pid = wintypes.DWORD()
+        user32.GetWindowThreadProcessId(hwnd, ctypes.byref(pid))
+        if pid.value in pid_set:
+            results.append(hwnd)
+        return True
+
+    user32.EnumWindows(WNDENUMPROC(callback), 0)
+    return results[0] if results else None
+
+
+def get_client_size(hwnd):
+    rect = wintypes.RECT()
+    if not user32.GetClientRect(hwnd, ctypes.byref(rect)):
+        return None
+    return rect.right - rect.left, rect.bottom - rect.top

assets/interface.json

@@ -4,7 +4,7 @@
     "description": "异环小助手，由MAAFramework强力驱动！",
     "contact": "QQ群: 1103323319",
     "license": "MIT",
-    "welcome": "首次使用！欢迎使用 MAANTE!",
+    "welcome": "首次使用！欢迎使用 MaaNTE!",
     "github": "https://github.com/1bananachicken/MaaNTE",
     "version": "0.0.1",
     "controller": [
@@ -13,7 +13,7 @@
             "type": "Win32",
             "win32": {
                 "class_regex": ".*",
-                "window_regex": "异环",
+                "window_regex": "^异环.*$",
                 "screencap": "FramePool",
                 "mouse": "SendMessageWithWindowPos",
                 "keyboard": "SendMessageWithWindowPos"