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

package/lib/index.js

@@ -1622,19 +1622,102 @@ Prompt: ${prompt}`);
       return error.message === "命令执行超时" ? "图片合成超时，请重试" : `图片合成失败：${error.message}`;
     });
   });
-  ctx.command(`${COMMANDS.RECHARGE} [content:text]`, "为用户充值次数（仅管理员）").action(async ({ session }, content) => {
+  ctx.command(`${COMMANDS.RECHARGE} [content:text]`, "为用户充值次数（仅管理员）").option("all", "-all 对所有使用过插件的用户充值").action(async ({ session, options }, content) => {
     if (!session?.userId) return "会话无效";
     if (!isAdmin(session.userId)) {
       return "权限不足，仅管理员可操作";
     }
-    const inputContent = content || await getPromptInput(session, "请输入充值信息，格式：\n@用户1 @用户2 充值次数 [备注]");
-    if (!inputContent) return "输入超时或无效";
-    const elements = import_koishi.h.parse(inputContent);
+    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 "请输入充值次数，例如：图像充值 -all 20";
+      }
+      const amount2 = parseInt(parts2[0]);
+      const note2 = parts2.slice(1).join(" ") || "全员充值";
+      if (!amount2 || amount2 <= 0) {
+        return "充值次数必须大于0";
+      }
+      try {
+        const usersData = await loadUsersData();
+        const rechargeHistory = await loadRechargeHistory();
+        const now = (/* @__PURE__ */ new Date()).toISOString();
+        const recordId = `recharge_${now.replace(/[-:T.]/g, "").slice(0, 14)}_${Math.random().toString(36).substr(2, 3)}`;
+        const targets = [];
+        const allUserIds = Object.keys(usersData);
+        if (allUserIds.length === 0) {
+          return "当前没有使用过插件的用户";
+        }
+        for (const userId of allUserIds) {
+          if (!userId) continue;
+          const userData = usersData[userId];
+          const userName = userData.userName || userId;
+          const beforeBalance = userData.remainingPurchasedCount;
+          userData.purchasedCount += amount2;
+          userData.remainingPurchasedCount += amount2;
+          targets.push({
+            userId,
+            userName,
+            amount: amount2,
+            beforeBalance,
+            afterBalance: userData.remainingPurchasedCount
+          });
+        }
+        await saveUsersData(usersData);
+        const record = {
+          id: recordId,
+          timestamp: now,
+          type: "batch",
+          operator: {
+            userId: session.userId,
+            userName: session.username || session.userId
+          },
+          targets,
+          totalAmount: amount2 * allUserIds.length,
+          note: note2 || "全员充值",
+          metadata: { all: true }
+        };
+        rechargeHistory.records.push(record);
+        await saveRechargeHistory(rechargeHistory);
+        return `✅ 全员充值成功
+目标用户数：${allUserIds.length}人
+充值次数：${amount2}次/人
+总充值：${record.totalAmount}次
+操作员：${record.operator.userName}
+备注：${record.note}`;
+      } catch (error) {
+        logger.error("全员充值操作失败", error);
+        return "充值失败，请稍后重试";
+      }
+    }
+    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();
     if (atElements.length === 0) {
-      return "未找到@用户，请使用@用户的方式";
+      return "未找到@用户，请使用@用户的方式，或使用 -all 参数进行全员充值";
     }
     const parts = text.split(/\s+/).filter((p) => p);
     if (parts.length === 0) {

package/package.json

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