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

package/lib/index.d.ts

@@ -49,6 +49,8 @@ export interface Config {
     styles: StyleConfig[];
     styleGroups?: Record<string, StyleGroupConfig>;
     logLevel: 'info' | 'debug';
+    securityBlockWindow: number;
+    securityBlockWarningThreshold: number;
 }
 export interface RechargeRecord {
     id: string;

package/lib/index.js

@@ -943,7 +943,10 @@ var Config = import_koishi.Schema.intersect([
     logLevel: import_koishi.Schema.union([
       import_koishi.Schema.const("info").description("普通信息"),
       import_koishi.Schema.const("debug").description("完整的debug信息")
-    ]).default("info").description("日志输出详细程度")
+    ]).default("info").description("日志输出详细程度"),
+    // 安全策略拦截设置
+    securityBlockWindow: import_koishi.Schema.number().default(600).min(60).max(3600).description("安全策略拦截追踪时间窗口（秒），在此时间窗口内连续触发拦截会被记录"),
+    securityBlockWarningThreshold: import_koishi.Schema.number().default(3).min(1).max(10).description("安全策略拦截警示阈值，连续触发此次数拦截后将发送警示消息，再次触发将被扣除积分")
   }),
   // 自定义风格命令配置
   import_koishi.Schema.object({
@@ -968,6 +971,8 @@ function apply(ctx, config) {
   const logger = ctx.logger("aka-ai-generator");
   const activeTasks = /* @__PURE__ */ new Map();
   const rateLimitMap = /* @__PURE__ */ new Map();
+  const securityBlockMap = /* @__PURE__ */ new Map();
+  const securityWarningMap = /* @__PURE__ */ new Map();
   const providerCache = /* @__PURE__ */ new Map();
   function getProviderInstance(providerType, modelId) {
     const cacheKey = `${providerType}:${modelId || "default"}`;
@@ -1393,6 +1398,40 @@ function apply(ctx, config) {
     });
   }
   __name(recordUserUsage, "recordUserUsage");
+  async function recordSecurityBlock(session, numImages = 1) {
+    const userId = session.userId;
+    if (!userId) return;
+    if (isAdmin(userId)) {
+      return;
+    }
+    const now = Date.now();
+    const windowMs = config.securityBlockWindow * 1e3;
+    const windowStart = now - windowMs;
+    let blockTimestamps = securityBlockMap.get(userId) || [];
+    blockTimestamps = blockTimestamps.filter((timestamp) => timestamp > windowStart);
+    blockTimestamps.push(now);
+    securityBlockMap.set(userId, blockTimestamps);
+    const blockCount = blockTimestamps.length;
+    const hasWarning = securityWarningMap.get(userId) || false;
+    logger.info("安全策略拦截记录", {
+      userId,
+      blockCount,
+      threshold: config.securityBlockWarningThreshold,
+      hasWarning,
+      numImages
+    });
+    if (blockCount >= config.securityBlockWarningThreshold && !hasWarning) {
+      securityWarningMap.set(userId, true);
+      await session.send(`⚠️ 安全策略警示
+您已连续${config.securityBlockWarningThreshold}次触发安全策略拦截，再次发送被拦截内容将被扣除积分`);
+      logger.warn("用户收到安全策略警示", { userId, blockCount, threshold: config.securityBlockWarningThreshold });
+    } else if (hasWarning) {
+      const commandName = "安全策略拦截";
+      await recordUserUsage(session, commandName, numImages);
+      logger.warn("用户因安全策略拦截被扣除积分", { userId, numImages });
+    }
+  }
+  __name(recordSecurityBlock, "recordSecurityBlock");
   async function getInputData(session, imgParam, mode) {
     const collectedImages = [];
     let collectedText = "";
@@ -1496,11 +1535,19 @@ function apply(ctx, config) {
       new Promise(
         (_, reject) => setTimeout(() => reject(new Error("命令执行超时")), config.commandTimeout * 1e3)
       )
-    ]).catch((error) => {
+    ]).catch(async (error) => {
       const userId = session.userId;
       if (userId) activeTasks.delete(userId);
       const sanitizedError = sanitizeError(error);
       logger.error("图像处理超时或失败", { userId, error: sanitizedError });
+      if (error?.message !== "命令执行超时") {
+        const errorMessage = error?.message || "";
+        const isSecurityBlock = errorMessage.includes("内容被安全策略拦截") || errorMessage.includes("内容被安全策略阻止") || errorMessage.includes("内容被阻止") || errorMessage.includes("被阻止") || errorMessage.includes("SAFETY") || errorMessage.includes("RECITATION");
+        if (isSecurityBlock) {
+          const imageCount = requestContext?.numImages || config.defaultNumImages;
+          await recordSecurityBlock(session, imageCount);
+        }
+      }
       const safeMessage = typeof error?.message === "string" ? sanitizeString(error.message) : "未知错误";
       return error.message === "命令执行超时" ? "图像处理超时，请重试" : `图像处理失败：${safeMessage}`;
     });
@@ -1589,6 +1636,11 @@ ${infoParts.join("\n")}`;
       activeTasks.delete(userId);
       const sanitizedError = sanitizeError(error);
       logger.error("图像处理失败", { userId, error: sanitizedError });
+      const errorMessage = error?.message || "";
+      const isSecurityBlock = errorMessage.includes("内容被安全策略拦截") || errorMessage.includes("内容被安全策略阻止") || errorMessage.includes("内容被阻止") || errorMessage.includes("被阻止") || errorMessage.includes("SAFETY") || errorMessage.includes("RECITATION");
+      if (isSecurityBlock) {
+        await recordSecurityBlock(session, imageCount);
+      }
       if (error?.message) {
         const safeMessage = sanitizeString(error.message);
         return `图像处理失败：${safeMessage}`;
@@ -1755,6 +1807,11 @@ Prompt: ${prompt}`);
           activeTasks.delete(userId);
           const sanitizedError = sanitizeError(error);
           logger.error("图片合成失败", { userId, error: sanitizedError });
+          const errorMessage = error?.message || "";
+          const isSecurityBlock = errorMessage.includes("内容被安全策略拦截") || errorMessage.includes("内容被安全策略阻止") || errorMessage.includes("内容被阻止") || errorMessage.includes("被阻止") || errorMessage.includes("SAFETY") || errorMessage.includes("RECITATION");
+          if (isSecurityBlock) {
+            await recordSecurityBlock(session, imageCount);
+          }
           if (error?.message) {
             const safeMessage = sanitizeString(error.message);
             return `图片合成失败：${safeMessage}`;
@@ -1765,11 +1822,19 @@ Prompt: ${prompt}`);
       new Promise(
         (_, reject) => setTimeout(() => reject(new Error("命令执行超时")), config.commandTimeout * 1e3)
       )
-    ]).catch((error) => {
+    ]).catch(async (error) => {
       const userId = session.userId;
       if (userId) activeTasks.delete(userId);
       const sanitizedError = sanitizeError(error);
       logger.error("图片合成超时或失败", { userId, error: sanitizedError });
+      if (error?.message !== "命令执行超时") {
+        const errorMessage = error?.message || "";
+        const isSecurityBlock = errorMessage.includes("内容被安全策略拦截") || errorMessage.includes("内容被安全策略阻止") || errorMessage.includes("内容被阻止") || errorMessage.includes("被阻止") || errorMessage.includes("SAFETY") || errorMessage.includes("RECITATION");
+        if (isSecurityBlock) {
+          const imageCount = options?.num || config.defaultNumImages;
+          await recordSecurityBlock(session, imageCount);
+        }
+      }
       const safeMessage = typeof error?.message === "string" ? sanitizeString(error.message) : "未知错误";
       return error.message === "命令执行超时" ? "图片合成超时，请重试" : `图片合成失败：${safeMessage}`;
     });

package/package.json

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