koishi-plugin-aka-ai-generator 0.5.5 → 0.6.1

package/lib/providers/index.d.ts

@@ -19,6 +19,5 @@ export interface ProviderFactoryConfig {
  */
 export declare function createImageProvider(config: ProviderFactoryConfig): ImageProvider;
 export { ImageProvider } from './types';
-export { YunwuProvider } from './yunwu';
 export { GptGodProvider } from './gptgod';
 export { GeminiProvider } from './gemini';

package/lib/providers/types.d.ts

@@ -14,11 +14,4 @@ export interface ProviderConfig {
     logger: any;
     ctx: any;
 }
-/**
- * 清理对象中的敏感信息（API KEY、密钥等）
- */
-export declare function sanitizeError(error: any): any;
-/**
- * 清理字符串中的 API KEY 模式
- */
-export declare function sanitizeString(str: string): string;
+export { sanitizeError, sanitizeString } from './utils';

package/lib/providers/utils.d.ts

@@ -0,0 +1,17 @@
+import { Context } from 'koishi';
+/**
+ * 清理对象中的敏感信息（API KEY、密钥等）
+ */
+export declare function sanitizeError(error: any): any;
+/**
+ * 清理字符串中的 API KEY 模式
+ */
+export declare function sanitizeString(str: string): string;
+/**
+ * 下载图片并转换为 Base64
+ * 包含 MIME 类型检测和大小限制
+ */
+export declare function downloadImageAsBase64(ctx: Context, url: string, timeout: number, logger: any, maxSize?: number): Promise<{
+    data: string;
+    mimeType: string;
+}>;

package/lib/services/UserManager.d.ts

@@ -0,0 +1,88 @@
+import { Config } from '../index';
+export interface UserData {
+    userId: string;
+    userName: string;
+    totalUsageCount: number;
+    dailyUsageCount: number;
+    lastDailyReset: string;
+    purchasedCount: number;
+    remainingPurchasedCount: number;
+    donationCount: number;
+    donationAmount: number;
+    lastUsed: string;
+    createdAt: string;
+}
+export interface UsersData {
+    [userId: string]: UserData;
+}
+export interface RechargeRecord {
+    id: string;
+    timestamp: string;
+    type: 'single' | 'batch' | 'all';
+    operator: {
+        userId: string;
+        userName: string;
+    };
+    targets: Array<{
+        userId: string;
+        userName: string;
+        amount: number;
+        beforeBalance: number;
+        afterBalance: number;
+    }>;
+    totalAmount: number;
+    note: string;
+    metadata: Record<string, any>;
+}
+export interface RechargeHistory {
+    version: string;
+    lastUpdate: string;
+    records: RechargeRecord[];
+}
+export declare class UserManager {
+    private dataDir;
+    private dataFile;
+    private backupFile;
+    private rechargeHistoryFile;
+    private logger;
+    private dataLock;
+    private historyLock;
+    private usersCache;
+    private activeTasks;
+    private rateLimitMap;
+    private securityBlockMap;
+    private securityWarningMap;
+    constructor(baseDir: string, logger: any);
+    startTask(userId: string): boolean;
+    endTask(userId: string): void;
+    isTaskActive(userId: string): boolean;
+    isAdmin(userId: string, config: Config): boolean;
+    private loadUsersData;
+    private saveUsersDataInternal;
+    getUserData(userId: string, userName: string): Promise<UserData>;
+    getAllUsers(): Promise<UsersData>;
+    updateUsersBatch(updates: (data: UsersData) => void): Promise<void>;
+    loadRechargeHistory(): Promise<RechargeHistory>;
+    addRechargeRecord(record: RechargeRecord): Promise<void>;
+    checkRateLimit(userId: string, config: Config): {
+        allowed: boolean;
+        message?: string;
+    };
+    updateRateLimit(userId: string): void;
+    checkDailyLimit(userId: string, config: Config, numImages?: number): Promise<{
+        allowed: boolean;
+        message?: string;
+        isAdmin?: boolean;
+    }>;
+    consumeQuota(userId: string, userName: string, commandName: string, numImages: number, config: Config): Promise<{
+        userData: UserData;
+        consumptionType: 'free' | 'purchased' | 'mixed';
+        freeUsed: number;
+        purchasedUsed: number;
+    }>;
+    recordSecurityBlock(userId: string, config: Config): Promise<{
+        shouldWarn: boolean;
+        shouldDeduct: boolean;
+        blockCount: number;
+    }>;
+}

package/lib/utils/parser.d.ts

@@ -0,0 +1,20 @@
+import { Argv } from 'koishi';
+import { ModelMappingConfig } from '../index';
+interface StyleCommandModifiers {
+    modelMapping?: ModelMappingConfig;
+    customPromptSuffix?: string;
+    customAdditions?: string[];
+}
+/**
+ * 规范化后缀名称
+ */
+export declare function normalizeSuffix(value?: string): string;
+/**
+ * 构建模型映射索引
+ */
+export declare function buildModelMappingIndex(mappings?: ModelMappingConfig[]): Map<string, ModelMappingConfig>;
+/**
+ * 解析风格命令修饰符
+ */
+export declare function parseStyleCommandModifiers(argv: Argv, imgParam: any, modelMappingIndex: Map<string, ModelMappingConfig>): StyleCommandModifiers;
+export {};

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.5.5",
+  "version": "0.6.1",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [