koishi-plugin-best-cave 1.5.3 → 1.5.5

package/lib/utils/AuditHandler.d.ts

@@ -0,0 +1,43 @@
+import { Context } from 'koishi';
+import { Config } from '../index';
+import { IdManager } from './IdManager';
+export interface CaveObject {
+    cave_id: number;
+    elements: Element[];
+    contributor_number: string;
+    contributor_name: string;
+}
+interface BaseElement {
+    type: 'text' | 'img' | 'video';
+    index: number;
+}
+interface TextElement extends BaseElement {
+    type: 'text';
+    content: string;
+}
+interface MediaElement extends BaseElement {
+    type: 'img' | 'video';
+    file?: string;
+    fileName?: string;
+    fileSize?: string;
+    filePath?: string;
+}
+type Element = TextElement | MediaElement;
+export interface PendingCave extends CaveObject {
+}
+export declare class AuditManager {
+    private ctx;
+    private config;
+    private caveDir;
+    private idManager;
+    private logger;
+    constructor(ctx: Context, config: Config, caveDir: string, idManager: IdManager);
+    processAudit(pendingData: PendingCave[], isApprove: boolean, caveFilePath: string, resourceDir: string, pendingFilePath: string, session: any, targetId?: number): Promise<string>;
+    private handleSingleAudit;
+    private handleBatchAudit;
+    sendAuditMessage(cave: PendingCave, content: string, session: any): Promise<void>;
+    private deleteMediaFiles;
+    private cleanElementsForSave;
+    private sendMessage;
+}
+export {};

package/lib/utils/ContentHasher.d.ts

@@ -0,0 +1,80 @@
+import { Buffer } from 'buffer';
+/**
+ * 图片哈希计算工具类
+ * 使用 DCT(离散余弦变换)方法计算图片的感知哈希值，可用于图片相似度比较
+ */
+export declare class ContentHasher {
+    /**
+     * 计算图片的感知哈希值
+     * @param imageBuffer - 图片的二进制数据
+     * @returns 返回64位的十六进制哈希字符串
+     * @throws 当图片处理失败时可能抛出错误
+     */
+    static calculateHash(imageBuffer: Buffer): Promise<string>;
+    /**
+     * 将二进制字符串转换为十六进制
+     * @param binary - 二进制字符串
+     * @returns 十六进制字符串
+     * @private
+     */
+    private static binaryToHex;
+    /**
+     * 将十六进制字符串转换为二进制
+     * @param hex - 十六进制字符串
+     * @returns 二进制字符串
+     * @private
+     */
+    private static hexToBinary;
+    /**
+     * 计算图像的DCT(离散余弦变换)
+     * @param data - 图像数据
+     * @param size - 图像尺寸
+     * @returns DCT变换后的矩阵
+     * @private
+     */
+    private static computeDCT;
+    /**
+     * 获取DCT系数
+     * @param index - 索引值
+     * @param size - 矩阵大小
+     * @returns DCT系数
+     * @private
+     */
+    private static getDCTCoefficient;
+    /**
+     * 计算数组的中位数
+     * @param arr - 输入数组
+     * @returns 中位数
+     * @private
+     */
+    private static calculateMedian;
+    /**
+     * 从DCT矩阵中提取特征值
+     * @param matrix - DCT矩阵
+     * @param size - 矩阵大小
+     * @returns 特征值数组
+     * @private
+     */
+    private static extractFeatures;
+    /**
+     * 计算两个哈希值之间的汉明距离
+     * @param hash1 - 第一个哈希值
+     * @param hash2 - 第二个哈希值
+     * @returns 汉明距离
+     * @throws 当两个哈希值长度不等时抛出错误
+     */
+    static calculateDistance(hash1: string, hash2: string): number;
+    /**
+     * 计算两个图片哈希值的相似度
+     * @param hash1 - 第一个哈希值
+     * @param hash2 - 第二个哈希值
+     * @returns 返回0-1之间的相似度值，1表示完全相同，0表示完全不同
+     */
+    static calculateSimilarity(hash1: string, hash2: string): number;
+    /**
+     * 计算文本的哈希值
+     * @param text - 输入文本
+     * @returns 文本的哈希值(36进制字符串)
+     */
+    static calculateTextHash(text: string): string;
+}

package/lib/utils/FileHandler.d.ts

@@ -0,0 +1,63 @@
+export declare class FileHandler {
+    private static locks;
+    private static readonly RETRY_COUNT;
+    private static readonly RETRY_DELAY;
+    private static readonly CONCURRENCY_LIMIT;
+    /**
+     * 并发控制
+     * @param operation 要执行的操作
+     * @param limit 并发限制
+     * @returns 操作结果
+     */
+    private static withConcurrencyLimit;
+    /**
+     * 文件操作包装器
+     * @param filePath 文件路径
+     * @param operation 要执行的操作
+     * @returns 操作结果
+     */
+    private static withFileOp;
+    /**
+     * 事务处理
+     * @param operations 要执行的操作数组
+     * @returns 操作结果数组
+     */
+    static withTransaction<T>(operations: Array<{
+        filePath: string;
+        operation: () => Promise<T>;
+        rollback?: () => Promise<void>;
+    }>): Promise<T[]>;
+    /**
+     * 读取 JSON 数据
+     * @param filePath 文件路径
+     * @returns JSON 数据
+     */
+    static readJsonData<T>(filePath: string): Promise<T[]>;
+    /**
+     * 写入 JSON 数据
+     * @param filePath 文件路径
+     * @param data 要写入的数据
+     */
+    static writeJsonData<T>(filePath: string, data: T[]): Promise<void>;
+    /**
+     * 确保目录存在
+     * @param dir 目录路径
+     */
+    static ensureDirectory(dir: string): Promise<void>;
+    /**
+     * 确保 JSON 文件存在
+     * @param filePath 文件路径
+     */
+    static ensureJsonFile(filePath: string): Promise<void>;
+    /**
+     * 保存媒体文件
+     * @param filePath 文件路径
+     * @param data 文件数据
+     */
+    static saveMediaFile(filePath: string, data: Buffer | string): Promise<void>;
+    /**
+     * 删除媒体文件
+     * @param filePath 文件路径
+     */
+    static deleteMediaFile(filePath: string): Promise<void>;
+}

package/lib/utils/HashManager.d.ts

@@ -0,0 +1,108 @@
+/**
+ * 哈希存储状态类型
+ */
+interface HashStorageStatus {
+    lastUpdated: string;
+    entries: Array<{
+        caveId: number;
+        imageHashes: string[];
+        textHashes: string[];
+    }>;
+}
+/**
+ * 图片哈希值存储管理类
+ * 负责管理和维护回声洞图片的哈希值
+ */
+export declare class HashManager {
+    private readonly caveDir;
+    private static readonly HASH_FILE;
+    private static readonly CAVE_FILE;
+    private static readonly BATCH_SIZE;
+    private imageHashes;
+    private textHashes;
+    private initialized;
+    /**
+     * 初始化HashManager实例
+     * @param caveDir 回声洞数据目录路径
+     */
+    constructor(caveDir: string);
+    private get filePath();
+    private get resourceDir();
+    private get caveFilePath();
+    /**
+     * 初始化哈希存储
+     * 读取现有哈希数据或重新构建哈希值
+     * @throws 初始化失败时抛出错误
+     */
+    initialize(): Promise<void>;
+    /**
+     * 获取当前哈希存储状态
+     * @returns 包含最后更新时间和所有条目的状态对象
+     */
+    getStatus(): Promise<HashStorageStatus>;
+    /**
+     * 更新指定回声洞的图片哈希值
+     * @param caveId 回声洞ID
+     * @param content 图片buffer数组
+     */
+    updateCaveContent(caveId: number, content: {
+        images?: Buffer[];
+        texts?: string[];
+    }): Promise<void>;
+    /**
+     * 更新所有回声洞的哈希值
+     * @param isInitialBuild 是否为初始构建
+     */
+    updateAllCaves(isInitialBuild?: boolean): Promise<void>;
+    /**
+     * 查找重复的图片
+     * @param content 待查找的图片buffer数组
+     * @param thresholds 相似度阈值
+     * @returns 匹配结果数组，包含索引、回声洞ID和相似度
+     */
+    findDuplicates(content: {
+        images?: Buffer[];
+        texts?: string[];
+    }, thresholds: {
+        image: number;
+        text: number;
+    }): Promise<Array<{
+        type: 'image' | 'text';
+        index: number;
+        caveId: number;
+        similarity: number;
+    } | null>>;
+    private findTextDuplicates;
+    private calculateTextSimilarity;
+    private findImageDuplicates;
+    /**
+     * 加载回声洞数据
+     * @returns 回声洞数据数组
+     * @private
+     */
+    private loadCaveData;
+    /**
+     * 保存哈希数据到文件
+     * @private
+     */
+    private saveContentHashes;
+    /**
+     * 构建初始哈希数据
+     * @private
+     */
+    private buildInitialHashes;
+    /**
+     * 更新缺失的哈希值
+     * @private
+     */
+    private updateMissingHashes;
+    /**
+     * 批量处理数组项
+     * @param items 待处理项数组
+     * @param processor 处理函数
+     * @param batchSize 批处理大小
+     * @private
+     */
+    private processBatch;
+}
+export {};

package/lib/utils/IdManager.d.ts

@@ -0,0 +1,69 @@
+/**
+ * ID管理器类
+ * 负责管理回声洞ID的分配、删除和统计信息
+ */
+export declare class IdManager {
+    private deletedIds;
+    private maxId;
+    private initialized;
+    private readonly statusFilePath;
+    private stats;
+    private usedIds;
+    /**
+     * 初始化ID管理器
+     * @param baseDir - 基础目录路径
+     */
+    constructor(baseDir: string);
+    /**
+     * 初始化ID管理系统
+     * @param caveFilePath - 正式回声洞数据文件路径
+     * @param pendingFilePath - 待处理回声洞数据文件路径
+     * @throws 当初始化失败时抛出错误
+     */
+    initialize(caveFilePath: string, pendingFilePath: string): Promise<void>;
+    /**
+     * 处理ID冲突
+     * @param conflicts - ID冲突映射表
+     * @param caveFilePath - 正式回声洞数据文件路径
+     * @param pendingFilePath - 待处理回声洞数据文件路径
+     * @param caveData - 正式回声洞数据
+     * @param pendingData - 待处理回声洞数据
+     * @private
+     */
+    private handleConflicts;
+    /**
+     * 获取下一个可用的ID
+     * @returns 下一个可用的ID
+     * @throws 当ID管理器未初始化时抛出错误
+     */
+    getNextId(): number;
+    /**
+     * 标记ID为已删除状态
+     * @param id - 要标记为删除的ID
+     * @throws 当ID管理器未初始化时抛出错误
+     */
+    markDeleted(id: number): Promise<void>;
+    /**
+     * 添加贡献统计
+     * @param contributorNumber - 贡献者编号
+     * @param caveId - 回声洞ID
+     */
+    addStat(contributorNumber: string, caveId: number): Promise<void>;
+    /**
+     * 移除贡献统计
+     * @param contributorNumber - 贡献者编号
+     * @param caveId - 回声洞ID
+     */
+    removeStat(contributorNumber: string, caveId: number): Promise<void>;
+    /**
+     * 获取所有贡献统计信息
+     * @returns 贡献者编号到回声洞ID列表的映射
+     */
+    getStats(): Record<string, number[]>;
+    /**
+     * 保存当前状态到文件
+     * @private
+     * @throws 当保存失败时抛出错误
+     */
+    private saveStatus;
+}

package/lib/utils/MediaHandle.d.ts

@@ -0,0 +1,42 @@
+import { Context } from 'koishi';
+export interface BaseElement {
+    type: 'text' | 'img' | 'video';
+    index: number;
+}
+export interface TextElement extends BaseElement {
+    type: 'text';
+    content: string;
+}
+export interface MediaElement extends BaseElement {
+    type: 'img' | 'video';
+    file?: string;
+    fileName?: string;
+    fileSize?: string;
+    filePath?: string;
+}
+export type Element = TextElement | MediaElement;
+export declare function extractMediaContent(originalContent: string, config: {
+    imageMaxSize: number;
+    videoMaxSize: number;
+}, session: any): Promise<{
+    imageUrls: string[];
+    imageElements: Array<{
+        type: 'img';
+        index: number;
+        fileName?: string;
+        fileSize?: string;
+    }>;
+    videoUrls: string[];
+    videoElements: Array<{
+        type: 'video';
+        index: number;
+        fileName?: string;
+        fileSize?: string;
+    }>;
+    textParts: Element[];
+}>;
+export declare function saveMedia(urls: string[], fileNames: (string | undefined)[], resourceDir: string, caveId: number, mediaType: 'img' | 'video', config: {
+    enableImageDuplicate: boolean;
+    imageDuplicateThreshold: number;
+    textDuplicateThreshold: number;
+}, ctx: Context, session: any, buffers?: Buffer[]): Promise<string[]>;

package/lib/utils/MediaHandler.d.ts

@@ -0,0 +1,52 @@
+import { Context } from 'koishi';
+export interface BaseElement {
+    type: 'text' | 'img' | 'video';
+    index: number;
+}
+interface CaveObject {
+    cave_id: number;
+    elements: Element[];
+    contributor_number: string;
+    contributor_name: string;
+}
+export interface TextElement extends BaseElement {
+    type: 'text';
+    content: string;
+}
+export interface MediaElement extends BaseElement {
+    type: 'img' | 'video';
+    file?: string;
+    fileName?: string;
+    fileSize?: string;
+    filePath?: string;
+}
+export type Element = TextElement | MediaElement;
+export declare function buildMessage(cave: CaveObject, resourceDir: string, session?: any): Promise<string>;
+export declare function sendMessage(session: any, key: string, params?: any[], isTemp?: boolean, timeout?: number): Promise<string>;
+export declare function processMediaFile(filePath: string, type: 'image' | 'video'): Promise<string | null>;
+export declare function extractMediaContent(originalContent: string, config: {
+    imageMaxSize: number;
+    videoMaxSize: number;
+}, session: any): Promise<{
+    imageUrls: string[];
+    imageElements: Array<{
+        type: 'img';
+        index: number;
+        fileName?: string;
+        fileSize?: string;
+    }>;
+    videoUrls: string[];
+    videoElements: Array<{
+        type: 'video';
+        index: number;
+        fileName?: string;
+        fileSize?: string;
+    }>;
+    textParts: Element[];
+}>;
+export declare function saveMedia(urls: string[], fileNames: (string | undefined)[], resourceDir: string, caveId: number, mediaType: 'img' | 'video', config: {
+    enableImageDuplicate: boolean;
+    imageDuplicateThreshold: number;
+    textDuplicateThreshold: number;
+}, ctx: Context, session: any, buffers?: Buffer[]): Promise<string[]>;
+export {};

package/lib/utils/ProcessHandle.d.ts

@@ -0,0 +1,7 @@
+import { Config } from '../index';
+import { IdManager } from './IdManager';
+import { HashManager } from './HashManager';
+export declare function processList(session: any, config: Config, idManager: IdManager, userId?: string, pageNum?: number): Promise<string>;
+export declare function processView(caveFilePath: string, resourceDir: string, session: any, options: any, content: string[]): Promise<string>;
+export declare function processRandom(caveFilePath: string, resourceDir: string, session: any): Promise<string | void>;
+export declare function processDelete(caveFilePath: string, resourceDir: string, pendingFilePath: string, session: any, config: Config, options: any, content: string[], idManager: IdManager, HashManager: HashManager): Promise<string>;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-best-cave",
   "description": "最好的 cave 插件，可开关的审核系统，可引用添加，支持图文混合内容，可查阅投稿列表，完美复刻你的 .cave 体验！",
-  "version": "1.5.3",
+  "version": "1.5.5",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],