using CounterStrikeSharp.API.ValveConstants.Protobuf;

namespace AntiDLL_CS2_SimpleAdmin;

using System.Text.Json.Serialization;
using CounterStrikeSharp.API;
using CounterStrikeSharp.API.Modules.Entities;
using CS2_SimpleAdminApi;
using CounterStrikeSharp.API.Core;
using CounterStrikeSharp.API.Core.Capabilities;
using Microsoft.Extensions.Logging;

using AntiDLL.API;

public class PluginConfig : IBasePluginConfig
{
    [JsonPropertyName("ConfigVersion")] public int Version { get; set; } = 1;
    [JsonPropertyName("Reason")] public string Reason { get; set; } = "Invalid event detected!";
    [JsonPropertyName("Duration")] public int Duration { get; set; } = 0;
    [JsonPropertyName("CommandToExecute")] public string CommandToExecute { get; set; } = "css_addban {steamid64} {duration} {reason}";
    [JsonPropertyName("BanType")] public string BanType { get; set; } = "auto";
}

public sealed class AntiDLL_CS2_SimpleAdmin : BasePlugin, IPluginConfig<PluginConfig>
{
    public PluginConfig Config { get; set; } = new();
    private readonly HashSet<int> _bannedPlayers = [];
    private static PluginCapability<IAntiDLL> AntiDll { get; } = new("AntiDLL");
    private static PluginCapability<ICS2_SimpleAdminApi> SimpleAdminApi { get; } = new("simpleadmin:api");
    private int _banType = 0;
    private static ICS2_SimpleAdminApi? _simpleAdminApi;
    
    public override string ModuleName => "AntiDLL [CS2-SimpleAdmin Module]";
    public override string ModuleDescription => "AntiDLL module for CS2-SimpleAdmin integration";
    public override string ModuleVersion => "1.0.0";
    public override string ModuleAuthor => "daffyy";

    public override void Load(bool hotReload)
    {
        RegisterListener<Listeners.OnClientDisconnect>(OnClientDisconnect);
    }
    
    public void OnConfigParsed(PluginConfig config)
    {
        Config = config;
    }
    
    public override void OnAllPluginsLoaded(bool hotReload)
    {
        try
        {
            var antidll = AntiDll.Get();
            if (antidll == null)
            {
                Logger.LogError("Failed to get AntiDLL API");
                Unload(false);
                return;
            }
            
            antidll.OnDetection += OnDetection;
        }
        catch (Exception)
        {
            Logger.LogError("Failed to get AntiDLL API");
            Unload(false);
        }

        try
        {
            _simpleAdminApi = SimpleAdminApi.Get();
            if (_simpleAdminApi != null)
                _banType = 1;
        }
        catch (Exception)
        {
            Logger.LogError("Failed to get CS2-SimpleAdmin API, using command as BanType");
        }
    }

    private void OnClientDisconnect(int playerSlot)
    {
        var player = Utilities.GetPlayerFromSlot(playerSlot);
        if (player == null || !player.IsValid || player.IsBot)
            return;

        _bannedPlayers.Remove(playerSlot);
    }

    private void OnDetection(CCSPlayerController player, string eventName)
    {
        if (!_bannedPlayers.Add(player.Slot))
            return;

        if (_banType == 1 && _simpleAdminApi != null)
        {
            _simpleAdminApi.IssuePenalty(new SteamID(player.SteamID), null, PenaltyType.Ban, Config.Reason, Config.Duration);
        }
        else if (Config.BanType == "kick")
        {
            player.Disconnect(NetworkDisconnectionReason.NETWORK_DISCONNECT_KICKED_UNTRUSTEDACCOUNT);
        }
        else
        {
            Server.ExecuteCommand(Config.CommandToExecute.Replace("{steamid64}", player.SteamID.ToString())
                .Replace("{duration}", Config.Duration.ToString().Replace("{reason}", $"\"{Config.Reason}\""))
                .Replace("{userid}", player.UserId.Value.ToString()));
        }
    }

    public override void Unload(bool hotReload)
    {
        RemoveListener<Listeners.OnClientDisconnect>(OnClientDisconnect);
        
        var antidll = AntiDll.Get();
        if (antidll != null)
        {
            antidll.OnDetection -= OnDetection;
        }
    }
}