Update to AutoCorrect Showdown

- Shows what was corrected.
bdawg1989 · Jun 18, 2024 · 7642af4 · 7642af4
1 parent 030a9bb
commit 7642af4
Show file tree

Hide file tree

Showing 3 changed files with 221 additions and 73 deletions.
diff --git a/SysBot.Pokemon.Discord/Commands/Bots/TradeModule.cs b/SysBot.Pokemon.Discord/Commands/Bots/TradeModule.cs
@@ -401,12 +401,24 @@ public async Task HideTradeAsync([Summary("Trade Code")] int code, [Summary("Sho
                 // Perform auto correct if it's on and send that shit through again
                 if (SysCord<T>.Runner.Config.Trade.AutoCorrectConfig.EnableAutoCorrect && !la.Valid)
                 {
-                    var correctedContent = await AutoCorrectShowdown<T>.PerformAutoCorrect(content, pkm, la);
+                    var (correctedContent, correctionMessages) = await AutoCorrectShowdown<T>.PerformAutoCorrect(content, pkm, la);
                     set = new ShowdownSet(correctedContent);
                     template = AutoLegalityWrapper.GetTemplate(set);
                     pkm = sav.GetLegal(template, out result);
                     la = new LegalityAnalysis(pkm);
                     setEdited = true;
+
+                    if (correctionMessages.Count > 0)
+                    {
+                        var userName = Context.User.Mention;
+                        var changesEmbed = new EmbedBuilder()
+                            .WithTitle("Showdown Set Corrections")
+                            .WithColor(Color.Orange)
+                            .WithDescription(string.Join("\n", correctionMessages))
+                            .Build();
+
+                        await ReplyAsync($"{userName}, here are the corrections made to your Showdown set:", embed: changesEmbed).ConfigureAwait(false);
+                    }
                 }
 
                 if (pkm is not T correctedPk || !la.Valid)
@@ -579,12 +591,24 @@ public async Task TradeAsync([Summary("Trade Code")] int code, [Summary("Showdow
                 // Perform auto correct if it's on and send that shit through again
                 if (SysCord<T>.Runner.Config.Trade.AutoCorrectConfig.EnableAutoCorrect && !la.Valid)
                 {
-                    var correctedContent = await AutoCorrectShowdown<T>.PerformAutoCorrect(content, pkm, la);
+                    var (correctedContent, correctionMessages) = await AutoCorrectShowdown<T>.PerformAutoCorrect(content, pkm, la);
                     set = new ShowdownSet(correctedContent);
                     template = AutoLegalityWrapper.GetTemplate(set);
                     pkm = sav.GetLegal(template, out result);
                     la = new LegalityAnalysis(pkm);
                     setEdited = true;
+
+                    if (correctionMessages.Count > 0)
+                    {
+                        var userName = Context.User.Mention;
+                        var changesEmbed = new EmbedBuilder()
+                            .WithTitle("Showdown Set Corrections")
+                            .WithColor(Color.Orange)
+                            .WithDescription(string.Join("\n", correctionMessages))
+                            .Build();
+
+                        await ReplyAsync($"{userName}, here are the corrections made to your Showdown set:", embed: changesEmbed).ConfigureAwait(false);
+                    }
                 }
 
                 if (pkm is not T correctedPk || !la.Valid)
@@ -860,12 +884,24 @@ private async Task ProcessSingleTradeAsync(string tradeContent, int batchTradeCo
                 // Perform auto correct if it's on and send that shit through again
                 if (SysCord<T>.Runner.Config.Trade.AutoCorrectConfig.EnableAutoCorrect && !la.Valid)
                 {
-                    var correctedContent = await AutoCorrectShowdown<T>.PerformAutoCorrect(tradeContent, pkm, la);
+                    var (correctedContent, correctionMessages) = await AutoCorrectShowdown<T>.PerformAutoCorrect(tradeContent, pkm, la);
                     set = new ShowdownSet(correctedContent);
                     template = AutoLegalityWrapper.GetTemplate(set);
                     pkm = sav.GetLegal(template, out result);
                     la = new LegalityAnalysis(pkm);
                     setEdited = true;
+
+                    if (correctionMessages.Count > 0)
+                    {
+                        var userName = Context.User.Mention;
+                        var changesEmbed = new EmbedBuilder()
+                            .WithTitle("Showdown Set Corrections")
+                            .WithColor(Color.Orange)
+                            .WithDescription(string.Join("\n", correctionMessages))
+                            .Build();
+
+                        await ReplyAsync($"{userName}, here are the corrections made to your Showdown set:", embed: changesEmbed).ConfigureAwait(false);
+                    }
                 }
 
                 if (pkm is not T correctedPk || !la.Valid)