Differentiate between MethodReplacer and ConstructorReplacer

pardeike · Jul 21, 2018 · a1cf7fc · a1cf7fc
1 parent 3b5f6da
commit a1cf7fc
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/Harmony/Transpilers.cs b/Harmony/Transpilers.cs
@@ -7,7 +7,7 @@ namespace Harmony
 {
 	public static class Transpilers
 	{
-		public static IEnumerable<CodeInstruction> MethodReplacer(this IEnumerable<CodeInstruction> instructions, MethodBase from, MethodBase to)
+		private static IEnumerable<CodeInstruction> Replacer(IEnumerable<CodeInstruction> instructions, MethodBase from, MethodBase to, OpCode opcode)
 		{
 			if (from == null)
 				throw new ArgumentException("Unexpected null argument", nameof(from));
@@ -19,13 +19,23 @@ public static IEnumerable<CodeInstruction> MethodReplacer(this IEnumerable<CodeI
 				var method = instruction.operand as MethodBase;
 				if (method == from)
 				{
-					instruction.opcode = OpCodes.Call;
+					instruction.opcode = opcode;
 					instruction.operand = to;
 				}
 				yield return instruction;
 			}
 		}
 
+		public static IEnumerable<CodeInstruction> MethodReplacer(this IEnumerable<CodeInstruction> instructions, MethodBase from, MethodBase to)
+		{
+			return Replacer(instructions, from, to, OpCodes.Call);
+		}
+
+		public static IEnumerable<CodeInstruction> ConstructorReplacer(this IEnumerable<CodeInstruction> instructions, MethodBase from, MethodBase to)
+		{
+			return Replacer(instructions, from, to, OpCodes.Newobj);
+		}
+
 		public static IEnumerable<CodeInstruction> DebugLogger(this IEnumerable<CodeInstruction> instructions, string text)
 		{
 			yield return new CodeInstruction(OpCodes.Ldstr, text);