NapCatQQ/packages/napcat-common/src/file-uuid.ts

import { Peer } from '@/napcat-core';
import { randomUUID } from 'crypto';

class TimeBasedCache<K, V> {
  private cache = new Map<K, { value: V, timestamp: number, frequency: number }>();
  private keyList = new Set<K>();
  private operationCount = 0;

  constructor (private maxCapacity: number, private ttl: number = 30 * 1000 * 60, private cleanupCount: number = 10) {}

  public put (key: K, value: V): void {
    const timestamp = Date.now();
    const cacheEntry = { value, timestamp, frequency: 1 };
    this.cache.set(key, cacheEntry);
    this.keyList.add(key);
    this.operationCount++;
    if (this.keyList.size > this.maxCapacity) this.evict();
    if (this.operationCount >= this.cleanupCount) this.cleanup(this.cleanupCount);
  }

  public get (key: K): V | undefined {
    const entry = this.cache.get(key);
    if (entry && Date.now() - entry.timestamp < this.ttl) {
      entry.timestamp = Date.now();
      entry.frequency++;
      this.operationCount++;
      if (this.operationCount >= this.cleanupCount) this.cleanup(this.cleanupCount);
      return entry.value;
    } else {
      this.deleteKey(key);
    }
    return undefined;
  }

  private cleanup (count: number): void {
    const currentTime = Date.now();
    let cleaned = 0;
    for (const key of this.keyList) {
      if (cleaned >= count) break;
      const entry = this.cache.get(key);
      if (entry && currentTime - entry.timestamp >= this.ttl) {
        this.deleteKey(key);
        cleaned++;
      }
    }
    this.operationCount = 0; // 重置操作计数器
  }

  private deleteKey (key: K): void {
    this.cache.delete(key);
    this.keyList.delete(key);
  }

  private evict (): void {
    while (this.keyList.size > this.maxCapacity) {
      let oldestKey: K | undefined;
      let minFrequency = Infinity;
      for (const key of this.keyList) {
        const entry = this.cache.get(key);
        if (entry && entry.frequency < minFrequency) {
          minFrequency = entry.frequency;
          oldestKey = key;
        }
      }
      if (oldestKey !== undefined) this.deleteKey(oldestKey);
    }
  }
}

interface FileUUIDData {
  peer: Peer;
  modelId?: string;
  fileId?: string;
  msgId?: string;
  elementId?: string;
  fileUUID?: string;
}

class FileUUIDManager {
  private cache: TimeBasedCache<string, FileUUIDData>;

  constructor (ttl: number) {
    this.cache = new TimeBasedCache<string, FileUUIDData>(5000, ttl);
  }

  public encode (data: FileUUIDData, endString: string = '', customUUID?: string): string {
    const uuid = customUUID || randomUUID().replace(/-/g, '') + endString;
    this.cache.put(uuid, data);
    return uuid;
  }

  public decode (uuid: string): FileUUIDData | undefined {
    return this.cache.get(uuid);
  }
}

export class FileNapCatOneBotUUIDWrap {
  private manager: FileUUIDManager;

  constructor (ttl: number = 86400000) {
    this.manager = new FileUUIDManager(ttl);
  }

  public encodeModelId (peer: Peer, modelId: string, fileId: string, fileUUID: string = '', endString: string = '', customUUID?: string): string {
    return this.manager.encode({ peer, modelId, fileId, fileUUID }, endString, customUUID);
  }

  public decodeModelId (uuid: string): FileUUIDData | undefined {
    return this.manager.decode(uuid);
  }

  public encode (peer: Peer, msgId: string, elementId: string, fileUUID: string = '', customUUID?: string): string {
    return this.manager.encode({ peer, msgId, elementId, fileUUID }, '', customUUID);
  }

  public decode (uuid: string): FileUUIDData | undefined {
    return this.manager.decode(uuid);
  }
}

export const FileNapCatOneBotUUID = new FileNapCatOneBotUUIDWrap();