feat: add HEARTBEAT listener

Author: ddiu8081

package.json

@@ -1,6 +1,6 @@
 {
   "name": "blive-message-listener",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Bilibili-live danmu listener with type",
   "type": "module",
   "main": "dist/index.js",

src/listener/index.ts

@@ -1,4 +1,5 @@
 import {
+  HEARTBEAT, type AttentionChangeMsgHandler,
   DANMU_MSG, type DanmuMsgHandler,
   GUARD_BUY, type GuardBuyHandler,
   SEND_GIFT, type GiftHandler,
@@ -9,6 +10,7 @@ import type { Message } from '../types/app'
 import type { KeepLiveTCP } from 'bilibili-live-ws'
 
 export type MsgHandler = Partial<
+  & AttentionChangeMsgHandler
   & DanmuMsgHandler
   & GuardBuyHandler
   & GiftHandler
@@ -31,6 +33,14 @@ const normalizeDanmu = <T>(msgType: string, data: T): Message<T> => {
 export const listenAll = (instance: KeepLiveTCP, handler?: MsgHandler) => {
   if (!handler) return
 
+  // HEARTBEAT
+  if (handler[HEARTBEAT.handlerName]) {
+    instance.on(HEARTBEAT.eventName, (data: any) => {
+      const parsedData = HEARTBEAT.parser(data)
+      handler[HEARTBEAT.handlerName]?.(normalizeDanmu(HEARTBEAT.eventName, parsedData))
+    })
+  }
+
   // DANMU_MSG
   if (handler[DANMU_MSG.handlerName]) {
     instance.on(DANMU_MSG.eventName, (data: any) => {

src/parser/HEARTBEAT.ts

@@ -0,0 +1,22 @@
+import type { Message } from '../types/app'
+
+export interface AttentionChangeMsg {
+  /** 直播间热度·*/
+  attention: number
+}
+
+const parser = (data: any): AttentionChangeMsg => {
+  return {
+    attention: data,
+  }
+}
+
+export const HEARTBEAT = {
+  parser,
+  eventName: 'heartbeat' as const,
+  handlerName: 'onAttentionChange' as const,
+}
+
+export type Handler = {
+  onAttentionChange: (data: Message<AttentionChangeMsg>) => void
+}

src/parser/index.ts

@@ -1,3 +1,4 @@
+export { HEARTBEAT, Handler as AttentionChangeMsgHandler, AttentionChangeMsg } from './HEARTBEAT'
 export { DANMU_MSG, Handler as DanmuMsgHandler, DanmuMsg } from './DANMU_MSG'
 export { GUARD_BUY, Handler as GuardBuyHandler, GuardBuyMsg } from './GUARD_BUY'
 export { SEND_GIFT, Handler as GiftHandler, GiftMsg } from './SEND_GIFT'

src/types/message.ts

@@ -1,4 +1,5 @@
 export type {
+  AttentionChangeMsg,
   DanmuMsg,
   GuardBuyMsg,
   GiftMsg,