Merge pull request #354 from LLOneBot/dev

3.29.6

manifest.json

@@ -4,7 +4,7 @@
   "name": "LLOneBot",
   "slug": "LLOneBot",
   "description": "实现 OneBot 11 协议，用于 QQ 机器人开发",
-  "version": "3.29.4",
+  "version": "3.29.6",
   "icon": "./icon.webp",
   "authors": [
     {

package.json

@@ -34,10 +34,10 @@
     "@types/fluent-ffmpeg": "^2.1.25",
     "@types/node": "^20.14.15",
     "@types/ws": "^8.5.12",
-    "electron": "^29.1.4",
+    "electron": "^31.4.0",
     "electron-vite": "^2.3.0",
     "typescript": "^5.5.4",
-    "vite": "^5.4.1",
+    "vite": "^5.4.2",
     "vite-plugin-cp": "^4.0.8"
   },
   "packageManager": "yarn@4.4.0"

scripts/test/test_db.ts

@@ -1,17 +0,0 @@
-import { Level } from 'level'
-
-const db = new Level(process.env['level_db_path'] as string, { valueEncoding: 'json' })
-
-async function getGroupNotify() {
-  let keys = await db.keys().all()
-  let result: string[] = []
-  for (const key of keys) {
-    // console.log(key)
-    if (key.startsWith('group_notify_')) {
-      result.push(key)
-    }
-  }
-  return result
-}
-
-getGroupNotify().then(console.log)

src/common/config.ts

@@ -5,9 +5,7 @@ import path from 'node:path'
 import { getSelfUin } from './data'
 import { DATA_DIR } from './utils'
 
-export const HOOK_LOG = false
-
-export const ALLOW_SEND_TEMP_MSG = false
+//export const HOOK_LOG = false
 
 export class ConfigUtil {
   private readonly configPath: string

src/common/data.ts

@@ -108,11 +108,10 @@ export function getSelfUin() {
 }
 
 const messages: Map<string, RawMessage> = new Map()
-let expire: number
 
 /** 缓存近期消息内容 */
 export async function addMsgCache(msg: RawMessage) {
-  expire ??= getConfigUtil().getConfig().msgCacheExpire! * 1000
+  const expire = getConfigUtil().getConfig().msgCacheExpire! * 1000
   if (expire === 0) {
     return
   }

src/common/server/http.ts

@@ -100,7 +100,7 @@ export abstract class HttpServerBase {
       } else if (req.query) {
         payload = { ...req.query, ...req.body }
       }
-      log('收到http请求', url, payload)
+      log('收到 HTTP 请求', url, payload)
       try {
         res.send(await handler(res, payload))
       } catch (e: any) {

src/common/server/websocket.ts

@@ -1,93 +0,0 @@
-import { WebSocket, WebSocketServer } from 'ws'
-import urlParse from 'url'
-import { IncomingMessage } from 'node:http'
-import { log } from '../utils/log'
-import { getConfigUtil } from '../config'
-import { llonebotError } from '../data'
-
-class WebsocketClientBase {
-  private wsClient: WebSocket | undefined
-
-  constructor() { }
-
-  send(msg: string) {
-    if (this.wsClient && this.wsClient.readyState == WebSocket.OPEN) {
-      this.wsClient.send(msg)
-    }
-  }
-
-  onMessage(msg: string) { }
-}
-
-export class WebsocketServerBase {
-  private ws: WebSocketServer | null = null
-
-  constructor() {
-    console.log(`llonebot websocket service started`)
-  }
-
-  start(port: number) {
-    try {
-      this.ws = new WebSocketServer({ port, maxPayload: 1024 * 1024 * 1024 })
-      llonebotError.wsServerError = ''
-    } catch (e: any) {
-      llonebotError.wsServerError = '正向ws服务启动失败, ' + e.toString()
-    }
-    this.ws?.on('connection', (wsClient, req) => {
-      const url = req.url?.split('?').shift()
-      this.authorize(wsClient, req)
-      this.onConnect(wsClient, url!, req)
-      wsClient.on('message', async (msg) => {
-        this.onMessage(wsClient, url!, msg.toString())
-      })
-    })
-  }
-
-  stop() {
-    llonebotError.wsServerError = ''
-    this.ws?.close((err) => {
-      log('ws server close failed!', err)
-    })
-    this.ws = null
-  }
-
-  restart(port: number) {
-    this.stop()
-    this.start(port)
-  }
-
-  authorize(wsClient: WebSocket, req) {
-    let token = getConfigUtil().getConfig().token
-    const url = req.url.split('?').shift()
-    log('ws connect', url)
-    let clientToken: string = ''
-    const authHeader = req.headers['authorization']
-    if (authHeader) {
-      clientToken = authHeader.split('Bearer ').pop()
-      log('receive ws header token', clientToken)
-    } else {
-      const parsedUrl = urlParse.parse(req.url, true)
-      const urlToken = parsedUrl.query.access_token
-      if (urlToken) {
-        if (Array.isArray(urlToken)) {
-          clientToken = urlToken[0]
-        } else {
-          clientToken = urlToken
-        }
-        log('receive ws url token', clientToken)
-      }
-    }
-    if (token && clientToken != token) {
-      this.authorizeFailed(wsClient)
-      return wsClient.close()
-    }
-  }
-
-  authorizeFailed(wsClient: WebSocket) { }
-
-  onConnect(wsClient: WebSocket, url: string, req: IncomingMessage) { }
-
-  onMessage(wsClient: WebSocket, url: string, msg: string) { }
-
-  sendHeart() { }
-}

src/common/utils/file.ts

@@ -1,9 +1,9 @@
 import fs from 'node:fs'
 import fsPromise from 'node:fs/promises'
 import path from 'node:path'
-import { log, TEMP_DIR } from './index'
-import * as fileType from 'file-type'
+import { TEMP_DIR } from './index'
 import { randomUUID, createHash } from 'node:crypto'
+import { fileURLToPath } from 'node:url'
 
 export function isGIF(path: string) {
   const buffer = Buffer.alloc(4)
@@ -32,31 +32,6 @@ export function checkFileReceived(path: string, timeout: number = 3000): Promise
   })
 }
 
-export async function file2base64(path: string) {
-  let result = {
-    err: '',
-    data: '',
-  }
-  try {
-    // 读取文件内容
-    // if (!fs.existsSync(path)){
-    //     path = path.replace("\\Ori\\", "\\Thumb\\");
-    // }
-    try {
-      await checkFileReceived(path, 5000)
-    } catch (e: any) {
-      result.err = e.toString()
-      return result
-    }
-    const data = await fsPromise.readFile(path)
-    // 转换为Base64编码
-    result.data = data.toString('base64')
-  } catch (err: any) {
-    result.err = err.toString()
-  }
-  return result
-}
-
 export function calculateFileMD5(filePath: string): Promise<string> {
   return new Promise((resolve, reject) => {
     // 创建一个流式读取器
@@ -109,112 +84,118 @@ export async function httpDownload(options: string | HttpDownloadOptions): Promi
   return Buffer.from(await fetchRes.arrayBuffer())
 }
 
+export enum FileUriType {
+  Unknown = 0,
+  FileURL = 1,
+  RemoteURL = 2,
+  OneBotBase64 = 3,
+  DataURL = 4,
+  Path = 5
+}
+
+export function checkUriType(uri: string): { type: FileUriType } {
+  if (uri.startsWith('base64://')) {
+    return { type: FileUriType.OneBotBase64 }
+  }
+  if (uri.startsWith('data:')) {
+    return { type: FileUriType.DataURL }
+  }
+  if (uri.startsWith('http://') || uri.startsWith('https://')) {
+    return { type: FileUriType.RemoteURL }
+  }
+  if (uri.startsWith('file://')) {
+    return { type: FileUriType.FileURL }
+  }
+  try {
+    if (fs.existsSync(uri)) return { type: FileUriType.Path }
+  } catch { }
+  return { type: FileUriType.Unknown }
+}
+
+interface FetchFileRes {
+  data: Buffer
+  url: string
+}
+
+async function fetchFile(url: string): Promise<FetchFileRes> {
+  const headers: Record<string, string> = {
+    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.71 Safari/537.36',
+    'Host': new URL(url).hostname
+  }
+  const raw = await fetch(url, { headers }).catch((err) => {
+    if (err.cause) {
+      throw err.cause
+    }
+    throw err
+  })
+  if (!raw.ok) throw new Error(`statusText: ${raw.statusText}`)
+  return {
+    data: Buffer.from(await raw.arrayBuffer()),
+    url: raw.url
+  }
+}
+
 type Uri2LocalRes = {
   success: boolean
   errMsg: string
   fileName: string
-  ext: string
   path: string
   isLocal: boolean
 }
 
-export async function uri2local(uri: string, fileName: string | null = null): Promise<Uri2LocalRes> {
-  let res = {
-    success: false,
-    errMsg: '',
-    fileName: '',
-    ext: '',
-    path: '',
-    isLocal: false,
-  }
-  if (!fileName) {
-    fileName = randomUUID()
-  }
-  let filePath = path.join(TEMP_DIR, fileName)
-  let url: URL | null = null
-  try {
-    url = new URL(uri)
-  } catch (e: any) {
-    res.errMsg = `uri ${uri} 解析失败,` + e.toString() + ` 可能${uri}不存在`
-    return res
+export async function uri2local(uri: string, filename?: string): Promise<Uri2LocalRes> {
+  const { type } = checkUriType(uri)
+
+  if (type === FileUriType.FileURL) {
+    const filePath = fileURLToPath(uri)
+    const fileName = path.basename(filePath)
+    return { success: true, errMsg: '', fileName, path: filePath, isLocal: true }
   }
 
-  // log("uri protocol", url.protocol, uri);
-  if (url.protocol == 'base64:') {
-    // base64转成文件
-    let base64Data = uri.split('base64://')[1]
+  if (type === FileUriType.Path) {
+    const fileName = path.basename(uri)
+    return { success: true, errMsg: '', fileName, path: uri, isLocal: true }
+  }
+
+  if (type === FileUriType.RemoteURL) {
     try {
-      const buffer = Buffer.from(base64Data, 'base64')
-      await fsPromise.writeFile(filePath, buffer)
-    } catch (e: any) {
-      res.errMsg = `base64文件下载失败,` + e.toString()
-      return res
-    }
-  } else if (url.protocol == 'http:' || url.protocol == 'https:') {
-    // 下载文件
-    let buffer: Buffer | null = null
-    try {
-      buffer = await httpDownload(uri)
-    } catch (e: any) {
-      res.errMsg = `${url}下载失败,` + e.toString()
-      return res
-    }
-    try {
-      const pathInfo = path.parse(decodeURIComponent(url.pathname))
-      if (pathInfo.name) {
-        fileName = pathInfo.name
-        if (pathInfo.ext) {
-          fileName += pathInfo.ext
-          // res.ext = pathInfo.ext
-        }
-      }
-      fileName = fileName.replace(/[/\\:*?"<>|]/g, '_')
-      res.fileName = fileName
-      filePath = path.join(TEMP_DIR, randomUUID() + fileName)
-      await fsPromise.writeFile(filePath, buffer)
-    } catch (e: any) {
-      res.errMsg = `${url}下载失败,` + e.toString()
-      return res
-    }
-  } else {
-    let pathname: string
-    if (url.protocol === 'file:') {
-      // await fs.copyFile(url.pathname, filePath);
-      pathname = decodeURIComponent(url.pathname)
-      if (process.platform === 'win32') {
-        filePath = pathname.slice(1)
+      const res = await fetchFile(uri)
+      const match = res.url.match(/.+\/([^/?]*)(?=\?)?/)
+      if (match?.[1]) {
+        filename ??= match[1].replace(/[/\\:*?"<>|]/g, '_')
       } else {
-        filePath = pathname
+        filename ??= randomUUID()
       }
+      const filePath = path.join(TEMP_DIR, filename)
+      await fsPromise.writeFile(filePath, res.data)
+      return { success: true, errMsg: '', fileName: filename, path: filePath, isLocal: false }
+    } catch (e: any) {
+      const errMsg = `${uri}下载失败,` + e.toString()
+      return { success: false, errMsg, fileName: '', path: '', isLocal: false }
     }
+  }
 
-    res.isLocal = true
+  if (type === FileUriType.OneBotBase64) {
+    filename ??= randomUUID()
+    const filePath = path.join(TEMP_DIR, filename)
+    const base64 = uri.replace(/^base64:\/\//, '')
+    await fsPromise.writeFile(filePath, base64, 'base64')
+    return { success: true, errMsg: '', fileName: filename, path: filePath, isLocal: false }
   }
-  // else{
-  //     res.errMsg = `不支持的file协议,` + url.protocol
-  //     return res
-  // }
-  // if (isGIF(filePath) && !res.isLocal) {
-  //     await fs.rename(filePath, filePath + ".gif");
-  //     filePath += ".gif";
-  // }
-  if (!res.isLocal && !res.ext) {
-    try {
-      const ext = (await fileType.fileTypeFromFile(filePath))?.ext
-      if (ext) {
-        log('获取文件类型', ext, filePath)
-        await fsPromise.rename(filePath, filePath + `.${ext}`)
-        filePath += `.${ext}`
-        res.fileName += `.${ext}`
-        res.ext = ext
-      }
-    } catch (e) {
-      // log("获取文件类型失败", filePath,e.stack)
+
+  if (type === FileUriType.DataURL) {
+    // https://developer.mozilla.org/en-US/docs/Web/HTTP/Basics_of_HTTP/MIME_types/Common_types
+    const capture = /^data:([\w/.+-]+);base64,(.*)$/.exec(uri)
+    if (capture) {
+      filename ??= randomUUID()
+      const [, _type, base64] = capture
+      const filePath = path.join(TEMP_DIR, filename)
+      await fsPromise.writeFile(filePath, base64, 'base64')
+      return { success: true, errMsg: '', fileName: filename, path: filePath, isLocal: false }
     }
   }
-  res.success = true
-  res.path = filePath
-  return res
+
+  return { success: false, errMsg: '未知文件类型', fileName: '', path: '', isLocal: false }
 }
 
 export async function copyFolder(sourcePath: string, destPath: string) {

src/main/main.ts

@@ -149,7 +149,6 @@ function onLoad() {
     const { debug, reportSelfMessage } = getConfigUtil().getConfig()
     for (let message of msgList) {
       // 过滤启动之前的消息
-      // log('收到新消息', message);
       if (parseInt(message.msgTime) < startTime / 1000) {
         continue
       }
@@ -190,13 +189,6 @@ function onLoad() {
           postOb11Event(privateEvent)
         }
       })
-      // OB11Constructor.FriendAddEvent(message).then((friendAddEvent) => {
-      //   log(message)
-      //   if (friendAddEvent) {
-      //     // log("post friend add event", friendAddEvent);
-      //     postOb11Event(friendAddEvent)
-      //   }
-      // })
     }
   }
 
@@ -376,7 +368,7 @@ function onLoad() {
   let startTime = 0 // 毫秒
 
   async function start(uid: string, uin: string) {
-    log('llonebot pid', process.pid)
+    log('process pid', process.pid)
     const config = getConfigUtil().getConfig()
     if (!config.enableLLOB) {
       llonebotError.otherError = 'LLOneBot 未启动'
@@ -391,7 +383,7 @@ function onLoad() {
     NTEventDispatch.init({ ListenerMap: wrapperConstructor, WrapperSession: getSession()! })
     MessageUnique.init(uin)
 
-    log('start activate group member info')
+    //log('start activate group member info')
     // 下面两个会导致CPU占用过高，QQ卡死
     // NTQQGroupApi.activateMemberInfoChange().then().catch(log)
     // NTQQGroupApi.activateMemberListChange().then().catch(log)

src/ntqqapi/hook.ts

@@ -16,7 +16,7 @@ import {
   setSelfInfo
 } from '@/common/data'
 import { postOb11Event } from '../onebot11/server/post-ob11-event'
-import { getConfigUtil, HOOK_LOG } from '@/common/config'
+import { getConfigUtil } from '@/common/config'
 import fs from 'node:fs'
 import { log } from '@/common/utils'
 import { randomUUID } from 'node:crypto'
@@ -80,52 +80,41 @@ let callHooks: Array<{
 export function hookNTQQApiReceive(window: BrowserWindow) {
   const originalSend = window.webContents.send
   const patchSend = (channel: string, ...args: NTQQApiReturnData) => {
-    // console.log("hookNTQQApiReceive", channel, args)
-    let isLogger = false
-    try {
-      isLogger = args[0]?.eventName?.startsWith('ns-LoggerApi')
-    } catch (e) { }
-    if (!isLogger) {
-      try {
-        HOOK_LOG && log(`received ntqq api message: ${channel}`, args)
-      } catch (e) {
-        log('hook log error', e, args)
+    /*try {
+      const isLogger = args[0]?.eventName?.startsWith('ns-LoggerApi')
+      if (!isLogger) {
+        log(`received ntqq api message: ${channel}`, args)
       }
-    }
-    try {
-      if (args?.[1] instanceof Array) {
-        for (let receiveData of args?.[1]) {
-          const ntQQApiMethodName = receiveData.cmdName
-          // log(`received ntqq api message: ${channel} ${ntQQApiMethodName}`, JSON.stringify(receiveData))
-          for (let hook of receiveHooks) {
-            if (hook.method.includes(ntQQApiMethodName)) {
-              new Promise((resolve, reject) => {
-                try {
-                  let _ = hook.hookFunc(receiveData.payload)
-                  if (hook.hookFunc.constructor.name === 'AsyncFunction') {
-                    ; (_ as Promise<void>).then()
-                  }
-                } catch (e: any) {
-                  log('hook error', ntQQApiMethodName, e.stack.toString())
-                }
-              }).then()
-            }
+    } catch { }*/
+    if (args?.[1] instanceof Array) {
+      for (const receiveData of args?.[1]) {
+        const ntQQApiMethodName = receiveData.cmdName
+        // log(`received ntqq api message: ${channel} ${ntQQApiMethodName}`, JSON.stringify(receiveData))
+        for (const hook of receiveHooks) {
+          if (hook.method.includes(ntQQApiMethodName)) {
+            new Promise((resolve, reject) => {
+              try {
+                hook.hookFunc(receiveData.payload)
+              } catch (e: any) {
+                log('hook error', ntQQApiMethodName, e.stack.toString())
+              }
+              resolve(undefined)
+            }).then()
           }
         }
       }
-      if (args[0]?.callbackId) {
-        // log("hookApiCallback", hookApiCallbacks, args)
-        const callbackId = args[0].callbackId
-        if (hookApiCallbacks[callbackId]) {
-          // log("callback found")
-          new Promise((resolve, reject) => {
-            hookApiCallbacks[callbackId](args[1])
-          }).then()
-          delete hookApiCallbacks[callbackId]
-        }
+    }
+    if (args[0]?.callbackId) {
+      // log("hookApiCallback", hookApiCallbacks, args)
+      const callbackId = args[0].callbackId
+      if (hookApiCallbacks[callbackId]) {
+        // log("callback found")
+        new Promise((resolve, reject) => {
+          hookApiCallbacks[callbackId](args[1])
+          resolve(undefined)
+        }).then()
+        delete hookApiCallbacks[callbackId]
       }
-    } catch (e: any) {
-      log('hookNTQQApiReceive error', e.stack.toString(), args)
     }
     originalSend.call(window.webContents, channel, ...args)
   }
@@ -145,9 +134,9 @@ export function hookNTQQApiCall(window: BrowserWindow) {
         isLogger = args[3][0].eventName.startsWith('ns-LoggerApi')
       } catch (e) { }
       if (!isLogger) {
-        try {
+        /*try {
           HOOK_LOG && log('call NTQQ api', thisArg, args)
-        } catch (e) { }
+        } catch (e) { }*/
         try {
           const _args: unknown[] = args[3][1]
           const cmdName: NTQQApiMethod = _args[0] as NTQQApiMethod
@@ -181,16 +170,16 @@ export function hookNTQQApiCall(window: BrowserWindow) {
   const proxyIpcInvoke = new Proxy(ipc_invoke_proxy, {
     apply(target, thisArg, args) {
       // console.log(args);
-      HOOK_LOG && log('call NTQQ invoke api', thisArg, args)
+      //HOOK_LOG && log('call NTQQ invoke api', thisArg, args)
       args[0]['_replyChannel']['sendReply'] = new Proxy(args[0]['_replyChannel']['sendReply'], {
         apply(sendtarget, sendthisArg, sendargs) {
           sendtarget.apply(sendthisArg, sendargs)
         },
       })
       let ret = target.apply(thisArg, args)
-      try {
+      /*try {
         HOOK_LOG && log('call NTQQ invoke api return', ret)
-      } catch (e) { }
+      } catch (e) { }*/
       return ret
     },
   })
@@ -395,7 +384,7 @@ export async function startHook() {
   }>(ReceiveCmdS.FRIENDS, (payload) => {
     // log("onBuddyListChange", payload)
     // let friendListV2: {userSimpleInfos: Map<string, SimpleInfo>} = []
-    type V2data = {userSimpleInfos: Map<string, SimpleInfo>}
+    type V2data = { userSimpleInfos: Map<string, SimpleInfo> }
     let friendList: User[] = [];
     if ((payload as any).userSimpleInfos) {
       // friendListV2 = payload as any
@@ -405,7 +394,7 @@ export async function startHook() {
         }
       })
     }
-    else{
+    else {
       for (const fData of payload.data) {
         friendList.push(...fData.buddyList)
       }

src/ntqqapi/ntcall.ts

@@ -1,7 +1,6 @@
 import { ipcMain } from 'electron'
 import { hookApiCallbacks, ReceiveCmd, ReceiveCmdS, registerReceiveHook, removeReceiveHook } from './hook'
 import { log } from '../common/utils/log'
-import { HOOK_LOG } from '../common/config'
 import { randomUUID } from 'node:crypto'
 
 export enum NTQQApiClass {
@@ -15,6 +14,7 @@ export enum NTQQApiClass {
   SKEY_API = 'ns-SkeyApi',
   GROUP_HOME_WORK = 'ns-GroupHomeWork',
   GROUP_ESSENCE = 'ns-GroupEssence',
+  NODE_STORE_API = 'ns-NodeStoreApi'
 }
 
 export enum NTQQApiMethod {
@@ -129,7 +129,7 @@ export function callNTQQApi<ReturnType>(params: NTQQApiParams) {
   timeout = timeout ?? 5
   afterFirstCmd = afterFirstCmd ?? true
   const uuid = randomUUID()
-  HOOK_LOG && log('callNTQQApi', channel, className, methodName, args, uuid)
+  //HOOK_LOG && log('callNTQQApi', channel, className, methodName, args, uuid)
   return new Promise((resolve: (data: ReturnType) => void, reject) => {
     // log("callNTQQApiPromise", channel, className, methodName, args, uuid)
     const _timeout = timeout * 1000
@@ -203,24 +203,3 @@ export interface GeneralCallResult {
   result: number // 0: success
   errMsg: string
 }
-
-export class NTQQApi {
-  static async call(className: NTQQApiClass, cmdName: string, args: any[]) {
-    return await callNTQQApi<GeneralCallResult>({
-      className,
-      methodName: cmdName,
-      args: [...args],
-    })
-  }
-
-  static async fetchUnitedCommendConfig() {
-    return await callNTQQApi<GeneralCallResult>({
-      methodName: NTQQApiMethod.FETCH_UNITED_COMMEND_CONFIG,
-      args: [
-        {
-          groups: ['100243'],
-        },
-      ],
-    })
-  }
-}

src/onebot11/action/group/GetGroupList.ts

@@ -12,7 +12,7 @@ class GetGroupList extends BaseAction<Payload, OB11Group[]> {
   actionName = ActionName.GetGroupList
 
   protected async _handle(payload: Payload) {
-    const groupList = await NTQQGroupApi.getGroups(payload?.no_cache === true || payload.no_cache === 'true')
+    const groupList = await NTQQGroupApi.getGroups(payload?.no_cache === true || payload?.no_cache === 'true')
     return OB11Constructor.groups(groupList)
   }
 }

src/onebot11/action/msg/SendMsg.ts

@@ -356,7 +356,13 @@ export class SendMsg extends BaseAction<OB11PostSendMsg, ReturnDataType> {
   }
 
   protected async _handle(payload: OB11PostSendMsg) {
-    const peer = await createContext(payload, ContextMode.Normal)
+    let contextMode = ContextMode.Normal
+    if (payload.message_type === 'group') {
+      contextMode = ContextMode.Group
+    } else if (payload.message_type === 'private') {
+      contextMode = ContextMode.Private
+    }
+    const peer = await createContext(payload, contextMode)
     const messages = convertMessage2List(
       payload.message,
       payload.auto_escape === true || payload.auto_escape === 'true',

src/onebot11/server/post-ob11-event.ts

@@ -27,9 +27,10 @@ export function unregisterWsEventSender(ws: WebSocketClass) {
 
 export function postWsEvent(event: PostEventType) {
   for (const ws of eventWSList) {
-    new Promise(() => {
+    new Promise((resolve) => {
       wsReply(ws, event)
-    }).then().catch(log)
+      resolve(undefined)
+    }).then()
   }
 }
 

src/onebot11/server/ws/ReverseWebsocket.ts

@@ -104,6 +104,10 @@ export class ReverseWebsocket {
 
     this.websocket.on('error', log)
 
+    this.websocket.on('ping',()=>{
+      this.websocket?.pong()
+    })
+
     const wsClientInterval = setInterval(() => {
       postWsEvent(new OB11HeartbeatEvent(selfInfo.online!, true, heartInterval!))
     }, heartInterval) // 心跳包

src/onebot11/server/ws/WebsocketServer.ts

@@ -1,70 +1,131 @@
-import { WebSocket } from 'ws'
+import BaseAction from '../../action/BaseAction'
+import { WebSocket, WebSocketServer } from 'ws'
 import { actionMap } from '../../action'
 import { OB11Response } from '../../action/OB11Response'
 import { postWsEvent, registerWsEventSender, unregisterWsEventSender } from '../post-ob11-event'
 import { ActionName } from '../../action/types'
-import BaseAction from '../../action/BaseAction'
 import { LifeCycleSubType, OB11LifeCycleEvent } from '../../event/meta/OB11LifeCycleEvent'
 import { OB11HeartbeatEvent } from '../../event/meta/OB11HeartbeatEvent'
-import { WebsocketServerBase } from '../../../common/server/websocket'
 import { IncomingMessage } from 'node:http'
 import { wsReply } from './reply'
-import { getSelfInfo } from '../../../common/data'
-import { log } from '../../../common/utils/log'
-import { getConfigUtil } from '../../../common/config'
+import { getSelfInfo } from '@/common/data'
+import { log } from '@/common/utils/log'
+import { getConfigUtil } from '@/common/config'
+import { llonebotError } from '@/common/data'
 
-class OB11WebsocketServer extends WebsocketServerBase {
-  authorizeFailed(wsClient: WebSocket) {
-    wsClient.send(JSON.stringify(OB11Response.res(null, 'failed', 1403, 'token验证失败')))
+export class OB11WebsocketServer {
+  private ws?: WebSocketServer
+
+  constructor() {
+    log(`llonebot websocket service started`)
   }
 
-  async handleAction(wsClient: WebSocket, actionName: string, params: any, echo?: any) {
+  start(port: number) {
+    try {
+      this.ws = new WebSocketServer({ port, maxPayload: 1024 * 1024 * 1024 })
+      llonebotError.wsServerError = ''
+    } catch (e: any) {
+      llonebotError.wsServerError = '正向 WebSocket 服务启动失败, ' + e.toString()
+      return
+    }
+    this.ws?.on('connection', (socket, req) => {
+      const url = req.url?.split('?').shift()
+      this.authorize(socket, req)
+      this.onConnect(socket, url!)
+    })
+  }
+
+  stop() {
+    llonebotError.wsServerError = ''
+    this.ws?.close(err => {
+      log('ws server close failed!', err)
+    })
+    this.ws = undefined
+  }
+
+  restart(port: number) {
+    this.stop()
+    this.start(port)
+  }
+
+  private authorize(socket: WebSocket, req: IncomingMessage) {
+    const { token } = getConfigUtil().getConfig()
+    const url = req.url?.split('?').shift()
+    log('ws connect', url)
+    let clientToken = ''
+    const authHeader = req.headers['authorization']
+    if (authHeader) {
+      clientToken = authHeader.split('Bearer ').pop()!
+      log('receive ws header token', clientToken)
+    } else {
+      const { searchParams } = new URL(`http://localhost${req.url}`)
+      const urlToken = searchParams.get('access_token')
+      if (urlToken) {
+        if (Array.isArray(urlToken)) {
+          clientToken = urlToken[0]
+        } else {
+          clientToken = urlToken
+        }
+        log('receive ws url token', clientToken)
+      }
+    }
+    if (token && clientToken !== token) {
+      this.authorizeFailed(socket)
+      return socket.close()
+    }
+  }
+
+  private authorizeFailed(socket: WebSocket) {
+    socket.send(JSON.stringify(OB11Response.res(null, 'failed', 1403, 'token验证失败')))
+  }
+
+  private async handleAction(socket: WebSocket, actionName: string, params: any, echo?: any) {
     const action: BaseAction<any, any> = actionMap.get(actionName)!
     if (!action) {
-      return wsReply(wsClient, OB11Response.error('不支持的api ' + actionName, 1404, echo))
+      return wsReply(socket, OB11Response.error('不支持的api ' + actionName, 1404, echo))
     }
     try {
-      let handleResult = await action.websocketHandle(params, echo)
+      const handleResult = await action.websocketHandle(params, echo)
       handleResult.echo = echo
-      wsReply(wsClient, handleResult)
+      wsReply(socket, handleResult)
     } catch (e: any) {
-      wsReply(wsClient, OB11Response.error(`api处理出错:${e.stack}`, 1200, echo))
+      wsReply(socket, OB11Response.error(`api处理出错:${e.stack}`, 1200, echo))
     }
   }
 
-  onConnect(wsClient: WebSocket, url: string, req: IncomingMessage) {
-    if (url == '/api' || url == '/api/' || url == '/') {
-      wsClient.on('message', async (msg) => {
+  private onConnect(socket: WebSocket, url: string) {
+    if (['/api', '/api/', '/'].includes(url)) {
+      socket.on('message', async (msg) => {
         let receiveData: { action: ActionName | null; params: any; echo?: any } = { action: null, params: {} }
-        let echo = null
+        let echo: any
         try {
           receiveData = JSON.parse(msg.toString())
           echo = receiveData.echo
           log('收到正向Websocket消息', receiveData)
         } catch (e) {
-          return wsReply(wsClient, OB11Response.error('json解析失败，请检查数据格式', 1400, echo))
+          return wsReply(socket, OB11Response.error('json解析失败，请检查数据格式', 1400, echo))
         }
-        this.handleAction(wsClient, receiveData.action!, receiveData.params, receiveData.echo).then()
+        this.handleAction(socket, receiveData.action!, receiveData.params, receiveData.echo)
       })
     }
-    if (url == '/event' || url == '/event/' || url == '/') {
-      registerWsEventSender(wsClient)
+    if (['/event', '/event/', '/'].includes(url)) {
+      registerWsEventSender(socket)
 
       log('event上报ws客户端已连接')
 
       try {
-        wsReply(wsClient, new OB11LifeCycleEvent(LifeCycleSubType.CONNECT))
+        wsReply(socket, new OB11LifeCycleEvent(LifeCycleSubType.CONNECT))
       } catch (e) {
         log('发送生命周期失败', e)
       }
       const { heartInterval } = getConfigUtil().getConfig()
-      const wsClientInterval = setInterval(() => {
+      const intervalId = setInterval(() => {
         postWsEvent(new OB11HeartbeatEvent(getSelfInfo().online!, true, heartInterval!))
       }, heartInterval) // 心跳包
-      wsClient.on('close', () => {
+      socket.on('close', () => {
         log('event上报ws客户端已断开')
-        clearInterval(wsClientInterval)
-        unregisterWsEventSender(wsClient)
+        clearInterval(intervalId)
+        unregisterWsEventSender(socket)
       })
     }
   }

src/version.ts

@@ -1 +1 @@
-export const version = '3.29.4'
+export const version = '3.29.6'