Moved source code under src folder - massive change

2026-04-25 22:36:59 -04:00 · 2013-10-02 18:01:32 -07:00
parent 2fc8123d6b
commit 5bf0e197ec
1499 changed files with 1054 additions and 1444 deletions
@@ -0,0 +1,76 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
+using NzbDrone.Common.EnvironmentInfo;
+using NzbDrone.Common.Messaging;
+using TinyIoC;
+
+
+namespace NzbDrone.Common.Composition
+{
+    public abstract class ContainerBuilderBase
+    {
+        private readonly List<Type> _loadedTypes;
+
+        public IContainer Container { get; private set; }
+
+        protected ContainerBuilderBase(IStartupArguments args, params string[] assemblies)
+        {
+
+
+            _loadedTypes = new List<Type>();
+
+            foreach (var assembly in assemblies)
+            {
+                _loadedTypes.AddRange(Assembly.Load(assembly).GetTypes());
+            }
+
+            Container = new Container(new TinyIoCContainer(), _loadedTypes);
+            AutoRegisterInterfaces();
+            Container.Register(args);
+        }
+
+        private void AutoRegisterInterfaces()
+        {
+            var loadedInterfaces = _loadedTypes.Where(t => t.IsInterface).ToList();
+            var implementedInterfaces = _loadedTypes.SelectMany(t => t.GetInterfaces());
+
+            var contracts = loadedInterfaces.Union(implementedInterfaces).Where(c => !c.IsGenericTypeDefinition && !string.IsNullOrWhiteSpace(c.FullName))
+                .Where(c => !c.FullName.StartsWith("System"))
+                .Except(new List<Type> { typeof(IMessage), typeof(IEvent), typeof(IContainer) }).Distinct().OrderBy(c => c.FullName);
+
+            foreach (var contract in contracts)
+            {
+                AutoRegisterImplementations(contract);
+            }
+        }
+
+        protected void AutoRegisterImplementations<TContract>()
+        {
+            AutoRegisterImplementations(typeof(TContract));
+        }
+
+        private void AutoRegisterImplementations(Type contractType)
+        {
+            var implementations = Container.GetImplementations(contractType).Where(c => !c.IsGenericTypeDefinition).ToList();
+
+
+
+            if (implementations.Count == 0)
+            {
+                return;
+            }
+            if (implementations.Count == 1)
+            {
+                var impl = implementations.Single();
+                Container.RegisterSingleton(contractType, impl);
+            }
+            else
+            {
+                Container.RegisterAllAsSingleton(contractType, implementations);
+            }
+        }
+
+    }
+}