aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Common/Plugins/PluginController.cs
blob: 1f83d485b0f4a671b23f10d1fae46a781dbfde50 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Threading.Tasks;
using MediaBrowser.Common.Kernel;

namespace MediaBrowser.Common.Plugins
{
    /// <summary>
    /// Manages Plugins within the PluginsPath directory
    /// </summary>
    public class PluginController
    {
        public string PluginsPath { get; set; }

        /// <summary>
        /// Gets the list of currently loaded plugins
        /// </summary>
        public IEnumerable<IPlugin> Plugins { get; private set; }
        
        /// <summary>
        /// Initializes the controller
        /// </summary>
        public void Init(KernelContext context)
        {
            AppDomain.CurrentDomain.AssemblyResolve -= new ResolveEventHandler(CurrentDomain_AssemblyResolve);
            AppDomain.CurrentDomain.AssemblyResolve += new ResolveEventHandler(CurrentDomain_AssemblyResolve);

            Plugins = GetAllPlugins();

            Parallel.For(0, Plugins.Count(), i =>
            {
                Plugins.ElementAt(i).Init();
            });
        }

        /// <summary>
        /// Gets all plugins within PluginsPath
        /// </summary>
        /// <returns></returns>
        private IEnumerable<IPlugin> GetAllPlugins()
        {
            if (!Directory.Exists(PluginsPath))
            {
                Directory.CreateDirectory(PluginsPath);
            }

            List<IPlugin> plugins = new List<IPlugin>();

            foreach (string folder in Directory.GetDirectories(PluginsPath, "*", SearchOption.TopDirectoryOnly))
            {
                IPlugin plugin = GetPluginFromDirectory(folder);

                plugin.Path = folder;

                if (plugin != null)
                {
                    plugins.Add(plugin);
                }
            }

            return plugins;
        }

        private IPlugin GetPluginFromDirectory(string path)
        {
            string dll = Directory.GetFiles(path, "*.dll", SearchOption.TopDirectoryOnly).FirstOrDefault();

            if (!string.IsNullOrEmpty(dll))
            {
                return GetPluginFromDll(dll);
            }

            return null;
        }

        private IPlugin GetPluginFromDll(string path)
        {
            return GetPluginFromDll(Assembly.Load(File.ReadAllBytes(path)));
        }

        private IPlugin GetPluginFromDll(Assembly assembly)
        {
            var plugin = assembly.GetTypes().Where(type => typeof(IPlugin).IsAssignableFrom(type)).FirstOrDefault();

            if (plugin != null)
            {
                return plugin.GetConstructor(Type.EmptyTypes).Invoke(null) as IPlugin;
            }

            return null;
        }

        Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args)
        {
            AssemblyName assemblyName = new AssemblyName(args.Name);

            IEnumerable<string> dllPaths = Directory.GetFiles(PluginsPath, "*.dll", SearchOption.AllDirectories);

            string dll = dllPaths.FirstOrDefault(f => Path.GetFileNameWithoutExtension(f) == assemblyName.Name);

            if (!string.IsNullOrEmpty(dll))
            {
                return Assembly.Load(File.ReadAllBytes(dll));
            }

            return null;
        }
    }
}