using System; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Linq; using System.Net.Http; using System.Reflection; using System.Threading.Tasks; using Discord.WebSocket; using PluginManager.Interfaces; using PluginManager.Loaders; using PluginManager.Online; using PluginManager.Others; using OperatingSystem = PluginManager.Others.OperatingSystem; namespace PluginManager.Items; public class ConsoleCommandsHandler { private static readonly PluginsManager manager = new("https://raw.githubusercontent.com/Wizzy69/installer/discord-bot-files/Plugins.txt"); private static readonly List commandList = new(); private static bool isDownloading; private static bool pluginsLoaded; private readonly DiscordSocketClient? client; public ConsoleCommandsHandler(DiscordSocketClient client) { this.client = client; InitializeBasicCommands(); //Console.WriteLine("Initialized console command handler !"); } private void InitializeBasicCommands() { commandList.Clear(); AddCommand("help", "Show help", "help ", args => { if (args.Length <= 1) { Console.WriteLine("Available commands:"); var items = new List(); items.Add(new[] { "-", "-", "-" }); items.Add(new[] { "Command", "Description", "Usage" }); items.Add(new[] { " ", " ", "Argument type: [required]" }); items.Add(new[] { "-", "-", "-" }); foreach (var command in commandList) { var pa = from p in command.Action.Method.GetParameters() where p.Name != null select p.ParameterType.FullName; items.Add(new[] { command.CommandName, command.Description, command.Usage }); } items.Add(new[] { "-", "-", "-" }); Console_Utilities.FormatAndAlignTable(items, TableFormat.DEFAULT); } else { foreach (var command in commandList) if (command.CommandName == args[1]) { Console.WriteLine("Command description: " + command.Description); Console.WriteLine("Command execution format:" + command.Usage); return; } Console.WriteLine("Command not found"); } } ); AddCommand("lp", "Load plugins", () => { if (pluginsLoaded) return; var loader = new PluginLoader(client!); var cc = Console.ForegroundColor; loader.onCMDLoad += (name, typeName, success, exception) => { if (name == null || name.Length < 2) name = typeName; if (success) { Console.ForegroundColor = ConsoleColor.Green; Console.WriteLine("[CMD] Successfully loaded command : " + name); } else { Console.ForegroundColor = ConsoleColor.Red; if (exception is null) Console.WriteLine("An error occured while loading: " + name); else Console.WriteLine("[CMD] Failed to load command : " + name + " because " + exception!.Message); } Console.ForegroundColor = cc; }; loader.onEVELoad += (name, typeName, success, exception) => { if (name == null || name.Length < 2) name = typeName; if (success) { Console.ForegroundColor = ConsoleColor.Green; Console.WriteLine("[EVENT] Successfully loaded event : " + name); } else { Console.ForegroundColor = ConsoleColor.Red; Console.WriteLine("[EVENT] Failed to load event : " + name + " because " + exception!.Message); } Console.ForegroundColor = cc; }; loader.LoadPlugins(); Console.ForegroundColor = cc; pluginsLoaded = true; } ); AddCommand("listplugs", "list available plugins", () => { manager.ListAvailablePlugins().Wait(); }); AddCommand("dwplug", "download plugin", "dwplug [name]", async args => { isDownloading = true; if (args.Length == 1) { isDownloading = false; Console.WriteLine("Please specify plugin name"); return; } var name = string.Join(' ', args, 1, args.Length - 1); // info[0] = plugin type // info[1] = plugin link // info[2] = if others are required, or string.Empty if none var info = await manager.GetPluginLinkByName(name); if (info[1] == null) // link is null { if (name == "") { isDownloading = false; Console_Utilities.WriteColorText("Name is invalid"); return; } isDownloading = false; Console_Utilities.WriteColorText($"Failed to find plugin &b{name} &c!" + " Use &glistplugs &ccommand to display all available plugins !"); return; } string path; if (info[0] == "Command" || info[0] == "Event") path = "./Data/Plugins/" + info[0] + "s/" + name + "." + (info[0] == "Command" ? PluginLoader.pluginCMDExtension : PluginLoader.pluginEVEExtension); else path = $"./{info[1].Split('/')[info[1].Split('/').Length - 1]}"; if (OperatingSystem.WINDOWS == Functions.GetOperatingSystem()) { await ServerCom.DownloadFileAsync(info[1], path); } else if (OperatingSystem.LINUX == Functions.GetOperatingSystem()) { var bar = new Console_Utilities.ProgressBar(ProgressBarType.NO_END); bar.Start(); await ServerCom.DownloadFileNoProgressAsync(info[1], path); bar.Stop("Plugin Downloaded !"); } if (info[0] == "Event") Config.PluginConfig.InstalledPlugins.Add(new Tuple(name, PluginType.Event)); else if (info[0] == "Command") Config.PluginConfig.InstalledPlugins.Add(new Tuple(name, PluginType.Command)); Console.WriteLine("\n"); // check requirements if any if (info.Length == 3 && info[2] != string.Empty && info[2] != null) { Console.WriteLine($"Downloading requirements for plugin : {name}"); var lines = await ServerCom.ReadTextFromURL(info[2]); foreach (var line in lines) { if (!(line.Length > 0 && line.Contains(","))) continue; var split = line.Split(','); Console.WriteLine($"\nDownloading item: {split[1]}"); if (File.Exists("./" + split[1])) File.Delete("./" + split[1]); if (OperatingSystem.WINDOWS == Functions.GetOperatingSystem()) { await ServerCom.DownloadFileAsync(split[0], "./" + split[1]); } else if (OperatingSystem.LINUX == Functions.GetOperatingSystem()) { var bar = new Console_Utilities.ProgressBar(ProgressBarType.NO_END); bar.Start(); await ServerCom.DownloadFileNoProgressAsync(split[0], "./" + split[1]); bar.Stop("Item downloaded !"); } Console.WriteLine(); if (split[0].EndsWith(".pak")) { File.Move("./" + split[1], "./Data/PAKS/" + split[1], true); } else if (split[0].EndsWith(".zip") || split[0].EndsWith(".pkg")) { Console.WriteLine($"Extracting {split[1]} ..."); var bar = new Console_Utilities.ProgressBar( ProgressBarType.NO_END); // { Max = 100f, Color = ConsoleColor.Green }; bar.Start(); await Functions.ExtractArchive("./" + split[1], "./", null, UnzipProgressType.PercentageFromTotalSize); bar.Stop("Extracted"); Console.WriteLine("\n"); File.Delete("./" + split[1]); } } Console.WriteLine(); } var ver = await ServerCom.GetVersionOfPackageFromWeb(name); if (ver is null) throw new Exception("Incorrect version"); Config.SetPluginVersion( name, $"{ver.PackageVersionID}.{ver.PackageMainVersion}.{ver.PackageCheckVersion}"); isDownloading = false; } ); AddCommand("value", "read value from VariableStack", "value [key]", args => { if (args.Length != 2) return; if (!Config.ContainsKey(args[1])) return; var data = Config.GetValue(args[1]); Console.WriteLine($"{args[1]} => {data}"); } ); AddCommand("add", "add variable to the system variables", "add [key] [value] [isReadOnly=true/false]", args => { if (args.Length < 4) return; var key = args[1]; var value = args[2]; var isReadOnly = args[3].Equals("true", StringComparison.CurrentCultureIgnoreCase); try { Config.GetAndAddValueToVariable(key, value, isReadOnly); Console.WriteLine($"Updated config file with the following command: {args[1]} => {value}"); } catch (Exception ex) { Console.WriteLine(ex.ToString()); } } ); AddCommand("remv", "remove variable from system variables", "remv [key]", args => { if (args.Length < 2) return; Config.RemoveKey(args[1]); } ); AddCommand("sd", "Shuts down the discord bot", async () => { if (client is null) return; var bar = new Console_Utilities.ProgressBar(ProgressBarType.NO_END); bar.Start(); await Config.SaveConfig(SaveType.NORMAL); await Config.SaveConfig(SaveType.BACKUP); bar.Stop("Saved config !"); Console.WriteLine(); await client.StopAsync(); await client.DisposeAsync(); await Task.Delay(1000); Environment.Exit(0); } ); AddCommand("import", "Load an external command", "import [pluginName]", async args => { if (args.Length <= 1) return; try { var pName = string.Join(' ', args, 1, args.Length - 1); var client = new HttpClient(); var url = (await manager.GetPluginLinkByName(pName))[1]; if (url is null) throw new Exception($"Invalid plugin name {pName}."); var s = await client.GetStreamAsync(url); var str = new MemoryStream(); await s.CopyToAsync(str); var asmb = Assembly.Load(str.ToArray()); var types = asmb.GetTypes(); foreach (var type in types) if (type.IsClass && typeof(DBEvent).IsAssignableFrom(type)) { var instance = (DBEvent)Activator.CreateInstance(type); instance.Start(this.client); Console.WriteLine($"[EVENT] Loaded external {type.FullName}!"); } else if (type.IsClass && typeof(DBCommand).IsAssignableFrom(type)) { var instance = (DBCommand)Activator.CreateInstance(type); Console.WriteLine($"[CMD] Instance: {type.FullName} loaded !"); } } catch (Exception ex) { Console.WriteLine(ex.Message); } }); AddCommand("remplug", "Remove a plugin", "remplug [plugName]", async args => { if (args.Length <= 1) return; isDownloading = true; var plugName = string.Join(' ', args, 1, args.Length - 1); if (pluginsLoaded) { if (Functions.GetOperatingSystem() == OperatingSystem.WINDOWS) { Process.Start("DiscordBot.exe", $"/remplug {plugName}"); await Task.Delay(100); Environment.Exit(0); } else { Process.Start("./DiscordBot", $"/remplug {plugName}"); await Task.Delay(100); Environment.Exit(0); } isDownloading = false; return; } var location = "./Data/Plugins/"; location = Config.PluginConfig.GetPluginType(plugName) switch { PluginType.Command => location + "Commands/" + plugName + "." + PluginLoader.pluginCMDExtension, PluginType.Event => location + "Events/" + plugName + "." + PluginLoader.pluginEVEExtension, PluginType.Unknown => "./", _ => throw new NotImplementedException("Plugin type incorrect") }; if (!File.Exists(location)) { Console.WriteLine("The plugin does not exist"); return; } File.Delete(location); if (Config.PluginConfig.Contains(plugName)) { var tuple = Config.PluginConfig.InstalledPlugins.Where(t => t.Item1 == plugName).FirstOrDefault(); Console.WriteLine("Found: " + tuple); Config.PluginConfig.InstalledPlugins.Remove(tuple); Config.RemovePluginVersion(plugName); await Config.SaveConfig(SaveType.NORMAL); } Console.WriteLine("Removed the plugin DLL. Checking for other files ..."); var info = await manager.GetPluginLinkByName(plugName); if (info[2] != string.Empty) { var lines = await ServerCom.ReadTextFromURL(info[2]); foreach (var line in lines) { if (!(line.Length > 0 && line.Contains(","))) continue; var split = line.Split(','); if (File.Exists("./" + split[1])) File.Delete("./" + split[1]); Console.WriteLine("Removed: " + split[1]); } if (Directory.Exists(plugName)) Directory.Delete(plugName, true); } isDownloading = false; Console.WriteLine(plugName + " has been successfully deleted !"); }); AddCommand("reload", "Reload the bot with all plugins", () => { if (Functions.GetOperatingSystem() == OperatingSystem.WINDOWS) { Process.Start("DiscordBot.exe", "lp"); HandleCommand("sd"); } else { Process.Start("./DiscordBot", "lp"); HandleCommand("sd"); } }); //AddCommand(""); //Sort the commands by name commandList.Sort((x, y) => x.CommandName.CompareTo(y.CommandName)); } public static void AddCommand(string command, string description, string usage, Action action) { commandList.Add(new ConsoleCommand { CommandName = command, Description = description, Action = action, Usage = usage }); Console.ForegroundColor = ConsoleColor.White; Console_Utilities.WriteColorText($"Command &r{command} &cadded to the list of commands"); } public static void AddCommand(string command, string description, Action action) { AddCommand(command, description, command, args => action()); } public static void RemoveCommand(string command) { commandList.RemoveAll(x => x.CommandName == command); } public static bool CommandExists(string command) { return GetCommand(command) is not null; } public static ConsoleCommand? GetCommand(string command) { return commandList.FirstOrDefault(t => t.CommandName == command); } public static async Task ExecuteCommad(string command) { var args = command.Split(' '); // Console.WriteLine(command); foreach (var item in commandList.ToList()) if (item.CommandName == args[0]) { item.Action.Invoke(args); Console.WriteLine(); while (isDownloading) await Task.Delay(1000); } } public bool HandleCommand(string command, bool removeCommandExecution = true) { Console.ForegroundColor = ConsoleColor.White; var args = command.Split(' '); foreach (var item in commandList.ToList()) if (item.CommandName == args[0]) { if (removeCommandExecution) { Console.SetCursorPosition(0, Console.CursorTop - 1); for (var i = 0; i < command.Length + 30; i++) Console.Write(" "); Console.SetCursorPosition(0, Console.CursorTop); } Console.WriteLine(); item.Action(args); return true; } return false; //Console.WriteLine($"Executing: {args[0]} with the following parameters: {args.MergeStrings(1)}"); } }