DSP_Mods/UXAssist/Common/PatchImpl.cs

using System;
using System.Linq;
using System.Reflection;
using HarmonyLib;

namespace UXAssist.Common;

[AttributeUsage(AttributeTargets.Class, Inherited = false)]
public class PatchGuidAttribute(string guid) : Attribute
{
    public string Guid { get; } = guid;
}

public enum PatchCallbackFlag
{
    // By default, OnEnable() is called After patch applied, set this flag to call it before patch is applied
    CallOnEnableBeforePatch,
    // By default, OnDisable() is called Before patch removed, set this flag to call it after patch is removed
    CallOnDisableAfterUnpatch,
}

[AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = true)]
public class PatchSetCallbackFlagAttribute(PatchCallbackFlag flag) : Attribute
{
    public PatchCallbackFlag Flag { get; } = flag;
}

public class PatchImpl<T> where T : PatchImpl<T>, new()
{
    protected static T Instance { get; } = new();

    protected Harmony _patch;

    public static void Enable(bool enable)
    {
        var thisInstance = Instance;
        if (enable)
        {
            if (thisInstance._patch != null) return;
            var guid = typeof(T).GetCustomAttribute<PatchGuidAttribute>()?.Guid ?? $"PatchImpl.{typeof(T).FullName ?? typeof(T).ToString()}";
            var callOnEnableBefore = typeof(T).GetCustomAttributes<PatchSetCallbackFlagAttribute>().Any(n => n.Flag == PatchCallbackFlag.CallOnEnableBeforePatch);
            if (callOnEnableBefore) thisInstance.OnEnable();
            thisInstance._patch = Harmony.CreateAndPatchAll(typeof(T), guid);
            if (!callOnEnableBefore) thisInstance.OnEnable();
            return;
        }
        if (thisInstance._patch == null) return;
        var callOnDisableAfter = typeof(T).GetCustomAttributes<PatchSetCallbackFlagAttribute>().Any(n => n.Flag == PatchCallbackFlag.CallOnDisableAfterUnpatch);
        if (!callOnDisableAfter) thisInstance.OnDisable();
        thisInstance._patch.UnpatchSelf();
        thisInstance._patch = null;
        if (callOnDisableAfter) thisInstance.OnDisable();
    }

    public static Harmony GetHarmony() => Instance._patch;

    protected virtual void OnEnable() { }
    protected virtual void OnDisable() { }
}