@neoline/hostbridge 2.0.3 → 2.0.5

package/src/server_control_mcp/remote_agent_bundle.py

@@ -0,0 +1,56 @@
+"""Build the deterministic, self-contained remote tunnel agent zipapp."""
+
+from __future__ import annotations
+
+import io
+import zipfile
+from pathlib import Path
+
+_PACKAGE_MODULES = (
+    "async_lifecycle.py",
+    "mux_connection.py",
+    "mux_protocol.py",
+    "mux_records.py",
+    "mux_rpc.py",
+    "mux_stream.py",
+    "remote_mux_agent.py",
+)
+_ZIP_TIMESTAMP = (1980, 1, 1, 0, 0, 0)
+_PACKAGE_INIT = b'"""Self-contained HostBridge remote tunnel agent."""\n'
+_ENTRYPOINT = b"""from __future__ import annotations
+
+import asyncio
+import os
+import sys
+
+from server_control_mcp.remote_mux_agent import run_stdio_agent
+
+try:
+    exit_code = asyncio.run(run_stdio_agent())
+except Exception as exc:
+    nonce = sys.argv[1].encode("ascii")
+    error = type(exc).__name__.encode("ascii", "replace")[:64]
+    os.write(1, b"\\0HBFAIL2:" + nonce + b":" + error + b"\\0")
+    raise
+raise SystemExit(exit_code)
+"""
+
+
+def build_remote_agent_bundle() -> bytes:
+    package_dir = Path(__file__).resolve().parent
+    files: list[tuple[str, bytes]] = [
+        ("__main__.py", _ENTRYPOINT),
+        ("server_control_mcp/__init__.py", _PACKAGE_INIT),
+    ]
+    for name in _PACKAGE_MODULES:
+        files.append((f"server_control_mcp/{name}", (package_dir / name).read_bytes()))
+
+    output = io.BytesIO()
+    with zipfile.ZipFile(output, mode="w", compression=zipfile.ZIP_STORED, strict_timestamps=True) as archive:
+        for path, content in files:
+            info = zipfile.ZipInfo(path, date_time=_ZIP_TIMESTAMP)
+            info.compress_type = zipfile.ZIP_STORED
+            info.create_system = 3
+            info.external_attr = 0o100644 << 16
+            archive.writestr(info, content)
+    return output.getvalue()