|
| 1 | +using System.Reflection; |
| 2 | +using System.Text; |
| 3 | +using Godot; |
| 4 | +using HarmonyLib; |
| 5 | +using FileAccess = System.IO.FileAccess; |
| 6 | + |
| 7 | +namespace STS2RitsuLib.Diagnostics |
| 8 | +{ |
| 9 | + /// <summary> |
| 10 | + /// Writes a UTF-8 text report of all Harmony-patched methods (similar to standalone dump mods). |
| 11 | + /// </summary> |
| 12 | + internal static class HarmonyPatchDumpWriter |
| 13 | + { |
| 14 | + /// <summary> |
| 15 | + /// Resolves <c>user://</c> / <c>res://</c> via Godot and returns an absolute filesystem path. |
| 16 | + /// </summary> |
| 17 | + internal static string? TryResolveFilesystemPath(string rawPath) |
| 18 | + { |
| 19 | + if (string.IsNullOrWhiteSpace(rawPath)) |
| 20 | + return null; |
| 21 | + |
| 22 | + var trimmed = rawPath.Trim(); |
| 23 | + if (trimmed.StartsWith("user://", StringComparison.OrdinalIgnoreCase) || |
| 24 | + trimmed.StartsWith("res://", StringComparison.OrdinalIgnoreCase)) |
| 25 | + return ProjectSettings.GlobalizePath(trimmed); |
| 26 | + |
| 27 | + return Path.GetFullPath(trimmed); |
| 28 | + } |
| 29 | + |
| 30 | + internal static bool TryWrite(string filesystemPath, out string? errorMessage) |
| 31 | + { |
| 32 | + errorMessage = null; |
| 33 | + try |
| 34 | + { |
| 35 | + var dir = Path.GetDirectoryName(filesystemPath); |
| 36 | + if (!string.IsNullOrEmpty(dir)) |
| 37 | + Directory.CreateDirectory(dir); |
| 38 | + |
| 39 | + using var fileStream = |
| 40 | + new FileStream(filesystemPath, FileMode.Create, FileAccess.Write, FileShare.Read); |
| 41 | + using var streamWriter = new StreamWriter(fileStream, Encoding.UTF8); |
| 42 | + WriteReport(streamWriter); |
| 43 | + return true; |
| 44 | + } |
| 45 | + catch (Exception ex) |
| 46 | + { |
| 47 | + errorMessage = ex.Message; |
| 48 | + return false; |
| 49 | + } |
| 50 | + } |
| 51 | + |
| 52 | + private static void WriteReport(StreamWriter streamWriter) |
| 53 | + { |
| 54 | + streamWriter.WriteLine("======================================================="); |
| 55 | + streamWriter.WriteLine("=== Harmony Patch Dump Report ==="); |
| 56 | + streamWriter.WriteLine("======================================================="); |
| 57 | + streamWriter.WriteLine($"Generated at: {DateTime.Now:O}"); |
| 58 | + streamWriter.WriteLine($"User data dir: {OS.GetUserDataDir()}"); |
| 59 | + streamWriter.WriteLine("======================================================="); |
| 60 | + streamWriter.WriteLine(); |
| 61 | + |
| 62 | + var allPatchedMethods = Harmony.GetAllPatchedMethods() |
| 63 | + .OrderBy(m => m.DeclaringType?.FullName ?? "Unknown") |
| 64 | + .ThenBy(m => m.Name) |
| 65 | + .ToList(); |
| 66 | + |
| 67 | + var methodCount = 0; |
| 68 | + var totalPrefixes = 0; |
| 69 | + var totalPostfixes = 0; |
| 70 | + var totalTranspilers = 0; |
| 71 | + var totalFinalizers = 0; |
| 72 | + |
| 73 | + foreach (var patchedMethod in allPatchedMethods) |
| 74 | + { |
| 75 | + methodCount++; |
| 76 | + var counts = LogPatchedMethodInfo(patchedMethod, streamWriter); |
| 77 | + totalPrefixes += counts.prefixes; |
| 78 | + totalPostfixes += counts.postfixes; |
| 79 | + totalTranspilers += counts.transpilers; |
| 80 | + totalFinalizers += counts.finalizers; |
| 81 | + streamWriter.WriteLine(); |
| 82 | + } |
| 83 | + |
| 84 | + streamWriter.WriteLine("======================================================="); |
| 85 | + streamWriter.WriteLine("=== Summary ==="); |
| 86 | + streamWriter.WriteLine("======================================================="); |
| 87 | + streamWriter.WriteLine($"Total Patched Methods: {methodCount}"); |
| 88 | + streamWriter.WriteLine($" - Prefix patches: {totalPrefixes}"); |
| 89 | + streamWriter.WriteLine($" - Postfix patches: {totalPostfixes}"); |
| 90 | + streamWriter.WriteLine($" - Transpiler patches: {totalTranspilers}"); |
| 91 | + streamWriter.WriteLine($" - Finalizer patches: {totalFinalizers}"); |
| 92 | + streamWriter.WriteLine( |
| 93 | + $" - Total patches: {totalPrefixes + totalPostfixes + totalTranspilers + totalFinalizers}"); |
| 94 | + streamWriter.WriteLine("======================================================="); |
| 95 | + } |
| 96 | + |
| 97 | + private static (int prefixes, int postfixes, int transpilers, int finalizers) LogPatchedMethodInfo( |
| 98 | + MethodBase methodBase, StreamWriter streamWriter) |
| 99 | + { |
| 100 | + var patchInfo = Harmony.GetPatchInfo(methodBase); |
| 101 | + if (patchInfo == null) return (0, 0, 0, 0); |
| 102 | + |
| 103 | + var declaringType = methodBase.DeclaringType?.FullName ?? "Unknown"; |
| 104 | + var methodSignature = GetMethodSignature(methodBase); |
| 105 | + var returnType = methodBase is MethodInfo mi ? mi.ReturnType.Name : "void"; |
| 106 | + |
| 107 | + streamWriter.WriteLine($"┌─ [{declaringType}]"); |
| 108 | + streamWriter.WriteLine($"│ Method: {returnType} {methodSignature}"); |
| 109 | + streamWriter.WriteLine("│"); |
| 110 | + |
| 111 | + var prefixCount = 0; |
| 112 | + var postfixCount = 0; |
| 113 | + var transpilerCount = 0; |
| 114 | + var finalizerCount = 0; |
| 115 | + |
| 116 | + if (patchInfo.Prefixes.Count > 0) |
| 117 | + { |
| 118 | + streamWriter.WriteLine($"│ ├─ Prefixes ({patchInfo.Prefixes.Count}):"); |
| 119 | + foreach (var patch in patchInfo.Prefixes.OrderBy(p => p.priority).ThenBy(p => p.owner)) |
| 120 | + { |
| 121 | + streamWriter.WriteLine($"│ │ {FormatPatchInfo(patch)}"); |
| 122 | + prefixCount++; |
| 123 | + } |
| 124 | + } |
| 125 | + |
| 126 | + if (patchInfo.Postfixes.Count > 0) |
| 127 | + { |
| 128 | + streamWriter.WriteLine($"│ ├─ Postfixes ({patchInfo.Postfixes.Count}):"); |
| 129 | + foreach (var patch in patchInfo.Postfixes.OrderBy(p => p.priority).ThenBy(p => p.owner)) |
| 130 | + { |
| 131 | + streamWriter.WriteLine($"│ │ {FormatPatchInfo(patch)}"); |
| 132 | + postfixCount++; |
| 133 | + } |
| 134 | + } |
| 135 | + |
| 136 | + if (patchInfo.Transpilers.Count > 0) |
| 137 | + { |
| 138 | + streamWriter.WriteLine($"│ ├─ Transpilers ({patchInfo.Transpilers.Count}):"); |
| 139 | + foreach (var patch in patchInfo.Transpilers.OrderBy(p => p.priority).ThenBy(p => p.owner)) |
| 140 | + { |
| 141 | + streamWriter.WriteLine($"│ │ {FormatPatchInfo(patch)}"); |
| 142 | + transpilerCount++; |
| 143 | + } |
| 144 | + } |
| 145 | + |
| 146 | + if (patchInfo.Finalizers.Count > 0) |
| 147 | + { |
| 148 | + streamWriter.WriteLine($"│ └─ Finalizers ({patchInfo.Finalizers.Count}):"); |
| 149 | + foreach (var patch in patchInfo.Finalizers.OrderBy(p => p.priority).ThenBy(p => p.owner)) |
| 150 | + { |
| 151 | + streamWriter.WriteLine($"│ {FormatPatchInfo(patch)}"); |
| 152 | + finalizerCount++; |
| 153 | + } |
| 154 | + } |
| 155 | + |
| 156 | + streamWriter.WriteLine("└─────────────────────────────────────────────────────────────────"); |
| 157 | + |
| 158 | + return (prefixCount, postfixCount, transpilerCount, finalizerCount); |
| 159 | + } |
| 160 | + |
| 161 | + private static string GetMethodSignature(MethodBase methodBase) |
| 162 | + { |
| 163 | + var parameters = methodBase.GetParameters(); |
| 164 | + var paramString = string.Join(", ", parameters.Select(p => $"{p.ParameterType.Name} {p.Name}")); |
| 165 | + return $"{methodBase.Name}({paramString})"; |
| 166 | + } |
| 167 | + |
| 168 | + private static string FormatPatchInfo(Patch patch) |
| 169 | + { |
| 170 | + var sb = new StringBuilder(); |
| 171 | + sb.Append($"├─ [Priority: {patch.priority}] "); |
| 172 | + sb.Append($"[{patch.owner}] "); |
| 173 | + var patchClass = patch.PatchMethod.DeclaringType?.FullName ?? "Unknown"; |
| 174 | + var patchMethodName = patch.PatchMethod.Name; |
| 175 | + sb.Append($"{patchClass}.{patchMethodName}"); |
| 176 | + |
| 177 | + try |
| 178 | + { |
| 179 | + var moduleName = Path.GetFileName(patch.PatchMethod.Module.FullyQualifiedName); |
| 180 | + if (!string.IsNullOrEmpty(moduleName) && moduleName != "<Unknown>") |
| 181 | + sb.Append($" (from {moduleName})"); |
| 182 | + } |
| 183 | + catch |
| 184 | + { |
| 185 | + // ignored |
| 186 | + } |
| 187 | + |
| 188 | + return sb.ToString(); |
| 189 | + } |
| 190 | + } |
| 191 | +} |
0 commit comments