Adds several new scripts

PkmnLib.Dynamic/ScriptHandling/ScriptSet.cs

@@ -1,4 +1,5 @@
 using System.Collections;
+using PkmnLib.Dynamic.ScriptHandling.Registry;
 using PkmnLib.Static.Utils;
 
 namespace PkmnLib.Dynamic.ScriptHandling;
@@ -26,7 +27,12 @@ public interface IScriptSet : IEnumerable<ScriptContainer>
     /// Gets a script from the set using its unique name.
     /// </summary>
     ScriptContainer? Get(StringKey scriptKey);
-
+    
+    /// <summary>
+    /// Gets a script from the set using its type.
+    /// </summary>
+    T? Get<T>() where T : Script;
+    
     /// <summary>
     /// Removes a script from the set using its unique name.
     /// </summary>
@@ -107,6 +113,9 @@ public class ScriptSet : IScriptSet
     /// <inheritdoc />
     public ScriptContainer? Get(StringKey scriptKey) => _scripts.FirstOrDefault(s => s.Script?.Name == scriptKey);
 
+    /// <inheritdoc />
+    public T? Get<T>() where T : Script => Get(ScriptUtils.ResolveName<T>())?.Script as T;
+
     /// <inheritdoc />
     public void Remove(StringKey scriptKey)
     {

Plugins/PkmnLib.Plugin.Gen7/Scripts/Moves/DoublePowerIfTargetDamagedInTurn.cs

@@ -28,11 +28,12 @@ public class DoublePowerIfTargetDamagedInTurn : Script
         var battle = move.User.BattleData?.Battle;
         if (battle == null)
             return;
-        var side = battle.Sides[target.BattleData!.SideIndex];
-        var script = side.VolatileScripts.Get(ScriptUtils.ResolveName<DoublePowerIfTargetDamagedInTurnData>());
-        if (script?.Script == null)
+        if (target.BattleData == null) 
+            return;
+        var side = battle.Sides[target.BattleData.SideIndex];
+        var data = side.VolatileScripts.Get<DoublePowerIfTargetDamagedInTurnData>();
+        if (data == null)
             return;
-        var data = (DoublePowerIfTargetDamagedInTurnData)script.Script;
         if (data._hitPokemon.Contains(target))
             basePower *= 2;
     }

Plugins/PkmnLib.Plugin.Gen7/Scripts/Moves/Drain.cs

@@ -0,0 +1,34 @@
+using System.Collections.Generic;
+using PkmnLib.Dynamic.Libraries;
+using PkmnLib.Dynamic.Models;
+using PkmnLib.Dynamic.ScriptHandling;
+using PkmnLib.Dynamic.ScriptHandling.Registry;
+using PkmnLib.Static.Utils;
+
+namespace PkmnLib.Plugin.Gen7.Scripts.Moves;
+
+[Script(ScriptCategory.Move, "drain")]
+public class Drain : Script
+{
+    public float DrainModifier { get; set; } = 0.5f;
+    
+    /// <inheritdoc />
+    public override void OnInitialize(IDynamicLibrary library, IReadOnlyDictionary<StringKey, object?>? parameters)
+    {
+        base.OnInitialize(library, parameters);
+        if (parameters == null)
+            return;
+        DrainModifier = parameters.GetValueOrDefault("drain_modifier") as float? ?? DrainModifier;
+    }
+
+    /// <inheritdoc />
+    public override void OnSecondaryEffect(IExecutingMove move, IPokemon target, byte hit)
+    {
+        var user = move.User;
+        var damage = move.GetHitData(target, hit).Damage;
+        var healed = (uint)(damage * DrainModifier);
+        if (move.User.HasHeldItem("big_root"))
+            healed = (uint)(healed * 1.3f);
+        user.Heal(healed, false);
+    }
+}

Plugins/PkmnLib.Plugin.Gen7/Scripts/Moves/Flinch.cs

@@ -0,0 +1,16 @@
+using PkmnLib.Dynamic.Models;
+using PkmnLib.Dynamic.ScriptHandling;
+using PkmnLib.Dynamic.ScriptHandling.Registry;
+using PkmnLib.Plugin.Gen7.Scripts.Pokemon;
+
+namespace PkmnLib.Plugin.Gen7.Scripts.Moves;
+
+[Script(ScriptCategory.Move, "flinch")]
+public class Flinch : Script
+{
+    /// <inheritdoc />
+    public override void OnSecondaryEffect(IExecutingMove move, IPokemon target, byte hit)
+    {
+        target.Volatile.Add(new FlinchEffect());
+    }
+}

Plugins/PkmnLib.Plugin.Gen7/Scripts/Pokemon/FlinchEffect.cs

@@ -0,0 +1,16 @@
+using PkmnLib.Dynamic.Models;
+using PkmnLib.Dynamic.ScriptHandling;
+using PkmnLib.Dynamic.ScriptHandling.Registry;
+
+namespace PkmnLib.Plugin.Gen7.Scripts.Pokemon;
+
+[Script(ScriptCategory.Pokemon, "flinch_effect")]
+public class FlinchEffect : Script
+{
+    /// <inheritdoc />
+    public override void PreventMove(IExecutingMove move, ref bool prevent)
+    {
+        prevent = true;
+        RemoveSelf();
+    }
+}