#!/usr/bin/env python
"""
WebSocket 开发服务器启动脚本
同时启动 Daphne (WebSocket) 和 Django runserver (开发服务器)
支持 Windows 和 Linux
"""

import os
import sys
import subprocess
import platform
import signal
from pathlib import Path
from threading import Thread
import time


def main():
    # 获取项目根目录
    base_dir = Path(__file__).resolve().parent
    os.chdir(base_dir)

    print("=" * 70)
    print("启动 Django 开发服务器 + WebSocket 服务器")
    print("=" * 70)
    print()

    # 检测操作系统
    is_windows = platform.system() == "Windows"

    # 检查虚拟环境（跨平台）
    if is_windows:
        # Windows: .venv/Scripts/python.exe
        venv_python = base_dir / ".venv" / "Scripts" / "python.exe"
    else:
        # Linux/Mac: .venv/bin/python
        venv_python = base_dir / ".venv" / "bin" / "python"

    if venv_python.exists():
        print("[✓] 使用虚拟环境: .venv")
        python_exec = str(venv_python)
    else:
        print("[!] 未找到 .venv 虚拟环境，使用全局 Python")
        print("[!] 建议创建虚拟环境: python -m venv .venv")
        python_exec = sys.executable

    # 检查 daphne 是否安装
    try:
        result = subprocess.run(
            [python_exec, "-m", "daphne", "--version"], capture_output=True, text=True
        )
        if result.returncode != 0 and result.returncode != 2:
            print("[✗] 错误: Daphne 未安装")
            print("请运行: pip install daphne channels channels-redis")
            sys.exit(1)
    except FileNotFoundError:
        print("[✗] 错误: 无法找到 Python 解释器")
        sys.exit(1)
    # 进程列表
    processes = []

    # 启动两个服务器
    try:
        # 启动 Django runserver (端口 8000)
        print("[*] 启动 Django 开发服务器 (端口 8000)...")
        runserver_cmd = ["uv", "run", "manage.py", "runserver", "0.0.0.0:8000"]
        runserver_process = subprocess.Popen(
            runserver_cmd,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
            text=True,
            bufsize=1,
            universal_newlines=True,
        )
        processes.append(("Django Runserver", runserver_process))

        # 等待一下，让 runserver 先启动
        time.sleep(1)

        # 启动 Daphne (端口 8001)
        print("[*] 启动 Daphne WebSocket 服务器 (端口 8001)...")
        daphne_cmd = [
            python_exec,
            "-m",
            "daphne",
            "-b",
            "0.0.0.0",
            "-p",
            "8001",
            "oj.asgi:application",
        ]
        daphne_process = subprocess.Popen(
            daphne_cmd,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
            text=True,
            bufsize=1,
            universal_newlines=True,
        )
        processes.append(("Daphne", daphne_process))

        print()
        print("[✓] 所有服务器已启动")
        print()

        # 创建输出线程
        def print_output(name, process):
            """打印进程输出"""
            for line in process.stdout:
                print(f"[{name}] {line}", end="")

        # 启动输出线程
        threads = []
        for name, process in processes:
            thread = Thread(target=print_output, args=(name, process), daemon=True)
            thread.start()
            threads.append(thread)

        # 等待进程（任意一个退出就退出）
        while True:
            for name, process in processes:
                if process.poll() is not None:
                    print(f"\n[!] {name} 已退出")
                    raise KeyboardInterrupt
            time.sleep(0.5)

    except KeyboardInterrupt:
        print()
        print()
        print("[*] 正在停止所有服务器...")

        # 终止所有进程
        for name, process in processes:
            try:
                if process.poll() is None:  # 如果进程还在运行
                    print(f"[*] 停止 {name}...")
                    if is_windows:
                        # Windows 使用 CTRL_C_EVENT
                        process.send_signal(signal.CTRL_C_EVENT)
                    else:
                        # Unix 使用 SIGTERM
                        process.terminate()

                    # 等待进程结束（最多 5 秒）
                    try:
                        process.wait(timeout=5)
                    except subprocess.TimeoutExpired:
                        print(f"[!] {name} 未响应，强制终止...")
                        process.kill()
                        process.wait()
            except Exception as e:
                print(f"[!] 停止 {name} 时出错: {e}")

        print()
        print("[✓] 所有服务器已停止")

    except Exception as e:
        print(f"[✗] 错误: {e}")

        # 清理所有进程
        for name, process in processes:
            try:
                if process.poll() is None:
                    process.kill()
                    process.wait()
            except Exception:
                pass

        sys.exit(1)


if __name__ == "__main__":
    main()