205 lines
7.2 KiB
C#
205 lines
7.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Threading.Tasks;
|
|
using DiscordBot.Utilities;
|
|
using PluginManager;
|
|
using PluginManager.Interfaces;
|
|
using PluginManager.Loaders;
|
|
using PluginManager.Online;
|
|
using PluginManager.Others;
|
|
using Spectre.Console;
|
|
|
|
namespace DiscordBot.Bot.Actions.Extra;
|
|
|
|
internal static class PluginMethods
|
|
{
|
|
private static readonly PluginsManager PluginsManager = new();
|
|
|
|
internal static async Task List()
|
|
{
|
|
var data = await ConsoleUtilities.ExecuteWithProgressBar(PluginsManager.GetAvailablePlugins(), "Loading plugins...");
|
|
|
|
TableData tableData = new(new List<string> { "Name", "Description", "Type", "Version" });
|
|
foreach (var plugin in data) tableData.AddRow(plugin);
|
|
|
|
tableData.HasRoundBorders = false;
|
|
tableData.PrintAsTable();
|
|
}
|
|
|
|
internal static async Task RefreshPlugins(bool quiet)
|
|
{
|
|
await Program.internalActionManager.Execute("plugin", "load", quiet ? "-q" : string.Empty);
|
|
await Program.internalActionManager.Refresh();
|
|
}
|
|
|
|
internal static async Task DownloadPlugin(PluginsManager manager, string pluginName)
|
|
{
|
|
var pluginData = await manager.GetPluginLinkByName(pluginName);
|
|
if (pluginData.Length == 0)
|
|
{
|
|
Console.WriteLine($"Plugin {pluginName} not found. Please check the spelling and try again.");
|
|
return;
|
|
}
|
|
|
|
var pluginType = pluginData[0];
|
|
var pluginLink = pluginData[1];
|
|
var pluginRequirements = pluginData[2];
|
|
|
|
|
|
await AnsiConsole.Progress()
|
|
.Columns(new ProgressColumn[]
|
|
{
|
|
new TaskDescriptionColumn(),
|
|
new ProgressBarColumn(),
|
|
new PercentageColumn()
|
|
})
|
|
.StartAsync(async ctx =>
|
|
{
|
|
var downloadTask = ctx.AddTask("Downloading plugin...");
|
|
|
|
IProgress<float> progress = new Progress<float>(p => { downloadTask.Value = p; });
|
|
|
|
await ServerCom.DownloadFileAsync(pluginLink, $"./Data/{pluginType}s/{pluginName}.dll", progress);
|
|
|
|
downloadTask.Increment(100);
|
|
|
|
ctx.Refresh();
|
|
});
|
|
|
|
if (pluginRequirements == string.Empty)
|
|
{
|
|
Console.WriteLine("Finished installing " + pluginName + " successfully");
|
|
await RefreshPlugins(false);
|
|
return;
|
|
}
|
|
|
|
List<string> requirementsUrLs = new();
|
|
|
|
await AnsiConsole.Progress()
|
|
.Columns(new ProgressColumn[]
|
|
{
|
|
new TaskDescriptionColumn(),
|
|
new ProgressBarColumn(),
|
|
new PercentageColumn()
|
|
})
|
|
.StartAsync(async ctx =>
|
|
{
|
|
var gatherInformationTask = ctx.AddTask("Gathering info...");
|
|
gatherInformationTask.IsIndeterminate = true;
|
|
requirementsUrLs = await ServerCom.ReadTextFromURL(pluginRequirements);
|
|
await Task.Delay(2000);
|
|
gatherInformationTask.Increment(100);
|
|
});
|
|
|
|
await AnsiConsole.Progress()
|
|
.Columns(new ProgressColumn[]
|
|
{
|
|
new TaskDescriptionColumn(),
|
|
new ProgressBarColumn(),
|
|
new PercentageColumn()
|
|
})
|
|
.StartAsync(async ctx =>
|
|
{
|
|
List<Tuple<ProgressTask, IProgress<float>, Task>> downloadTasks = new();
|
|
|
|
foreach (var info in requirementsUrLs)
|
|
{
|
|
if (info.Length < 2) continue;
|
|
string[] data = info.Split(',');
|
|
string url = data[0];
|
|
string fileName = data[1];
|
|
|
|
var task = ctx.AddTask($"Downloading {fileName}...");
|
|
IProgress<float> progress = new Progress<float>(p =>
|
|
{
|
|
task.Value = p;
|
|
});
|
|
|
|
var downloadTask = ServerCom.DownloadFileAsync(url, $"./{fileName}", progress);
|
|
downloadTasks.Add(new Tuple<ProgressTask, IProgress<float>, Task>(task, progress, downloadTask));
|
|
}
|
|
|
|
foreach (var task in downloadTasks)
|
|
{
|
|
await task.Item3;
|
|
}
|
|
|
|
});
|
|
|
|
await RefreshPlugins(false);
|
|
}
|
|
|
|
internal static async Task<bool> LoadPlugins(string[] args)
|
|
{
|
|
var loader = new PluginLoader(Config.DiscordBot.client);
|
|
if (args.Length == 2 && args[1] == "-q")
|
|
{
|
|
loader.LoadPlugins();
|
|
return true;
|
|
}
|
|
|
|
var cc = Console.ForegroundColor;
|
|
loader.onCMDLoad += (name, typeName, success, exception) =>
|
|
{
|
|
if (name == null || name.Length < 2)
|
|
name = typeName;
|
|
if (success)
|
|
{
|
|
Config.Logger.Log("Successfully loaded command : " + name, source: typeof(ICommandAction),
|
|
type: LogType.INFO);
|
|
}
|
|
|
|
else
|
|
{
|
|
Config.Logger.Log("Failed to load command : " + name + " because " + exception?.Message,
|
|
source: typeof(ICommandAction), type: LogType.ERROR);
|
|
}
|
|
|
|
Console.ForegroundColor = cc;
|
|
};
|
|
loader.onEVELoad += (name, typeName, success, exception) =>
|
|
{
|
|
if (name == null || name.Length < 2)
|
|
name = typeName;
|
|
|
|
if (success)
|
|
{
|
|
Config.Logger.Log("Successfully loaded event : " + name, source: typeof(ICommandAction),
|
|
type: LogType.INFO);
|
|
}
|
|
else
|
|
{
|
|
Config.Logger.Log("Failed to load event : " + name + " because " + exception?.Message,
|
|
source: typeof(ICommandAction), type: LogType.ERROR);
|
|
}
|
|
|
|
Console.ForegroundColor = cc;
|
|
};
|
|
|
|
loader.onSLSHLoad += (name, typeName, success, exception) =>
|
|
{
|
|
if (name == null || name.Length < 2)
|
|
name = typeName;
|
|
|
|
if (success)
|
|
{
|
|
Config.Logger.Log("Successfully loaded slash command : " + name, source: typeof(ICommandAction),
|
|
type: LogType.INFO);
|
|
}
|
|
else
|
|
{
|
|
Config.Logger.Log("Failed to load slash command : " + name + " because " + exception?.Message,
|
|
source: typeof(ICommandAction), type: LogType.ERROR);
|
|
}
|
|
|
|
Console.ForegroundColor = cc;
|
|
};
|
|
|
|
loader.LoadPlugins();
|
|
Console.ForegroundColor = cc;
|
|
return true;
|
|
}
|
|
|
|
|
|
}
|