diff --git "a/Scripts/\347\233\264\346\222\255ws\350\204\232\346\234\254/\347\233\264\346\222\255\345\274\271\345\271\225\347\233\221\345\220\254 \350\275\254\345\217\221\350\207\263\346\234\254\345\234\260WS\346\234\215\345\212\241\347\253\257.js" "b/Scripts/\347\233\264\346\222\255ws\350\204\232\346\234\254/\347\233\264\346\222\255\345\274\271\345\271\225\347\233\221\345\220\254 \350\275\254\345\217\221\350\207\263\346\234\254\345\234\260WS\346\234\215\345\212\241\347\253\257.js" new file mode 100644 index 00000000..60035806 --- /dev/null +++ "b/Scripts/\347\233\264\346\222\255ws\350\204\232\346\234\254/\347\233\264\346\222\255\345\274\271\345\271\225\347\233\221\345\220\254 \350\275\254\345\217\221\350\207\263\346\234\254\345\234\260WS\346\234\215\345\212\241\347\253\257.js" @@ -0,0 +1,206 @@ +// ==UserScript== +// @name 直播弹幕监听 转发至本地WS服务端 +// @namespace http://tampermonkey.net/ +// @version 0.1 +// @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 +// ==/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连接打开"); + + // 向服务器发送一条消息 + socket.send("ws连接成功"); + }); + + // 当收到消息时触发 + 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); +})();