MaaFw 3.0.4__py3-none-win_arm64.whl → 4.0.0b4__py3-none-win_arm64.whl

maa/controller.py

@@ -22,7 +22,7 @@ __all__ = [
 class Controller:
     _notification_handler: Optional[NotificationHandler]
     _handle: MaaControllerHandle
-    _own: bool = False
+    _own: bool
 
     def __init__(
         self,
@@ -39,40 +39,40 @@ class Controller:
 
     def __del__(self):
         if self._handle and self._own:
-            Library.framework.MaaControllerDestroy(self._handle)
+            Library.framework().MaaControllerDestroy(self._handle)
 
     def post_connection(self) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostConnection(self._handle)
+        ctrl_id = Library.framework().MaaControllerPostConnection(self._handle)
         return self._gen_ctrl_job(ctrl_id)
 
     def post_click(self, x: int, y: int) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostClick(self._handle, x, y)
+        ctrl_id = Library.framework().MaaControllerPostClick(self._handle, x, y)
         return self._gen_ctrl_job(ctrl_id)
 
     def post_swipe(self, x1: int, y1: int, x2: int, y2: int, duration: int) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostSwipe(
+        ctrl_id = Library.framework().MaaControllerPostSwipe(
             self._handle, x1, y1, x2, y2, duration
         )
         return self._gen_ctrl_job(ctrl_id)
 
     def post_press_key(self, key: int) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostPressKey(self._handle, key)
+        ctrl_id = Library.framework().MaaControllerPostPressKey(self._handle, key)
         return self._gen_ctrl_job(ctrl_id)
 
     def post_input_text(self, text: str) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostInputText(
+        ctrl_id = Library.framework().MaaControllerPostInputText(
             self._handle, text.encode()
         )
         return self._gen_ctrl_job(ctrl_id)
 
     def post_start_app(self, intent: str) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostStartApp(
+        ctrl_id = Library.framework().MaaControllerPostStartApp(
             self._handle, intent.encode()
         )
         return self._gen_ctrl_job(ctrl_id)
 
     def post_stop_app(self, intent: str) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostStopApp(
+        ctrl_id = Library.framework().MaaControllerPostStopApp(
             self._handle, intent.encode()
         )
         return self._gen_ctrl_job(ctrl_id)
@@ -80,7 +80,7 @@ class Controller:
     def post_touch_down(
         self, x: int, y: int, contact: int = 0, pressure: int = 1
     ) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostTouchDown(
+        ctrl_id = Library.framework().MaaControllerPostTouchDown(
             self._handle, contact, x, y, pressure
         )
         return self._gen_ctrl_job(ctrl_id)
@@ -88,17 +88,17 @@ class Controller:
     def post_touch_move(
         self, x: int, y: int, contact: int = 0, pressure: int = 1
     ) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostTouchMove(
+        ctrl_id = Library.framework().MaaControllerPostTouchMove(
             self._handle, contact, x, y, pressure
         )
         return self._gen_ctrl_job(ctrl_id)
 
     def post_touch_up(self, contact: int = 0) -> Job:
-        ctrl_id = Library.framework.MaaControllerPostTouchUp(self._handle, contact)
+        ctrl_id = Library.framework().MaaControllerPostTouchUp(self._handle, contact)
         return self._gen_ctrl_job(ctrl_id)
 
     def post_screencap(self) -> JobWithResult:
-        ctrl_id = Library.framework.MaaControllerPostScreencap(self._handle)
+        ctrl_id = Library.framework().MaaControllerPostScreencap(self._handle)
         return JobWithResult(
             ctrl_id,
             self._status,
@@ -109,7 +109,7 @@ class Controller:
     @property
     def cached_image(self) -> numpy.ndarray:
         image_buffer = ImageBuffer()
-        if not Library.framework.MaaControllerCachedImage(
+        if not Library.framework().MaaControllerCachedImage(
             self._handle, image_buffer._handle
         ):
             raise RuntimeError("Failed to get cached image.")
@@ -117,19 +117,19 @@ class Controller:
 
     @property
     def connected(self) -> bool:
-        return bool(Library.framework.MaaControllerConnected(self._handle))
+        return bool(Library.framework().MaaControllerConnected(self._handle))
 
     @property
     def uuid(self) -> str:
         buffer = StringBuffer()
-        if not Library.framework.MaaControllerGetUuid(self._handle, buffer._handle):
+        if not Library.framework().MaaControllerGetUuid(self._handle, buffer._handle):
             raise RuntimeError("Failed to get UUID.")
         return buffer.get()
 
     def set_screenshot_target_long_side(self, long_side: int) -> bool:
         cint = ctypes.c_int32(long_side)
         return bool(
-            Library.framework.MaaControllerSetOption(
+            Library.framework().MaaControllerSetOption(
                 self._handle,
                 MaaOption(MaaCtrlOptionEnum.ScreenshotTargetLongSide),
                 ctypes.pointer(cint),
@@ -140,7 +140,7 @@ class Controller:
     def set_screenshot_target_short_side(self, short_side: int) -> bool:
         cint = ctypes.c_int32(short_side)
         return bool(
-            Library.framework.MaaControllerSetOption(
+            Library.framework().MaaControllerSetOption(
                 self._handle,
                 MaaOption(MaaCtrlOptionEnum.ScreenshotTargetShortSide),
                 ctypes.pointer(cint),
@@ -151,7 +151,7 @@ class Controller:
     def set_screenshot_use_raw_size(self, enable: bool) -> bool:
         cbool = MaaBool(enable)
         return bool(
-            Library.framework.MaaControllerSetOption(
+            Library.framework().MaaControllerSetOption(
                 self._handle,
                 MaaOption(MaaCtrlOptionEnum.ScreenshotUseRawSize),
                 ctypes.pointer(cbool),
@@ -162,10 +162,10 @@ class Controller:
     ### private ###
 
     def _status(self, maaid: int) -> MaaStatus:
-        return Library.framework.MaaControllerStatus(self._handle, maaid)
+        return Library.framework().MaaControllerStatus(self._handle, maaid)
 
     def _wait(self, maaid: int) -> MaaStatus:
-        return Library.framework.MaaControllerWait(self._handle, maaid)
+        return Library.framework().MaaControllerWait(self._handle, maaid)
 
     def _get_screencap(self, _: int) -> numpy.ndarray:
         return self.cached_image
@@ -185,29 +185,29 @@ class Controller:
             return
         Controller._api_properties_initialized = True
 
-        Library.framework.MaaControllerDestroy.restype = None
-        Library.framework.MaaControllerDestroy.argtypes = [MaaControllerHandle]
+        Library.framework().MaaControllerDestroy.restype = None
+        Library.framework().MaaControllerDestroy.argtypes = [MaaControllerHandle]
 
-        Library.framework.MaaControllerSetOption.restype = MaaBool
-        Library.framework.MaaControllerSetOption.argtypes = [
+        Library.framework().MaaControllerSetOption.restype = MaaBool
+        Library.framework().MaaControllerSetOption.argtypes = [
             MaaControllerHandle,
             MaaCtrlOption,
             MaaOptionValue,
             MaaOptionValueSize,
         ]
 
-        Library.framework.MaaControllerPostConnection.restype = MaaCtrlId
-        Library.framework.MaaControllerPostConnection.argtypes = [MaaControllerHandle]
+        Library.framework().MaaControllerPostConnection.restype = MaaCtrlId
+        Library.framework().MaaControllerPostConnection.argtypes = [MaaControllerHandle]
 
-        Library.framework.MaaControllerPostClick.restype = MaaCtrlId
-        Library.framework.MaaControllerPostClick.argtypes = [
+        Library.framework().MaaControllerPostClick.restype = MaaCtrlId
+        Library.framework().MaaControllerPostClick.argtypes = [
             MaaControllerHandle,
             c_int32,
             c_int32,
         ]
 
-        Library.framework.MaaControllerPostSwipe.restype = MaaCtrlId
-        Library.framework.MaaControllerPostSwipe.argtypes = [
+        Library.framework().MaaControllerPostSwipe.restype = MaaCtrlId
+        Library.framework().MaaControllerPostSwipe.argtypes = [
             MaaControllerHandle,
             c_int32,
             c_int32,
@@ -216,37 +216,37 @@ class Controller:
             c_int32,
         ]
 
-        Library.framework.MaaControllerPostPressKey.restype = MaaCtrlId
-        Library.framework.MaaControllerPostPressKey.argtypes = [
+        Library.framework().MaaControllerPostPressKey.restype = MaaCtrlId
+        Library.framework().MaaControllerPostPressKey.argtypes = [
             MaaControllerHandle,
             c_int32,
         ]
 
-        Library.framework.MaaControllerPostInputText.restype = MaaCtrlId
-        Library.framework.MaaControllerPostInputText.argtypes = [
+        Library.framework().MaaControllerPostInputText.restype = MaaCtrlId
+        Library.framework().MaaControllerPostInputText.argtypes = [
             MaaControllerHandle,
             ctypes.c_char_p,
         ]
 
-        Library.framework.MaaControllerPostScreencap.restype = MaaCtrlId
-        Library.framework.MaaControllerPostScreencap.argtypes = [
+        Library.framework().MaaControllerPostScreencap.restype = MaaCtrlId
+        Library.framework().MaaControllerPostScreencap.argtypes = [
             MaaControllerHandle,
         ]
 
-        Library.framework.MaaControllerPostStartApp.restype = MaaCtrlId
-        Library.framework.MaaControllerPostStartApp.argtypes = [
+        Library.framework().MaaControllerPostStartApp.restype = MaaCtrlId
+        Library.framework().MaaControllerPostStartApp.argtypes = [
             MaaControllerHandle,
             ctypes.c_char_p,
         ]
 
-        Library.framework.MaaControllerPostStopApp.restype = MaaCtrlId
-        Library.framework.MaaControllerPostStopApp.argtypes = [
+        Library.framework().MaaControllerPostStopApp.restype = MaaCtrlId
+        Library.framework().MaaControllerPostStopApp.argtypes = [
             MaaControllerHandle,
             ctypes.c_char_p,
         ]
 
-        Library.framework.MaaControllerPostTouchDown.restype = MaaCtrlId
-        Library.framework.MaaControllerPostTouchDown.argtypes = [
+        Library.framework().MaaControllerPostTouchDown.restype = MaaCtrlId
+        Library.framework().MaaControllerPostTouchDown.argtypes = [
             MaaControllerHandle,
             c_int32,
             c_int32,
@@ -254,8 +254,8 @@ class Controller:
             c_int32,
         ]
 
-        Library.framework.MaaControllerPostTouchMove.restype = MaaCtrlId
-        Library.framework.MaaControllerPostTouchMove.argtypes = [
+        Library.framework().MaaControllerPostTouchMove.restype = MaaCtrlId
+        Library.framework().MaaControllerPostTouchMove.argtypes = [
             MaaControllerHandle,
             c_int32,
             c_int32,
@@ -263,34 +263,34 @@ class Controller:
             c_int32,
         ]
 
-        Library.framework.MaaControllerPostTouchUp.restype = MaaCtrlId
-        Library.framework.MaaControllerPostTouchUp.argtypes = [
+        Library.framework().MaaControllerPostTouchUp.restype = MaaCtrlId
+        Library.framework().MaaControllerPostTouchUp.argtypes = [
             MaaControllerHandle,
             c_int32,
         ]
-        Library.framework.MaaControllerStatus.restype = MaaStatus
-        Library.framework.MaaControllerStatus.argtypes = [
+        Library.framework().MaaControllerStatus.restype = MaaStatus
+        Library.framework().MaaControllerStatus.argtypes = [
             MaaControllerHandle,
             MaaCtrlId,
         ]
 
-        Library.framework.MaaControllerWait.restype = MaaStatus
-        Library.framework.MaaControllerWait.argtypes = [
+        Library.framework().MaaControllerWait.restype = MaaStatus
+        Library.framework().MaaControllerWait.argtypes = [
             MaaControllerHandle,
             MaaCtrlId,
         ]
 
-        Library.framework.MaaControllerConnected.restype = MaaBool
-        Library.framework.MaaControllerConnected.argtypes = [MaaControllerHandle]
+        Library.framework().MaaControllerConnected.restype = MaaBool
+        Library.framework().MaaControllerConnected.argtypes = [MaaControllerHandle]
 
-        Library.framework.MaaControllerCachedImage.restype = MaaBool
-        Library.framework.MaaControllerCachedImage.argtypes = [
+        Library.framework().MaaControllerCachedImage.restype = MaaBool
+        Library.framework().MaaControllerCachedImage.argtypes = [
             MaaControllerHandle,
             MaaImageBufferHandle,
         ]
 
-        Library.framework.MaaControllerGetUuid.restype = MaaBool
-        Library.framework.MaaControllerGetUuid.argtypes = [
+        Library.framework().MaaControllerGetUuid.restype = MaaBool
+        Library.framework().MaaControllerGetUuid.argtypes = [
             MaaControllerHandle,
             MaaStringBufferHandle,
         ]
@@ -317,7 +317,7 @@ class AdbController(Controller):
 
         self._notification_handler = notification_handler
 
-        self._handle = Library.framework.MaaAdbControllerCreate(
+        self._handle = Library.framework().MaaAdbControllerCreate(
             str(adb_path).encode(),
             address.encode(),
             MaaAdbScreencapMethod(screencap_methods),
@@ -332,8 +332,8 @@ class AdbController(Controller):
 
     def _set_adb_api_properties(self):
 
-        Library.framework.MaaAdbControllerCreate.restype = MaaControllerHandle
-        Library.framework.MaaAdbControllerCreate.argtypes = [
+        Library.framework().MaaAdbControllerCreate.restype = MaaControllerHandle
+        Library.framework().MaaAdbControllerCreate.argtypes = [
             ctypes.c_char_p,
             ctypes.c_char_p,
             MaaAdbScreencapMethod,
@@ -349,7 +349,7 @@ class Win32Controller(Controller):
 
     def __init__(
         self,
-        hWnd: Optional[ctypes.c_void_p],
+        hWnd: Union[ctypes.c_void_p, int, None],
         screencap_method: int = MaaWin32ScreencapMethodEnum.DXGI_DesktopDup,
         input_method: int = MaaWin32InputMethodEnum.Seize,
         notification_handler: Optional[NotificationHandler] = None,
@@ -358,7 +358,7 @@ class Win32Controller(Controller):
         self._set_win32_api_properties()
 
         self._notification_handler = notification_handler
-        self._handle = Library.framework.MaaWin32ControllerCreate(
+        self._handle = Library.framework().MaaWin32ControllerCreate(
             hWnd,
             MaaWin32ScreencapMethod(screencap_method),
             MaaWin32InputMethod(input_method),
@@ -369,8 +369,8 @@ class Win32Controller(Controller):
             raise RuntimeError("Failed to create Win32 controller.")
 
     def _set_win32_api_properties(self):
-        Library.framework.MaaWin32ControllerCreate.restype = MaaControllerHandle
-        Library.framework.MaaWin32ControllerCreate.argtypes = [
+        Library.framework().MaaWin32ControllerCreate.restype = MaaControllerHandle
+        Library.framework().MaaWin32ControllerCreate.argtypes = [
             ctypes.c_void_p,
             MaaWin32ScreencapMethod,
             MaaWin32InputMethod,
@@ -393,7 +393,7 @@ class DbgController(Controller):
         self._set_dbg_api_properties()
 
         self._notification_handler = notification_handler
-        self._handle = Library.framework.MaaDbgControllerCreate(
+        self._handle = Library.framework().MaaDbgControllerCreate(
             str(read_path).encode(),
             str(write_path).encode(),
             MaaDbgControllerType(dbg_type),
@@ -405,8 +405,8 @@ class DbgController(Controller):
             raise RuntimeError("Failed to create DBG controller.")
 
     def _set_dbg_api_properties(self):
-        Library.framework.MaaDbgControllerCreate.restype = MaaControllerHandle
-        Library.framework.MaaDbgControllerCreate.argtypes = [
+        Library.framework().MaaDbgControllerCreate.restype = MaaControllerHandle
+        Library.framework().MaaDbgControllerCreate.argtypes = [
             ctypes.c_char_p,
             ctypes.c_char_p,
             MaaDbgControllerType,
@@ -444,7 +444,7 @@ class CustomController(Controller):
             CustomController._c_input_text_agent,
         )
 
-        self._handle = Library.framework.MaaCustomControllerCreate(
+        self._handle = Library.framework().MaaCustomControllerCreate(
             self.c_handle,
             self.c_arg,
             *NotificationHandler._gen_c_param(self._notification_handler)
@@ -735,8 +735,8 @@ class CustomController(Controller):
         return int(self.input_text(c_text.decode()))
 
     def _set_custom_api_properties(self):
-        Library.framework.MaaCustomControllerCreate.restype = MaaControllerHandle
-        Library.framework.MaaCustomControllerCreate.argtypes = [
+        Library.framework().MaaCustomControllerCreate.restype = MaaControllerHandle
+        Library.framework().MaaCustomControllerCreate.argtypes = [
             ctypes.POINTER(MaaCustomControllerCallbacks),
             ctypes.c_void_p,
             MaaNotificationCallback,

maa/define.py

@@ -1,10 +1,11 @@
 import ctypes
 import platform
 from dataclasses import dataclass
-from enum import IntEnum, Enum
+from enum import IntEnum
 from typing import List, Tuple, Union, Dict, Optional
 
 import numpy
+from strenum import StrEnum  # For Python 3.9/3.10
 
 MaaBool = ctypes.c_uint8
 MaaSize = ctypes.c_size_t
@@ -52,7 +53,7 @@ MaaCtrlOption = MaaOption
 MaaResOption = MaaOption
 
 
-class MaaGlobalOptionEnum:
+class MaaGlobalOptionEnum(IntEnum):
     Invalid = 0
 
     # Log dir
@@ -88,7 +89,7 @@ class MaaGlobalOptionEnum:
     DebugMode = 6
 
 
-class MaaCtrlOptionEnum:
+class MaaCtrlOptionEnum(IntEnum):
     Invalid = 0
 
     # Only one of long and short side can be set, and the other is automatically scaled according to the aspect ratio.
@@ -110,13 +111,13 @@ class MaaCtrlOptionEnum:
     Recording = 5
 
 
-class MaaInferenceDeviceEnum:
+class MaaInferenceDeviceEnum(IntEnum):
     CPU = -2
     Auto = -1
     # and more gpu id or flag...
 
 
-class MaaInferenceExecutionProviderEnum:
+class MaaInferenceExecutionProviderEnum(IntEnum):
     # I don't recommend setting up MaaResOption_InferenceDevice in this case,
     # because you don't know which EP will be used on different user devices.
     Auto = 0
@@ -139,7 +140,7 @@ class MaaInferenceExecutionProviderEnum:
     CUDA = 4
 
 
-class MaaResOptionEnum:
+class MaaResOptionEnum(IntEnum):
     Invalid = 0
 
     # Use the specified inference device.
@@ -160,7 +161,7 @@ class MaaResOptionEnum:
 MaaAdbScreencapMethod = ctypes.c_uint64
 
 
-class MaaAdbScreencapMethodEnum:
+class MaaAdbScreencapMethodEnum(IntEnum):
     """
     Use bitwise OR to set the method you need
     MaaFramework will test their speed and use the fastest one.
@@ -183,7 +184,7 @@ class MaaAdbScreencapMethodEnum:
 MaaAdbInputMethod = ctypes.c_uint64
 
 
-class MaaAdbInputMethodEnum:
+class MaaAdbInputMethodEnum(IntEnum):
     """
     Use bitwise OR to set the method you need
     MaaFramework will select the available ones according to priority.
@@ -205,7 +206,7 @@ MaaWin32ScreencapMethod = ctypes.c_uint64
 
 
 # No bitwise OR, just set it
-class MaaWin32ScreencapMethodEnum:
+class MaaWin32ScreencapMethodEnum(IntEnum):
     Null = 0
 
     GDI = 1
@@ -217,7 +218,7 @@ MaaWin32InputMethod = ctypes.c_uint64
 
 
 # No bitwise OR, just set it
-class MaaWin32InputMethodEnum:
+class MaaWin32InputMethodEnum(IntEnum):
     Null = 0
 
     Seize = 1
@@ -228,7 +229,7 @@ class MaaWin32InputMethodEnum:
 MaaDbgControllerType = ctypes.c_uint64
 
 
-class MaaDbgControllerTypeEnum:
+class MaaDbgControllerTypeEnum(IntEnum):
     Null = 0
 
     CarouselImage = 1
@@ -273,6 +274,8 @@ MaaToolkitAdbDeviceHandle = ctypes.c_void_p
 MaaToolkitDesktopWindowListHandle = ctypes.c_void_p
 MaaToolkitDesktopWindowHandle = ctypes.c_void_p
 
+MaaAgentClientHandle = ctypes.c_void_p
+
 
 class MaaCustomControllerCallbacks(ctypes.Structure):
     ConnectFunc = FUNCTYPE(
@@ -446,7 +449,7 @@ RectType = Union[
 ]
 
 
-class AlgorithmEnum(str, Enum):
+class AlgorithmEnum(StrEnum):
     DirectHit = "DirectHit"
     TemplateMatch = "TemplateMatch"
     FeatureMatch = "FeatureMatch"

maa/library.py

@@ -7,38 +7,126 @@ from .define import *
 
 
 class Library:
+    _is_agent_server: bool = False
+
+    _framework: ctypes.CDLL = None
+    _toolkit: ctypes.CDLL = None
+    _agent_client: ctypes.CDLL = None
+    _agent_server: ctypes.CDLL = None
+    _lib_type = None
 
     @staticmethod
-    def open(path: pathlib.Path):
+    def open(path: pathlib.Path, agent_server: bool = False):
+        if Library._api_properties_initialized:
+            return
+
         if not path.exists():
             raise FileNotFoundError(f"`{path}` does not exist.")
 
-        platform_values = {
-            "windows": ("MaaFramework.dll", "MaaToolkit.dll"),
-            "darwin": ("libMaaFramework.dylib", "libMaaToolkit.dylib"),
-            "linux": ("libMaaFramework.so", "libMaaToolkit.so"),
+        WINDOWS = "windows"
+        MACOS = "darwin"
+        LINUX = "linux"
+
+        Library._is_agent_server = agent_server
+
+        if not Library.is_agent_server():
+            framework_library = {
+                WINDOWS: "MaaFramework.dll",
+                MACOS: "libMaaFramework.dylib",
+                LINUX: "libMaaFramework.so",
+            }
+            agent_client_library = {
+                WINDOWS: "MaaAgentClient.dll",
+                MACOS: "libMaaAgentClient.dylib",
+                LINUX: "libMaaAgentClient.so",
+            }
+        else:
+            agent_server_library = {
+                WINDOWS: "MaaAgentServer.dll",
+                MACOS: "libMaaAgentServer.dylib",
+                LINUX: "libMaaAgentServer.so",
+            }
+
+        toolkit_library = {
+            WINDOWS: "MaaToolkit.dll",
+            MACOS: "libMaaToolkit.dylib",
+            LINUX: "libMaaToolkit.so",
         }
 
         platform_type = platform.system().lower()
 
         if platform_type == "windows":
-            lib_import = ctypes.WinDLL
+            Library._lib_type = ctypes.WinDLL
         else:
-            lib_import = ctypes.CDLL
+            Library._lib_type = ctypes.CDLL
 
-        Library.framework_libpath = path / platform_values[platform_type][0]
-        Library.framework = lib_import(str(Library.framework_libpath))
+        if not Library.is_agent_server():
+            Library.framework_libpath = path / framework_library[platform_type]
+            Library.agent_client_libpath = path / agent_client_library[platform_type]
+        else:
+            Library.agent_server_libpath = path / agent_server_library[platform_type]
 
-        Library.toolkit_libpath = path / platform_values[platform_type][1]
-        Library.toolkit = lib_import(str(Library.toolkit_libpath))
+        Library.toolkit_libpath = path / toolkit_library[platform_type]
 
-        Library._set_api_properties()
+    @staticmethod
+    def framework() -> ctypes.CDLL:
+        if not Library.is_agent_server():
+            if not Library._framework:
+                Library._framework = Library._lib_type(str(Library.framework_libpath))
+
+            return Library._framework
+        else:
+            return Library.agent_server()
+
+    @staticmethod
+    def toolkit() -> ctypes.CDLL:
+        if not Library._toolkit:
+            Library._toolkit = Library._lib_type(str(Library.toolkit_libpath))
+
+        return Library._toolkit
+
+    @staticmethod
+    def agent_client() -> ctypes.CDLL:
+        if Library.is_agent_server():
+            raise ValueError("Agent server is not available in the current context.")
+
+        if not Library._agent_client:
+            Library._agent_client = Library._lib_type(
+                str(Library.agent_client_libpath)
+            )
+
+        return Library._agent_client
+
+    @staticmethod
+    def agent_server() -> ctypes.CDLL:
+        if not Library.is_agent_server():
+            raise ValueError("Agent client is not available in the current context.")
+
+        if not Library._agent_server:
+            Library._agent_server = Library._lib_type(
+                str(Library.agent_server_libpath)
+            )
+
+        return Library._agent_server
+
+    @staticmethod
+    def is_agent_server() -> bool:
+        return Library._is_agent_server
 
     @staticmethod
     def version() -> str:
-        return Library.framework.MaaVersion().decode()
+        Library._set_api_properties()
+
+        return Library.framework().MaaVersion().decode()
+
+    _api_properties_initialized: bool = False
 
     @staticmethod
     def _set_api_properties():
-        Library.framework.MaaVersion.restype = ctypes.c_char_p
-        Library.framework.MaaVersion.argtypes = []
+        if Library._api_properties_initialized:
+            return
+
+        Library._api_properties_initialized = True
+
+        Library.framework().MaaVersion.restype = ctypes.c_char_p
+        Library.framework().MaaVersion.argtypes = []

maa/notification_handler.py

@@ -2,13 +2,13 @@ import ctypes
 import json
 from abc import ABC
 from typing import Optional, Tuple
-from enum import Enum
+from enum import IntEnum
 from dataclasses import dataclass
 
 from .define import MaaNotificationCallback
 
 
-# class NotificationEvent(Enum):
+# class NotificationEvent(IntEnum):
 #     ResourceLoading = 1
 #     ControllerAction = 2
 #     TaskerTask = 3
@@ -17,7 +17,7 @@ from .define import MaaNotificationCallback
 #     TaskAction = 6
 
 
-class NotificationType(Enum):
+class NotificationType(IntEnum):
     Unknown = 0
     Starting = 1
     Succeeded = 2