koishi-plugin-aka-ai-generator 0.4.4 → 0.5.0

package/lib/index.d.ts

@@ -25,6 +25,8 @@ export interface UserData {
     donationAmount: number;
     lastUsed: string;
     createdAt: string;
+    safetyBlockCount: number;
+    safetyBlockHistory: string[];
 }
 export interface UsersData {
     [userId: string]: UserData;

package/lib/index.js

@@ -1207,6 +1207,46 @@ function apply(ctx, config) {
     return { allowed: true, isAdmin: false };
   }
   __name(checkDailyLimit, "checkDailyLimit");
+  function isSafetyBlockError(error) {
+    if (!error) return false;
+    const errorMessage = typeof error === "string" ? error : error?.message || "";
+    if (!errorMessage) return false;
+    const lowerMessage = errorMessage.toLowerCase();
+    const safetyKeywords = [
+      "安全策略",
+      "被阻止",
+      "被拦截",
+      "blocked",
+      "safety",
+      "prohibited",
+      "recitation",
+      "内容被阻止",
+      "内容被拦截",
+      "安全策略阻止",
+      "安全策略拦截"
+    ];
+    return safetyKeywords.some((keyword) => lowerMessage.includes(keyword.toLowerCase()));
+  }
+  __name(isSafetyBlockError, "isSafetyBlockError");
+  async function checkSafetyBlockLimit(userId) {
+    const usersData = await loadUsersData();
+    const userData = usersData[userId];
+    if (!userData || !userData.safetyBlockHistory || userData.safetyBlockHistory.length === 0) {
+      return { allowed: true };
+    }
+    const windowSize = 10;
+    const recentBlocks = userData.safetyBlockHistory.slice(-windowSize);
+    const blockRate = recentBlocks.length / windowSize;
+    const threshold = 0.5;
+    if (blockRate > threshold) {
+      return {
+        allowed: false,
+        message: `检测到您最近多次发送被安全策略拦截的内容（拦截率：${(blockRate * 100).toFixed(0)}%），请检查您的内容是否符合使用规范。如继续发送违规内容，可能会被进一步限制。`
+      };
+    }
+    return { allowed: true };
+  }
+  __name(checkSafetyBlockLimit, "checkSafetyBlockLimit");
   async function getPromptInput(session, message) {
     await session.send(message);
     const input = await session.prompt(3e4);
@@ -1286,15 +1326,24 @@ function apply(ctx, config) {
         donationCount: 0,
         donationAmount: 0,
         lastUsed: (/* @__PURE__ */ new Date()).toISOString(),
-        createdAt: (/* @__PURE__ */ new Date()).toISOString()
+        createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+        safetyBlockCount: 0,
+        safetyBlockHistory: []
       };
       await saveUsersData(usersData);
       logger.info("创建新用户数据", { userId, userName });
+    } else {
+      if (usersData[userId].safetyBlockCount === void 0) {
+        usersData[userId].safetyBlockCount = 0;
+      }
+      if (usersData[userId].safetyBlockHistory === void 0) {
+        usersData[userId].safetyBlockHistory = [];
+      }
     }
     return usersData[userId];
   }
   __name(getUserData, "getUserData");
-  async function updateUserData(userId, userName, commandName, numImages = 1) {
+  async function updateUserData(userId, userName, commandName, numImages = 1, isSafetyBlock = false) {
     const usersData = await loadUsersData();
     const now = (/* @__PURE__ */ new Date()).toISOString();
     const today = (/* @__PURE__ */ new Date()).toDateString();
@@ -1310,11 +1359,19 @@ function apply(ctx, config) {
         donationCount: 0,
         donationAmount: 0,
         lastUsed: now,
-        createdAt: now
+        createdAt: now,
+        safetyBlockCount: isSafetyBlock ? numImages : 0,
+        safetyBlockHistory: isSafetyBlock ? [now] : []
       };
       await saveUsersData(usersData);
       return { userData: usersData[userId], consumptionType: "free", freeUsed: numImages, purchasedUsed: 0 };
     }
+    if (usersData[userId].safetyBlockCount === void 0) {
+      usersData[userId].safetyBlockCount = 0;
+    }
+    if (usersData[userId].safetyBlockHistory === void 0) {
+      usersData[userId].safetyBlockHistory = [];
+    }
     usersData[userId].totalUsageCount += numImages;
     usersData[userId].lastUsed = now;
     const lastReset = new Date(usersData[userId].lastDailyReset || usersData[userId].createdAt).toDateString();
@@ -1338,6 +1395,13 @@ function apply(ctx, config) {
       purchasedUsed = purchasedToUse;
       remainingToConsume -= purchasedToUse;
     }
+    if (isSafetyBlock) {
+      usersData[userId].safetyBlockCount += numImages;
+      usersData[userId].safetyBlockHistory.push(now);
+      if (usersData[userId].safetyBlockHistory.length > 20) {
+        usersData[userId].safetyBlockHistory = usersData[userId].safetyBlockHistory.slice(-20);
+      }
+    }
     await saveUsersData(usersData);
     let consumptionType;
     if (freeUsed > 0 && purchasedUsed > 0) {
@@ -1350,16 +1414,17 @@ function apply(ctx, config) {
     return { userData: usersData[userId], consumptionType, freeUsed, purchasedUsed };
   }
   __name(updateUserData, "updateUserData");
-  async function recordUserUsage(session, commandName, numImages = 1) {
+  async function recordUserUsage(session, commandName, numImages = 1, isSafetyBlock = false) {
     const userId = session.userId;
     const userName = session.username || session.userId || "未知用户";
     if (!userId) return;
-    const { userData, consumptionType, freeUsed, purchasedUsed } = await updateUserData(userId, userName, commandName, numImages);
+    const { userData, consumptionType, freeUsed, purchasedUsed } = await updateUserData(userId, userName, commandName, numImages, isSafetyBlock);
     if (isAdmin(userId)) {
+      const blockInfo = isSafetyBlock ? "\n⚠️ 本次请求被安全策略拦截，已扣除使用次数" : "";
       await session.send(`📊 使用统计 [管理员]
 用户：${userData.userName}
 总调用次数：${userData.totalUsageCount}次
-状态：无限制使用`);
+状态：无限制使用${blockInfo}`);
     } else {
       const remainingToday = Math.max(0, config.dailyFreeLimit - userData.dailyUsageCount);
       let consumptionText = "";
@@ -1370,13 +1435,14 @@ function apply(ctx, config) {
       } else {
         consumptionText = `充值次数 -${purchasedUsed}`;
       }
+      const blockInfo = isSafetyBlock ? "\n⚠️ 本次请求被安全策略拦截，已扣除使用次数。请检查内容是否符合使用规范。" : "";
       await session.send(`📊 使用统计
 用户：${userData.userName}
 本次生成：${numImages}张图片
 本次消费：${consumptionText}
 总调用次数：${userData.totalUsageCount}次
 今日剩余免费：${remainingToday}次
-充值剩余：${userData.remainingPurchasedCount}次`);
+充值剩余：${userData.remainingPurchasedCount}次${blockInfo}`);
     }
     logger.info("用户调用记录", {
       userId,
@@ -1389,6 +1455,8 @@ function apply(ctx, config) {
       totalUsageCount: userData.totalUsageCount,
       dailyUsageCount: userData.dailyUsageCount,
       remainingPurchasedCount: userData.remainingPurchasedCount,
+      isSafetyBlock,
+      safetyBlockCount: userData.safetyBlockCount,
       isAdmin: isAdmin(userId)
     });
   }
@@ -1589,6 +1657,18 @@ ${infoParts.join("\n")}`;
       activeTasks.delete(userId);
       const sanitizedError = sanitizeError(error);
       logger.error("图像处理失败", { userId, error: sanitizedError });
+      const isSafetyBlock = isSafetyBlockError(error);
+      if (isSafetyBlock) {
+        try {
+          await recordUserUsage(session, styleName, imageCount, true);
+          const blockLimitCheck = await checkSafetyBlockLimit(userId);
+          if (!blockLimitCheck.allowed) {
+            return `图像处理失败：内容被安全策略拦截。${blockLimitCheck.message}`;
+          }
+        } catch (recordError) {
+          logger.error("记录安全拦截使用次数失败", { userId, error: recordError });
+        }
+      }
       if (error?.message) {
         const safeMessage = sanitizeString(error.message);
         return `图像处理失败：${safeMessage}`;
@@ -1755,6 +1835,18 @@ Prompt: ${prompt}`);
           activeTasks.delete(userId);
           const sanitizedError = sanitizeError(error);
           logger.error("图片合成失败", { userId, error: sanitizedError });
+          const isSafetyBlock = isSafetyBlockError(error);
+          if (isSafetyBlock) {
+            try {
+              await recordUserUsage(session, COMMANDS.COMPOSE_IMAGE, imageCount, true);
+              const blockLimitCheck = await checkSafetyBlockLimit(userId);
+              if (!blockLimitCheck.allowed) {
+                return `图片合成失败：内容被安全策略拦截。${blockLimitCheck.message}`;
+              }
+            } catch (recordError) {
+              logger.error("记录安全拦截使用次数失败", { userId, error: recordError });
+            }
+          }
           if (error?.message) {
             const safeMessage = sanitizeString(error.message);
             return `图片合成失败：${safeMessage}`;
@@ -1825,7 +1917,9 @@ Prompt: ${prompt}`);
             donationCount: 0,
             donationAmount: 0,
             lastUsed: now,
-            createdAt: now
+            createdAt: now,
+            safetyBlockCount: 0,
+            safetyBlockHistory: []
           };
         }
         const beforeBalance = usersData[userId].remainingPurchasedCount;

package/package.json

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