Implementation of Pokeballs

PkmnLib.Dynamic/ScriptHandling/Registry/ScriptRegistry.cs

@@ -2,6 +2,7 @@ using System.Linq.Expressions;
 using System.Reflection;
 using JetBrains.Annotations;
 using PkmnLib.Dynamic.Libraries;
+using PkmnLib.Static;
 using PkmnLib.Static.Utils;
 
 namespace PkmnLib.Dynamic.ScriptHandling.Registry;
@@ -13,10 +14,11 @@ namespace PkmnLib.Dynamic.ScriptHandling.Registry;
 public class ScriptRegistry
 {
     private readonly Dictionary<(ScriptCategory category, StringKey name), Func<Script>> _scriptTypes = new();
-    private readonly Dictionary<StringKey, Func<ItemScript>> _itemScriptTypes = new();
+    private readonly Dictionary<StringKey, Func<IItem, ItemScript>> _itemScriptTypes = new();
     private IBattleStatCalculator? _battleStatCalculator;
     private IDamageCalculator? _damageCalculator;
     private IMiscLibrary? _miscLibrary;
+    private ICaptureLibrary? _captureLibrary;
 
     /// <summary>
     /// Automatically register all scripts in the given assembly that have the <see cref="ScriptAttribute"/>, and
@@ -73,13 +75,15 @@ public class ScriptRegistry
             throw new ArgumentNullException(nameof(type));
 
         var constructor = type.GetConstructor(BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance,
-            null, Type.EmptyTypes, null);
+            null, [typeof(IItem)], null);
         if (constructor == null)
-            throw new ArgumentException($"Type {type} does not have a parameterless constructor.");
+            throw new ArgumentException($"Type {type} does not have a constructor that takes an IItem.");
 
         // We create a lambda that creates a new instance of the script type.
         // This is more performant than using Activator.CreateInstance.
-        _itemScriptTypes[name] = Expression.Lambda<Func<ItemScript>>(Expression.New(constructor)).Compile();
+        var parameterExpression = Expression.Parameter(typeof(IItem), "item");
+        var newExpression = Expression.New(constructor, parameterExpression);
+        _itemScriptTypes[name] = Expression.Lambda<Func<IItem, ItemScript>>(newExpression, parameterExpression).Compile();
     }
 
     /// <summary>
@@ -99,10 +103,17 @@ public class ScriptRegistry
     /// </summary>
     public void RegisterMiscLibrary<T>(T miscLibrary) where T : IMiscLibrary
         => _miscLibrary = miscLibrary;
+    
+    /// <summary>
+    /// Register a capture library.
+    /// </summary>
+    public void RegisterCaptureLibrary<T>(T captureLibrary) where T : ICaptureLibrary
+        => _captureLibrary = captureLibrary;
 
     internal IReadOnlyDictionary<(ScriptCategory category, StringKey name), Func<Script>> ScriptTypes => _scriptTypes;
-    internal IReadOnlyDictionary<StringKey, Func<ItemScript>> ItemScriptTypes => _itemScriptTypes;
+    internal IReadOnlyDictionary<StringKey, Func<IItem, ItemScript>> ItemScriptTypes => _itemScriptTypes;
     internal IBattleStatCalculator? BattleStatCalculator => _battleStatCalculator;
     internal IDamageCalculator? DamageCalculator => _damageCalculator;
     internal IMiscLibrary? MiscLibrary => _miscLibrary;
+    internal ICaptureLibrary? CaptureLibrary => _captureLibrary;
 }