😏
This commit is contained in:
Executable
+1426
File diff suppressed because it is too large
Load Diff
Executable
+77
@@ -0,0 +1,77 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Small Switchyard-compatible health/models stand-in for integration tests."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import signal
|
||||
import tomllib
|
||||
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def route_ids(path: Path) -> list[str]:
|
||||
with path.open("rb") as handle:
|
||||
data = tomllib.load(handle)
|
||||
return [str(value["id"]) for value in data.get("routes", {}).values()]
|
||||
|
||||
|
||||
def main() -> int:
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--version", action="version", version="switchyard-server 0.2.0")
|
||||
parser.add_argument("--config", type=Path, required=True)
|
||||
parser.add_argument("--host", default="127.0.0.1")
|
||||
parser.add_argument("--port", type=int, default=4000)
|
||||
parser.add_argument("--routing-log-file")
|
||||
parser.add_argument("--dry-run", action="store_true")
|
||||
args = parser.parse_args()
|
||||
models = route_ids(args.config)
|
||||
if args.dry_run:
|
||||
print(json.dumps({"valid": True, "routes": models}))
|
||||
return 0
|
||||
|
||||
class Handler(BaseHTTPRequestHandler):
|
||||
def log_message(self, _format: str, *_args: object) -> None:
|
||||
return
|
||||
|
||||
def respond(self, status: int, value: object) -> None:
|
||||
body = json.dumps(value).encode()
|
||||
self.send_response(status)
|
||||
self.send_header("Content-Type", "application/json")
|
||||
self.send_header("Content-Length", str(len(body)))
|
||||
self.end_headers()
|
||||
self.wfile.write(body)
|
||||
|
||||
def do_GET(self) -> None: # noqa: N802
|
||||
if self.path == "/health":
|
||||
self.respond(200, {"status": "ok"})
|
||||
elif self.path in {"/v1/models", "/models"}:
|
||||
self.respond(
|
||||
200,
|
||||
{
|
||||
"object": "list",
|
||||
"data": [{"id": item, "object": "model"} for item in models],
|
||||
},
|
||||
)
|
||||
else:
|
||||
self.respond(404, {"error": "not found"})
|
||||
|
||||
server = ThreadingHTTPServer((args.host, args.port), Handler)
|
||||
|
||||
def stop(_signum: int, _frame: object) -> None:
|
||||
raise KeyboardInterrupt
|
||||
|
||||
signal.signal(signal.SIGTERM, stop)
|
||||
signal.signal(signal.SIGINT, stop)
|
||||
try:
|
||||
server.serve_forever(poll_interval=0.1)
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
finally:
|
||||
server.server_close()
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
@@ -0,0 +1,46 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Enter an allocated pseudo-terminal and invoke the real MMO TUI launcher."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import fcntl
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import termios
|
||||
|
||||
from mmo_runtime import launch_interactive
|
||||
|
||||
|
||||
def main() -> int:
|
||||
try:
|
||||
# Popen(start_new_session=True) made this process a session leader. The
|
||||
# slave PTY descriptors were inherited, so explicitly claim the terminal
|
||||
# before MMO performs normal foreground process-group handoff.
|
||||
fcntl.ioctl(sys.stdin.fileno(), termios.TIOCSCTTY, 0)
|
||||
os.tcsetpgrp(sys.stdin.fileno(), os.getpgrp())
|
||||
initial = termios.tcgetattr(sys.stdin.fileno())
|
||||
exit_code = launch_interactive(profile=sys.argv[1], cwd=sys.argv[2])
|
||||
restored = termios.tcgetattr(sys.stdin.fileno())
|
||||
print(
|
||||
json.dumps(
|
||||
{
|
||||
"event": "launcher_returned",
|
||||
"exit_code": exit_code,
|
||||
"echo_enabled": bool(restored[3] & termios.ECHO),
|
||||
"attributes_restored": restored == initial,
|
||||
"pgrp": os.getpgrp(),
|
||||
"foreground_pgrp": os.tcgetpgrp(sys.stdin.fileno()),
|
||||
},
|
||||
sort_keys=True,
|
||||
),
|
||||
flush=True,
|
||||
)
|
||||
return exit_code
|
||||
except BaseException as exc:
|
||||
print(json.dumps({"event": "launcher_error", "error": repr(exc)}), flush=True)
|
||||
return 97
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
Reference in New Issue
Block a user