function getWsUrl() {
  const protocol = window.location.protocol === "https:" ? "wss" : "ws";
  const host = window.location.host; // 包含主机和端口（如 localhost:5000）
  const path = "/api/ws"; // 你的 WebSocket 路由
  return `${protocol}://${host}${path}`;
}

function initWebSocket() {
  // 替换成你实际的 WebSocket 地址，例如 ws://localhost:5000/ws
  wsUrl = getWsUrl();
  const ws = new WebSocket(wsUrl);
  // 设置接收二进制数据为 ArrayBuffer
  ws.binaryType = "arraybuffer";

  ws.onopen = function() {
    console.log("WebSocket 已连接");
    // 发送登录数据包，假设格式为 JSON（后端可以根据需要解析此包） --暂时默认0只有一个
    ws.send(JSON.stringify({ user_id: 0 }));
  };

  ws.onmessage = function(event) {
    // 解析收到的二进制数据
    const buffer = event.data;
    const dataView = new DataView(buffer);
    // 读取前四个字节，验证魔数 "TFTF"
    let magic = "";
    for (let i = 0; i < 4; i++) {
      magic += String.fromCharCode(dataView.getUint8(i));
    }
    if (magic !== "TFTF") {
      console.error("收到非法格式数据:", magic);
      return;
    }
    // 读取数据头：接下来的 8 字节分别为 idata_type 和 idata_len（均为 32 位无符号整数，假设大端）
    const idata_type = dataView.getUint32(4, false);
    const idata_len = dataView.getUint32(8, false);

    // 数据体从偏移量 12 开始，长度为 idata_len
    const bodyBytes = new Uint8Array(buffer, 12, idata_len);
    const decoder = new TextDecoder("utf-8");
    const bodyText = decoder.decode(bodyBytes);
    // 假设数据体为 JSON 字符串
    let bodyData;
    try {
      bodyData = JSON.parse(bodyText);
    } catch (error) {
      console.error("解析数据体出错:", error);
      return;
    }

    // 针对 idata_type 为 1 的处理：更新节点
    if(idata_type === 0){
      console.log(bodyData);
    }
    else if (idata_type === 1) {//type-1为更新节点的node_workstatus
      // bodyData 应该包含 node_path 和 node_workstatus 两个字段
      updateTreeNode(bodyData.node_path, bodyData.node_workstatus);//node_tree.js
    }
    else if(idata_type === 2){
      if(cur_task_id === 0){return;}
      // 清空选中状态
      selectedNodeData = null;
      //刷新界面内容
      update_select_node_data_show("-","-","-","-","-",false)
      // 重新加载节点树数据
      loadNodeTree(cur_task_id);
    }
    else {
      console.error("未知的数据类型");
    }
  };

  ws.onerror = function(error) {
    console.error("WebSocket 错误:", error);
  };

  ws.onclose = function() {
    console.log("WebSocket 连接关闭");
  };

  return ws;
}