DamageAPI: Adding support for ChainGunOrb risk-of-thunder#395

KingEnderBrine · Jun 3, 2022 · 6cf664a · 6cf664a
1 parent 46f8ce3
commit 6cf664a
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/R2API/DamageAPI.cs b/R2API/DamageAPI.cs
@@ -49,6 +49,7 @@ internal static void SetHooks() {
             IL.RoR2.Orbs.DamageOrb.OnArrival += DamageOrbOnArrivalIL;
             IL.RoR2.Orbs.GenericDamageOrb.OnArrival += GenericDamageOrbOnArrivalIL;
             IL.RoR2.Orbs.LightningOrb.OnArrival += LightningOrbOnArrivalIL;
+            IL.RoR2.Orbs.ChainGunOrb.OnArrival += ChainGunOrbOnArrivalIL;
 
             IL.RoR2.Projectile.DeathProjectile.FixedUpdate += DeathProjectileFixedUpdateIL;
             IL.RoR2.Projectile.ProjectileDotZone.ResetOverlap += ProjectileDotZoneResetOverlapIL;
@@ -99,6 +100,7 @@ internal static void UnsetHooks() {
             IL.RoR2.Orbs.DamageOrb.OnArrival -= DamageOrbOnArrivalIL;
             IL.RoR2.Orbs.GenericDamageOrb.OnArrival -= GenericDamageOrbOnArrivalIL;
             IL.RoR2.Orbs.LightningOrb.OnArrival -= LightningOrbOnArrivalIL;
+            IL.RoR2.Orbs.ChainGunOrb.OnArrival -= ChainGunOrbOnArrivalIL;
 
             IL.RoR2.Projectile.DeathProjectile.FixedUpdate -= DeathProjectileFixedUpdateIL;
             IL.RoR2.Projectile.ProjectileDotZone.ResetOverlap -= ProjectileDotZoneResetOverlapIL;
@@ -222,6 +224,26 @@ private static void DamageOrbOnArrivalIL(ILContext il) {
 
             EmitCopyCall(c);
         }
+
+        private static void ChainGunOrbOnArrivalIL(ILContext il) {
+            var c = new ILCursor(il);
+            var damageInfoIndex = GotoDamageInfo(c);
+
+            c.Emit(OpCodes.Ldarg_0);
+            c.Emit(OpCodes.Ldloc, damageInfoIndex);
+
+            EmitCopyCall(c);
+
+            var chainGunOrbIndex = -1;
+            c.GotoNext(
+                x => x.MatchNewobj<ChainGunOrb>(),
+                x => x.MatchStloc(out chainGunOrbIndex));
+
+            c.Emit(OpCodes.Ldarg_0);
+            c.Emit(OpCodes.Ldloc, chainGunOrbIndex);
+
+            EmitCopyCall(c);
+        }
         #endregion
 
         #region Projectiles