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

package/lib/index.d.ts

@@ -53,7 +53,7 @@ export interface Config {
 export interface RechargeRecord {
     id: string;
     timestamp: string;
-    type: 'single' | 'batch';
+    type: 'single' | 'batch' | 'all';
     operator: {
         userId: string;
         userName: string;

package/lib/index.js

@@ -748,6 +748,7 @@ var COMMANDS = {
   PIXELATE: "变像素",
   QUERY_QUOTA: "图像额度",
   RECHARGE: "图像充值",
+  RECHARGE_ALL: "活动充值",
   RECHARGE_HISTORY: "图像充值记录",
   FUNCTION_LIST: "图像功能",
   IMAGE_COMMANDS: "图像指令"
@@ -979,6 +980,7 @@ function apply(ctx, config) {
     // 管理员指令
     adminCommands: [
       { name: COMMANDS.RECHARGE, description: "为用户充值次数（仅管理员）" },
+      { name: COMMANDS.RECHARGE_ALL, description: "为所有用户充值次数（活动派发，仅管理员）" },
       { name: COMMANDS.RECHARGE_HISTORY, description: "查看充值历史记录（仅管理员）" }
     ]
   };
@@ -1622,102 +1624,19 @@ Prompt: ${prompt}`);
       return error.message === "命令执行超时" ? "图片合成超时，请重试" : `图片合成失败：${error.message}`;
     });
   });
-  ctx.command(`${COMMANDS.RECHARGE} [content:text]`, "为用户充值次数（仅管理员）").option("all", "-all 对所有使用过插件的用户充值").action(async ({ session, options }, content) => {
+  ctx.command(`${COMMANDS.RECHARGE} [content:text]`, "为用户充值次数（仅管理员）").action(async ({ session }, content) => {
     if (!session?.userId) return "会话无效";
     if (!isAdmin(session.userId)) {
       return "权限不足，仅管理员可操作";
     }
-    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 inputContent = content || await getPromptInput(session, "请输入充值信息，格式：\n@用户1 @用户2 充值次数 [备注]");
+    if (!inputContent) return "输入超时或无效";
+    const elements = import_koishi.h.parse(inputContent);
     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 "未找到@用户，请使用@用户的方式，或使用 -all 参数进行全员充值";
+      return "未找到@用户，请使用@用户的方式";
     }
     const parts = text.split(/\s+/).filter((p) => p);
     if (parts.length === 0) {
@@ -1798,6 +1717,77 @@ Prompt: ${prompt}`);
       return "充值失败，请稍后重试";
     }
   });
+  ctx.command(`${COMMANDS.RECHARGE_ALL} [content:text]`, "为所有用户充值次数（活动派发，仅管理员）").action(async ({ session }, content) => {
+    if (!session?.userId) return "会话无效";
+    if (!isAdmin(session.userId)) {
+      return "权限不足，仅管理员可操作";
+    }
+    const inputContent = content || await getPromptInput(session, "请输入活动充值信息，格式：\n充值次数 [备注]\n例如：20 或 20 春节活动奖励");
+    if (!inputContent) return "输入超时或无效";
+    const elements = import_koishi.h.parse(inputContent);
+    const textElements = import_koishi.h.select(elements, "text");
+    const text = textElements.map((el) => el.attrs.content).join(" ").trim();
+    const parts = text.split(/\s+/).filter((p) => p);
+    if (parts.length === 0) {
+      return "请输入充值次数，例如：图像活动充值 20 或 图像活动充值 20 活动名称";
+    }
+    const amount = parseInt(parts[0]);
+    const note = parts.slice(1).join(" ") || "活动充值";
+    if (!amount || amount <= 0) {
+      return "充值次数必须大于0";
+    }
+    try {
+      const usersData = await loadUsersData();
+      const rechargeHistory = await loadRechargeHistory();
+      const now = (/* @__PURE__ */ new Date()).toISOString();
+      const recordId = `recharge_all_${now.replace(/[-:T.]/g, "").slice(0, 14)}_${Math.random().toString(36).substr(2, 3)}`;
+      const allUserIds = Object.keys(usersData).filter((userId) => userId && usersData[userId]);
+      if (allUserIds.length === 0) {
+        return "当前没有使用过插件的用户，无法进行活动充值";
+      }
+      const targets = [];
+      for (const userId of allUserIds) {
+        if (!userId) continue;
+        const userData = usersData[userId];
+        const userName = userData.userName || userId;
+        const beforeBalance = userData.remainingPurchasedCount;
+        userData.purchasedCount += amount;
+        userData.remainingPurchasedCount += amount;
+        targets.push({
+          userId,
+          userName,
+          amount,
+          beforeBalance,
+          afterBalance: userData.remainingPurchasedCount
+        });
+      }
+      await saveUsersData(usersData);
+      const record = {
+        id: recordId,
+        timestamp: now,
+        type: "all",
+        operator: {
+          userId: session.userId,
+          userName: session.username || session.userId
+        },
+        targets,
+        totalAmount: amount * allUserIds.length,
+        note: note || "活动充值",
+        metadata: { all: true }
+      };
+      rechargeHistory.records.push(record);
+      await saveRechargeHistory(rechargeHistory);
+      return `✅ 活动充值成功
+目标用户数：${allUserIds.length}人
+充值次数：${amount}次/人
+总充值：${record.totalAmount}次
+操作员：${record.operator.userName}
+备注：${record.note}`;
+    } catch (error) {
+      logger.error("活动充值操作失败", error);
+      return "活动充值失败，请稍后重试";
+    }
+  });
   ctx.command(`${COMMANDS.QUERY_QUOTA} [target:text]`, "查询用户额度信息").action(async ({ session }, target) => {
     if (!session?.userId) return "会话无效";
     const userIsAdmin = isAdmin(session.userId);

package/package.json

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