koishi-plugin-aka-ai-generator 0.8.8 → 0.8.9

package/lib/index.d.ts

@@ -33,6 +33,7 @@ export interface Config {
     commandTimeout: number;
     defaultNumImages: number;
     dailyFreeLimit: number;
+    unlimitedPlatforms: string[];
     rateLimitWindow: number;
     rateLimitMax: number;
     adminUsers: string[];

package/lib/index.js

@@ -1636,6 +1636,39 @@ var UserManager = class {
       return { allowed: true, reservationId };
     });
   }
+  // 只记录调用次数，不扣减配额（用于管理员和平台免配额用户）
+  async recordUsageOnly(userId, userName, commandName, numImages) {
+    await this.loadUsersData();
+    return await this.dataLock.acquire(async () => {
+      if (!this.usersCache) {
+        this.logger.error("recordUsageOnly: usersCache 为空，这不应该发生");
+        this.usersCache = {};
+      }
+      let userData = this.usersCache[userId];
+      const now = (/* @__PURE__ */ new Date()).toISOString();
+      if (!userData) {
+        userData = {
+          userId,
+          userName: userName || userId,
+          totalUsageCount: 0,
+          dailyUsageCount: 0,
+          lastDailyReset: now,
+          purchasedCount: 0,
+          remainingPurchasedCount: 0,
+          donationCount: 0,
+          donationAmount: 0,
+          lastUsed: now,
+          createdAt: now
+        };
+        this.usersCache[userId] = userData;
+      }
+      userData.totalUsageCount += numImages;
+      userData.lastUsed = now;
+      await this.saveUsersDataInternal();
+      this.logger.debug("recordUsageOnly: 仅记录调用次数", { userId, userName, commandName, numImages, totalUsageCount: userData.totalUsageCount });
+      return userData;
+    });
+  }
   // 扣减额度并记录使用
   async consumeQuota(userId, userName, commandName, numImages, config) {
     await this.loadUsersData();
@@ -1898,6 +1931,7 @@ var Config = import_koishi2.Schema.intersect([
   // ===== 6. 限流与配额 =====
   import_koishi2.Schema.object({
     dailyFreeLimit: import_koishi2.Schema.number().default(5).min(1).max(100).description("每日免费调用次数"),
+    unlimitedPlatforms: import_koishi2.Schema.array(import_koishi2.Schema.string()).default(["lark"]).description("不受配额限制的平台列表（如 lark, onebot, discord 等）"),
     rateLimitWindow: import_koishi2.Schema.number().default(300).min(60).max(3600).description("限流时间窗口（秒）"),
     rateLimitMax: import_koishi2.Schema.number().default(3).min(1).max(20).description("限流窗口内最大调用次数")
   }).description("🚦 限流与配额"),
@@ -2090,11 +2124,19 @@ function apply(ctx, config) {
     return input || null;
   }
   __name(getPromptInput, "getPromptInput");
-  function buildStatsMessage(userData, numImages, consumptionType, freeUsed, purchasedUsed, config2) {
-    if (userManager.isAdmin(userData.userId, config2)) {
+  function buildStatsMessage(userData, numImages, consumptionType, freeUsed, purchasedUsed, config2, platform) {
+    const isAdmin = userManager.isAdmin(userData.userId, config2);
+    const isPlatformExempt = platform && config2.unlimitedPlatforms?.includes(platform);
+    if (isAdmin) {
       return `📊 使用统计 [管理员]
 用户：${userData.userName}
 总调用次数：${userData.totalUsageCount}次
+状态：无限制使用`;
+    }
+    if (isPlatformExempt) {
+      return `📊 使用统计
+用户：${userData.userName}
+总调用次数：${userData.totalUsageCount}次
 状态：无限制使用`;
     }
     const remainingToday = Math.max(0, config2.dailyFreeLimit - userData.dailyUsageCount);
@@ -2118,8 +2160,23 @@ function apply(ctx, config) {
   async function recordUserUsage(session, commandName, numImages = 1, sendStatsImmediately = true) {
     const userId = session.userId;
     const userName = session.username || session.userId || "未知用户";
+    const platform = session.platform;
     if (!userId) return;
-    const { userData, consumptionType, freeUsed, purchasedUsed } = await userManager.consumeQuota(userId, userName, commandName, numImages, config);
+    const isPlatformExempt = platform && config.unlimitedPlatforms?.includes(platform);
+    const isAdmin = userManager.isAdmin(userId, config);
+    let userData;
+    let consumptionType = "free";
+    let freeUsed = 0;
+    let purchasedUsed = 0;
+    if (isAdmin || isPlatformExempt) {
+      userData = await userManager.recordUsageOnly(userId, userName, commandName, numImages);
+    } else {
+      const result = await userManager.consumeQuota(userId, userName, commandName, numImages, config);
+      userData = result.userData;
+      consumptionType = result.consumptionType;
+      freeUsed = result.freeUsed;
+      purchasedUsed = result.purchasedUsed;
+    }
     logger.info("用户调用记录", {
       userId,
       userName: userData.userName,
@@ -2131,11 +2188,13 @@ function apply(ctx, config) {
       totalUsageCount: userData.totalUsageCount,
       dailyUsageCount: userData.dailyUsageCount,
       remainingPurchasedCount: userData.remainingPurchasedCount,
-      isAdmin: userManager.isAdmin(userId, config)
+      isAdmin,
+      isPlatformExempt,
+      platform
     });
     if (sendStatsImmediately) {
       try {
-        const statsMessage = buildStatsMessage(userData, numImages, consumptionType, freeUsed, purchasedUsed, config);
+        const statsMessage = buildStatsMessage(userData, numImages, consumptionType, freeUsed, purchasedUsed, config, platform);
         await session.send(statsMessage);
       } catch (error) {
         logger.warn("发送统计信息失败", { userId, error: sanitizeError(error) });
@@ -2143,7 +2202,7 @@ function apply(ctx, config) {
     } else {
       setImmediate(async () => {
         try {
-          const statsMessage = buildStatsMessage(userData, numImages, consumptionType, freeUsed, purchasedUsed, config);
+          const statsMessage = buildStatsMessage(userData, numImages, consumptionType, freeUsed, purchasedUsed, config, platform);
           await session.send(statsMessage);
           logger.debug("统计信息已异步发送", { userId, commandName });
         } catch (error) {

package/lib/services/UserManager.d.ts

@@ -130,6 +130,7 @@ export declare class UserManager {
         message?: string;
         reservationId?: string;
     }>;
+    recordUsageOnly(userId: string, userName: string, commandName: string, numImages: number): Promise<UserData>;
     consumeQuota(userId: string, userName: string, commandName: string, numImages: number, config: Config): Promise<{
         userData: UserData;
         consumptionType: 'free' | 'purchased' | 'mixed';

package/package.json

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