add harmony_summary command

2 files changed, 168 insertions, 0 deletions

diff --git a/src/SMAPI/Framework/Commands/HarmonySummaryCommand.cs b/src/SMAPI/Framework/Commands/HarmonySummaryCommand.cs
new file mode 100644
index 00000000..bc8f4aa2
--- /dev/null
+++ b/src/SMAPI/Framework/Commands/HarmonySummaryCommand.cs
@@ -0,0 +1,167 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
+using System.Text;
+using HarmonyLib;
+
+namespace StardewModdingAPI.Framework.Commands
+{
+    /// <summary>The 'harmony_summary' SMAPI console command.</summary>
+    internal class HarmonySummaryCommand : IInternalCommand
+    {
+        /*********
+        ** Accessors
+        *********/
+        /// <summary>The command name, which the user must type to trigger it.</summary>
+        public string Name { get; } = "harmony_summary";
+
+        /// <summary>The human-readable documentation shown when the player runs the built-in 'help' command.</summary>
+        public string Description { get; } = "Harmony is a library which rewrites game code, used by SMAPI and some mods. This command lists current Harmony patches.\n\nUsage: harmony_summary\nList all Harmony patches.\n\nUsage: harmony_summary <search>\n- search: one more more words to search. If any word matches a method name, the method and all its patchers will be listed; otherwise only matching patchers will be listed for the method.";
+
+
+        /*********
+        ** Public methods
+        *********/
+        /// <summary>Handle the console command when it's entered by the user.</summary>
+        /// <param name="args">The command arguments.</param>
+        /// <param name="monitor">Writes messages to the console.</param>
+        public void HandleCommand(string[] args, IMonitor monitor)
+        {
+            SearchResult[] matches = this.FilterPatches(args).OrderBy(p => p.Method).ToArray();
+
+            StringBuilder result = new StringBuilder();
+
+            if (!matches.Any())
+                result.AppendLine("No current patches match your search.");
+            else
+            {
+                result.AppendLine(args.Any() ? "Harmony patches which match your search terms:" : "Current Harmony patches:");
+                result.AppendLine();
+                foreach (var match in matches)
+                {
+                    result.AppendLine($"   {match.Method}");
+                    foreach (var ownerGroup in match.PatchTypesByOwner)
+                    {
+                        var sortedTypes = ownerGroup.Value
+                            .OrderBy(p => p switch { PatchType.Prefix => 0, PatchType.Postfix => 1, PatchType.Finalizer => 2, PatchType.Transpiler => 3, _ => 4 });
+
+                        result.AppendLine($"      - {ownerGroup.Key} ({string.Join(", ", sortedTypes).ToLower()})");
+                    }
+                }
+            }
+
+            monitor.Log(result.ToString(), LogLevel.Info);
+        }
+
+
+        /*********
+        ** Private methods
+        *********/
+        /// <summary>Get all current Harmony patches matching any of the given search terms.</summary>
+        /// <param name="searchTerms">The search terms to match.</param>
+        private IEnumerable<SearchResult> FilterPatches(string[] searchTerms)
+        {
+            bool hasSearch = searchTerms.Any();
+            bool IsMatch(string target) => searchTerms.Any(search => target != null && target.IndexOf(search, StringComparison.OrdinalIgnoreCase) > -1);
+
+            foreach (var patch in this.GetAllPatches())
+            {
+                if (!hasSearch)
+                    yield return patch;
+
+                // matches entire patch
+                if (IsMatch(patch.Method))
+                {
+                    yield return patch;
+                    continue;
+                }
+
+                // matches individual patchers
+                foreach (var pair in patch.PatchTypesByOwner.ToArray())
+                {
+                    if (!IsMatch(pair.Key) && !pair.Value.Any(type => IsMatch(type.ToString())))
+                        patch.PatchTypesByOwner.Remove(pair.Key);
+                }
+
+                if (patch.PatchTypesByOwner.Any())
+                    yield return patch;
+            }
+        }
+
+        /// <summary>Get all current Harmony patches.</summary>
+        private IEnumerable<SearchResult> GetAllPatches()
+        {
+            foreach (MethodBase method in Harmony.GetAllPatchedMethods())
+            {
+                // get metadata for method
+                string methodLabel = method.FullDescription();
+                HarmonyLib.Patches patchInfo = Harmony.GetPatchInfo(method);
+                IDictionary<PatchType, IReadOnlyCollection<Patch>> patchGroups = new Dictionary<PatchType, IReadOnlyCollection<Patch>>
+                {
+                    [PatchType.Prefix] = patchInfo.Prefixes,
+                    [PatchType.Postfix] = patchInfo.Postfixes,
+                    [PatchType.Finalizer] = patchInfo.Finalizers,
+                    [PatchType.Transpiler] = patchInfo.Transpilers
+                };
+
+                // get patch types by owner
+                var typesByOwner = new Dictionary<string, ISet<PatchType>>();
+                foreach (var group in patchGroups)
+                {
+                    foreach (var patch in group.Value)
+                    {
+                        if (!typesByOwner.TryGetValue(patch.owner, out ISet<PatchType> patchTypes))
+                            typesByOwner[patch.owner] = patchTypes = new HashSet<PatchType>();
+                        patchTypes.Add(group.Key);
+                    }
+                }
+
+                // create search result
+                yield return new SearchResult(methodLabel, typesByOwner);
+            }
+        }
+
+        /// <summary>A Harmony patch type.</summary>
+        private enum PatchType
+        {
+            /// <summary>A prefix patch.</summary>
+            Prefix,
+
+            /// <summary>A postfix patch.</summary>
+            Postfix,
+
+            /// <summary>A finalizer patch.</summary>
+            Finalizer,
+
+            /// <summary>A transpiler patch.</summary>
+            Transpiler
+        }
+
+        /// <summary>A patch search result for a method.</summary>
+        private class SearchResult
+        {
+            /*********
+            ** Accessors
+            *********/
+            /// <summary>A detailed human-readable label for the patched method.</summary>
+            public string Method { get; }
+
+            /// <summary>The patch types by the Harmony instance ID that added them.</summary>
+            public IDictionary<string, ISet<PatchType>> PatchTypesByOwner { get; }
+
+
+            /*********
+            ** Public methods
+            *********/
+            /// <summary>Construct an instance.</summary>
+            /// <param name="method">A detailed human-readable label for the patched method.</param>
+            /// <param name="patchTypesByOwner">The patch types by the Harmony instance ID that added them.</param>
+            public SearchResult(string method, IDictionary<string, ISet<PatchType>> patchTypesByOwner)
+            {
+                this.Method = method;
+                this.PatchTypesByOwner = patchTypesByOwner;
+            }
+        }
+    }
+}
diff --git a/src/SMAPI/Framework/SCore.cs b/src/SMAPI/Framework/SCore.cs
index a89616a3..9d96bad1 100644
--- a/src/SMAPI/Framework/SCore.cs
+++ b/src/SMAPI/Framework/SCore.cs
@@ -511,6 +511,7 @@ namespace StardewModdingAPI.Framework
             this.Monitor.Log("Type 'help' for help, or 'help <cmd>' for a command's usage", LogLevel.Info);
             this.GameInstance.CommandManager
                 .Add(new HelpCommand(this.GameInstance.CommandManager), this.Monitor)
+                .Add(new HarmonySummaryCommand(), this.Monitor)
                 .Add(new ReloadI18nCommand(this.ReloadTranslations), this.Monitor);
 
             // start handling command line input