Add busy system for command handler

acid-chicken · Dec 19, 2017 · 8e7e477 · 8e7e477
1 parent 4c85d0f
commit 8e7e477
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 4 deletions.
diff --git a/Samurai/Models/Config.cs b/Samurai/Models/Config.cs
@@ -8,6 +8,9 @@ namespace AcidChicken.Samurai.Models
     [JsonObject]
     public class Config
     {
+        [JsonProperty("busy")]
+        public ulong Busy { get; set; } = ulong.MaxValue;
+
         [JsonProperty("default_settings")]
         public Dictionary<string, string> DefaultSettings { get; set; } = new Dictionary<string, string>();
 

diff --git a/Samurai/Modules/ModuleManager.cs b/Samurai/Modules/ModuleManager.cs
@@ -16,6 +16,8 @@ public static class ModuleManager
     {
         public const string Prefix = "./";
 
+        public static ulong Busy;
+
         static ModuleManager()
         {
             ServiceConfig = new CommandServiceConfig();
@@ -51,23 +53,43 @@ ref position
             var context = new CommandContext(DiscordClient, message);
             using (var typing = context.Channel.EnterTypingState())
             {
-                var result = await Service.ExecuteAsync(context, position);
-                if (!result.IsSuccess)
+                if (++Busy > ApplicationConfig.Busy)
                 {
                     await context.Channel.SendMessageAsync
                     (
                         text: context.User.Mention,
                         embed:
                             new EmbedBuilder()
-                                .WithTitle("コマンドエラー")
-                                .WithDescription(result.ErrorReason)
+                                .WithTitle("ビジー状態")
+                                .WithDescription("現在Botが混み合っています。時間を空けてから再度お試し下さい。")
                                 .WithCurrentTimestamp()
+                                .WithColor(Colors.Orange)
                                 .WithColor(Colors.Red)
                                 .WithFooter(EmbedManager.CurrentFooter)
                                 .WithAuthor(context.User)
                     );
                 }
+                else
+                {
+                    var result = await Service.ExecuteAsync(context, position);
+                    if (!result.IsSuccess)
+                    {
+                        await context.Channel.SendMessageAsync
+                        (
+                            text: context.User.Mention,
+                            embed:
+                                new EmbedBuilder()
+                                    .WithTitle("コマンドエラー")
+                                    .WithDescription(result.ErrorReason)
+                                    .WithCurrentTimestamp()
+                                    .WithColor(Colors.Red)
+                                    .WithFooter(EmbedManager.CurrentFooter)
+                                    .WithAuthor(context.User)
+                        );
+                    }
+                }
             }
+            Busy--;
         }
     }
 }