from .connect_manager import ConnectionManager
from models import WebSocket
import json, asyncio
from mcu.DeviceControl import DeviceControl, checkMcuConnection
from mcu.BlueToothMode import BlueToothMode

# socket消息发送逻辑处理方法
async def handlerSend(
    manager: ConnectionManager, receiveData: str, websocket: WebSocket
):
    receiveData = json.loads(receiveData)
    # 处理消息发送逻辑
    receiveData = json.loads(receiveData.get("text"))
    jsonType = receiveData.get("type")
    code = receiveData.get("code")
    msg = receiveData.get("msg")
    print("receiveData", receiveData)
    print("jsonType", jsonType)
    match jsonType:
        case "ping":
            """发送心跳"""
            data = manager.jsonMessage("pong")
            await manager.send_personal_message(data, websocket)
        case "pong":
            """发送心跳"""
            pass
        case "forward_message":
            data = receiveData.get("data")
            dictMsg = {"code":code,"msg":msg,"data":data}
            await manager.broadcast(dictMsg)
        case "connect_mcu":
            device_ctrl = DeviceControl(websocket_manager=manager)
            # if device_ctrl.serial_ins.check_connect():
            #     print("未连接")
            loop = asyncio.get_event_loop()
            loop.create_task(checkMcuConnection(device_ctrl), name="mcu")
        case "connect_bluetooth":
            blue_tooth = BlueToothMode(websocket_manager=manager)
            # await  blue_tooth.main_func()
            loop = asyncio.get_event_loop()
            loop.create_task(blue_tooth.main_func(), name="blue_tooth")
            # loop.close()