add banbanmini backend

talkingq-url/api/ota.py

@@ -0,0 +1,103 @@
+from fastapi import APIRouter, HTTPException, Depends, Path, Query
+from pydantic import BaseModel
+from typing import Dict, Any, Optional
+from api.auth import api_auth, client_auth
+from api.response import Response
+from services.ota_flow_controller import ota_flow_controller
+from utils.logger import session_logger
+from services.connection_manager import connection_manager
+from services.system_config_manager import system_config_manager
+
+router = APIRouter(prefix="/api/ota", tags=["OTA"])
+
+class OTAStatusResponse(BaseModel):
+    status: str
+    progress: float
+    version: str
+
+class FirmwareUpdateSettings(BaseModel):
+    version: str
+    url: str
+
+@router.get("/check/{device_id}", response_model=Response)
+async def check_update(
+    device_id: str = Path(..., description="设备ID"),
+    authenticated: bool = Depends(api_auth)
+):
+    """检查设备是否需要更新 (APP调用)"""
+    try:
+        result = await ota_flow_controller.start_update_flow(device_id)
+        if result["status"] == "error":
+            return Response(code=-1, msg=result["message"], data={})
+        return Response(code=0, msg="success", data=result)
+    except Exception as e:
+        session_logger.error(device_id, "ota", f"检查更新失败: {str(e)}")
+        return Response(code=-1, msg=f"检查更新失败: {str(e)}", data={})
+
+@router.post("/start/{device_id}", response_model=Response)
+async def start_update(
+    device_id: str = Path(..., description="设备ID"),
+    authenticated: bool = Depends(api_auth)
+):
+    """启动设备更新 (APP调用)"""
+    try:
+        result = await ota_flow_controller.execute_update(device_id)
+        if result["status"] == "error":
+            return Response(code=-1, msg=result["message"], data={})
+        return Response(code=0, msg="success", data={"updating": True})
+    except Exception as e:
+        session_logger.error(device_id, "ota", f"启动更新失败: {str(e)}")
+        return Response(code=-1, msg=f"启动更新失败: {str(e)}", data={"updating": False})
+
+@router.get("/status/{device_id}", response_model=Response)
+async def get_update_status(
+    device_id: str = Path(..., description="设备ID"),
+    authenticated: bool = Depends(api_auth)
+):
+    """获取设备更新状态 (APP调用)"""
+    try:
+        status = await ota_flow_controller.get_update_status(device_id)
+        websocket = await connection_manager.get_connection(device_id)
+        status["device_online"] = websocket is not None
+        return Response(code=0, msg="success", data=status)
+    except Exception as e:
+        session_logger.error(device_id, "ota", f"获取更新状态失败: {str(e)}")
+        return Response(code=-1, msg=f"获取更新状态失败: {str(e)}", data={
+            "status": "error",
+            "progress": 0,
+            "version": "unknown",
+            "device_online": False
+        })
+
+@router.post("/config/firmware", response_model=Response)
+async def update_firmware_settings(
+    settings: FirmwareUpdateSettings,
+    admin_api_key: str = Depends(api_auth)
+):
+    """更新固件配置信息（版本和URL）"""
+    try:
+        await system_config_manager.update_config('latest_firmware_version', settings.version)
+        await system_config_manager.update_config('update_firmware_url', settings.url)
+        return Response(code=0, msg="固件配置已更新", data={
+            "version": settings.version,
+            "url": settings.url
+        })
+    except Exception as e:
+        session_logger.error("system", "ota", f"更新固件配置失败: {str(e)}")
+        return Response(code=-1, msg=f"更新固件配置失败: {str(e)}", data={})
+
+@router.get("/config/firmware", response_model=Response)
+async def get_firmware_settings(
+    admin_api_key: str = Depends(api_auth)
+):
+    """获取当前固件配置信息"""
+    try:
+        version_config = await system_config_manager.get_config('latest_firmware_version')
+        url_config = await system_config_manager.get_config('update_firmware_url')
+        return Response(code=0, msg="success", data={
+            "version": version_config.config_value if version_config else "",
+            "url": url_config.config_value if url_config else ""
+        })
+    except Exception as e:
+        session_logger.error("system", "ota", f"获取固件配置失败: {str(e)}")
+        return Response(code=-1, msg=f"获取固件配置失败: {str(e)}", data={})