feat 新增 MYBAN 命令

chr233 · Nov 25, 2024 · f43ace4 · f43ace4
1 parent cd3092b
commit f43ace4
Show file tree

Hide file tree

Showing 4 changed files with 194 additions and 86 deletions.
diff --git a/ASFEnhance/ASFEnhance.cs b/ASFEnhance/ASFEnhance.cs
@@ -325,7 +325,10 @@ public Task OnLoaded()
 
                 "REGISTEDATE" when access >= EAccess.Operator =>
                     Account.Command.ResponseGetRegisteDate(bot),
-
+
+                "MYBAN" when access >= EAccess.Operator =>
+                    Account.Command.ResponseGetMyBans(bot),
+
                 //Cart
                 "CART" or
                 "C" when access >= EAccess.Operator =>
@@ -624,6 +627,9 @@ public Task OnLoaded()
                 "RL" when access >= EAccess.Master =>
                     Account.Command.ResponseRemoveFreeLicenses(bot, args[1]),
 
+                "MYBAN" when access >= EAccess.Operator =>
+                    Account.Command.ResponseGetMyBans(Utilities.GetArgsAsText(args, 1, ",")),
+
                 //Cart
                 "CART" or
                 "C" when access >= EAccess.Operator =>

diff --git a/ASFEnhance/Account/Command.cs b/ASFEnhance/Account/Command.cs
@@ -1334,6 +1334,56 @@ async Task workThread(uint subId)
 
         var responses = new List<string?>(results.Where(result => !string.IsNullOrEmpty(result)));
 
+        return responses.Count > 0 ? string.Join(Environment.NewLine, responses) : null;
+    }
+
+     /// <summary>
+    /// 移除免费许可证
+    /// </summary>
+    /// <param name="bot"></param>
+    /// <param name="query"></param>
+    /// <returns></returns>
+    internal static async Task<string?> ResponseGetMyBans(Bot bot)
+    {
+        if (!bot.IsConnectedAndLoggedOn)
+        {
+            return bot.FormatBotResponse(Strings.BotNotConnected);
+        }
+
+        var sb = new StringBuilder();
+        sb.AppendLine(bot.FormatBotResponse(Langs.MultipleLineResult));
+
+        var response = await WebRequest.GetMyBans(bot).ConfigureAwait(false);
+        sb.AppendLine(response);
+
+        return sb.ToString();
+    }
+
+    /// <summary>
+    /// 移除免费许可证 (多个Bot)
+    /// </summary>
+    /// <param name="botNames"></param>
+    /// <param name="query"></param>
+    /// <returns></returns>
+    /// <exception cref="ArgumentNullException"></exception>
+    internal static async Task<string?> ResponseGetMyBans(string botNames)
+    {
+        if (string.IsNullOrEmpty(botNames))
+        {
+            throw new ArgumentNullException(nameof(botNames));
+        }
+
+        var bots = Bot.GetBots(botNames);
+
+        if ((bots == null) || (bots.Count == 0))
+        {
+            return FormatStaticResponse(Strings.BotNotFound, botNames);
+        }
+
+        var results = await Utilities.InParallel(bots.Select(ResponseGetMyBans)).ConfigureAwait(false);
+
+        var responses = new List<string?>(results.Where(result => !string.IsNullOrEmpty(result)));
+
         return responses.Count > 0 ? string.Join(Environment.NewLine, responses) : null;
     }
 }