NapCatQQ

src/onebot11/action/system/CleanCache.ts

@@ -0,0 +1,105 @@
+import BaseAction from '../BaseAction';
+import { ActionName } from '../types';
+import fs from 'fs';
+import Path from 'path';
+import {
+  ChatType,
+  ChatCacheListItemBasic,
+  CacheFileType
+} from '../../../ntqqapi/types';
+import { dbUtil } from '../../../common/db';
+import { NTQQFileApi, NTQQFileCacheApi } from '@/core/qqnt/apis/file';
+
+export default class CleanCache extends BaseAction<void, void> {
+  actionName = ActionName.CleanCache;
+
+  protected _handle(): Promise<void> {
+    return new Promise<void>(async (res, rej) => {
+      try {
+        // dbUtil.clearCache();
+        const cacheFilePaths: string[] = [];
+
+        await NTQQFileCacheApi.setCacheSilentScan(false);
+
+        cacheFilePaths.push((await NTQQFileCacheApi.getHotUpdateCachePath()));
+        cacheFilePaths.push((await NTQQFileCacheApi.getDesktopTmpPath()));
+        (await NTQQFileCacheApi.getCacheSessionPathList()).forEach(e => cacheFilePaths.push(e.value));
+
+        // await NTQQApi.addCacheScannedPaths(); // XXX: 调用就崩溃，原因目前还未知
+        const cacheScanResult = await NTQQFileCacheApi.scanCache();
+        const cacheSize = parseInt(cacheScanResult.size[6]);
+
+        if (cacheScanResult.result !== 0) {
+          throw('Something went wrong while scanning cache. Code: ' + cacheScanResult.result);
+        }
+
+        await NTQQFileCacheApi.setCacheSilentScan(true);
+        if (cacheSize > 0 && cacheFilePaths.length > 2) { // 存在缓存文件且大小不为 0 时执行清理动作
+          // await NTQQApi.clearCache([ 'tmp', 'hotUpdate', ...cacheScanResult ]) // XXX: 也是调用就崩溃，调用 fs 删除得了
+          deleteCachePath(cacheFilePaths);
+        }
+
+        // 获取聊天记录列表
+        // NOTE: 以防有人不需要删除聊天记录，暂时先注释掉，日后加个开关
+        // const privateChatCache = await getCacheList(ChatType.friend); // 私聊消息
+        // const groupChatCache = await getCacheList(ChatType.group); // 群聊消息
+        // const chatCacheList = [ ...privateChatCache, ...groupChatCache ];
+        const chatCacheList: ChatCacheListItemBasic[] = [];
+
+        // 获取聊天缓存文件列表
+        const cacheFileList: string[] = [];
+                
+        for (const name in CacheFileType) {
+          if (!isNaN(parseInt(name))) continue;
+
+          const fileTypeAny: any = CacheFileType[name];
+          const fileType: CacheFileType = fileTypeAny;
+
+          cacheFileList.push(...(await NTQQFileCacheApi.getFileCacheInfo(fileType)).infos.map(file => file.fileKey));
+        }
+
+        // 一并清除
+        await NTQQFileCacheApi.clearChatCache(chatCacheList, cacheFileList);
+        res();
+      } catch(e) {
+        console.error('清理缓存时发生了错误');
+        rej(e);
+      }
+    });
+  }
+}
+
+function deleteCachePath(pathList: string[]) {
+  const emptyPath = (path: string) => {
+    if (!fs.existsSync(path)) return;
+    const files = fs.readdirSync(path);
+    files.forEach(file => {
+      const filePath = Path.resolve(path, file);
+      const stats = fs.statSync(filePath);
+      if (stats.isDirectory()) emptyPath(filePath);
+      else fs.unlinkSync(filePath);
+    });
+    fs.rmdirSync(path);
+  };
+
+  for (const path of pathList) {
+    emptyPath(path);
+  }
+}
+
+function getCacheList(type: ChatType) { // NOTE: 做这个方法主要是因为目前还不支持针对频道消息的清理
+  return new Promise<Array<ChatCacheListItemBasic>>((res, rej) => {
+    NTQQFileCacheApi.getChatCacheList(type, 1000, 0)
+      .then(data => {
+        const list = data.infos.filter(e => e.chatType === type && parseInt(e.basicChatCacheInfo.chatSize) > 0);
+        const result = list.map(e => {
+          const result = { ...e.basicChatCacheInfo };
+          result.chatType = type;
+          result.isChecked = true;
+          return result;
+        });
+        res(result);
+      })
+      .catch(e => rej(e));
+  });
+}