claudePySnake/run_server.py

"""Run the Snake game server."""

import asyncio
import argparse
from pathlib import Path
from src.server.game_server import GameServer
from src.server.http_server import HTTPServer
from src.shared.constants import DEFAULT_HOST, DEFAULT_PORT, DEFAULT_WS_PORT, DEFAULT_HTTP_PORT


async def main() -> None:
    """Run the server with command line arguments."""
    parser = argparse.ArgumentParser(description="Run the Snake game server")
    parser.add_argument(
        "--host",
        default=DEFAULT_HOST,
        help=f"Host address to bind to (default: {DEFAULT_HOST})",
    )
    parser.add_argument(
        "--port",
        type=int,
        default=DEFAULT_PORT,
        help=f"TCP port number (default: {DEFAULT_PORT})",
    )
    parser.add_argument(
        "--ws-port",
        type=int,
        default=DEFAULT_WS_PORT,
        help=f"WebSocket port (default: {DEFAULT_WS_PORT}, 0 to disable)",
    )
    parser.add_argument(
        "--http-port",
        type=int,
        default=DEFAULT_HTTP_PORT,
        help=f"HTTP server port (default: {DEFAULT_HTTP_PORT}, 0 to disable)",
    )
    parser.add_argument(
        "--web-dir",
        default="web",
        help="Directory containing web client files (default: web)",
    )
    parser.add_argument(
        "--name",
        default="Snake Server",
        help="Server name for discovery (default: Snake Server)",
    )
    parser.add_argument(
        "--no-discovery",
        action="store_true",
        help="Disable multicast discovery beacon",
    )
    parser.add_argument(
        "--no-websocket",
        action="store_true",
        help="Disable WebSocket server",
    )
    parser.add_argument(
        "--no-http",
        action="store_true",
        help="Disable HTTP server (for production with external web server)",
    )

    args = parser.parse_args()

    # Determine WebSocket port
    ws_port = None if args.no_websocket or args.ws_port == 0 else args.ws_port

    # Create game server
    server = GameServer(
        host=args.host,
        port=args.port,
        server_name=args.name,
        enable_discovery=not args.no_discovery,
        ws_port=ws_port,
    )

    # Start HTTP server if enabled
    http_task = None
    if not args.no_http and args.http_port > 0:
        web_dir = Path(args.web_dir)
        if web_dir.exists():
            http_server = HTTPServer(web_dir, args.http_port, "0.0.0.0")
            await http_server.start()
            http_task = asyncio.create_task(asyncio.Future())  # Keep running
        else:
            print(f"Warning: Web directory '{web_dir}' not found. HTTP server disabled.")

    # Start game server
    try:
        await server.start()
    finally:
        if http_task:
            http_task.cancel()


if __name__ == "__main__":
    asyncio.run(main())