cs2-SimpleAdmin/CS2-SimpleAdmin/CS2-SimpleAdmin.cs

using CounterStrikeSharp.API.Core;
using CounterStrikeSharp.API.Core.Attributes;
using CounterStrikeSharp.API.Core.Capabilities;
using CounterStrikeSharp.API.Modules.Commands;
using CounterStrikeSharp.API.Modules.Commands.Targeting;
using CS2_SimpleAdmin.Managers;
using CS2_SimpleAdminApi;
using Microsoft.Extensions.Logging;
using MySqlConnector;

using Menu;

namespace CS2_SimpleAdmin;

// TODO:
// Speed fix
// Gravity fix

[MinimumApiVersion(340)]
public partial class CS2_SimpleAdmin : BasePlugin, IPluginConfig<CS2_SimpleAdminConfig>
{
    internal static CS2_SimpleAdmin Instance { get; private set; } = new();

    public override string ModuleName => "CS2-SimpleAdmin" + (Helper.IsDebugBuild ? " (DEBUG)" : " (RELEASE)");
    public override string ModuleDescription => "Simple admin plugin for Counter-Strike 2 :)";
    public override string ModuleAuthor => "daffyy, Dliix66, ShiNxz & Cruze";
    public override string ModuleVersion => "1.7.8-alpha";

    public override void Load(bool hotReload)
    {
        Instance = this;

        Menu = new KitsuneMenu(this);

        RegisterEvents();

        if (hotReload)
        {
            ServerLoaded = false;
            _serverLoading = false;

            CacheManager = new CacheManager();
            OnGameServerSteamAPIActivated();
            OnMapStart(string.Empty);

            AddTimer(2.0f, () =>
            {
                if (Database == null) return;

                var playerManager = new PlayerManager();

                foreach (var player in Helper.GetValidPlayers())
                {
                    playerManager.LoadPlayerData(player);
                }
                ;
            });
        }

        SimpleAdminApi = new Api.CS2_SimpleAdminApi();
        Capabilities.RegisterPluginCapability(ICS2_SimpleAdminApi.PluginCapability, () => SimpleAdminApi);

        new PlayerManager().CheckPlayersTimer();

        ChatManager = new ChatManager();

        RegisterCommands.InitializeCommands();
    }

    public void OnConfigParsed(CS2_SimpleAdminConfig config)
    {
        Instance = this;
        _logger = Logger;

        if (config.DatabaseHost.Length < 1 || config.DatabaseName.Length < 1 || config.DatabaseUser.Length < 1)
        {
            throw new Exception("[CS2-SimpleAdmin] You need to setup Database credentials in config!");
        }

        MySqlConnectionStringBuilder builder = new()
        {
            Server = config.DatabaseHost,
            Database = config.DatabaseName,
            UserID = config.DatabaseUser,
            Password = config.DatabasePassword,
            Port = (uint)config.DatabasePort,
            SslMode = Enum.TryParse(config.DatabaseSSlMode, true, out MySqlSslMode sslMode) ? sslMode : MySqlSslMode.Preferred,
            Pooling = true,
            MinimumPoolSize = 0,
            MaximumPoolSize = 640,
        };

        DbConnectionString = builder.ConnectionString;
        Database = new Database.Database(DbConnectionString);

        if (!Database.CheckDatabaseConnection(out var exception))
        {
            if (exception != null)
                Logger.LogError("Problem with database connection! \n{exception}", exception);

            Unload(false);
            return;
        }

        Task.Run(() => Database.DatabaseMigration());

        Config = config;
        Helper.UpdateConfig(config);

        if (!Directory.Exists(ModuleDirectory + "/data"))
        {
            Directory.CreateDirectory(ModuleDirectory + "/data");
        }

        _localizer = Localizer;

        if (!string.IsNullOrEmpty(Config.Discord.DiscordLogWebhook))
            DiscordWebhookClientLog = new DiscordManager(Config.Discord.DiscordLogWebhook);

        PluginInfo.ShowAd(ModuleVersion);
        if (Config.EnableUpdateCheck)
            Task.Run(async () => await PluginInfo.CheckVersion(ModuleVersion, _logger));

        PermissionManager = new PermissionManager(Database);
        BanManager = new BanManager(Database);
        MuteManager = new MuteManager(Database);
        WarnManager = new WarnManager(Database);
    }

    private static TargetResult? GetTarget(CommandInfo command)
    {
        var matches = command.GetArgTargetResult(1);

        if (!matches.Any())
        {
            command.ReplyToCommand($"Target {command.GetArg(1)} not found.");
            return null;
        }

        if (command.GetArg(1).StartsWith('@'))
            return matches;

        if (matches.Count() == 1)
            return matches;

        command.ReplyToCommand($"Multiple targets found for \"{command.GetArg(1)}\".");
        return null;
    }

    public override void Unload(bool hotReload)
    {
        CacheManager?.Dispose();
        CacheManager = null;
    }
}