Initial Commit

2026-03-16 14:38:46 +02:00
commit b8f7327a21
2327 changed files with 253610 additions and 0 deletions
@@ -0,0 +1,74 @@
+#if !NOT_UNITY3D
+
+using System;
+using System.Collections.Generic;
+using ModestTree;
+
+namespace Zenject
+{
+    [NoReflectionBaking]
+    public class GetFromPrefabComponentProvider : IProvider
+    {
+        readonly IPrefabInstantiator _prefabInstantiator;
+        readonly Type _componentType;
+        readonly bool _matchSingle;
+
+        // if concreteType is null we use the contract type from inject context
+        public GetFromPrefabComponentProvider(
+            Type componentType,
+            IPrefabInstantiator prefabInstantiator, bool matchSingle)
+        {
+            _prefabInstantiator = prefabInstantiator;
+            _componentType = componentType;
+            _matchSingle = matchSingle;
+        }
+
+        public bool IsCached
+        {
+            get { return false; }
+        }
+
+        public bool TypeVariesBasedOnMemberType
+        {
+            get { return false; }
+        }
+
+        public Type GetInstanceType(InjectContext context)
+        {
+            return _componentType;
+        }
+
+        public void GetAllInstancesWithInjectSplit(
+            InjectContext context, List<TypeValuePair> args, out Action injectAction, List<object> buffer)
+        {
+            Assert.IsNotNull(context);
+
+            var gameObject = _prefabInstantiator.Instantiate(context, args, out injectAction);
+
+            // NOTE: Need to set includeInactive to true here, because prefabs are always
+            // instantiated as disabled until injection occurs, so that Awake / OnEnabled is executed
+            // after injection has occurred
+
+            if (_matchSingle)
+            {
+                var match = gameObject.GetComponentInChildren(_componentType, true);
+
+                Assert.IsNotNull(match, "Could not find component with type '{0}' on prefab '{1}'",
+                _componentType, _prefabInstantiator.GetPrefab(context).name);
+
+                buffer.Add(match);
+                return;
+            }
+
+            var allComponents = gameObject.GetComponentsInChildren(_componentType, true);
+
+            Assert.That(allComponents.Length >= 1,
+                "Expected to find at least one component with type '{0}' on prefab '{1}'",
+                _componentType, _prefabInstantiator.GetPrefab(context).name);
+
+            buffer.AllocFreeAddRange(allComponents);
+        }
+    }
+}
+
+#endif