NapCatQQ source snapshot

packages/napcat-core/packet/highway/uploader/highwayUploader.ts

@@ -0,0 +1,64 @@
+// import * as tea from '@/napcat-core/packet/utils/crypto/tea';
+import { NapProtoMsg } from 'napcat-protobuf';
+import { PacketHighwayTrans } from '@/napcat-core/packet/highway/client';
+import { PacketLogger } from '@/napcat-core/packet/context/loggerContext';
+import * as proto from '@/napcat-core/packet/transformer/proto';
+
+export abstract class IHighwayUploader {
+  readonly trans: PacketHighwayTrans;
+  readonly logger: PacketLogger;
+
+  constructor (trans: PacketHighwayTrans, logger: PacketLogger) {
+    this.trans = trans;
+    this.logger = logger;
+  }
+
+  // TODO: 没用到的加密方法先注释掉
+  // private encryptTransExt (key: Uint8Array) {
+  //   if (!this.trans.encrypt) return;
+  //   this.trans.ext = tea.encrypt(Buffer.from(this.trans.ext), Buffer.from(key));
+  // }
+
+  protected timeout (): Promise<void> {
+    return new Promise<void>((_resolve, reject) => {
+      setTimeout(() => {
+        reject(new Error(`[Highway] timeout after ${this.trans.timeout}s`));
+      }, (this.trans.timeout ?? Infinity) * 1000
+      );
+    });
+  }
+
+  buildPicUpHead (offset: number, bodyLength: number, bodyMd5: Uint8Array): Uint8Array {
+    return new NapProtoMsg(proto.ReqDataHighwayHead).encode({
+      msgBaseHead: {
+        version: 1,
+        uin: this.trans.uin,
+        command: 'PicUp.DataUp',
+        seq: 0,
+        retryTimes: 0,
+        appId: 1600001604,
+        dataFlag: 16,
+        commandId: this.trans.cmd,
+      },
+      msgSegHead: {
+        serviceId: 0,
+        filesize: BigInt(this.trans.size),
+        dataOffset: BigInt(offset),
+        dataLength: bodyLength,
+        serviceTicket: this.trans.ticket,
+        md5: bodyMd5,
+        fileMd5: this.trans.sum,
+        cacheAddr: 0,
+        cachePort: 0,
+      },
+      bytesReqExtendInfo: this.trans.ext,
+      timestamp: BigInt(0),
+      msgLoginSigHead: {
+        uint32LoginSigType: 8,
+        appId: 1600001604,
+      },
+    });
+  }
+
+  abstract upload (): Promise<void>;
+}