ss

d2ypp2/tools/render_neko_compose.py

@@ -0,0 +1,122 @@
+#!/usr/bin/env python3
+"""Render a dynamic Neko docker-compose file from the current environment."""
+from __future__ import annotations
+
+import argparse
+import json
+import os
+from pathlib import Path
+
+
+def env(name: str, default: str) -> str:
+    value = os.environ.get(name)
+    return value if value not in {None, ""} else default
+
+
+def q(value: str) -> str:
+    return json.dumps(str(value), ensure_ascii=False)
+
+
+def instance_value(prefix: str, idx: int, default: str) -> str:
+    return env(f"{prefix}_{idx}", default)
+
+
+def default_udp_range(idx: int) -> str:
+    if idx == 1:
+        return "52000-52099"
+    if idx == 2:
+        return "52101-52199"
+    if idx == 3:
+        return "52201-52299"
+    start = 52400 + ((idx - 4) * 100)
+    return f"{start}-{start + 99}"
+
+
+def render(count: int) -> str:
+    image = env("NEKO_IMAGE", "ghcr.io/m1k1o/neko/chromium:3.0.0")
+    shm = env("NEKO_SHM_SIZE", "512mb")
+    screen = env("NEKO_DESKTOP_SCREEN", "1280x720@15")
+    user_pass = env("NEKO_USER_PASS", "12345678")
+    admin_pass = env("NEKO_ADMIN_PASS", "12345678")
+    icelite = env("NEKO_WEBRTC_ICELITE", "0")
+    nat1to1 = env("NEKO_WEBRTC_NAT1TO1", "")
+    max_fps = env("NEKO_MAX_FPS", "15")
+    video_bitrate = env("NEKO_VIDEO_BITRATE", "1200")
+    plugins_enabled = env("NEKO_PLUGINS_ENABLED", "true")
+    start_url = env("NEKO_START_URL", "https://accounts.google.com/ServiceLogin?continue=https://www.youtube.com/")
+    render_mode = env("NEKO_BROWSER_RENDER_MODE", "auto")
+    supervisor_conf = env("NEKO_BROWSER_SUPERVISOR_CONF", "chromium.conf")
+    base_profile = env("NEKO_PROFILE_HOST_DIR", str(Path.cwd() / "data" / "chromium-profile-v2"))
+    http_base = int(env("NEKO_HTTP_PORT", "9200"))
+    mux_base = int(env("NEKO_WEBRTC_TCPMUX", "52300"))
+
+    lines = [
+        "# Generated by tools/render_neko_compose.py - do not hand-edit",
+        "services:",
+    ]
+    for idx in range(1, count + 1):
+        http_port = instance_value("NEKO_HTTP_PORT", idx, str(http_base + idx - 1))
+        tcp_mux = instance_value("NEKO_WEBRTC_TCPMUX", idx, str(mux_base + idx - 1))
+        udp_mux = instance_value("NEKO_WEBRTC_UDPMUX", idx, str(mux_base + idx - 1))
+        udp_range = instance_value("NEKO_WEBRTC_UDP_RANGE", idx, default_udp_range(idx))
+        epr = instance_value("NEKO_WEBRTC_EPR", idx, udp_range)
+        profile = instance_value("NEKO_PROFILE_HOST_DIR", idx, base_profile if idx == 1 else f"{base_profile}-{idx}")
+        lines.extend(
+            [
+                f"  neko{idx}:",
+                f"    image: {q(image)}",
+                "    restart: unless-stopped",
+                f"    shm_size: {q(shm)}",
+                "    cap_add:",
+                "      - SYS_ADMIN",
+                f"    container_name: live-neko-{idx}",
+                "    environment:",
+                '      DISPLAY: ":99.0"',
+                f"      NEKO_DESKTOP_SCREEN: {q(screen)}",
+                '      NEKO_SERVER_BIND: ":8080"',
+                '      NEKO_MEMBER_PROVIDER: "multiuser"',
+                f"      NEKO_MEMBER_MULTIUSER_USER_PASSWORD: {q(user_pass)}",
+                f"      NEKO_MEMBER_MULTIUSER_ADMIN_PASSWORD: {q(admin_pass)}",
+                f"      NEKO_WEBRTC_ICELITE: {q(icelite)}",
+                f"      NEKO_WEBRTC_NAT1TO1: {q(nat1to1)}",
+                f"      NEKO_MAX_FPS: {q(max_fps)}",
+                f"      NEKO_VIDEO_BITRATE: {q(video_bitrate)}",
+                f"      NEKO_PLUGINS_ENABLED: {q(plugins_enabled)}",
+                '      NEKO_PLUGINS_DIR: "/etc/neko/plugins/"',
+                f"      NEKO_START_URL: {q(start_url)}",
+                f"      NEKO_BROWSER_RENDER_MODE: {q(render_mode)}",
+                f"      NEKO_WEBRTC_EPR: {q(epr)}",
+                f"      NEKO_WEBRTC_TCPMUX: {q(tcp_mux)}",
+                f"      NEKO_WEBRTC_UDPMUX: {q(udp_mux)}",
+                "    ports:",
+                f"      - {q(f'{http_port}:8080')}",
+                f"      - {q(f'{tcp_mux}:{tcp_mux}/tcp')}",
+                f"      - {q(f'{udp_mux}:{udp_mux}/udp')}",
+                f"      - {q(f'{udp_range}:{udp_range}/udp')}",
+                "    volumes:",
+                f"      - {q(f'{profile}:/home/neko/.config/chromium')}",
+                f"      - {q(f'{profile}:/home/neko/.config/google-chrome')}",
+                '      - "./policies:/etc/chromium/policies/managed:ro"',
+                '      - "./policies:/etc/opt/chrome/policies/managed:ro"',
+                '      - "./provisioning:/var/www/provisioning:ro"',
+                '      - "./browser-launch.sh:/etc/neko/browser-launch.sh:ro"',
+                f"      - {q(f'./{supervisor_conf}:/etc/neko/supervisord/{supervisor_conf}:ro')}",
+            ]
+        )
+    return "\n".join(lines) + "\n"
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description="Render Neko docker compose")
+    parser.add_argument("--count", type=int, required=True)
+    parser.add_argument("--output", required=True)
+    args = parser.parse_args()
+    count = max(2, min(64, args.count))
+    output = Path(args.output)
+    output.parent.mkdir(parents=True, exist_ok=True)
+    output.write_text(render(count), encoding="utf-8")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())