312 lines
11 KiB
JavaScript
312 lines
11 KiB
JavaScript
/**
|
||
* 抖音直播间消息处理脚本
|
||
* 消息类型说明:
|
||
* 1[普通弹幕]|2[点赞消息]|3[进入直播间]|4[关注消息]|5[礼物消息]|6[统计消息]|7[粉丝团消息]|8[直播间分享]|9[下播]
|
||
*
|
||
* 全局统计变量:
|
||
* Live_Total - 本场点赞数
|
||
* Live_user - 本场总观看人数
|
||
* Live_man - 本场男生人数
|
||
* Live_woman - 本场女生人数
|
||
*/
|
||
|
||
const WebSocket = require('ws');
|
||
const chalk = require('chalk');
|
||
const { PackMsgType, Gender, LiveStats } = require('./entities');
|
||
const { MessageParser } = require('./parsers');
|
||
|
||
// 颜色定义
|
||
const COLOR_MAP = {
|
||
[PackMsgType.弹幕消息]: chalk.cyan, // 青色
|
||
[PackMsgType.点赞消息]: chalk.yellow, // 黄色
|
||
[PackMsgType.进直播间]: chalk.green, // 绿色
|
||
[PackMsgType.关注消息]: chalk.magenta, // 洋红色
|
||
[PackMsgType.礼物消息]: chalk.red, // 红色
|
||
[PackMsgType.直播间统计]: chalk.blue, // 蓝色
|
||
[PackMsgType.粉丝团消息]: chalk.blue.bright, // 浅蓝色
|
||
[PackMsgType.直播间分享]: chalk.magenta.bright, // 浅洋红色
|
||
[PackMsgType.下播]: chalk.red, // 红色
|
||
};
|
||
|
||
// 初始化统计对象
|
||
const liveStats = new LiveStats();
|
||
|
||
/**
|
||
* 返回当前时间戳，格式：HH:MM:SS
|
||
* @returns {string} 格式化的时间戳
|
||
*/
|
||
function getTimestamp() {
|
||
const now = new Date();
|
||
return now.toTimeString().split(' ')[0];
|
||
}
|
||
|
||
/**
|
||
* 打印带颜色的消息
|
||
* @param {number} msgType - 消息类型
|
||
* @param {string} content - 消息内容
|
||
* @param {string} [author="未知"] - 消息来源作者
|
||
*/
|
||
function printColoredMessage(msgType, content, author = "未知") {
|
||
const colorFunc = COLOR_MAP[msgType] || chalk.white;
|
||
const timestamp = getTimestamp();
|
||
const authorStr = `[${author}]`; // 使用作者信息
|
||
console.log(colorFunc(`${timestamp} ${authorStr} ${content}`));
|
||
}
|
||
|
||
/**
|
||
* 统一获取消息所属的主播/房间信息
|
||
* @param {Object} msg - 解析后的消息对象
|
||
* @returns {string} 主播昵称或房间ID
|
||
*/
|
||
function getMsgOwner(msg) {
|
||
if (msg && msg.Owner && msg.Owner.Nickname) {
|
||
return msg.Owner.Nickname;
|
||
} else if (msg && msg.WebRoomId) {
|
||
return msg.WebRoomId;
|
||
} else if (msg && msg.RoomId) {
|
||
return msg.RoomId;
|
||
}
|
||
return "未知";
|
||
}
|
||
|
||
/**
|
||
* 处理用户发送的弹幕消息
|
||
* @param {Object} dataDict - 弹幕消息数据
|
||
*/
|
||
function handleDanmaku(dataDict) {
|
||
const msg = MessageParser.parseDanmaku(dataDict);
|
||
const genderStr = msg.User.genderToString();
|
||
const content = `[弹幕消息] [${genderStr}] ${msg.User.Nickname}: ${msg.Content}`;
|
||
const author = getMsgOwner(msg);
|
||
printColoredMessage(PackMsgType.弹幕消息, content, author);
|
||
}
|
||
|
||
/**
|
||
* 处理用户点赞消息并更新全局统计
|
||
* @param {Object} dataDict - 点赞消息数据
|
||
*/
|
||
function handleDianzanku(dataDict) {
|
||
const likeMsg = MessageParser.parseLike(dataDict);
|
||
liveStats.total_likes = likeMsg.Total;
|
||
const genderStr = likeMsg.User.genderToString();
|
||
const content = `[点赞消息] [${genderStr}] ${likeMsg.User.Nickname} 为主播点了${likeMsg.Count}个赞, 总点赞${likeMsg.Total}`;
|
||
const author = getMsgOwner(likeMsg);
|
||
printColoredMessage(PackMsgType.点赞消息, content, author);
|
||
}
|
||
|
||
/**
|
||
* 处理用户进入直播间的消息，并根据性别更新统计
|
||
* @param {Object} dataDict - 进入直播间消息数据
|
||
*/
|
||
function handleUserentry(dataDict) {
|
||
const memberMsg = MessageParser.parseMember(dataDict);
|
||
|
||
// 根据性别更新全局计数
|
||
if (memberMsg.User.Gender === Gender.男) {
|
||
liveStats.male_users += 1;
|
||
} else if (memberMsg.User.Gender === Gender.女) {
|
||
liveStats.female_users += 1;
|
||
}
|
||
|
||
const genderStr = memberMsg.User.genderToString();
|
||
const content = `[进直播间] [${genderStr}] ${memberMsg.User.Nickname} $来了 直播间人数:${memberMsg.CurrentCount}`;
|
||
const author = getMsgOwner(memberMsg);
|
||
printColoredMessage(PackMsgType.进直播间, content, author);
|
||
}
|
||
|
||
/**
|
||
* 处理用户关注主播的消息
|
||
* @param {Object} dataDict - 关注消息数据
|
||
*/
|
||
function handleFollow(dataDict) {
|
||
try {
|
||
const userNickname = dataDict.User?.Nickname || "匿名用户";
|
||
const gender = dataDict.User?.Gender || 0;
|
||
const genderStr = Gender.toString(gender);
|
||
const content = `[关注消息] [${genderStr}] ${userNickname} 关注了主播`;
|
||
const author = getMsgOwner(dataDict);
|
||
printColoredMessage(PackMsgType.关注消息, content, author);
|
||
} catch (e) {
|
||
console.log(`处理关注消息出错: ${e.message}`);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 处理用户赠送礼物的消息
|
||
* @param {Object} dataDict - 礼物消息数据
|
||
*/
|
||
function handleGift(dataDict) {
|
||
try {
|
||
const userNickname = dataDict.User?.Nickname || "匿名用户";
|
||
const gender = dataDict.User?.Gender || 0;
|
||
const genderStr = Gender.toString(gender);
|
||
const giftName = dataDict.GiftName || "未知礼物";
|
||
const giftCount = dataDict.GiftCount || 1;
|
||
const giftValue = dataDict.GiftValue || 0;
|
||
const content = `[礼物消息] [${genderStr}] ${userNickname} 送出了 ${giftCount}个${giftName}, 价值${giftValue}抖币`;
|
||
const author = getMsgOwner(dataDict);
|
||
printColoredMessage(PackMsgType.礼物消息, content, author);
|
||
} catch (e) {
|
||
console.log(`处理礼物消息出错: ${e.message}`);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 处理直播间统计信息
|
||
* @param {Object} dataDict - 统计消息数据
|
||
*/
|
||
function handleStatistics(dataDict) {
|
||
const statsMsg = MessageParser.parseStatistics(dataDict);
|
||
liveStats.total_users = parseInt(statsMsg.TotalUserCount);
|
||
const content = `[直播间统计] 当前直播间人数 ${statsMsg.OnlineUserCountStr}, 累计直播间人数 ${statsMsg.TotalUserCountStr}万`;
|
||
const author = getMsgOwner(statsMsg);
|
||
printColoredMessage(PackMsgType.直播间统计, content, author);
|
||
}
|
||
|
||
/**
|
||
* 处理粉丝团相关消息
|
||
* @param {Object} dataDict - 粉丝团消息数据
|
||
*/
|
||
function handleFansclub(dataDict) {
|
||
try {
|
||
const userNickname = dataDict.User?.Nickname || "匿名用户";
|
||
const gender = dataDict.User?.Gender || 0;
|
||
const genderStr = Gender.toString(gender);
|
||
const clubName = dataDict.FansClubName || "";
|
||
const content = `[粉丝团消息] [${genderStr}] ${userNickname} 加入了${clubName}粉丝团`;
|
||
const author = getMsgOwner(dataDict);
|
||
printColoredMessage(PackMsgType.粉丝团消息, content, author);
|
||
} catch (e) {
|
||
console.log(`处理粉丝团消息出错: ${e.message}`);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 处理用户分享直播间的消息
|
||
* @param {Object} dataDict - 分享消息数据
|
||
*/
|
||
function handleShare(dataDict) {
|
||
try {
|
||
const userNickname = dataDict.User?.Nickname || "匿名用户";
|
||
const gender = dataDict.User?.Gender || 0;
|
||
const genderStr = Gender.toString(gender);
|
||
const content = `[直播间分享] [${genderStr}] ${userNickname} 分享了直播间`;
|
||
const author = getMsgOwner(dataDict);
|
||
printColoredMessage(PackMsgType.直播间分享, content, author);
|
||
} catch (e) {
|
||
console.log(`处理分享消息出错: ${e.message}`);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 处理直播结束事件，显示并重置统计数据
|
||
*/
|
||
function handleLiveExit() {
|
||
const content = `[下播] 直播结束：累计观看 ${liveStats.total_users} 人，累计点赞 ${liveStats.total_likes}，` +
|
||
`男生 ${liveStats.male_users} 女生 ${liveStats.female_users}`;
|
||
printColoredMessage(PackMsgType.下播, content, "系统消息");
|
||
liveStats.reset();
|
||
console.log(`直播统计数据已清空~`);
|
||
}
|
||
|
||
/**
|
||
* 处理从WebSocket接收到的单条消息
|
||
* @param {string} message - WebSocket接收到的消息
|
||
*/
|
||
function processMessage(message) {
|
||
try {
|
||
const data = JSON.parse(message);
|
||
|
||
if (!("Type" in data)) {
|
||
console.log("⚠️ 收到无效消息：缺少Type字段");
|
||
return;
|
||
}
|
||
|
||
const msgType = data.Type;
|
||
|
||
// 检查是否需要处理Data字段
|
||
if (msgType === PackMsgType.下播) {
|
||
handleLiveExit();
|
||
return;
|
||
}
|
||
|
||
if (!("Data" in data)) {
|
||
console.log(`⚠️ 消息类型 ${Object.keys(PackMsgType).find(key => PackMsgType[key] === msgType)} 缺少Data字段`);
|
||
return;
|
||
}
|
||
|
||
try {
|
||
const dataDict = JSON.parse(data.Data);
|
||
|
||
// 消息处理映射表
|
||
const handlers = {
|
||
[PackMsgType.弹幕消息]: handleDanmaku,
|
||
[PackMsgType.点赞消息]: handleDianzanku,
|
||
[PackMsgType.进直播间]: handleUserentry,
|
||
[PackMsgType.关注消息]: handleFollow,
|
||
[PackMsgType.礼物消息]: handleGift,
|
||
[PackMsgType.直播间统计]: handleStatistics,
|
||
[PackMsgType.粉丝团消息]: handleFansclub,
|
||
[PackMsgType.直播间分享]: handleShare,
|
||
};
|
||
|
||
// 分发消息到相应处理器
|
||
if (handlers[msgType]) {
|
||
handlers[msgType](dataDict);
|
||
}
|
||
|
||
} catch (e) {
|
||
if (e instanceof SyntaxError) {
|
||
console.log(`⚠️ 无法解析Data字段内容：${data.Data.substring(0, 100)}...`);
|
||
} else {
|
||
throw e;
|
||
}
|
||
}
|
||
|
||
} catch (e) {
|
||
if (e instanceof SyntaxError) {
|
||
console.log(`⚠️ 无效的JSON格式: ${message.substring(0, 100)}...`);
|
||
} else {
|
||
console.log(`❌ 消息处理错误: ${e.message}`);
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 连接到WebSocket服务器并处理接收到的消息
|
||
*/
|
||
function connectAndPrint() {
|
||
const uri = "ws://127.0.0.1:8888";
|
||
let reconnectInterval = 5000; // 重连间隔，单位毫秒
|
||
|
||
console.log(`正在连接到服务器 ${uri}...`);
|
||
|
||
function connect() {
|
||
const ws = new WebSocket(uri);
|
||
|
||
ws.on('open', function open() {
|
||
console.log(`✅ 已成功连接到服务器 ${uri}`);
|
||
});
|
||
|
||
ws.on('message', function incoming(message) {
|
||
processMessage(message.toString());
|
||
});
|
||
|
||
ws.on('close', function close() {
|
||
console.log(`⚠️ 与服务器的连接已关闭`);
|
||
console.log(`正在尝试重新连接...`);
|
||
setTimeout(connect, reconnectInterval);
|
||
});
|
||
|
||
ws.on('error', function error(err) {
|
||
console.log(`❌ 发生错误: ${err.message}`);
|
||
// 错误处理由 close 事件处理重连
|
||
});
|
||
}
|
||
|
||
connect();
|
||
}
|
||
|
||
// 程序入口
|
||
console.log("启动抖音直播消息监听...");
|
||
connectAndPrint(); |