Adds admin option to set custom replacements

Author: benbierens

Tools/BiblioTech/CommandHandler.cs

@@ -10,12 +10,14 @@ namespace BiblioTech
     public class CommandHandler
     {
         private readonly DiscordSocketClient client;
+        private readonly CustomReplacement replacement;
         private readonly BaseCommand[] commands;
         private readonly ILog log;
 
-        public CommandHandler(ILog log, DiscordSocketClient client, params BaseCommand[] commands)
+        public CommandHandler(ILog log, DiscordSocketClient client, CustomReplacement replacement, params BaseCommand[] commands)
         {
             this.client = client;
+            this.replacement = replacement;
             this.commands = commands;
             this.log = log;
             client.Ready += Client_Ready;
@@ -31,7 +33,7 @@ private async Task Client_Ready()
             var adminChannels = guild.TextChannels.Where(Program.AdminChecker.IsAdminChannel).ToArray();
             if (adminChannels == null || !adminChannels.Any()) throw new Exception("No admin message channel");
             Program.AdminChecker.SetAdminChannel(adminChannels.First());
-            Program.RoleDriver = new RoleDriver(client, log);
+            Program.RoleDriver = new RoleDriver(client, log, replacement);
 
             var builders = commands.Select(c =>
             {

Tools/BiblioTech/Commands/AdminCommand.cs

@@ -1,4 +1,5 @@
 using BiblioTech.Options;
+using BiblioTech.Rewards;
 
 namespace BiblioTech.Commands
 {
@@ -10,12 +11,14 @@ public class AdminCommand : BaseCommand
         private readonly AddSprCommand addSprCommand;
         private readonly ClearSprsCommand clearSprsCommand;
         private readonly GetSprCommand getSprCommand;
+        private readonly LogReplaceCommand logReplaceCommand;
 
-        public AdminCommand(SprCommand sprCommand)
+        public AdminCommand(SprCommand sprCommand, CustomReplacement replacement)
         {
             addSprCommand = new AddSprCommand(sprCommand);
             clearSprsCommand = new ClearSprsCommand(sprCommand);
             getSprCommand = new GetSprCommand(sprCommand);
+            logReplaceCommand = new LogReplaceCommand(replacement);
         }
 
         public override string Name => "admin";
@@ -29,7 +32,8 @@ public AdminCommand(SprCommand sprCommand)
             whoIsCommand,
             addSprCommand,
             clearSprsCommand,
-            getSprCommand
+            getSprCommand,
+            logReplaceCommand
         };
 
         protected override async Task Invoke(CommandContext context)
@@ -52,6 +56,7 @@ protected override async Task Invoke(CommandContext context)
             await addSprCommand.CommandHandler(context);
             await clearSprsCommand.CommandHandler(context);
             await getSprCommand.CommandHandler(context);
+            await logReplaceCommand.CommandHandler(context);
         }
 
         public class ClearUserAssociationCommand : SubCommandOption
@@ -194,7 +199,7 @@ protected override async Task onSubCommand(CommandContext context)
             }
         }
 
-        public class GetSprCommand: SubCommandOption
+        public class GetSprCommand : SubCommandOption
         {
             private readonly SprCommand sprCommand;
 
@@ -210,5 +215,56 @@ protected override async Task onSubCommand(CommandContext context)
                 await context.Followup("SPRs: " + string.Join(", ", sprCommand.Get().Select(s => $"'{s}'")));
             }
         }
+
+        public class LogReplaceCommand : SubCommandOption
+        {
+            private readonly CustomReplacement replacement;
+            private readonly StringOption fromOption = new StringOption("from", "string to replace", true);
+            private readonly StringOption toOption = new StringOption("to", "string to replace with", false);
+
+            public LogReplaceCommand(CustomReplacement replacement)
+                : base(name: "logreplace",
+               description: "Replaces all occurances of 'from' with 'to' in ChainEvent messages. Leave 'to' empty to remove a replacement.")
+            {
+                this.replacement = replacement;
+            }
+
+            public override CommandOption[] Options => new[] { fromOption, toOption };
+
+            protected override async Task onSubCommand(CommandContext context)
+            {
+                var from = await fromOption.Parse(context);
+                var to = await toOption.Parse(context);
+
+                if (string.IsNullOrEmpty(from))
+                {
+                    await context.Followup("'from' not received");
+                    return;
+                }
+
+                if (from.Length < 5)
+                {
+                    await context.Followup("'from' must be length 5 or greater.");
+                    return;
+                }
+                
+                if (string.IsNullOrEmpty(to))
+                {
+                    replacement.Remove(from);
+                    await context.Followup($"Replace for '{from}' removed.");
+                }
+                else
+                {
+                    if (to.Length < 5)
+                    {
+                        await context.Followup("'to' must be length 5 or greater.");
+                        return;
+                    }
+
+                    replacement.Add(from, to);
+                    await context.Followup($"Replace added '{from}' -->> '{to}'.");
+                }
+            }
+        }
     }
 }

Tools/BiblioTech/Program.cs

@@ -11,6 +11,7 @@ namespace BiblioTech
     public class Program
     {
         private DiscordSocketClient client = null!;
+        private readonly CustomReplacement replacement = new CustomReplacement();
 
         public static Configuration Config { get; private set; } = null!;
         public static UserRepo UserRepo { get; } = new UserRepo();
@@ -73,13 +74,13 @@ private async Task StartDiscordBot()
             var notifyCommand = new NotifyCommand();
             var associateCommand = new UserAssociateCommand(notifyCommand);
             var sprCommand = new SprCommand();
-            var handler = new CommandHandler(Log, client,
+            var handler = new CommandHandler(Log, client, replacement,
                 new GetBalanceCommand(associateCommand),
                 new MintCommand(associateCommand),
                 sprCommand,
                 associateCommand,
                 notifyCommand,
-                new AdminCommand(sprCommand),
+                new AdminCommand(sprCommand, replacement),
                 new MarketCommand()
             );
 

Tools/BiblioTech/Rewards/ChainEventsSender.cs

@@ -6,11 +6,13 @@ namespace BiblioTech.Rewards
     public class ChainEventsSender
     {
         private readonly ILog log;
-        private SocketTextChannel? eventsChannel;
+        private readonly CustomReplacement replacement;
+        private readonly SocketTextChannel? eventsChannel;
 
-        public ChainEventsSender(ILog log, SocketTextChannel? eventsChannel)
+        public ChainEventsSender(ILog log, CustomReplacement replacement, SocketTextChannel? eventsChannel)
         {
             this.log = log;
+            this.replacement = replacement;
             this.eventsChannel = eventsChannel;
         }
 
@@ -43,6 +45,7 @@ await Task.Run(async () =>
         private string ApplyReplacements(UserData[] users, string msg)
         {
             var result = ApplyUserAddressReplacements(users, msg);
+            result = ApplyCustomReplacements(result);
             return result;
         }
 
@@ -60,5 +63,10 @@ private string ApplyUserAddressReplacements(UserData[] users, string msg)
 
             return msg;
         }
+
+        private string ApplyCustomReplacements(string result)
+        {
+            return replacement.Apply(result);
+        }
     }
 }

Tools/BiblioTech/Rewards/CustomReplacement.cs

@@ -0,0 +1,34 @@
+namespace BiblioTech.Rewards
+{
+    public class CustomReplacement
+    {
+        private readonly Dictionary<string, string> replacements = new Dictionary<string, string>();
+
+        public void Add(string from, string to)
+        {
+            if (replacements.ContainsKey(from))
+            {
+                replacements[from] = to;
+            }
+            else
+            {
+                replacements.Add(from, to);
+            }
+        }
+
+        public void Remove(string from)
+        {
+            replacements.Remove(from);
+        }
+
+        public string Apply(string msg)
+        {
+            var result = msg;
+            foreach (var pair in  replacements)
+            {
+                result.Replace(pair.Key, pair.Value);
+            }
+            return result;
+        }
+    }
+}

Tools/BiblioTech/Rewards/RoleDriver.cs

@@ -14,12 +14,12 @@ public class RoleDriver : IDiscordRoleDriver
         private readonly ChainEventsSender eventsSender;
         private readonly RewardRepo repo = new RewardRepo();
 
-        public RoleDriver(DiscordSocketClient client, ILog log)
+        public RoleDriver(DiscordSocketClient client, ILog log, CustomReplacement replacement)
         {
             this.client = client;
             this.log = log;
             rewardsChannel = GetChannel(Program.Config.RewardsChannelId);
-            eventsSender = new ChainEventsSender(log, GetChannel(Program.Config.ChainEventsChannelId));
+            eventsSender = new ChainEventsSender(log, replacement, GetChannel(Program.Config.ChainEventsChannelId));
         }
 
         public async Task GiveRewards(GiveRewardsCommand rewards)