desungongpai/app/routers/commands.py

"""
Commands Router - 指令管理接口
API endpoints for sending commands / messages to devices and viewing command history.
"""

import math

from fastapi import APIRouter, Depends, HTTPException, Query
from pydantic import BaseModel, Field
from sqlalchemy.ext.asyncio import AsyncSession

from app.database import get_db
from app.schemas import (
    APIResponse,
    CommandResponse,
    PaginatedList,
)
from app.services import command_service, device_service

router = APIRouter(prefix="/api/commands", tags=["Commands / 指令管理"])


# ---------------------------------------------------------------------------
# Request schemas specific to this router
# ---------------------------------------------------------------------------


class SendCommandRequest(BaseModel):
    """Request body for sending a command to a device."""
    device_id: int | None = Field(None, description="设备ID / Device ID (provide device_id or imei)")
    imei: str | None = Field(None, description="IMEI号 / IMEI number (provide device_id or imei)")
    command_type: str = Field(..., max_length=30, description="指令类型 / Command type")
    command_content: str = Field(..., description="指令内容 / Command content")


class SendMessageRequest(BaseModel):
    """Request body for sending a message (0x82) to a device."""
    device_id: int | None = Field(None, description="设备ID / Device ID (provide device_id or imei)")
    imei: str | None = Field(None, description="IMEI号 / IMEI number (provide device_id or imei)")
    message: str = Field(..., max_length=500, description="消息内容 / Message content")


class SendTTSRequest(BaseModel):
    """Request body for sending a TTS voice broadcast to a device."""
    device_id: int | None = Field(None, description="设备ID / Device ID (provide device_id or imei)")
    imei: str | None = Field(None, description="IMEI号 / IMEI number (provide device_id or imei)")
    text: str = Field(..., min_length=1, max_length=200, description="语音播报文本 / TTS text content")


# ---------------------------------------------------------------------------
# Helper
# ---------------------------------------------------------------------------


async def _resolve_device(
    db: AsyncSession,
    device_id: int | None,
    imei: str | None,
):
    """Resolve a device from either device_id or imei. Returns the Device ORM instance."""
    if device_id is None and imei is None:
        raise HTTPException(
            status_code=400,
            detail="Either device_id or imei must be provided / 必须提供 device_id 或 imei",
        )

    if device_id is not None:
        device = await device_service.get_device(db, device_id)
    else:
        device = await device_service.get_device_by_imei(db, imei)

    if device is None:
        identifier = f"ID={device_id}" if device_id else f"IMEI={imei}"
        raise HTTPException(
            status_code=404,
            detail=f"Device {identifier} not found / 未找到设备 {identifier}",
        )
    return device


# ---------------------------------------------------------------------------
# Endpoints
# ---------------------------------------------------------------------------


@router.get(
    "",
    response_model=APIResponse[PaginatedList[CommandResponse]],
    summary="获取指令历史 / List command history",
)
async def list_commands(
    device_id: int | None = Query(default=None, description="设备ID / Device ID"),
    status: str | None = Query(default=None, description="指令状态 / Command status (pending/sent/success/failed)"),
    page: int = Query(default=1, ge=1, description="页码 / Page number"),
    page_size: int = Query(default=20, ge=1, le=100, description="每页数量 / Items per page"),
    db: AsyncSession = Depends(get_db),
):
    """
    获取指令历史记录，支持按设备和状态过滤。
    List command history with optional device and status filters.
    """
    commands, total = await command_service.get_commands(
        db, device_id=device_id, status=status, page=page, page_size=page_size
    )
    return APIResponse(
        data=PaginatedList(
            items=[CommandResponse.model_validate(c) for c in commands],
            total=total,
            page=page,
            page_size=page_size,
            total_pages=math.ceil(total / page_size) if total else 0,
        )
    )


@router.post(
    "/send",
    response_model=APIResponse[CommandResponse],
    status_code=201,
    summary="发送指令 / Send command to device",
)
async def send_command(body: SendCommandRequest, db: AsyncSession = Depends(get_db)):
    """
    向设备发送指令（通过TCP连接下发）。
    Send a command to a device via the TCP connection.
    Requires the device to be online.
    """
    device = await _resolve_device(db, body.device_id, body.imei)

    # Import tcp_manager lazily to avoid circular imports
    from app.tcp_server import tcp_manager

    # Check if device is connected
    if device.imei not in tcp_manager.connections:
        raise HTTPException(
            status_code=400,
            detail=f"Device {device.imei} is not online / 设备 {device.imei} 不在线",
        )

    # Create command log entry
    command_log = await command_service.create_command(
        db,
        device_id=device.id,
        command_type=body.command_type,
        command_content=body.command_content,
    )

    # Send command via TCP
    try:
        await tcp_manager.send_command(
            device.imei, body.command_type, body.command_content
        )
    except Exception as e:
        command_log.status = "failed"
        await db.flush()
        await db.refresh(command_log)
        raise HTTPException(
            status_code=500,
            detail=f"Failed to send command / 指令发送失败: {str(e)}",
        )

    command_log.status = "sent"
    await db.flush()
    await db.refresh(command_log)

    return APIResponse(
        message="Command sent successfully / 指令发送成功",
        data=CommandResponse.model_validate(command_log),
    )


@router.post(
    "/message",
    response_model=APIResponse[CommandResponse],
    status_code=201,
    summary="发送留言 / Send message to device (0x82)",
)
async def send_message(body: SendMessageRequest, db: AsyncSession = Depends(get_db)):
    """
    向设备发送留言消息（协议号 0x82）。
    Send a text message to a device using protocol 0x82.
    """
    device = await _resolve_device(db, body.device_id, body.imei)

    from app.tcp_server import tcp_manager

    if device.imei not in tcp_manager.connections:
        raise HTTPException(
            status_code=400,
            detail=f"Device {device.imei} is not online / 设备 {device.imei} 不在线",
        )

    # Create command log for the message
    command_log = await command_service.create_command(
        db,
        device_id=device.id,
        command_type="message",
        command_content=body.message,
    )

    try:
        await tcp_manager.send_message(device.imei, body.message)
    except Exception as e:
        command_log.status = "failed"
        await db.flush()
        await db.refresh(command_log)
        raise HTTPException(
            status_code=500,
            detail=f"Failed to send message / 留言发送失败: {str(e)}",
        )

    command_log.status = "sent"
    await db.flush()
    await db.refresh(command_log)

    return APIResponse(
        message="Message sent successfully / 留言发送成功",
        data=CommandResponse.model_validate(command_log),
    )


@router.post(
    "/tts",
    response_model=APIResponse[CommandResponse],
    status_code=201,
    summary="语音下发 / Send TTS voice broadcast to device",
)
async def send_tts(body: SendTTSRequest, db: AsyncSession = Depends(get_db)):
    """
    向设备发送 TTS 语音播报（通过 0x80 在线指令，TTS 命令格式）。
    Send a TTS voice broadcast to a device via online command (0x80).
    The device will use its built-in TTS engine to speak the text aloud.
    """
    device = await _resolve_device(db, body.device_id, body.imei)

    from app.tcp_server import tcp_manager

    if device.imei not in tcp_manager.connections:
        raise HTTPException(
            status_code=400,
            detail=f"Device {device.imei} is not online / 设备 {device.imei} 不在线",
        )

    tts_command = f"TTS,{body.text}"

    # Create command log entry
    command_log = await command_service.create_command(
        db,
        device_id=device.id,
        command_type="tts",
        command_content=tts_command,
    )

    try:
        await tcp_manager.send_command(device.imei, "tts", tts_command)
    except Exception as e:
        command_log.status = "failed"
        await db.flush()
        await db.refresh(command_log)
        raise HTTPException(
            status_code=500,
            detail=f"Failed to send TTS / 语音下发失败: {str(e)}",
        )

    command_log.status = "sent"
    await db.flush()
    await db.refresh(command_log)

    return APIResponse(
        message="TTS sent successfully / 语音下发成功",
        data=CommandResponse.model_validate(command_log),
    )


@router.get(
    "/{command_id}",
    response_model=APIResponse[CommandResponse],
    summary="获取指令详情 / Get command details",
)
async def get_command(command_id: int, db: AsyncSession = Depends(get_db)):
    """
    按ID获取指令详情。
    Get command log details by ID.
    """
    command = await command_service.get_command(db, command_id)
    if command is None:
        raise HTTPException(status_code=404, detail=f"Command {command_id} not found / 未找到指令{command_id}")
    return APIResponse(data=CommandResponse.model_validate(command))