koishi-plugin-aka-ai-generator 0.3.7 → 0.3.8

package/lib/index.js

@@ -1627,15 +1627,31 @@ Prompt: ${prompt}`);
     if (!isAdmin(session.userId)) {
       return "权限不足，仅管理员可操作";
     }
-    if (options?.all) {
-      const inputContent2 = content || await getPromptInput(session, "请输入充值信息，格式：\n充值次数 [备注]");
-      if (!inputContent2) return "输入超时或无效";
-      const elements2 = import_koishi.h.parse(inputContent2);
-      const textElements2 = import_koishi.h.select(elements2, "text");
-      const text2 = textElements2.map((el) => el.attrs.content).join(" ").trim();
+    let useAll = options?.all;
+    let inputContent = content;
+    if (!useAll && inputContent) {
+      const contentStr = typeof inputContent === "string" ? inputContent : inputContent ? import_koishi.h.select(import_koishi.h.parse(inputContent), "text").map((e) => e.attrs.content).join(" ") : "";
+      if (contentStr && /-all\b/i.test(contentStr)) {
+        useAll = true;
+        inputContent = contentStr.replace(/-all\s*/gi, "").trim();
+      }
+    }
+    if (useAll) {
+      if (!inputContent || typeof inputContent === "string" && !inputContent.trim()) {
+        inputContent = await getPromptInput(session, "请输入充值次数 [备注]，例如：20 或 20 活动奖励");
+      }
+      if (!inputContent) return "输入超时或无效";
+      let text2 = "";
+      if (typeof inputContent === "string") {
+        text2 = inputContent.trim();
+      } else {
+        const elements2 = import_koishi.h.parse(inputContent);
+        const textElements2 = import_koishi.h.select(elements2, "text");
+        text2 = textElements2.map((el) => el.attrs.content).join(" ").trim();
+      }
       const parts2 = text2.split(/\s+/).filter((p) => p);
       if (parts2.length === 0) {
-        return "请输入充值次数";
+        return "请输入充值次数，例如：图像充值 -all 20";
       }
       const amount2 = parseInt(parts2[0]);
       const note2 = parts2.slice(1).join(" ") || "全员充值";
@@ -1694,9 +1710,9 @@ Prompt: ${prompt}`);
         return "充值失败，请稍后重试";
       }
     }
-    const inputContent = content || await getPromptInput(session, "请输入充值信息，格式：\n@用户1 @用户2 充值次数 [备注]");
-    if (!inputContent) return "输入超时或无效";
-    const elements = import_koishi.h.parse(inputContent);
+    const rechargeContent = content || await getPromptInput(session, "请输入充值信息，格式：\n@用户1 @用户2 充值次数 [备注]");
+    if (!rechargeContent) return "输入超时或无效";
+    const elements = import_koishi.h.parse(rechargeContent);
     const atElements = import_koishi.h.select(elements, "at");
     const textElements = import_koishi.h.select(elements, "text");
     const text = textElements.map((el) => el.attrs.content).join(" ").trim();

package/package.json

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