SethDiscordBot/PluginManager/Loaders/PluginLoader.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Threading.Tasks;

using Discord.WebSocket;

using PluginManager.Interfaces;
using PluginManager.Online;
using PluginManager.Online.Updates;
using PluginManager.Others;

namespace PluginManager.Loaders;

public class PluginLoader
{
    public delegate void CMDLoaded(string name, string typeName, bool success, Exception? e = null);

    public delegate void EVELoaded(string name, string typeName, bool success, Exception? e = null);

    private const string pluginCMDFolder = @"./Data/Plugins/Commands/";
    private const string pluginEVEFolder = @"./Data/Plugins/Events/";

    internal const string pluginCMDExtension = "dll";
    internal const string pluginEVEExtension = "dll";
    private readonly DiscordSocketClient _client;

    /// <summary>
    ///     Event that is fired when a <see cref="DBCommand" /> is successfully loaded into commands list
    /// </summary>
    public CMDLoaded? onCMDLoad;

    /// <summary>
    ///     Event that is fired when a <see cref="DBEvent" /> is successfully loaded into events list
    /// </summary>
    public EVELoaded? onEVELoad;

    /// <summary>
    ///     The Plugin Loader constructor
    /// </summary>
    /// <param name="discordSocketClient">The discord bot client where the plugins will pe attached to</param>
    public PluginLoader(DiscordSocketClient discordSocketClient)
    {
        _client = discordSocketClient;
    }


    /// <summary>
    ///     A list of <see cref="DBCommand" /> commands
    /// </summary>
    public static List<DBCommand>? Commands { get; set; }

    /// <summary>
    ///     A list of <see cref="DBEvent" /> commands
    /// </summary>
    public static List<DBEvent>? Events { get; set; }

    /// <summary>
    ///     The main mathod that is called to load all events
    /// </summary>
    public async void LoadPlugins()
    {
        //Check for updates in commands
        foreach (var file in Directory.GetFiles("./Data/Plugins/Commands", $"*.{pluginCMDExtension}",
                                                SearchOption.AllDirectories))
            await Task.Run(async () =>
            {
                var name = new FileInfo(file).Name.Split('.')[0];
                var version = await ServerCom.GetVersionOfPackageFromWeb(name);
                if (version is null)
                    return;
                if (!Config.PluginVersionsContainsKey(name))
                    Config.SetPluginVersion(
                        name, (version.PackageVersionID + ".0.0"));

                if (await PluginUpdater.CheckForUpdates(name))
                    await PluginUpdater.Download(name);
            });

        //Check for updates in events
        foreach (var file in Directory.GetFiles("./Data/Plugins/Events", $"*.{pluginEVEExtension}",
                                                SearchOption.AllDirectories))
            await Task.Run(async () =>
            {
                var name = new FileInfo(file).Name.Split('.')[0];
                var version = await ServerCom.GetVersionOfPackageFromWeb(name);
                if (version is null)
                    return;
                if (!Config.PluginVersionsContainsKey(name))
                    Config.SetPluginVersion(
                        name, (version.PackageVersionID + ".0.0"));

                if (await PluginUpdater.CheckForUpdates(name))
                    await PluginUpdater.Download(name);
            });


        //Save the new config file (after the updates)
        await Config.SaveConfig(SaveType.NORMAL);


        //Load all plugins

        Commands = new List<DBCommand>();
        Events = new List<DBEvent>();

        Functions.WriteLogFile("Starting plugin loader ... Client: " + _client.CurrentUser.Username);
        Console.WriteLine("Loading plugins");

        var commandsLoader = new Loader<DBCommand>(pluginCMDFolder, pluginCMDExtension);
        var eventsLoader = new Loader<DBEvent>(pluginEVEFolder, pluginEVEExtension);


        commandsLoader.FileLoaded += OnCommandFileLoaded;
        commandsLoader.PluginLoaded += OnCommandLoaded;

        eventsLoader.FileLoaded += EventFileLoaded;
        eventsLoader.PluginLoaded += OnEventLoaded;

        Commands = commandsLoader.Load();
        Events = eventsLoader.Load();

    }

    private void EventFileLoaded(LoaderArgs e)
    {
        if (!e.IsLoaded)
            Functions.WriteLogFile($"[EVENT] Event from file [{e.PluginName}] has been successfully created !");
    }

    private void OnCommandFileLoaded(LoaderArgs e)
    {
        if (!e.IsLoaded)
            Functions.WriteLogFile($"[CMD] Command from file [{e.PluginName}] has been successfully loaded !");
    }

    private void OnEventLoaded(LoaderArgs e)
    {
        try
        {
            if (e.IsLoaded)
                ((DBEvent)e.Plugin!).Start(_client);

            onEVELoad?.Invoke(((DBEvent)e.Plugin!).Name, e.TypeName!, e.IsLoaded, e.Exception);
        }
        catch (Exception ex)
        {
            Console.WriteLine(ex.ToString());
            Console.WriteLine("Plugin: " + e.PluginName);
            Console.WriteLine("Type: " + e.TypeName);
            Console.WriteLine("IsLoaded: " + e.IsLoaded);
        }
    }

    private void OnCommandLoaded(LoaderArgs e)
    {
        onCMDLoad?.Invoke(((DBCommand)e.Plugin!).Command, e.TypeName!, e.IsLoaded, e.Exception);
    }
}