[Add:API] Traits.StopEffect

Author: TacoConKvass

ExampleMod/TraitAddition.cs

@@ -27,6 +27,14 @@ void IRegistrable.Register() {
 		Player.PostUpdateStats.Event += PostUpdateStats_Event;
 
 		Traits.ApplyEffect.Event += ApplyEffect_Event;
+
+		Traits.StopEffect.Event += StopEffect_Event;
+	}
+
+	private void StopEffect_Event(TraitType type) {
+		if (type != MyTrait) return;
+
+		Mod.Log("TestTrait stopeed!");
 	}
 
 	void Awake() {

RL2.API/Endpoints/Traits.cs

@@ -6,6 +6,8 @@
 using System.Reflection;
 using Rewired.Utils.Libraries.TinyJson;
 using UnityEngine;
+using MonoMod.Cil;
+using Mono.Cecil.Cil;
 
 namespace RL2.API;
 
@@ -19,6 +21,7 @@ public static class Traits
 		LoadData.Hook,
 		LoadContent.Hook,
 		ApplyEffect.Hook,
+		StopEffect.ILHook,
 		ModifyData.Hook,
 		ModifyTraitObj.Hook,
 		ExtendTypeArray.Hook,
@@ -205,6 +208,39 @@ internal static void Method(Action<TraitManager, TraitType, bool> orig, TraitMan
 		}
 	}
 
+	/// <summary>
+	/// Triggered when a Trait is disabled on player death
+	/// </summary>
+	public static class StopEffect
+	{
+		/// <inheritdoc cref="StopEffect"/>
+		/// <param name="type"></param>
+		public delegate void Definition(TraitType type);
+
+		/// <inheritdoc cref="Definition"/>
+		public static event Definition? Event;
+
+		internal static ILHook ILHook = new ILHook(
+			typeof(PlayerDeathWindowController).GetMethod(nameof(PlayerDeathWindowController.OnOpen), BindingFlags.NonPublic | BindingFlags.Instance),
+			Method,
+			new ILHookConfig() {
+				ID = "RL2.API::Traits.ApplyEffect",
+				ManualApply = true,
+			}
+		);
+
+		internal static void Method(ILContext il) {
+			ILCursor cursor = new ILCursor(il);
+
+			if (cursor.TryGotoNext(MoveType.After, i => i.MatchCallvirt<BaseTrait>(nameof(BaseTrait.DisableOnDeath)))) {
+				cursor.Emit(OpCodes.Ldloc_3);
+				cursor.EmitDelegate((List<BaseTrait>.Enumerator enumerator) => {
+					Event?.Invoke(enumerator.Current.TraitType);
+				});
+			}
+		}
+	}
+
 	/// <summary>
 	/// Allows extending the <see cref="TraitType_RL.TypeArray"/>
 	/// </summary>