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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Runtime.Loader;
using Impostor.Api.Plugins;
using Impostor.Server.Utils;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.FileSystemGlobbing;
using Microsoft.Extensions.Hosting;
using Serilog;
namespace Impostor.Server.Plugins
{
public static class PluginLoader
{
private static readonly ILogger Logger = Log.ForContext(typeof(PluginLoader));
public static IHostBuilder UsePluginLoader(this IHostBuilder builder, PluginConfig config)
{
var assemblyInfos = new List<IAssemblyInformation>();
var context = AssemblyLoadContext.Default;
// Add the plugins and libraries.
var pluginPaths = new List<string>(config.Paths);
var libraryPaths = new List<string>(config.LibraryPaths);
var rootFolder = Directory.GetCurrentDirectory();
pluginPaths.Add(Path.Combine(rootFolder, "plugins"));
libraryPaths.Add(Path.Combine(rootFolder, "libraries"));
var matcher = new Matcher(StringComparison.OrdinalIgnoreCase);
matcher.AddInclude("*.dll");
matcher.AddExclude("Impostor.Api.dll");
RegisterAssemblies(pluginPaths, matcher, assemblyInfos, true);
RegisterAssemblies(libraryPaths, matcher, assemblyInfos, false);
// Register the resolver to the current context.
// TODO: Move this to a new context so we can unload/reload plugins.
context.Resolving += (loadContext, name) =>
{
Logger.Verbose("Loading assembly {0} v{1}", name.Name, name.Version);
// Some plugins may be referencing another Impostor.Api version and try to load it.
// We want to only use the one shipped with the server.
if (name.Name.Equals("Impostor.Api"))
{
return typeof(IPlugin).Assembly;
}
var info = assemblyInfos.FirstOrDefault(a => a.AssemblyName.Name == name.Name);
return info?.Load(loadContext);
};
// TODO: Catch uncaught exceptions.
var assemblies = assemblyInfos
.Where(a => a.IsPlugin)
.Select(a => context.LoadFromAssemblyName(a.AssemblyName))
.ToList();
// Find all plugins.
var plugins = new List<PluginInformation>();
foreach (var assembly in assemblies)
{
// Find plugin startup.
var pluginStartup = assembly
.GetTypes()
.Where(t => typeof(IPluginStartup).IsAssignableFrom(t) && t.IsClass)
.ToList();
if (pluginStartup.Count > 1)
{
Logger.Warning("A plugin may only define zero or one IPluginStartup implementation ({0}).", assembly);
continue;
}
// Find plugin.
var plugin = assembly
.GetTypes()
.Where(t => typeof(IPlugin).IsAssignableFrom(t)
&& t.IsClass
&& !t.IsAbstract
&& t.GetCustomAttribute<ImpostorPluginAttribute>() != null)
.ToList();
if (plugin.Count != 1)
{
Logger.Warning("A plugin must define exactly one IPlugin or PluginBase implementation ({0}).", assembly);
continue;
}
// Save plugin.
plugins.Add(new PluginInformation(
pluginStartup
.Select(Activator.CreateInstance)
.Cast<IPluginStartup>()
.FirstOrDefault(),
plugin.First()));
}
foreach (var plugin in plugins.Where(plugin => plugin.Startup != null))
{
plugin.Startup.ConfigureHost(builder);
}
builder.ConfigureServices(services =>
{
services.AddHostedService(provider => ActivatorUtilities.CreateInstance<PluginLoaderService>(provider, plugins));
foreach (var plugin in plugins.Where(plugin => plugin.Startup != null))
{
plugin.Startup.ConfigureServices(services);
}
});
return builder;
}
private static void RegisterAssemblies(
IEnumerable<string> paths,
Matcher matcher,
ICollection<IAssemblyInformation> assemblyInfos,
bool isPlugin)
{
foreach (var path in paths.SelectMany(matcher.GetResultsInFullPath))
{
AssemblyName assemblyName;
try
{
assemblyName = AssemblyName.GetAssemblyName(path);
}
catch (BadImageFormatException)
{
continue;
}
assemblyInfos.Add(new AssemblyInformation(assemblyName, path, isPlugin));
}
}
}
}
|
