aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Common/Plugins/PluginController.cs
blob: c26275436bbc1cab239351e5006dd546dee7a2ac (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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
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<BasePlugin> 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 =>
            {
                var plugin = Plugins.ElementAt(i);

                plugin.ReloadConfiguration();

                if (plugin.Enabled)
                {
                    if (context == KernelContext.Server)
                    {
                        plugin.InitInServer();
                    }
                    else
                    {
                        plugin.InitInUI();
                    }
                }
            });
        }

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

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

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

                plugin.Path = folder;

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

            return plugins;
        }

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

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

            return null;
        }

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

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

            if (plugin != null)
            {
                BasePlugin instance = plugin.GetConstructor(Type.EmptyTypes).Invoke(null) as BasePlugin;

                instance.Version = assembly.GetName().Version;

                return instance;
            }

            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;
        }
    }
}