@alemonjs/qq-bot 2.1.31 → 2.1.33

package/lib/upload.d.ts

@@ -0,0 +1,21 @@
+import type { QQBotAPI } from './sdk/api';
+import { FileType } from './sdk/typing';
+export declare const CHUNK_THRESHOLD = 10002432;
+export declare const fileDataToBuffer: (fileData: any) => Buffer | null;
+export declare const computeFileDigests: (filePath: string) => {
+    buffer: NonSharedBuffer;
+    md5: string;
+    sha1: string;
+    md5_10m: string;
+};
+export declare const chunkedUpload: (client: QQBotAPI, scope: "user" | "group", openId: string, file: string | Buffer, options: {
+    file_type: FileType;
+    file_name?: string;
+    srv_send_msg?: boolean;
+}) => Promise<{
+    file_uuid: string;
+    file_info: string;
+    ttl: number;
+    id?: string;
+    raw_url?: string;
+}>;

package/lib/upload.js

@@ -0,0 +1,60 @@
+import { createHash } from 'crypto';
+import { readFileSync } from 'fs';
+
+const CHUNK_THRESHOLD = 10002432;
+const fileDataToBuffer = (fileData) => {
+    if (!fileData) {
+        return null;
+    }
+    if (Buffer.isBuffer(fileData)) {
+        return fileData;
+    }
+    if (typeof fileData === 'string') {
+        if (fileData.indexOf('base64://') === 0) {
+            return Buffer.from(fileData.replace('base64://', ''), 'base64');
+        }
+        return Buffer.from(fileData, 'base64');
+    }
+    return null;
+};
+const computeFileDigests = (filePath) => {
+    const buffer = readFileSync(filePath);
+    const md5 = createHash('md5').update(buffer).digest('hex');
+    const sha1 = createHash('sha1').update(buffer).digest('hex');
+    const md5_10m = createHash('md5').update(buffer.subarray(0, CHUNK_THRESHOLD)).digest('hex');
+    return { buffer, md5, sha1, md5_10m };
+};
+const chunkedUpload = async (client, scope, openId, file, options) => {
+    const buffer = typeof file === 'string' ? readFileSync(file) : file;
+    const md5 = createHash('md5').update(buffer).digest('hex');
+    const sha1 = createHash('sha1').update(buffer).digest('hex');
+    const md5_10m = createHash('md5').update(buffer.subarray(0, CHUNK_THRESHOLD)).digest('hex');
+    const file_name = options.file_name ?? (typeof file === 'string' ? (file.split(/[\\/]/).pop() ?? 'file') : 'file');
+    const prepareData = {
+        file_type: options.file_type,
+        file_name,
+        file_size: String(buffer.byteLength),
+        md5,
+        sha1,
+        md5_10m
+    };
+    const prepare = await (scope === 'user' ? client.usersUploadPrepare(openId, prepareData) : client.groupUploadPrepare(openId, prepareData));
+    const blockSize = Number(prepare.block_size);
+    for (const part of prepare.parts) {
+        const start = part.index * blockSize;
+        const chunk = Buffer.from(buffer.subarray(start, start + Number(part.block_size)));
+        const partMd5 = createHash('md5').update(chunk).digest('hex');
+        await client.uploadPartDirect(part.presigned_url, chunk);
+        const finishData = {
+            upload_id: prepare.upload_id,
+            part_index: part.index,
+            block_size: part.block_size,
+            md5: partMd5
+        };
+        await (scope === 'user' ? client.usersUploadPartFinish(openId, finishData) : client.groupUploadPartFinish(openId, finishData));
+    }
+    const merge = { file_type: options.file_type, upload_id: prepare.upload_id, file_name, srv_send_msg: options.srv_send_msg };
+    return scope === 'user' ? client.postRichMediaByUser(openId, merge) : client.postRichMediaByGroup(openId, merge);
+};
+
+export { CHUNK_THRESHOLD, chunkedUpload, computeFileDigests, fileDataToBuffer };

package/package.json

@@ -1,13 +1,14 @@
 {
   "name": "@alemonjs/qq-bot",
-  "version": "2.1.31",
-  "description": "阿柠檬qq-bot平台连接",
+  "version": "2.1.33",
+  "description": "QQ官方平台",
   "author": "lemonade",
   "license": "MIT",
   "type": "module",
   "main": "lib/index.js",
   "scripts": {
-    "build": "lvy build"
+    "build": "lvy build",
+    "test:ws": "yarn build && node tests/websocket.integration.mjs"
   },
   "dependencies": {
     "@noble/curves": "^1.7.0",
@@ -54,7 +55,7 @@
       }
     ],
     "config-source": {
-      "readme": "https://github.com/lemonade-lab/alemonjs/blob/main/packages/qq-bot/README.md",
+      "readme": "https://raw.githubusercontent.com/lemonade-lab/alemonjs/main/packages/qq-bot/README.md",
       "official": "https://bot.q.qq.com/wiki/develop/api-v2/",
       "platform": "https://q.qq.com/"
     },
@@ -93,5 +94,5 @@
     "type": "git",
     "url": "https://github.com/lemonade-lab/alemonjs.git"
   },
-  "gitHead": "0425a2fbe4d6ca852f19096526219a68221f6bf5"
+  "gitHead": "6c93b0455b8db78ffa1a83becc1c2de19ba544db"
 }