From 58f544862b5be2f6afbd8c61a80e1cc31260c3d7 Mon Sep 17 00:00:00 2001 From: linyuchen Date: Mon, 29 Apr 2024 11:57:58 +0800 Subject: [PATCH] refactor: private/group image rkey --- src/ntqqapi/api/file.ts | 148 ++++++++++++++-------------------------- 1 file changed, 50 insertions(+), 98 deletions(-) diff --git a/src/ntqqapi/api/file.ts b/src/ntqqapi/api/file.ts index e8763ef..52d15c7 100644 --- a/src/ntqqapi/api/file.ts +++ b/src/ntqqapi/api/file.ts @@ -12,13 +12,15 @@ import { import path from "path"; import fs from "fs"; import {ReceiveCmdS} from "../hook"; -import {log} from "../../common/utils/log"; +import {log} from "../../common/utils"; import http from "http"; import {sleep} from "../../common/utils"; import {hookApi} from "../external/moehook/hook"; let privateImageRKey = ''; let groupImageRKey = ''; +let lastGetRKeyTime = 0; +const rkeyExpireTime = 1000 * 60 * 30; export class NTQQFileApi { static async getFileType(filePath: string) { @@ -139,6 +141,7 @@ export class NTQQFileApi { } static async getImageUrl(msg: RawMessage) { + const isPrivateImage = msg.chatType !== ChatType.group; const msgElement = msg.elements.find(e => !!e.picElement); if (!msgElement) { return ''; @@ -147,118 +150,67 @@ export class NTQQFileApi { const md5HexStr = msgElement.picElement.md5HexStr; const fileMd5 = msgElement.picElement.md5HexStr; const fileUuid = msgElement.picElement.fileUuid; - let imageUrl = ''; - // let currentRKey = "CAQSKAB6JWENi5LMk0kc62l8Pm3Jn1dsLZHyRLAnNmHGoZ3y_gDZPqZt-64" if (url) { if (url.startsWith('/download')) { // console.log('rkey', rkey); if (url.includes('&rkey=')) { - imageUrl = IMAGE_HTTP_HOST_NT + url; - } else { - if (!hookApi.isAvailable()) { - log('hookApi is not available'); - return ''; - } - let isPrivateImage = false; - if (url.indexOf('appid=1406') !== -1) { - // 私聊图片 - isPrivateImage = true; - } - let rkey = ''; - if (isPrivateImage) { - rkey = privateImageRKey; + return IMAGE_HTTP_HOST_NT + url; + } + + if (!hookApi.isAvailable()) { + log('hookApi is not available'); + return ''; + } + + const saveRKey = (rkey: string)=>{ + if(isPrivateImage){ + privateImageRKey = rkey; } else { - rkey = groupImageRKey; + groupImageRKey = rkey; } - rkey = rkey || hookApi.getRKey(); - const saveRKey = ()=>{ - if (isPrivateImage){ - privateImageRKey = rkey; - }else { - groupImageRKey = rkey; - } - }; - const refreshRKey = async () => { - log('正在获取图片rkey...'); - NTQQFileApi.downloadMedia(msg.msgId, msg.chatType, msg.peerUid, msgElement.elementId, '', msgElement.picElement.sourcePath, true).then().catch(() => { - }); - await sleep(300); - const _rkey = hookApi.getRKey(); - if (_rkey) { - rkey = _rkey; - log('图片rkey获取成功', rkey); - } - }; - if (!rkey) { - // 下载一次图片获取rkey - try { - await refreshRKey(); - } catch (e) { - log('获取图片rkey失败', e); - return ''; - } - } - imageUrl = IMAGE_HTTP_HOST_NT + url + `${rkey}`; - // 调用head请求获取图片rkey是否正常 - const checkUrl = new Promise((resolve, reject) => { - const uri = new URL(imageUrl); - const options = { - method: 'GET', - host: uri.host, - path: uri.pathname + uri.search, - headers: { - 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36', - 'Accept': '*/*', - 'Range': 'bytes=0-0' - } - }; - const req = http.request(options, (res) => { - console.log(`Check rkey status: ${res.statusCode}`); - console.log(`Check rkey headers: ${JSON.stringify(res.headers)}`); - if (res.statusCode == 200 || res.statusCode === 206) { - // console.log('The image URL is accessible.'); - resolve('ok'); - } else { - reject('The image URL is not accessible.'); - } - }); + lastGetRKeyTime = Date.now(); + }; - req.setTimeout(3000, () => { - req.destroy(); - reject('Check rkey request timed out'); - }); - - req.on('error', (e) => { - console.error(`Problem with rkey request: ${e.message}`); - reject(e.message); - }); - req.end(); + const refreshRKey = async () => { + log('获取图片rkey...'); + NTQQFileApi.downloadMedia(msg.msgId, msg.chatType, msg.peerUid, msgElement.elementId, '', msgElement.picElement.sourcePath, true).then().catch(() => { }); - try { - const start = Date.now(); - await checkUrl; - const end = Date.now(); - log('Check rkey request time:', end - start); - saveRKey(); - } catch (e) { - try { - await refreshRKey(); - saveRKey(); - imageUrl = IMAGE_HTTP_HOST_NT + url + `${rkey}`; - } catch (e) { - log('获取rkey失败', e); - } + await sleep(1000); + const _rkey = hookApi.getRKey(); + if (_rkey) { + log('图片rkey获取成功', _rkey); + saveRKey(_rkey); + return _rkey; + } + }; + + const existsRKey = isPrivateImage ? privateImageRKey : groupImageRKey; + if ((Date.now() - lastGetRKeyTime > rkeyExpireTime) || !existsRKey) { + // rkey过期 + const newRKey = await refreshRKey(); + if (newRKey){ + return IMAGE_HTTP_HOST_NT + url + `${newRKey}`; + } + else{ + log('图片rkey获取失败', url); + return ''; } } + // 使用未过期的rkey + if (existsRKey){ + return IMAGE_HTTP_HOST_NT + url + `${existsRKey}`; + } } else { - imageUrl = IMAGE_HTTP_HOST + url; + // 老的图片url，不需要rkey + return IMAGE_HTTP_HOST + url; } } else if (fileMd5 || md5HexStr) { - imageUrl = `${IMAGE_HTTP_HOST}/gchatpic_new/0/0-0-${(fileMd5 || md5HexStr)!.toUpperCase()}/0`; + // 没有url，需要自己拼接 + return `${IMAGE_HTTP_HOST}/gchatpic_new/0/0-0-${(fileMd5 || md5HexStr)!.toUpperCase()}/0`; } - - return imageUrl; + log('图片url获取失败', msg); + return ''; } }