//
// Created by Michal Přikryl on 10.10.2025.
// Copyright (c) 2025 slynxcz. All rights reserved.
//

using System.Buffers;
using System.Numerics;
using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using System.Runtime.Loader;
using System.Text;
using System.Text.Json;
using System.Text.RegularExpressions;
using CounterStrikeSharp.API;
using CounterStrikeSharp.API.Core;
using CounterStrikeSharp.API.Core.Attributes.Registration;
using CounterStrikeSharp.API.Core.Capabilities;
using CounterStrikeSharp.API.Modules.Commands;
using CounterStrikeSharp.API.Modules.Memory;
using CounterStrikeSharp.API.Modules.Utils;
using RayTraceAPI;
using Vector = CounterStrikeSharp.API.Modules.Utils.Vector;
using TraceOptions = RayTraceAPI.TraceOptions;
using TraceResult = RayTraceAPI.TraceResult;

namespace RayTraceImpl;

// ReSharper disable once InconsistentNaming
// ReSharper disable once UnusedType.Global
public class RayTraceImpl : BasePlugin
{
    public override string ModuleName => "RayTraceImpl";
    public override string ModuleVersion => "v1.0.0";
    public override string ModuleAuthor => "Slynx";

    internal static PluginCapability<CRayTraceInterface> RayTraceInterface { get; } =
        new("raytrace:craytraceinterface");

    public override void Load(bool hotReload)
    {
        Capabilities.RegisterPluginCapability(RayTraceInterface, () => new CRayTrace());
        RegisterListener<Listeners.OnMetamodAllPluginsLoaded>(OnMetamodAllPluginsLoaded);
    }

    public override void Unload(bool hotReload)
    {
        RemoveListener<Listeners.OnMetamodAllPluginsLoaded>(OnMetamodAllPluginsLoaded);
    }

    private void OnMetamodAllPluginsLoaded()
    {
        if (!NativeBridge.Initialize())
        {
            Prints.ServerLog("[RayTraceImpl] Native bridge initialization failed.", ConsoleColor.Red);
            return;
        }

        Prints.ServerLog("[RayTraceImpl] Managed side initialized.", ConsoleColor.Green);
    }
}

public class CRayTrace : CRayTraceInterface
{
    public bool TraceShape(Vector start, QAngle angles, CEntityInstance? ignore, TraceOptions options, out TraceResult result)
    {
        unsafe
        {
            result = default;

            if (!NativeBridge.Loaded)
                return false;

            TraceResult resultBuffer = default;
            TraceOptions optionsBuffer = options;

            bool success = NativeBridge.TraceShape!(NativeBridge.Handle,
                start.Handle,
                angles.Handle,
                ignore?.Handle ?? nint.Zero,
                (nint)(&optionsBuffer),
                (nint)(&resultBuffer));

            result = resultBuffer;
            return success;
        }
    }

    public bool TraceEndShape(Vector start, Vector end, CEntityInstance? ignore, TraceOptions options, out TraceResult result)
    {
        unsafe
        {
            result = default;

            if (!NativeBridge.Loaded)
                return false;

            TraceResult resultBuffer = default;
            TraceOptions optionsBuffer = options;

            bool success = NativeBridge.TraceEndShape!(NativeBridge.Handle,
                start.Handle,
                end.Handle,
                ignore?.Handle ?? nint.Zero,
                (nint)(&optionsBuffer),
                (nint)(&resultBuffer));

            result = resultBuffer;
            return success;
        }
    }

    public bool TraceHullShape(Vector start, Vector end, Vector mins, Vector maxs, CEntityInstance? ignore, TraceOptions options, out TraceResult result)
    {
        unsafe
        {
            result = default;

            if (!NativeBridge.Loaded)
                return false;

            TraceResult resultBuffer = default;
            TraceOptions optionsBuffer = options;

            bool success = NativeBridge.TraceHullShape!(NativeBridge.Handle,
                start.Handle,
                end.Handle,
                mins.Handle,
                maxs.Handle,
                ignore?.Handle ?? nint.Zero,
                (nint)(&optionsBuffer),
                (nint)(&resultBuffer));

            result = resultBuffer;
            return success;
        }
    }

    public bool TraceShapeEx(Vector start, Vector end, nint filter, nint ray, out TraceResult result)
    {
        unsafe
        {
            result = default;

            if (!NativeBridge.Loaded)
                return false;

            TraceResult resultBuffer = default;

            bool success = NativeBridge.TraceShapeEx!(NativeBridge.Handle,
                start.Handle,
                end.Handle,
                filter,
                ray,
                (nint)(&resultBuffer));

            result = resultBuffer;
            return success;
        }
    }
}

public static class NativeBridge
{
    public static nint Handle;
    public static bool Loaded;

    public static unsafe Func<nint, nint, nint, nint, nint, nint, bool>? TraceShape;
    public static unsafe Func<nint, nint, nint, nint, nint, nint, bool>? TraceEndShape;
    public static unsafe Func<nint, nint, nint, nint, nint, nint, nint, nint, bool>? TraceHullShape;
    public static unsafe Func<nint, nint, nint, nint, nint, nint, bool>? TraceShapeEx;

    public static bool Initialize()
    {
        Handle = (nint)Utilities.MetaFactory("CRayTraceInterface002")!;

        if (Handle == 0)
            return false;

        Bind();
        Loaded = true;
        return true;
    }

    private static void Bind()
    {
        int shape, end, hull, ex;

        if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
        {
            shape = 1;
            end = 2;
            hull = 3;
            ex = 4;
        }
        else
        {
            shape = 2;
            end = 3;
            hull = 4;
            ex = 5;
        }

        TraceShape = VirtualFunction.Create<nint, nint, nint, nint, nint, nint, bool>(Handle, shape);
        TraceEndShape = VirtualFunction.Create<nint, nint, nint, nint, nint, nint, bool>(Handle, end);
        TraceHullShape = VirtualFunction.Create<nint, nint, nint, nint, nint, nint, nint, nint, bool>(Handle, hull);
        TraceShapeEx = VirtualFunction.Create<nint, nint, nint, nint, nint, nint, bool>(Handle, ex);
    }
}

public static class Prints
{
    public static void ServerLog(string msg, ConsoleColor color = ConsoleColor.White)
    {
        Console.ForegroundColor = color;
        Console.WriteLine(msg);
        Console.ResetColor();
    }
}