// ==UserScript==
// @name 虎牙弹幕统计机
// @namespace http://tampermonkey.net/
// @version 2024-05-26
// @description 1.虎牙实时统计高频弹幕，2.礼物、人进入、弹幕数据持久化到IndexDB
// @author 黎曼
// @match https://www.huya.com/*
// @icon https://www.google.com/s2/favicons?sz=64&domain=huya.com
// @grant GM_log
// @license MIT
// ==/UserScript==
(function () {
'use strict';
// Your code here...
var room_name;
// 等待页面加载完成
window.addEventListener('load', function() {
room_name = document.querySelector('.host-name').textContent.trim();
}, false);
var IDBStore;
var total = 0;
// 创建一个浮动的 div 元素用于显示统计结果
const statsDiv = document.createElement('div');
statsDiv.style.position = 'fixed';
statsDiv.style.top = '50px';
statsDiv.style.right = '10px';
statsDiv.style.width = '300px';
statsDiv.style.height = '500px';
statsDiv.style.backgroundColor = 'rgba(0, 0, 0, 0.8)';
statsDiv.style.color = 'white';
statsDiv.style.padding = '10px';
statsDiv.style.overflowY = 'auto';
statsDiv.style.zIndex = '10000';
statsDiv.style.font_size = '36px';
document.body.appendChild(statsDiv);
// 创建一个对象用于记录弹幕频率
var barrageFrequency = {};
const barrageContainer = document.querySelector('.chat-room__list');
// 观察弹幕容器的变化
const observer = new MutationObserver(mutations => {
mutations.forEach(mutation => {
if (mutation.addedNodes.length > 0) {
mutation.addedNodes.forEach(node => {
if (node.nodeType === Node.ELEMENT_NODE) {
room_name = document.querySelector('.host-name').textContent.trim();
const barrageHtml = node.querySelector('.msg-normal');
if (barrageHtml) {
deal_barrage(barrageHtml);
deal_barrage_frequency(barrageHtml);
}
const peopleInHtml = node.querySelector('[class^="msg--"]');
if (peopleInHtml) {
deal_people(peopleInHtml);
}
const giftHtml = node.querySelector('.tit-h-send');
if (giftHtml) {
deal_gift(giftHtml);
}
}
});
}
});
});
function deal_gift(giftHtml) {
const username = giftHtml.querySelector('.cont-item.name.J_userMenu').textContent.trim();
const giftname =giftHtml.querySelector('.cont-item.send-gift img').getAttribute('alt');
// 使用正则表达式提取数字
const num = giftHtml.querySelector('.send-gift').nextElementSibling.textContent.trim();
const match = num.match(/\d+/);
const num_n = match[0];
store_gift(username, giftname, num_n);
}
function deal_people(peopleInHtml) {
const name = peopleInHtml.querySelector('[class^="name--"]').textContent.trim();
store_people(name);
}
function deal_barrage(barrageHtml) {
// 1. 获取弹幕文本类
const name = barrageHtml.querySelector('.name.J_userMenu').textContent.trim();;
const msg = barrageHtml.querySelector('.msg').textContent.trim();;
store_barrage(name, msg);
if (msg in barrageFrequency) {
barrageFrequency[msg] += 1;
} else {
barrageFrequency[msg] = 1;
}
}
function deal_barrage_frequency(barrageHtml) {
const msg = barrageHtml.querySelector('.msg').textContent.trim();;
if (msg in barrageFrequency) {
barrageFrequency[msg] += 1;
} else {
barrageFrequency[msg] = 1;
}
}
observer.observe(barrageContainer, { childList: true, subtree: true });
// 每隔一定时间（例如5秒）输出弹幕频率统计结果
const startTime = new Date();
const timeString = startTime.toLocaleTimeString();
const interval = 5000;
setInterval(() => {
statsDiv.innerHTML = `<h1 style="color: red;">弹幕频率统计, ${interval/1000}秒一次</h1>`;
statsDiv.innerHTML += `<p style="color: green;"> start ${startTime.toLocaleTimeString()}</p>`;
statsDiv.innerHTML += `<p style="color: green;"> end ${new Date().toLocaleTimeString()}</p>`;
total += Object.keys(barrageFrequency).length;
statsDiv.innerHTML += `<p style="color: yellow;"> total ${total}`;
const sortedBarrage = Object.entries(barrageFrequency).sort((a, b) => b[1] - a[1]).slice(0, 20);
sortedBarrage.forEach(([text, count]) => {
const p = document.createElement('p');
p.textContent = `${text}: ${count}`;
statsDiv.appendChild(p);
});
barrageFrequency = {};
}, interval);
setInterval(() => {
console.clear();
}, 500000);
store_init();
IDBStore('gift_price', { gift_name: "虎粮", price: 1});
function store_gift(username, giftname, number){
let gift = {room_name:room_name, username: username, giftname: giftname, number:number, money: number * 1, time: new Date().toLocaleString()};
IDBStore("gift", gift);
}
function store_barrage(name, msg) {
//console.log(`store barrage ${name} : ${msg}`);
let barrage = {room_name:room_name,name: name, message: msg, time: new Date().toLocaleString()};
IDBStore("barrage", barrage);
}
function store_people(name){
let people = {room_name:room_name,name: name, time: new Date().toLocaleString()};
IDBStore("people", people);
}
function store_init() {
// 在 Tampermonkey 脚本中使用 IndexedDB
console.log("store init.");
// 打开或者创建一个名为 myDatabase 的数据库
var request = indexedDB.open('myDatabase', 1);
// 当数据库需要升级时（包括首次创建数据库），触发这个事件
request.onupgradeneeded = function (event) {
var db = event.target.result;
// 创建一个名为 barrage 的对象存储，主键为自动递增的 id
if (!db.objectStoreNames.contains('barrage')) {
db.createObjectStore('barrage', { keyPath: 'id', autoIncrement: true });
}
// 创建一个名为 gift 的对象存储，主键为自动递增的 id
if (!db.objectStoreNames.contains('gift')) {
db.createObjectStore('gift', { keyPath: 'id', autoIncrement: true });
}
// 创建一个名为 people 的对象存储，主键为自动递增的 id
if (!db.objectStoreNames.contains('people')) {
db.createObjectStore('people', { keyPath: 'id', autoIncrement: true });
}
if (!db.objectStoreNames.contains('gift_price')) {
db.createObjectStore('gift_price', { keyPath: 'gift_name'});
}
console.log('对象存储 barrage gift people创建成功');
};
// 当数据库打开成功时的回调函数
request.onsuccess = function (event) {
// 获取数据库对象
var db = event.target.result;
IDBStore = function (storeName, data) {
operateOnStore(db, storeName, data);
}
};
// 当数据库打开失败时的回调函数
request.onerror = function (event) {
console.error('数据库打开失败');
};
}
function getGiftPrice(giftname) {
//console.log("price",giftname);
return new Promise((resolve, reject) => {
IDBStore("gift_price").get(giftname).then(data => {
if (data) {
resolve(data[0].price);
} else {
reject("Gift price not found");
}
}).catch(error => {
console.log(error);
reject(error);
});
});
}
function operateOnStore(db, storeName, data) {
const transaction = db.transaction([storeName], 'readwrite');
const objectStore = transaction.objectStore(storeName);
// 如果传入了 data 参数，则向对象存储空间写入数据
if (data) {
objectStore.add(data); // 添加数据到对象存储空间
//console.log('已写入数据:', data);
} else { // 否则从对象存储空间读取数据
const getRequest = objectStore.getAll();
getRequest.onsuccess = function(event) {
const items = event.target.result;
console.log('已读取数据:', items);
};
}
}
})();
