#   AUTO-MAS: A Multi-Script, Multi-Config Management and Automation Software
#   Copyright © 2024-2025 DLmaster361
#   Copyright © 2025 MoeSnowyFox

#   This file is part of AUTO-MAS.

#   AUTO-MAS is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published
#   by the Free Software Foundation, either version 3 of the License,
#   or (at your option) any later version.

#   AUTO-MAS is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty
#   of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
#   the GNU General Public License for more details.

#   You should have received a copy of the GNU General Public License
#   along with AUTO-MAS. If not, see <https://www.gnu.org/licenses/>.

#   Contact: DLmaster_361@163.com


import time
import asyncio
from fastapi import APIRouter, WebSocket, WebSocketDisconnect

from app.core import Config, Broadcast, TaskManager
from app.services import System
from app.models.schema import *

router = APIRouter(prefix="/api/core", tags=["核心信息"])


@router.websocket("/ws")
async def connect_websocket(websocket: WebSocket):

    if Config.websocket is not None:
        await websocket.close(code=1000, reason="已有连接")
        return

    await websocket.accept()
    Config.websocket = websocket
    last_pong = time.monotonic()
    last_ping = time.monotonic()
    data = {}

    await TaskManager.start_startup_queue()

    while True:

        try:

            data = await asyncio.wait_for(websocket.receive_json(), timeout=1000005.0)
            if data.get("type") == "Signal" and "Pong" in data.get("data", {}):
                last_pong = time.monotonic()
            elif data.get("type") == "Signal" and "Ping" in data.get("data", {}):
                await websocket.send_json(
                    WebSocketMessage(
                        id="Main", type="Signal", data={"Pong": "无描述"}
                    ).model_dump()
                )
            else:
                await Broadcast.put(data)

        except asyncio.TimeoutError:

            if last_pong < last_ping:
                await websocket.close(code=1000, reason="Ping超时")
                break
            await websocket.send_json(
                WebSocketMessage(
                    id="Main", type="Signal", data={"Ping": "无描述"}
                ).model_dump()
            )
            last_ping = time.monotonic()

        except WebSocketDisconnect:
            break

    Config.websocket = None
    await System.set_power("KillSelf")


@router.post("/close")
async def close():
    """关闭后端程序"""

    try:
        await System.set_power("KillSelf")
    except Exception as e:
        return OutBase(
            code=500, status="error", message=f"{type(e).__name__}: {str(e)}"
        )
    return OutBase()