/// <summary>Creates patches by manually specifying the methods</summary>
/// <param name="original">The original method</param>
/// <param name="prefix">An optional prefix method wrapped in a HarmonyMethod object</param>
/// <param name="postfix">An optional postfix method wrapped in a HarmonyMethod object</param>
/// <param name="transpiler">An optional transpiler method wrapped in a HarmonyMethod object</param>
/// <param name="finalizer">An optional finalizer method wrapped in a HarmonyMethod object</param>
/// <returns>The dynamic method that was created to patch the original method</returns>
///
public DynamicMethod Patch(MethodBase original, HarmonyMethod prefix = null, HarmonyMethod postfix = null, HarmonyMethod transpiler = null, HarmonyMethod finalizer = null)
{
var processor = new PatchProcessor(this, original);
processor.AddPrefix(prefix);
processor.AddPostfix(postfix);
processor.AddTranspiler(transpiler);
processor.AddFinalizer(finalizer);
return(processor.Patch().FirstOrDefault());
}
/// <summary>Creates patches by manually specifying the methods</summary>
/// <param name="original">The original method</param>
/// <param name="prefix">An optional prefix method wrapped in a HarmonyMethod object</param>
/// <param name="postfix">An optional postfix method wrapped in a HarmonyMethod object</param>
/// <param name="transpiler">An optional transpiler method wrapped in a HarmonyMethod object</param>
/// <param name="finalizer">An optional finalizer method wrapped in a HarmonyMethod object</param>
/// <returns>The dynamic method that was created to patch the original method</returns>
///
public DynamicMethod Patch(MethodBase original, HarmonyMethod prefix = null, HarmonyMethod postfix = null,
HarmonyMethod transpiler = null, HarmonyMethod finalizer = null)
{
// TODO: Figure out allow MethodInfo return type (maybe shimming?)
var processor = new PatchProcessor(this, original);
processor.AddPrefix(prefix);
processor.AddPostfix(postfix);
processor.AddTranspiler(transpiler);
processor.AddFinalizer(finalizer);
return(processor.Patch().FirstOrDefault());
}
/// <summary>
/// Applies all patches specified in the type.
/// </summary>
/// <param name="type">The type to scan.</param>
public void PatchAll(Type type)
{
if (type == null)
{
throw new ArgumentNullException(nameof(type));
}
foreach (var method in type.GetMethods(BindingFlags.Static | BindingFlags.Public | BindingFlags.NonPublic))
{
var patchAttributeMethods = HarmonyMethodExtensions.GetFromMethod(method);
if (patchAttributeMethods != null && patchAttributeMethods.Any())
{
var attributes = method.GetCustomAttributes(true);
var combinedInfo = HarmonyMethod.Merge(patchAttributeMethods);
var completeMethods = patchAttributeMethods.Where(x => x.declaringType != null && x.methodName != null).ToList();
if (patchAttributeMethods.All(x => x.declaringType != combinedInfo.declaringType && x.methodName != combinedInfo.methodName))
{
completeMethods.Add(combinedInfo);
}
var originalMethods = new List <MethodBase>();
foreach (var methodToPatch in completeMethods)
{
foreach (var index in attributes.OfType <ParameterByRefAttribute>().SelectMany(x => x.ParameterIndices))
{
if (!methodToPatch.argumentTypes[index].IsByRef)
{
methodToPatch.argumentTypes[index] = methodToPatch.argumentTypes[index].MakeByRefType();
}
}
if (!methodToPatch.methodType.HasValue)
{
methodToPatch.methodType = MethodType.Normal;
}
if (methodToPatch.method == null)
{
methodToPatch.method = method;
}
var originalMethod = PatchProcessor.GetOriginalMethod(methodToPatch);
if (originalMethod != null)
{
originalMethods.Add(originalMethod);
}
}
var processor = new PatchProcessor(this);
foreach (var originalMethod in originalMethods)
{
processor.AddOriginal(originalMethod);
}
if (attributes.Any(x => x is HarmonyPrefix))
{
processor.AddPrefix(new HarmonyMethod(method));
}
if (attributes.Any(x => x is HarmonyTranspiler))
{
processor.AddTranspiler(new HarmonyMethod(method));
}
if (attributes.Any(x => x is HarmonyPostfix))
{
processor.AddPostfix(new HarmonyMethod(method));
}
if (attributes.Any(x => x is HarmonyFinalizer))
{
processor.AddFinalizer(new HarmonyMethod(method));
}
processor.Patch();
}
else
{
// Only check when logging warnings
if ((Logger.ChannelFilter & Logger.LogChannel.Warn) != 0)
{
if (method.GetCustomAttributes(typeof(HarmonyAttribute), true).Any())
{
Logger.LogText(Logger.LogChannel.Warn, "Method " + method.FullDescription() + " has an invalid combination of Harmony attributes and will be ignored");
}
}
}
}
}
