BotSharp/src/Infrastructure/BotSharp.Core/Plugins/PluginLoader.cs

126 lines
4.1 KiB
C#
Raw Normal View History

2023-10-11 04:11:06 +00:00
using BotSharp.Abstraction.Plugins.Models;
using Microsoft.AspNetCore.Builder;
2023-10-12 11:36:36 +00:00
using Microsoft.EntityFrameworkCore.Metadata.Internal;
2023-10-11 04:11:06 +00:00
using Microsoft.Extensions.Configuration;
using System.Drawing;
using System.IO;
using System.Reflection;
using System.Xml;
namespace BotSharp.Core.Plugins;
public class PluginLoader
{
private readonly IServiceCollection _services;
private readonly IConfiguration _config;
private readonly PluginSettings _settings;
private static List<IBotSharpPlugin> _modules = new List<IBotSharpPlugin>();
private static List<PluginDef> _plugins = new List<PluginDef>();
private static string _executingDir;
public PluginLoader(IServiceCollection services,
IConfiguration config,
PluginSettings settings)
{
_services = services;
_config = config;
_settings = settings;
}
public void Load(Action<Assembly> loaded)
{
_executingDir = Directory.GetParent(Assembly.GetEntryAssembly().Location).FullName;
_settings.Assemblies.ToList().ForEach(assemblyName =>
{
var assemblyPath = Path.Combine(_executingDir, assemblyName + ".dll");
if (File.Exists(assemblyPath))
{
var assembly = Assembly.Load(assemblyName);
var modules = assembly.GetTypes()
.Where(x => x.GetInterface(nameof(IBotSharpPlugin)) != null)
.Select(x => Activator.CreateInstance(x) as IBotSharpPlugin)
.ToList();
foreach (var module in modules)
{
module.RegisterDI(_services, _config);
2023-10-12 11:36:36 +00:00
// string classSummary = GetSummaryComment(module.GetType());
var name = string.IsNullOrEmpty(module.Name) ? module.GetType().Name : module.Name;
2023-10-11 04:11:06 +00:00
_modules.Add(module);
_plugins.Add(new PluginDef
{
2023-11-13 16:47:27 +00:00
Id = module.GetType().FullName,
2023-10-12 11:36:36 +00:00
Name = name,
Description = module.Description,
2023-10-11 04:11:06 +00:00
Assembly = assemblyName
});
Console.Write($"Loaded plugin ");
2023-10-12 11:36:36 +00:00
Console.Write(name, Color.Green);
2023-10-11 04:11:06 +00:00
Console.WriteLine($" from {assemblyName}.");
2023-10-12 11:36:36 +00:00
if (!string.IsNullOrEmpty(module.Description))
2023-10-11 04:11:06 +00:00
{
2023-10-12 11:36:36 +00:00
Console.WriteLine(module.Description);
2023-10-11 04:11:06 +00:00
}
}
loaded(assembly);
}
else
{
Console.WriteLine($"Can't find assemble {assemblyPath}.");
}
});
}
public List<PluginDef> GetPlugins()
{
return _plugins;
}
public string GetSummaryComment(Type member)
{
string summary = string.Empty;
XmlDocument xmlDoc = new XmlDocument();
// Load the XML documentation file
var xmlFile = Path.Combine(_executingDir, $"{member.Module.Assembly.FullName.Split(',')[0]}.xml");
if (!File.Exists(xmlFile))
{
return "";
}
xmlDoc.Load(xmlFile); // Replace with your actual XML documentation file name
// Construct the XPath query to find the summary comment
string memberName = $"T:{member.FullName}";
string xpath = $"/doc/members/member[@name='{memberName}']/summary";
// Find the summary comment using the XPath query
XmlNode summaryNode = xmlDoc.SelectSingleNode(xpath);
if (summaryNode != null)
{
summary = summaryNode.InnerXml.Trim();
}
return summary;
}
public void Configure(IApplicationBuilder app)
{
if (_modules.Count == 0)
{
Console.WriteLine($"No plugin loaded. Please check whether the Load() method is called.", Color.Yellow);
}
_modules.ForEach(module =>
{
if (module.GetType().GetInterface(nameof(IBotSharpAppPlugin)) != null)
{
(module as IBotSharpAppPlugin).Configure(app);
}
});
}
}