koishi-plugin-aka-ai-generator 0.6.13 → 0.6.15

package/lib/index.js

@@ -1056,8 +1056,12 @@ var UserManager = class {
   }
   // 批量更新用户数据 (用于充值)
   async updateUsersBatch(updates) {
+    await this.loadUsersData();
     await this.dataLock.acquire(async () => {
-      await this.loadUsersData();
+      if (!this.usersCache) {
+        this.logger.error("updateUsersBatch: usersCache 为空，这不应该发生");
+        this.usersCache = {};
+      }
       updates(this.usersCache);
       await this.saveUsersDataInternal();
     });
@@ -1158,20 +1162,26 @@ var UserManager = class {
       return { ...rateLimitCheck };
     }
     this.updateRateLimit(userId);
+    await this.loadUsersData();
+    const userData = await this.getUserData(userId, userName);
     return await this.dataLock.acquire(async () => {
-      const userData = await this.getUserData(userId, userName);
+      if (!this.usersCache) {
+        this.logger.error("checkAndReserveQuota: usersCache 为空，这不应该发生");
+        this.usersCache = {};
+      }
+      const cachedUserData = this.usersCache[userId] || userData;
       const today = (/* @__PURE__ */ new Date()).toDateString();
-      const lastReset = new Date(userData.lastDailyReset || userData.createdAt).toDateString();
-      let dailyCount = userData.dailyUsageCount;
+      const lastReset = new Date(cachedUserData.lastDailyReset || cachedUserData.createdAt).toDateString();
+      let dailyCount = cachedUserData.dailyUsageCount;
       if (today !== lastReset) {
         dailyCount = 0;
       }
       const remainingToday = Math.max(0, config.dailyFreeLimit - dailyCount);
-      const totalAvailable = remainingToday + userData.remainingPurchasedCount;
+      const totalAvailable = remainingToday + cachedUserData.remainingPurchasedCount;
       if (totalAvailable < numImages) {
         return {
           allowed: false,
-          message: `生成 ${numImages} 张图片需要 ${numImages} 次可用次数，但您的可用次数不足（今日免费剩余：${remainingToday}次，充值剩余：${userData.remainingPurchasedCount}次，共${totalAvailable}次）`
+          message: `生成 ${numImages} 张图片需要 ${numImages} 次可用次数，但您的可用次数不足（今日免费剩余：${remainingToday}次，充值剩余：${cachedUserData.remainingPurchasedCount}次，共${totalAvailable}次）`
         };
       }
       const reservationId = `${userId}_${Date.now()}_${Math.random()}`;
@@ -1180,8 +1190,12 @@ var UserManager = class {
   }
   // 扣减额度并记录使用
   async consumeQuota(userId, userName, commandName, numImages, config) {
+    await this.loadUsersData();
     return await this.dataLock.acquire(async () => {
-      await this.loadUsersData();
+      if (!this.usersCache) {
+        this.logger.error("consumeQuota: usersCache 为空，这不应该发生");
+        this.usersCache = {};
+      }
       let userData = this.usersCache[userId];
       const now = (/* @__PURE__ */ new Date()).toISOString();
       const today = (/* @__PURE__ */ new Date()).toDateString();

package/package.json

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