// ==UserScript==
// @name         直播弹幕监听 转发至本地WS服务端
// @namespace    http://tampermonkey.net/
// @version      0.5
// @description  观察指定 DOM 节点的变化以将数据发送到连接的WebSocket服务端
// @description  Github：https://github.com/Ikaros-521/AI-Vtuber/tree/main/Scripts/%E7%9B%B4%E6%92%ADws%E8%84%9A%E6%9C%AC
// @author       Ikaros
// @match        https://www.douyu.com/*
// @match        https://live.kuaishou.com/u/*
// @grant        none
// @namespace    https://greasyfork.org/scripts/490966
// @license      GPL-3.0
// @downloadURL none
// ==/UserScript==

(function () {
  "use strict";

  let socket = null;
  let wsUrl = "ws://127.0.0.1:5000";
  let targetNode = null;
  let observer = null;

  const hostname = window.location.hostname;

  if (hostname === "www.douyu.com") {
    wsUrl = "ws://127.0.0.1:5000";
  } else if (hostname === "live.kuaishou.com") {
    wsUrl = "ws://127.0.0.1:5000";
  }

  function connectWebSocket() {
    // 创建 WebSocket 连接，适配服务端
    socket = new WebSocket(wsUrl);

    // 当连接建立时触发
    socket.addEventListener("open", (event) => {
      console.log("ws连接打开");

      // 向服务器发送一条消息
      const data = {
        type: "info",
        content: "ws连接成功",
      };
      console.log(data);
      socket.send(JSON.stringify(data));
    });

    // 当收到消息时触发
    socket.addEventListener("message", (event) => {
      console.log("收到服务器数据:", event.data);
    });

    // 当连接关闭时触发
    socket.addEventListener("close", (event) => {
      console.log("WS连接关闭");

      // 重连
      setTimeout(() => {
        connectWebSocket();
      }, 1000); // 延迟 1 秒后重连
    });
  }

  // 初始连接
  connectWebSocket();
  if (hostname === "www.douyu.com") {
    // 选择需要观察变化的节点
    targetNode = document.querySelector(".Barrage-list");

    // 创建观察器实例
    observer = new MutationObserver((mutations) => {
      mutations.forEach((mutation) => {
        // 这里处理新增的DOM元素
        if (mutation.type === "childList") {
          mutation.addedNodes.forEach((node) => {
            // 判断是否是新增的弹幕消息
            if (node.classList.contains("Barrage-listItem")) {
              // 新增的动态DOM元素处理
              // console.log('Added node:', node);

              const spans = node.getElementsByTagName("span");

              let username = "";
              let content = "";

              for (let span of spans) {
                //console.log(span);
                if (span.classList.contains("Barrage-nickName")) {
                  const targetSpan = span;
                  // 获取用户名
                  let tmp = targetSpan.textContent.trim().slice(0, -1);
                  if (tmp != "")
                    username = targetSpan.textContent.trim().slice(0, -1);
                } else if (span.classList.contains("Barrage-content")) {
                  const targetSpan = span;
                  // 获取弹幕内容
                  content = targetSpan.textContent.trim();
                }
              }

              console.log(username + ":" + content);

              // 获取到弹幕数据
              if (username != "" && content != "") {
                const data = {
                  type: "comment",
                  username: username,
                  content: content,
                };
                console.log(data);
                socket.send(JSON.stringify(data));
              }
            }
          });
        }
      });
    });
  } else if (hostname === "live.kuaishou.com") {
    // 选择需要观察变化的节点
    targetNode = document.querySelector(".history");

    // 创建观察器实例
    observer = new MutationObserver((mutations) => {
      mutations.forEach((mutation) => {
        // 这里处理新增的DOM元素
        if (mutation.type === "childList") {
          mutation.addedNodes.forEach((node) => {
            // 判断是否是新增的弹幕消息
            if (node.classList.contains("chat-info")) {
              // 新增的动态DOM元素处理
              console.log("Added node:", node);

              const usernameElement = node.querySelector(".username");
              const commentElement = node.querySelector(".comment");

              // 礼物数据
              const giftCommentElement = node.querySelector(".gift-comment");
              const giftImgElement = node.querySelector(".gift-img");

              const likeElement = node.querySelector(".like");

              if (usernameElement && giftCommentElement) {
                // 礼物数据处理
                const username = usernameElement.textContent.trim();
                console.log(username + "送出了礼物");

                // 如果 socket 已经初始化，可以在这里发送礼物数据
                if (socket) {
                  const data = {
                    type: "gift",
                    username: username,
                    // 可以根据需要添加其他礼物相关数据
                  };
                  console.log(data);
                  socket.send(JSON.stringify(data));
                }
              } else if (usernameElement && likeElement) {
                const username = usernameElement.textContent.trim();
                console.log(username + "点了个赞");

                // 如果 socket 已经初始化，可以在这里发送礼物数据
                if (socket) {
                  const data = {
                    type: "like",
                    username: username,
                    // 可以根据需要添加其他礼物相关数据
                  };
                  console.log(data);
                  socket.send(JSON.stringify(data));
                }
              } else if (
                usernameElement &&
                commentElement &&
                !giftCommentElement &&
                !likeElement
              ) {
                const username = usernameElement.textContent.trim().slice(0, -1);
                const content = commentElement.textContent.trim();

                console.log(username + ":" + content);

                // 获取到弹幕数据
                if (username !== "" && content !== "") {
                  const data = {
                    type: "comment",
                    username: username,
                    content: content,
                  };
                  console.log(data);
                  // 如果 socket 已经初始化，可以在这里发送数据
                  if (socket) {
                    socket.send(JSON.stringify(data));
                  }
                }
              }
            }
          });
        }
      });
    });
  }

  // 配置观察选项
  const config = {
    childList: true,
    subtree: true,
  };

  // 开始观察
  observer.observe(targetNode, config);
})();