PluginManager/Bot/Boot.cs

@@ -0,0 +1,158 @@
+using System.Net.Mime;
+using System;
+using System.Threading.Tasks;
+
+using Discord;
+using Discord.Commands;
+using Discord.WebSocket;
+
+namespace PluginManager.Bot;
+
+public class Boot
+{
+    /// <summary>
+    ///     The bot prefix
+    /// </summary>
+    public readonly string botPrefix;
+
+    /// <summary>
+    ///     The bot token
+    /// </summary>
+    public readonly string botToken;
+
+    /// <summary>
+    ///     The bot client
+    /// </summary>
+    public DiscordSocketClient client;
+
+    /// <summary>
+    ///     The bot command handler
+    /// </summary>
+    private CommandHandler commandServiceHandler;
+
+    /// <summary>
+    ///     The command service
+    /// </summary>
+    private CommandService service;
+
+    /// <summary>
+    ///     The main Boot constructor
+    /// </summary>
+    /// <param name="botToken">The bot token</param>
+    /// <param name="botPrefix">The bot prefix</param>
+    public Boot(string botToken, string botPrefix)
+    {
+        this.botPrefix = botPrefix;
+        this.botToken = botToken;
+    }
+
+
+    /// <summary>
+    ///     Checks if the bot is ready
+    /// </summary>
+    /// <value> true if the bot is ready, othwerwise false </value>
+    public bool isReady { get; private set; }
+
+    /// <summary>
+    ///     The start method for the bot. This method is used to load the bot
+    /// </summary>
+    /// <param name="config">The discord socket config. If null then the default one will be applied (AlwaysDownloadUsers=true, UseInteractionSnowflakeDate=false, GatewayIntents=GatewayIntents.All)</param>
+    /// <returns>Task</returns>
+    public async Task Awake(DiscordSocketConfig? config = null)
+    {
+        if(config is null)
+        config = new DiscordSocketConfig
+        {
+
+            AlwaysDownloadUsers = true,
+
+            //Disable system clock checkup (for responses at slash commands)
+            UseInteractionSnowflakeDate = false,
+
+            GatewayIntents = GatewayIntents.All
+        };
+
+        client = new DiscordSocketClient(config);
+        service = new CommandService();
+
+        CommonTasks();
+
+        await client.LoginAsync(TokenType.Bot, botToken);
+
+        await client.StartAsync();
+
+        commandServiceHandler = new CommandHandler(client, service, botPrefix);
+
+        await commandServiceHandler.InstallCommandsAsync();
+
+
+
+        await Task.Delay(2000);
+        while (!isReady) ;
+    }
+
+    private void CommonTasks()
+    {
+        if (client == null) return;
+        client.LoggedOut += Client_LoggedOut;
+        client.Log += Log;
+        client.LoggedIn += LoggedIn;
+        client.Ready += Ready;
+        client.Disconnected += Client_Disconnected;
+    }
+
+    private async Task Client_Disconnected(Exception arg)
+    {
+        if (arg.Message.Contains("401"))
+        {
+            Config.Variables.RemoveKey("token");
+            Logger.LogError("The token is invalid. Please restart the bot and enter a valid token.");
+            await Task.Delay(4000);
+            Environment.Exit(0);
+        }
+
+        Logger.WriteErrFile(arg);
+    }
+
+    private async Task Client_LoggedOut()
+    {
+        Logger.WriteLine("Successfully Logged Out");
+        await Log(new LogMessage(LogSeverity.Info, "Boot", "Successfully logged out from discord !"));
+    }
+
+    private Task Ready()
+    {
+        Console.Title = "ONLINE";
+        isReady = true;
+        return Task.CompletedTask;
+    }
+
+    private Task LoggedIn()
+    {
+        Console.Title = "CONNECTED";
+        return Task.CompletedTask;
+    }
+
+    private Task Log(LogMessage message)
+    {
+        switch (message.Severity)
+        {
+            case LogSeverity.Error:
+            case LogSeverity.Critical:
+                Logger.WriteErrFile(message.Message);
+                Logger.WriteColored(message.Message + "\n", ConsoleColor.Red);
+
+                break;
+
+            case LogSeverity.Info:
+            case LogSeverity.Debug:
+                Logger.WriteLogFile(message.Message);
+                Logger.WriteColored(message.Message + "\n", ConsoleColor.White);
+
+
+                break;
+        }
+
+        return Task.CompletedTask;
+    }
+}

PluginManager/Bot/CommandHandler.cs

@@ -0,0 +1,160 @@
+using System;
+using System.Linq;
+using System.Reflection;
+using System.Threading.Tasks;
+
+using Discord.Commands;
+using Discord.WebSocket;
+using PluginManager.Interfaces;
+using PluginManager.Loaders;
+using PluginManager.Others;
+using PluginManager.Others.Permissions;
+
+namespace PluginManager.Bot;
+
+internal class CommandHandler
+{
+    private readonly string botPrefix;
+    private readonly DiscordSocketClient client;
+    private readonly CommandService commandService;
+
+    /// <summary>
+    ///     Command handler constructor
+    /// </summary>
+    /// <param name="client">The discord bot client</param>
+    /// <param name="commandService">The discord bot command service</param>
+    /// <param name="botPrefix">The prefix to watch for</param>
+    public CommandHandler(DiscordSocketClient client, CommandService commandService, string botPrefix)
+    {
+        this.client = client;
+        this.commandService = commandService;
+        this.botPrefix = botPrefix;
+
+    }
+
+    /// <summary>
+    ///     The method to initialize all commands
+    /// </summary>
+    /// <returns></returns>
+    public async Task InstallCommandsAsync()
+    {
+        client.MessageReceived += MessageHandler;
+        client.SlashCommandExecuted += Client_SlashCommandExecuted;
+        await commandService.AddModulesAsync(Assembly.GetEntryAssembly(), null);
+    }
+
+    private Task Client_SlashCommandExecuted(SocketSlashCommand arg)
+    {
+        try
+        {
+            var plugin = PluginLoader.SlashCommands!
+             .Where(p => p.Name == arg.Data.Name)
+             .FirstOrDefault();
+
+            if (plugin is null)
+                throw new Exception("Failed to run command. !");
+
+
+            if (arg.Channel is SocketDMChannel)
+                plugin.ExecuteDM(arg);
+            else plugin.ExecuteServer(arg);
+        }
+        catch (Exception ex)
+        {
+
+            Console.WriteLine(ex.ToString());
+            ex.WriteErrFile();
+        }
+
+        return Task.CompletedTask;
+
+    }
+
+    /// <summary>
+    ///     The message handler for the bot
+    /// </summary>
+    /// <param name="Message">The message got from the user in discord chat</param>
+    /// <returns></returns>
+    private async Task MessageHandler(SocketMessage Message)
+    {
+
+        try
+        {
+            if (Message.Author.IsBot) 
+                return;
+
+            if (Message as SocketUserMessage == null)
+                return;
+
+            var message = Message as SocketUserMessage;
+
+            if (message is null)
+                return;
+
+            var argPos = 0;
+
+            if (!message.Content.StartsWith(botPrefix) && !message.HasMentionPrefix(client.CurrentUser, ref argPos))
+                return;
+
+            var context = new SocketCommandContext(client, message);
+
+            await commandService.ExecuteAsync(context, argPos, null);
+
+            DBCommand plugin;
+            string cleanMessage = "";
+
+            if (message.HasMentionPrefix(client.CurrentUser, ref argPos))
+            {
+                string mentionPrefix = "<@" + client.CurrentUser.Id + ">";
+
+                plugin = PluginLoader.Commands!
+                   .Where
+                   (
+                       plug => plug.Command == message.Content.Substring(mentionPrefix.Length+1).Split(' ')[0] ||
+                               (
+                                   plug.Aliases is not null &&
+                                       plug.Aliases.Contains(message.CleanContent.Substring(mentionPrefix.Length+1).Split(' ')[0])
+                               )
+                   )
+                   .FirstOrDefault();
+
+                cleanMessage = message.Content.Substring(mentionPrefix.Length + 1);
+            }
+
+            else
+            {
+                plugin = PluginLoader.Commands!
+                             .Where(
+                                  p => p.Command == message.Content.Split(' ')[0].Substring(botPrefix.Length) ||
+                                       (p.Aliases is not null &&
+                                        p.Aliases.Contains(
+                                            message.Content.Split(' ')[0].Substring(botPrefix.Length))))
+                             .FirstOrDefault();
+                cleanMessage = message.Content.Substring(botPrefix.Length);
+            }
+            if (plugin is null) 
+                throw new Exception($"Failed to run command ! " + message.CleanContent);
+
+            if (plugin.requireAdmin && !context.Message.Author.isAdmin())
+                return;
+
+            string[] split = cleanMessage.Split(' ');
+
+            string[] argsClean = null;
+            if(split.Length > 1)
+                argsClean = string.Join(' ', split, 1, split.Length-1).Split(' ');
+
+            CmdArgs cmd = new(context, cleanMessage, split[0], argsClean);
+
+            if (context.Channel is SocketDMChannel)
+                plugin.ExecuteDM(cmd);
+            else plugin.ExecuteServer(cmd);
+        }
+        catch (Exception ex)
+        {
+            ex.WriteErrFile();
+
+            Console.WriteLine(ex.ToString());
+        }
+    }
+}