84 lines
2.7 KiB
C#
84 lines
2.7 KiB
C#
using DiscordBotCore;
|
|
using DiscordBotCore.Bot;
|
|
using DiscordBotCore.Loaders;
|
|
using DiscordBotCore.Others;
|
|
using DiscordBotCore.Others.Exceptions;
|
|
|
|
namespace DiscordBotWebUI.DiscordBot;
|
|
|
|
public class DiscordBotStartup
|
|
{
|
|
public required Func<ModuleRequirement, Task> RequirementsSolver { get; set; }
|
|
|
|
internal delegate void LogEventHandler(string message, LogType type);
|
|
internal LogEventHandler Log;
|
|
|
|
private void WriteLog(string message, LogType logType)
|
|
{
|
|
Log?.Invoke(message, logType);
|
|
}
|
|
|
|
public async Task CreateApplication()
|
|
{
|
|
await Application.CreateApplication(RequirementsSolver);
|
|
}
|
|
|
|
public bool LoadComponents()
|
|
{
|
|
Application.Logger.SetOutFunction(WriteLog);
|
|
|
|
return Application.CurrentApplication.ApplicationEnvironmentVariables.ContainsKey("ServerID") &&
|
|
Application.CurrentApplication.ApplicationEnvironmentVariables.ContainsKey("token") &&
|
|
Application.CurrentApplication.ApplicationEnvironmentVariables.ContainsKey("prefix");
|
|
|
|
}
|
|
|
|
public async Task PrepareBot()
|
|
{
|
|
var token = Application.CurrentApplication.ApplicationEnvironmentVariables.Get<string>("token");
|
|
var prefix = Application.CurrentApplication.ApplicationEnvironmentVariables.Get<string>("prefix");
|
|
|
|
DiscordBotApplication app = new DiscordBotApplication(token, prefix);
|
|
await app.StartAsync();
|
|
}
|
|
|
|
public async Task RefreshPlugins(bool quiet)
|
|
{
|
|
await LoadPlugins(quiet ? ["-q"] : null);
|
|
await InitializeInternalActionManager();
|
|
}
|
|
|
|
private async Task LoadPlugins(string[]? args)
|
|
{
|
|
var loader = new PluginLoader(Application.CurrentApplication.DiscordBotClient.Client);
|
|
if (args is not null && args.Contains("-q"))
|
|
{
|
|
await loader.LoadPlugins();
|
|
}
|
|
|
|
loader.OnCommandLoaded += (command) => {
|
|
Application.Logger.Log($"Command {command.Command} loaded successfully", LogType.Info);
|
|
};
|
|
|
|
loader.OnEventLoaded += (eEvent) => {
|
|
Application.Logger.Log($"Event {eEvent.Name} loaded successfully", LogType.Info);
|
|
};
|
|
|
|
loader.OnActionLoaded += (action) => {
|
|
Application.Logger.Log($"Action {action.ActionName} loaded successfully", LogType.Info);
|
|
};
|
|
|
|
loader.OnSlashCommandLoaded += (slashCommand) => {
|
|
Application.Logger.Log($"Slash Command {slashCommand.Name} loaded successfully", LogType.Info);
|
|
};
|
|
|
|
await loader.LoadPlugins();
|
|
}
|
|
|
|
private async Task InitializeInternalActionManager()
|
|
{
|
|
await Application.CurrentApplication.InternalActionManager.Initialize();
|
|
}
|
|
|
|
}
|