Squashed 'astrbot-sdk/' changes from 955470aee..b817715b9

b817715b9 chore: refresh vendor snapshot [skip ci]
231bd5da2 Merge pull request #99 from united-pooh:dev
2637cbbdb feat: 添加 register_file_url 方法以支持文件 URL 注册，并更新相关文档
bb4a57aec fix: 更新 MessageEvent 类中的方法文档，增加中文说明以提升可读性
REVERT: 955470aee chore: refresh vendor snapshot [skip ci]

git-subtree-dir: astrbot-sdk
git-subtree-split: b817715b9179b359ea1a1ea5a7d1bf198f0be798

Author: whatevertogo

src/astrbot_sdk/clients/files.py

@@ -34,6 +34,17 @@ async def register_file(
         )
         return FileRegistration.from_payload(output).token
 
+    async def register_file_url(
+        self,
+        path: str,
+        timeout: float | None = None,
+    ) -> str:
+        output = await self._proxy.call(
+            "system.file.register",
+            {"path": str(path), "timeout": timeout},
+        )
+        return FileRegistration.from_payload(output).url
+
     async def handle_file(self, token: str) -> str:
         output = await self._proxy.call(
             "system.file.handle",
@@ -46,8 +57,4 @@ async def _register_file_url(
         path: str,
         timeout: float | None = None,
     ) -> str:
-        output = await self._proxy.call(
-            "system.file.register",
-            {"path": str(path), "timeout": timeout},
-        )
-        return FileRegistration.from_payload(output).url
+        return await self.register_file_url(path, timeout=timeout)

src/astrbot_sdk/context.py

@@ -344,7 +344,7 @@ async def _register_file_url(
         path: str,
         timeout: float | None = None,
     ) -> str:
-        return await self.files._register_file_url(path, timeout=timeout)
+        return await self.files.register_file_url(path, timeout=timeout)
 
     async def text_to_image(
         self,

src/astrbot_sdk/runtime/transport.py

@@ -59,6 +59,7 @@
 from loguru import logger
 
 MessageHandler = Callable[[str], Awaitable[None]]
+STDIO_SUBPROCESS_STREAM_LIMIT = 8 * 1024 * 1024
 
 
 def _get_aiohttp():
@@ -73,15 +74,19 @@ def _get_web():
     return web
 
 
-def _frame_stdio_payload(payload: str) -> str:
-    body = payload
-    if body.endswith("\r\n"):
-        body = body[:-2]
-    elif body.endswith(("\n", "\r")):
-        body = body[:-1]
-    if "\n" in body or "\r" in body:
-        raise ValueError("STDIO payload 不允许包含原始换行符")
-    return f"{body}\n"
+def _frame_stdio_payload(payload: str) -> bytes:
+    body = payload.encode("utf-8")
+    return f"{len(body)}\n".encode("ascii") + body
+
+
+def _parse_stdio_header(raw_header: bytes) -> int:
+    header = raw_header.decode("ascii").strip()
+    if not header:
+        raise ValueError("STDIO frame header is empty")
+    try:
+        return int(header)
+    except ValueError as exc:
+        raise ValueError(f"Invalid STDIO frame header: {header!r}") from exc
 
 
 # TODO 一个更好的解决方案？
@@ -169,6 +174,7 @@ async def _start_subprocess_with_retry(self) -> asyncio.subprocess.Process:
                     stdin=asyncio.subprocess.PIPE,
                     stdout=asyncio.subprocess.PIPE,
                     stderr=sys.stderr,
+                    limit=STDIO_SUBPROCESS_STREAM_LIMIT,
                 )
             except Exception as exc:
                 last_error = exc
@@ -205,11 +211,11 @@ async def stop(self) -> None:
         self._closed.set()
 
     async def send(self, payload: str) -> None:
-        line = _frame_stdio_payload(payload)
+        frame = _frame_stdio_payload(payload)
         if self._process is not None:
             if self._process.stdin is None:
                 raise RuntimeError("STDIO subprocess stdin 不可用")
-            self._process.stdin.write(line.encode("utf-8"))
+            self._process.stdin.write(frame)
             await self._process.stdin.drain()
             return
 
@@ -218,8 +224,11 @@ async def send(self, payload: str) -> None:
 
         def _write() -> None:
             assert self._stdout is not None
-            self._stdout.write(line)
-            self._stdout.flush()
+            binary_stdout = getattr(self._stdout, "buffer", None)
+            if binary_stdout is None:
+                raise RuntimeError("STDIO stdout 必须提供可写入 bytes 的 buffer")
+            binary_stdout.write(frame)
+            binary_stdout.flush()
 
         await asyncio.to_thread(_write)
 
@@ -228,21 +237,30 @@ async def _read_process_loop(self) -> None:
         assert self._process.stdout is not None
         try:
             while True:
-                raw = await self._process.stdout.readline()
-                if not raw:
+                raw_header = await self._process.stdout.readline()
+                if not raw_header:
                     break
-                await self._dispatch(raw.decode("utf-8").rstrip("\r\n"))
+                payload_size = _parse_stdio_header(raw_header)
+                raw = await self._process.stdout.readexactly(payload_size)
+                await self._dispatch(raw.decode("utf-8"))
         finally:
             self._closed.set()
 
     async def _read_file_loop(self) -> None:
         assert self._stdin is not None
         try:
             while True:
-                raw = await asyncio.to_thread(self._stdin.readline)
-                if not raw:
+                binary_stdin = getattr(self._stdin, "buffer", None)
+                if binary_stdin is None:
+                    raise RuntimeError("STDIO stdin 必须提供可读取 bytes 的 buffer")
+                raw_header = await asyncio.to_thread(binary_stdin.readline)
+                if not raw_header:
                     break
-                await self._dispatch(raw.rstrip("\r\n"))
+                payload_size = _parse_stdio_header(raw_header)
+                raw = await asyncio.to_thread(binary_stdin.read, payload_size)
+                if len(raw) != payload_size:
+                    raise EOFError("STDIO frame truncated before payload completed")
+                await self._dispatch(raw.decode("utf-8"))
         finally:
             self._closed.set()