Greasy Fork is available in English.
观察指定 DOM 节点的变化以将数据发送到连接的WebSocket服务端
目前为
// ==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 http://greasyfork.icu/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连接打开");
// 向服务器发送一条消息
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);
})();