logs.py

"""
CLI command for serving logs with file watching and real-time updates.
"""

import sys
from pathlib import Path

from ..utils.logs_server import serve_logs


def logs_command(args):
    """Serve logs with file watching and real-time updates"""

    port = args.port
    print("🚀 Starting Eval Protocol Logs Server")
    print(f"🌐 URL: http://localhost:{port}")
    print(f"🔌 WebSocket: ws://localhost:{port}/ws")
    print(f"👀 Watching paths: {['current directory']}")
    print(f"🔍 Debug mode: {args.debug}")
    print("Press Ctrl+C to stop the server")
    print("-" * 50)

    # Setup Elasticsearch based on flags
    elasticsearch_config = None
    try:
        if getattr(args, "use_env_elasticsearch_config", False):
            # Use environment variables for configuration
            print("⚙️ Using environment variables for Elasticsearch config")
            from eval_protocol.pytest.remote_rollout_processor import (
                create_elasticsearch_config_from_env,
            )

            elasticsearch_config = create_elasticsearch_config_from_env()
        elif not getattr(args, "disable_elasticsearch_setup", False):
            # Default behavior: start or connect to local Elasticsearch via Docker helper
            from eval_protocol.pytest.elasticsearch_setup import ElasticsearchSetup

            print("🧰 Auto-configuring local Elasticsearch (Docker)")
            elasticsearch_config = ElasticsearchSetup().setup_elasticsearch()
        else:
            print("🚫 Elasticsearch setup disabled; running without Elasticsearch integration")
    except Exception as e:
        print(f"❌ Failed to configure Elasticsearch: {e}")
        return 1

    try:
        serve_logs(port=args.port, elasticsearch_config=elasticsearch_config, debug=args.debug)
        return 0
    except KeyboardInterrupt:
        print("\n🛑 Server stopped by user")
        return 0
    except Exception as e:
        print(f"❌ Error starting server: {e}")
        return 1