from .connect_manager import ConnectionManager
from models import WebSocket
import json

# 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)