feat: 增加通用的连接成功失败handler

README.md

@@ -32,7 +32,9 @@ const handler: MsgHandler = {
   },
 }
 
-startListen(652581, handler)
+const instance = startListen(652581, handler)
+
+instance.close()
 ```
 
 ## Handlers & Type Definitions
@@ -88,6 +90,50 @@ export interface User {
 }
 ```
 
+### handler.onOpen
+
+连接成功
+
+```ts
+export type Handler = {
+  /** 连接成功 */
+  onOpen: () => void,
+}
+```
+
+### handler.onClose
+
+连接关闭
+
+```ts
+export type Handler = {
+  /** 连接成功 */
+  onClose: () => void,
+}
+```
+
+### handler.onError
+
+连接错误
+
+```ts
+export type Handler = {
+  /** 连接错误 */
+  onError: (e: Error) => void,
+}
+```
+
+### handler.onStartListen
+
+开始监听消息
+
+```ts
+export type Handler = {
+  /** 开始监听消息 */
+  onStartListen: () => void,
+}
+```
+
 ### handler.onIncomeDanmu
 
 收到普通弹幕消息

src/listener/index.ts

@@ -11,6 +11,16 @@ import type { Message } from '../types/app'
 import type { KeepLiveTCP } from 'bilibili-live-ws'
 
 export type MsgHandler = Partial<
+  {
+    /** 连接成功 */
+    onOpen: () => void,
+    /** 连接关闭 */
+    onClose: () => void,
+    /** 连接错误 */
+    onError: (e: Error) => void,
+    /** 开始监听消息 */
+    onStartListen: () => void,
+  }
   & AttentionChangeMsgHandler
   & DanmuMsgHandler
   & GuardBuyHandler
@@ -36,6 +46,23 @@ const normalizeDanmu = <T>(msgType: string, body: T): Message<T> => {
 export const listenAll = (instance: KeepLiveTCP, roomId: number, handler?: MsgHandler) => {
   if (!handler) return
 
+  // Common
+  if (handler.onOpen) {
+    instance.on('open', () => {
+      handler.onOpen?.()
+    })
+  }
+  if (handler.onClose) {
+    instance.on('close', () => {
+      handler.onClose?.()
+    })
+  }
+  if (handler.onStartListen) {
+    instance.on('live', () => {
+      handler.onStartListen?.()
+    })
+  }
+
   // HEARTBEAT
   if (handler[HEARTBEAT.handlerName]) {
     instance.on(HEARTBEAT.eventName, (data: any) => {