SashegDev
120 lines
4.3 KiB
Python
120 lines
4.3 KiB
Python
# cli.py
|
|
import argparse
|
|
import sys
|
|
import asyncio
|
|
from pathlib import Path
|
|
import structlog
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description="ZernMC Launcher Server")
|
|
|
|
# Mode selection (mutually exclusive)
|
|
mode_group = parser.add_mutually_exclusive_group()
|
|
mode_group.add_argument("--dev", action="store_true", help="Development mode with auto-reload")
|
|
mode_group.add_argument("--prod", action="store_true", help="Production mode with 4 workers")
|
|
mode_group.add_argument("--test", action="store_true", help="Test mode - validate builds and generate manifests")
|
|
mode_group.add_argument("--sync", action="store_true", help="Sync mode - sync with main server as mirror")
|
|
|
|
# Additional options
|
|
parser.add_argument("--host", default="0.0.0.0", help="Host to bind to (default: 0.0.0.0)")
|
|
parser.add_argument("--port", type=int, default=1582, help="Port to bind to (default: 1582)")
|
|
parser.add_argument("--workers", type=int, default=1, help="Number of workers for production mode (default: 1, more causes file download slowdown)")
|
|
parser.add_argument("--reload", action="store_true", help="Enable auto-reload (development)")
|
|
|
|
return parser.parse_args()
|
|
|
|
async def run_test_mode():
|
|
"""Validate all packs and generate/update manifests"""
|
|
logger.info("Running in TEST mode - validating builds and generating manifests")
|
|
|
|
from pack_manager import scan_pack, PACKS_DIR
|
|
|
|
pack_count = 0
|
|
error_count = 0
|
|
|
|
for pack_dir in PACKS_DIR.iterdir():
|
|
if pack_dir.is_dir():
|
|
try:
|
|
logger.info(f"Validating pack: {pack_dir.name}")
|
|
meta = await scan_pack(pack_dir.name)
|
|
pack_count += 1
|
|
logger.info(f"✓ Pack validated: {pack_dir.name} v{meta.version}, {len(meta.files)} files")
|
|
except Exception as e:
|
|
error_count += 1
|
|
logger.error(f"✗ Failed to validate pack {pack_dir.name}", error=str(e), exc_info=True)
|
|
|
|
logger.info(f"Test mode completed: {pack_count} packs validated, {error_count} errors")
|
|
|
|
if error_count > 0:
|
|
logger.error("Some packs failed validation")
|
|
sys.exit(1)
|
|
else:
|
|
logger.info("All packs validated successfully")
|
|
sys.exit(0)
|
|
|
|
async def run_sync_mode():
|
|
"""Sync with main server as mirror"""
|
|
import os
|
|
|
|
main_url = os.environ.get("MAIN_SERVER_URL")
|
|
if not main_url:
|
|
logger.error("MAIN_SERVER_URL not set. Run: MAIN_SERVER_URL=http://main:1582 python cli.py --sync")
|
|
sys.exit(1)
|
|
|
|
logger.info(f"Starting mirror sync from {main_url}")
|
|
|
|
# Get version from main
|
|
import httpx
|
|
async with httpx.AsyncClient() as client:
|
|
# Get version
|
|
try:
|
|
resp = await client.get(f"{main_url}/launcher/version")
|
|
data = resp.json()
|
|
version = data.get("version")
|
|
logger.info(f"Main server version: {version}")
|
|
except Exception as e:
|
|
logger.error(f"Failed to get version from main: {e}")
|
|
sys.exit(1)
|
|
|
|
# Get sync manifest
|
|
try:
|
|
resp = await client.get(f"{main_url}/launcher/sync/{version}")
|
|
sync_data = resp.json()
|
|
logger.info(f"Files to sync: {len(sync_data.get('files', []))}")
|
|
except Exception as e:
|
|
logger.error(f"Failed to get sync manifest: {e}")
|
|
sys.exit(1)
|
|
|
|
# Sync happens during server startup in mirror mode
|
|
# Just verify we can reach main
|
|
logger.info("Mirror sync configured. Server will sync on startup.")
|
|
|
|
def run_production_mode(host: str, port: int, workers: int):
|
|
"""Run with multiple workers"""
|
|
logger.info(f"Starting in PRODUCTION mode with {workers} workers on {host}:{port}")
|
|
|
|
import uvicorn
|
|
uvicorn.run(
|
|
"main:app",
|
|
host=host,
|
|
port=port,
|
|
workers=workers,
|
|
log_config=None,
|
|
access_log=False # We have our own logging middleware
|
|
)
|
|
|
|
def run_development_mode(host: str, port: int, reload: bool = True):
|
|
"""Run with auto-reload for development"""
|
|
logger.info(f"Starting in DEVELOPMENT mode with reload={reload} on {host}:{port}")
|
|
|
|
import uvicorn
|
|
uvicorn.run(
|
|
"main:app",
|
|
host=host,
|
|
port=port,
|
|
reload=reload,
|
|
log_config=None,
|
|
access_log=False
|
|
) |