// 集群调度WS
import func from '@/utils/func'
import { getWebsocketUrl } from '@/utils/websocket/config'
import { useConnectWebSocket } from '@/utils/websocket/connect-websocket'

/**
 * 使用无人机WebSocket连接，根据工作区ID动态管理WebSocket连接。
 * model_type
 *
 * @param {Ref<string>} workspaceIdRef - 工作区ID的引用，用于动态更新WebSocket连接。
 * @returns  {Object} - 包含wsInfo和removeWS的响应式对象。
 */
export const useClusterSchedulingWS = (workspaceIdRef, modelType) => {
	// console.log('55555', workspaceIdRef.value)
	const wsInfo = ref({})
	let droneWebSocket = null

	// ws消息钩子 todo后面加节流参数处理
	const messageHandler = result => {
		let payload = JSON.parse(result)
		// console.log('payload', payload)
		wsInfo.value = payload
		// if (payload.length > 0) {
		// 	wsInfo.value = payload
		// }
	}

	function init() {
		let webSocketUrl = getWebsocketUrl() + '&workspace-id=' + workspaceIdRef.value + '&model_type=' + modelType
		// 监听ws 消息
		droneWebSocket = useConnectWebSocket(messageHandler, webSocketUrl)
	}

	watch(workspaceIdRef, (newValue, oldValue) => {
		removeWS()
		if (workspaceIdRef.value) {
			init()
		}
	})

	// 移除ws 事件
	function removeWS() {
		droneWebSocket?.close()
		droneWebSocket = null
		wsInfo.value = []
	}

	function sendWSMessage(message) {
		console.log('发送消息', droneWebSocket)
		droneWebSocket.sendMessage(message)
	}

	onBeforeUnmount(() => {
		removeWS()
	})

	return { wsInfo, removeWS, sendWSMessage }
}