MaaFw 3.0.4__py3-none-macosx_13_0_x86_64.whl → 4.0.0a1__py3-none-macosx_13_0_x86_64.whl

maa/__init__.py

@@ -9,4 +9,4 @@ if env_path:
 else:
     __PATH = Path(Path(__file__).parent, "bin")
 
-Library.open(__PATH)
+Library.open(__PATH, agent_server=False)

maa/agent/__init__.py

@@ -0,0 +1,12 @@
+import os
+from pathlib import Path
+
+from ..library import Library
+
+env_path = os.environ.get("MAAFW_BINARY_PATH")
+if env_path:
+    __PATH = Path(env_path)
+else:
+    __PATH = Path(Path(__file__).parent.parent, "bin")
+
+Library.open(__PATH, agent_server=True)

maa/agent/agent_server.py

@@ -0,0 +1,132 @@
+import ctypes
+import sys
+
+from ..define import *
+from ..library import Library
+from ..buffer import StringBuffer
+
+
+class AgentServer:
+
+    @staticmethod
+    def custom_recognition(name: str):
+
+        def wrapper_recognition(recognition):
+            AgentServer.register_custom_recognition(
+                name=name, recognition=recognition()
+            )
+            return recognition
+
+        return wrapper_recognition
+
+    _custom_recognition_holder = {}
+
+    @staticmethod
+    def register_custom_recognition(
+        name: str, recognition: "CustomRecognition"  # type: ignore
+    ) -> bool:
+
+        AgentServer._set_api_properties()
+
+        # avoid gc
+        AgentServer._custom_recognition_holder[name] = recognition
+
+        return bool(
+            Library.agent_server().MaaAgentServerRegisterCustomRecognition(
+                name.encode(),
+                recognition.c_handle,
+                recognition.c_arg,
+            )
+        )
+
+    @staticmethod
+    def custom_action(name: str):
+
+        def wrapper_action(action):
+            AgentServer.register_custom_action(name=name, action=action())
+            return action
+
+        return wrapper_action
+
+    _custom_action_holder = {}
+
+    @staticmethod
+    def register_custom_action(name: str, action: "CustomAction") -> bool:  # type: ignore
+
+        AgentServer._set_api_properties()
+
+        # avoid gc
+        AgentServer._custom_action_holder[name] = action
+
+        return bool(
+            Library.agent_server().MaaAgentServerRegisterCustomAction(
+                name.encode(),
+                action.c_handle,
+                action.c_arg,
+            )
+        )
+
+    @staticmethod
+    def start_up(identifier: str) -> bool:
+
+        AgentServer._set_api_properties()
+
+        return bool(Library.agent_server().MaaAgentServerStartUp(identifier.encode()))
+
+    @staticmethod
+    def shut_down() -> None:
+
+        AgentServer._set_api_properties()
+
+        Library.agent_server().MaaAgentServerShutDown()
+
+    @staticmethod
+    def join() -> None:
+
+        AgentServer._set_api_properties()
+
+        Library.agent_server().MaaAgentServerJoin()
+
+    @staticmethod
+    def detach(self) -> None:
+
+        AgentServer._set_api_properties()
+
+        Library.agent_server().MaaAgentServerDetach()
+
+    _api_properties_initialized: bool = False
+
+    @staticmethod
+    def _set_api_properties():
+        if AgentServer._api_properties_initialized:
+            return
+
+        AgentServer._api_properties_initialized = True
+
+        Library.agent_server().MaaAgentServerRegisterCustomRecognition.restype = MaaBool
+        Library.agent_server().MaaAgentServerRegisterCustomRecognition.argtypes = [
+            ctypes.c_char_p,
+            MaaCustomRecognitionCallback,
+            ctypes.c_void_p,
+        ]
+
+        Library.agent_server().MaaAgentServerRegisterCustomAction.restype = MaaBool
+        Library.agent_server().MaaAgentServerRegisterCustomAction.argtypes = [
+            ctypes.c_char_p,
+            MaaCustomActionCallback,
+            ctypes.c_void_p,
+        ]
+
+        Library.agent_server().MaaAgentServerStartUp.restype = MaaBool
+        Library.agent_server().MaaAgentServerStartUp.argtypes = [
+            ctypes.c_char_p,
+        ]
+
+        Library.agent_server().MaaAgentServerShutDown.restype = None
+        Library.agent_server().MaaAgentServerShutDown.argtypes = []
+
+        Library.agent_server().MaaAgentServerJoin.restype = None
+        Library.agent_server().MaaAgentServerJoin.argtypes = []
+
+        Library.agent_server().MaaAgentServerDetach.restype = None
+        Library.agent_server().MaaAgentServerDetach.argtypes = []

maa/agent_client.py

@@ -0,0 +1,83 @@
+import ctypes
+
+from .define import *
+from .library import Library
+from .resource import Resource
+from .buffer import StringBuffer
+
+
+class AgentClient:
+    _handle: MaaAgentClientHandle
+
+    def __init__(self):
+        self._set_api_properties()
+
+        self._handle = Library.agent_client().MaaAgentClientCreate()
+        if not self._handle:
+            raise RuntimeError("Failed to create agent client.")
+
+    def __del__(self):
+        if self._handle:
+            Library.agent_client().MaaAgentClientDestroy(self._handle)
+
+    def bind(self, resource: Resource) -> bool:
+        # avoid gc
+        self._resource = resource
+
+        return bool(
+            Library.agent_client().MaaAgentClientBindResource(
+                self._handle, resource._handle
+            )
+        )
+
+    def create_socket(self, identifier: str = "") -> Optional[str]:
+        id_buffer = StringBuffer()
+        id_buffer.set(identifier)
+
+        ret = bool(
+            Library.agent_client().MaaAgentClientCreateSocket(
+                self._handle, id_buffer._handle
+            )
+        )
+        if not ret:
+            return None
+
+        return id_buffer.get()
+
+    def connect(self) -> bool:
+        return bool(Library.agent_client().MaaAgentClientConnect(self._handle))
+
+    _api_properties_initialized: bool = False
+
+    @staticmethod
+    def _set_api_properties():
+        if AgentClient._api_properties_initialized:
+            return
+
+        if Library.is_agent_server():
+            raise RuntimeError("AgentClient is not available in AgentServer.")
+
+        AgentClient._api_properties_initialized = True
+
+        Library.agent_client().MaaAgentClientCreate.restype = MaaAgentClientHandle
+        Library.agent_client().MaaAgentClientCreate.argtypes = []
+
+        Library.agent_client().MaaAgentClientDestroy.restype = None
+        Library.agent_client().MaaAgentClientDestroy.argtypes = [MaaAgentClientHandle]
+
+        Library.agent_client().MaaAgentClientBindResource.restype = MaaBool
+        Library.agent_client().MaaAgentClientBindResource.argtypes = [
+            MaaAgentClientHandle,
+            MaaResourceHandle,
+        ]
+
+        Library.agent_client().MaaAgentClientCreateSocket.restype = MaaBool
+        Library.agent_client().MaaAgentClientCreateSocket.argtypes = [
+            MaaAgentClientHandle,
+            MaaStringBufferHandle,
+        ]
+
+        Library.agent_client().MaaAgentClientConnect.restype = MaaBool
+        Library.agent_client().MaaAgentClientConnect.argtypes = [
+            MaaAgentClientHandle,
+        ]