NapCatQQ source snapshot

packages/napcat-core/apis/packet.ts

@@ -0,0 +1,93 @@
+import * as os from 'os';
+import offset from '@/napcat-core/external/napi2native.json';
+import { InstanceContext, NapCatCore } from '@/napcat-core/index';
+import { PacketClientSession } from '@/napcat-core/packet/clientSession';
+import { napCatVersion } from 'napcat-common/src/version';
+import { LogWrapper } from '../helper/log';
+
+interface OffsetType {
+  [key: string]: {
+    recv: string;
+    send: string;
+  };
+}
+
+const typedOffset: OffsetType = offset;
+
+export class NTQQPacketApi {
+  context: InstanceContext;
+  core: NapCatCore;
+  logger: LogWrapper;
+  qqVersion: string | undefined;
+  pkt!: PacketClientSession;
+  errStack: string[] = [];
+  packetStatus: boolean = false;
+
+  constructor (context: InstanceContext, core: NapCatCore) {
+    this.context = context;
+    this.core = core;
+    this.logger = core.context.logger;
+  }
+
+  async initApi () {
+    this.packetStatus = (await this.InitSendPacket(this.context.basicInfoWrapper.getFullQQVersion())
+      .then((result) => {
+        return result;
+      })
+      .catch((err) => {
+        this.logger.logError(err);
+        this.pushError(err);
+        return false;
+      })) && this.pkt?.available;
+  }
+
+  get available (): boolean {
+    return this.pkt?.available ?? false;
+  }
+
+  get clientLogStack () {
+    return [this.pkt?.clientLogStack, ...this.errStack].filter(Boolean).join('\n');
+  }
+
+  private pushError (error: unknown) {
+    if (error instanceof Error) {
+      this.errStack.push(error.stack || error.message);
+      return;
+    }
+    this.errStack.push(String(error));
+  }
+
+  async InitSendPacket (qqVer: string) {
+    this.qqVersion = qqVer;
+    const table = typedOffset[qqVer + '-' + os.arch()];
+    if (!table) {
+      const err = `[Core] [Packet] PacketBackend 不支持当前QQ版本架构：${qqVer}-${os.arch()}，
+            请参照 https://github.com/NapNeko/NapCatQQ/releases/tag/v${napCatVersion} 配置正确的QQ版本！`;
+      this.logger.logError(err);
+      this.errStack.push(err);
+      return false;
+    }
+    if (this.core.configLoader.configData.packetBackend === 'disable') {
+      const err = '[Core] [Packet] 已禁用PacketBackend，NapCat.Packet将不会加载！';
+      this.logger.logError(err);
+      this.errStack.push(err);
+      return false;
+    }
+    this.pkt = new PacketClientSession(this.core);
+    await this.pkt.init(process.pid, table.recv, table.send);
+    if (!this.pkt.available) {
+      const err = '[Core] [Packet] NativePacketClient 初始化失败，PacketBackend 发包能力不可用！';
+      this.logger.logError(err);
+      this.errStack.push(err);
+      return false;
+    }
+    try {
+      await this.pkt.operation.FetchRkey(3000);
+    } catch (error) {
+      this.logger.logError('测试Packet状态异常', error);
+      this.pushError(error);
+      return false;
+    }
+    return true;
+  }
+}