NapCatQQ/src/onebot11/action/system/CleanCache.ts

import BaseAction from '../BaseAction';
import { ActionName } from '../types';
import fs from 'fs';
import Path from 'path';
import {
  ChatType,
  ChatCacheListItemBasic,
  CacheFileType
} from '@/core/entities';
import { dbUtil } from '../../../common/db';
import { NTQQFileApi, NTQQFileCacheApi } from '@/core/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));
  });
}