entari-plugin-server 0.1.0__tar.gz

entari_plugin_server-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 ARCLET
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

entari_plugin_server-0.1.0/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.1
+Name: entari-plugin-server
+Version: 0.1.0
+Summary: Entari plugin for running Satori Server
+Author-Email: RF-Tar-Railt <rf_tar_railt@qq.com>
+License: MIT
+Project-URL: homepage, https://github.com/ArcletProject/entari-plugin-server
+Project-URL: repository, https://github.com/ArcletProject/entari-plugin-server
+Requires-Python: >=3.9
+Requires-Dist: arclet-entari>=0.12.1
+Requires-Dist: satori-python-server>=0.16.3
+Description-Content-Type: text/markdown
+
+# entari-plugin-server
+Entari plugin for running Satori Server
+
+## Example Usage
+
+```yaml
+plugins:
+  server:
+    adapters:
+      - $path: package.module:AdapterClass
+        # Following are adapter's configuration
+        key1: value1
+        key2: value2
+    host: 127.0.0.1
+    port: 5140
+```

entari_plugin_server-0.1.0/README.md

@@ -0,0 +1,16 @@
+# entari-plugin-server
+Entari plugin for running Satori Server
+
+## Example Usage
+
+```yaml
+plugins:
+  server:
+    adapters:
+      - $path: package.module:AdapterClass
+        # Following are adapter's configuration
+        key1: value1
+        key2: value2
+    host: 127.0.0.1
+    port: 5140
+```

entari_plugin_server-0.1.0/pyproject.toml

@@ -0,0 +1,35 @@
+[project]
+name = "entari-plugin-server"
+version = "0.1.0"
+description = "Entari plugin for running Satori Server"
+authors = [
+    { name = "RF-Tar-Railt", email = "rf_tar_railt@qq.com" },
+]
+dependencies = [
+    "arclet-entari>=0.12.1",
+    "satori-python-server>=0.16.3",
+]
+requires-python = ">=3.9"
+readme = "README.md"
+
+[project.license]
+text = "MIT"
+
+[project.urls]
+homepage = "https://github.com/ArcletProject/entari-plugin-server"
+repository = "https://github.com/ArcletProject/entari-plugin-server"
+
+[build-system]
+requires = [
+    "pdm-backend",
+]
+build-backend = "pdm.backend"
+
+[tool.pdm]
+distribution = true
+
+[tool.pdm.dev-dependencies]
+dev = [
+    "arclet-entari[yaml]>=0.12.1",
+    "nekobox @ git+https://github.com/wyapx/nekobox.git",
+]

entari_plugin_server-0.1.0/src/entari_plugin_server/__init__.py

@@ -0,0 +1,79 @@
+import re
+from functools import reduce
+from importlib import import_module
+
+from arclet.entari import plugin
+from arclet.entari import logger as log_m
+from arclet.entari.config import BasicConfModel, field
+from graia.amnesia.builtins import asgi
+from satori.server import Adapter, Server
+
+from .patch import DirectAdapterServer
+
+asgi.LoguruHandler = log_m.LoguruHandler
+
+
+class Config(BasicConfModel):
+    direct_adapter: bool = False
+    """是否使用直连适配器"""
+    adapters: list[dict] = field(default_factory=list)
+    host: str = "127.0.0.1"
+    port: int = 5140
+    path: str = "satori"
+    version: str = "v1"
+    token: str | None = None
+    stream_threshold: int = 16 * 1024 * 1024
+    stream_chunk_size: int = 64 * 1024
+
+
+conf = plugin.get_config(Config)
+logger = log_m.log.wrapper("[Server]")
+
+if conf.direct_adapter:
+    server = DirectAdapterServer(conf.host, conf.port, conf.path, conf.version, conf.token, None, conf.stream_threshold, conf.stream_chunk_size)
+else:
+    server = Server(conf.host, conf.port, conf.path, conf.version, conf.token, None, conf.stream_threshold, conf.stream_chunk_size)
+
+
+pattern = re.compile(r"(?P<module>[\w.]+)\s*(:\s*(?P<attr>[\w.]+)\s*)?((?P<extras>\[.*\])\s*)?$")
+
+
+def _load_adapter(adapter_config: dict):
+    if "$path" not in adapter_config:
+        logger.warning(f"Adapter config missing `$path`: {adapter_config}")
+        return None
+    path = adapter_config["$path"]
+    if path.startswith("@."):
+        path = f"satori.adapters{path[1:]}"
+    elif path.startswith("@"):
+        path = f"satori.adapters.{path[1:]}"
+    match = pattern.match(path)
+    if not match:
+        logger.warning(f"Invalid adapter path: {path}")
+        return None
+    try:
+        module = import_module(match.group("module"))
+    except ImportError:
+        logger.warning(f"Could not import module {match.group('module')}")
+        return None
+    try:
+        attrs = filter(None, (match.group("attr") or "Adapter").split("."))
+        ext = reduce(getattr, attrs, module)
+    except AttributeError:
+        logger.warning(f"Could not find adapter in {module.__name__}")
+        return None
+    if isinstance(ext, type) and issubclass(ext, Adapter):
+        return ext(**{k: v for k, v in adapter_config.items() if k != "$path"})  # type: ignore
+    elif isinstance(ext, Adapter):
+        return ext
+    logger.warning(f"Invalid adapter in {module.__name__}")
+    return None
+
+
+adapters: list[Adapter] = [*filter(None, map(_load_adapter, conf.adapters))]
+
+for adapter in adapters:
+    logger.debug(f"Applying adapter {adapter}")
+    server.apply(adapter)
+
+plugin.add_service(server)

entari_plugin_server-0.1.0/src/entari_plugin_server/patch.py

@@ -0,0 +1,83 @@
+import json
+from io import BytesIO
+
+from arclet.entari import Entari
+from arclet.entari.session import EntariProtocol
+from loguru import logger
+from satori import Api, Event
+from satori.client import Account, ApiInfo
+from satori.exception import ActionFailed
+from satori.server import Server
+from starlette.datastructures import FormData, Headers, UploadFile
+from starlette.requests import Request
+from starlette.responses import JSONResponse
+
+
+class DirectAdapterProtocol(EntariProtocol):
+    server: Server
+
+    async def call_api(
+        self, action: str | Api, params: dict | None = None, multipart: bool = False, method: str = "POST"
+    ):
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.account.config.token or ''}",
+            "X-Platform": self.account.platform,
+            "X-Self-ID": self.account.self_id,
+            "Satori-Platform": self.account.platform,
+            "Satori-User-ID": self.account.self_id,
+        }
+        _form = None
+        if multipart:
+            if params is None:
+                raise TypeError("multipart requires params")
+            forms = []
+            headers.pop("Content-Type")
+            for k, v in params.items():
+                if isinstance(v, dict):
+                    forms.append(
+                        (
+                            k,
+                            UploadFile(
+                                BytesIO(v["value"]),
+                                filename=v.get("filename"),
+                                headers=Headers({"content-type": v["content_type"]})
+                            )
+                        )
+                    )
+                else:
+                    forms.append((k, v))
+            _form = FormData(forms)
+        _headers = [(k.lower().encode("latin-1"), v.encode("latin-1")) for k, v in headers.items()]
+        req = Request({"type": "http", "method": method, "path_params": {"action": action}, "headers": _headers})
+        if _form:
+            req._form = _form
+        else:
+            req._json = params or {}
+        resp = await self.server.http_server_handler(req)
+        if isinstance(resp, JSONResponse):
+            return json.loads(resp.body)  # type: ignore
+        raise ActionFailed(resp.body.decode())  # type: ignore
+
+
+class DirectAdapterServer(Server):
+    async def event_callback(self, event: Event):
+        app = Entari.current()
+        proxy_urls = []
+        for provider in self.providers:
+            proxy_urls.extend(provider.proxy_urls())
+        await super().event_callback(event)
+        login_sn = f"{event.login.user.id}@{id(self)}"
+        if login_sn not in app.accounts:
+            acc = Account(
+                event.login,
+                ApiInfo(),
+                proxy_urls,
+                DirectAdapterProtocol
+            )
+            acc.protocol.server = self
+            app.accounts[login_sn] = acc
+            logger.info(f"account added: {acc}")
+        else:
+            acc = app.accounts[login_sn]
+        await app.handle_event(acc, event)