koishi-plugin-hhs-huatu 0.3.1 → 0.3.2

package/lib/index.js

@@ -3246,9 +3246,6 @@ function apply(ctx, config) {
             resolve: /* @__PURE__ */ __name((value) => {
               queueSystem.userTasks[currentUserId]--;
               targetBot.sendMessage(targetChannelId, value);
-              if (config.membershipEnabled) {
-                membershipSystem.incrementUsage(currentUserId, 1);
-              }
               if (Array.isArray(ctx.config.token)) {
                 if (config.tokenStrategy === "parallel" && ctx.runtime?.tokenUsage) {
                   const usedTokens = Object.entries(ctx.runtime.tokenUsage).filter(([_, used]) => used).map(([index2]) => index2);
@@ -3493,13 +3490,29 @@ function apply(ctx, config) {
     ]);
   });
   if (config.memberDebugCommandEnabled) {
-    ctx.command("novelai.member-debug", "会员系统调试指令").userFields(["authority"]).alias("会员调试").option("cleanup", "-c 立即执行会员信息清理").option("remind", "-r 立即执行会员到期提醒").option("status", "-s 查看定时任务状态").action(async ({ session, options: options2 }) => {
+    ctx.command("novelai.member-debug", "会员系统调试指令").userFields(["authority"]).alias("会员调试").option("cleanup", "-c 立即执行会员信息清理").option("remind", "-r 立即执行会员到期提醒").option("status", "-s 查看定时任务状态").option("resetUsage", "-u <userId:string> 重置指定用户的使用次数").action(async ({ session, options: options2 }) => {
       if (session.user.authority < config.memberDebugCommandAuthLv) {
         return `权限不足，需要权限等级 ${config.memberDebugCommandAuthLv} 或以上`;
       }
       if (!config.membershipEnabled) {
         return "会员系统未启用";
       }
+      if (options2.resetUsage) {
+        const targetId = options2.resetUsage;
+        if (!userData[targetId]) {
+          return `用户 ${targetId} 不存在`;
+        }
+        userData[targetId].dailyUsage = 0;
+        await membershipSystem.saveUserData();
+        const user = userData[targetId];
+        const dailyLimit = user.isMember ? config.memberDailyLimit : config.nonMemberDailyLimit;
+        const remaining = dailyLimit - user.dailyUsage;
+        return `✅ 已重置用户 ${targetId} 的使用次数
+当前状态：${user.isMember ? "会员" : "非会员"}
+每日限额：${dailyLimit} 次
+已使用：${user.dailyUsage} 次
+剩余：${remaining} 次`;
+      }
       if (options2.status) {
         let statusMsg = "【会员系统状态】\n\n";
         if (config.memberCleanupEnabled) {
@@ -3595,7 +3608,7 @@ function apply(ctx, config) {
         await membershipSystem.checkAndRemindExpiringMembers();
         return "✅ 提醒完成！请查看控制台日志获取详细信息。";
       }
-      return "请使用以下选项：\n-c 立即执行会员信息清理\n-r 立即执行会员到期提醒\n-s 查看定时任务状态";
+      return "请使用以下选项：\n-c 立即执行会员信息清理\n-r 立即执行会员到期提醒\n-s 查看定时任务状态\n-u 重置指定用户的使用次数";
     });
   }
   if (config.directorToolsEnabled !== false) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-hhs-huatu",
   "description": "Generate images by diffusion models",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [