#include <sourcemod>
#include <sdktools>

int g_HaloSprite;
int g_LaserBeam;

public void OnPluginStart()
{
    RegConsoleCmd( "sm_hitboxtest", Command_HitboxTest );
    RegConsoleCmd( "sm_bboxtest", Command_BBoxTest );
    RegConsoleCmd( "sm_clipraytest", Command_ClipRayTest );
}

public void OnMapStart()
{
    g_HaloSprite = PrecacheModel( "materials/sprites/glow01.vmt" );
    g_LaserBeam = PrecacheModel( "materials/sprites/laserbeam.vmt" );
}

public Action Command_HitboxTest( int client, int args )
{
    float startpos[3];
    GetClientEyePosition( client, startpos );
    float angles[3];
    GetClientEyeAngles( client, angles );

    TR_EnumerateEntities( startpos, angles, false, RayType_Infinite, HitPlayerHBox, client );
}

public Action Command_BBoxTest( int client, int args )
{
    float startpos[3];
    GetClientEyePosition( client, startpos );
    float angles[3];
    GetClientEyeAngles( client, angles );

    TR_EnumerateEntities( startpos, angles, false, RayType_Infinite, HitPlayerBBox, client );
}

public Action Command_ClipRayTest( int client, int args )
{
    float start[3];
    GetClientAbsOrigin( client, start );
    float end[3];
    GetClientEyePosition( client, end );

    TR_ClipRayToEntity( start, end, MASK_ALL, RayType_EndPoint, client );
    TR_GetEndPosition( end );

    PrintToChatAll( "Hit myself at (%.2f, %.2f, %.2f)", end[0], end[1], end[2] );

    return Plugin_Handled;
}

public bool HitPlayerBBox( int entity, int client )
{
    if( 0 < entity <= MaxClients && entity != client )
    {
        TR_ClipCurrRayToEntity( MASK_PLAYERSOLID, entity );

        float pos[3];
        TR_GetEndPosition( pos );

        PrintToChatAll( "Hit %N (%i)! Collided at (%.2f, %.2f, %.2f) | Fraction: %.2f", entity, entity, pos[0], pos[1], pos[2], TR_GetFraction() );

        float startpos[3];
        GetClientEyePosition( client, startpos );

        TE_SetupBeamPoints( startpos, pos, g_LaserBeam, g_HaloSprite, 0, 0, 20.0, 1.0, 1.0, 0, 0.0, { 0, 255, 0, 255 }, 0 );
        TE_SendToAll( 0.0 );

        return false;
    }

    return true;
}

public bool HitPlayerHBox( int entity, int client )
{
    if( 0 < entity <= MaxClients && entity != client )
    {
        Handle tr = TR_ClipCurrRayToEntityEx( MASK_ALL, entity );

        float pos[3];
        TR_GetEndPosition( pos, tr );

        PrintToChatAll( "Hit %N (%i)! Collided at (%.2f, %.2f, %.2f) | Fraction: %.2f", entity, entity, pos[0], pos[1], pos[2], TR_GetFraction( tr ) );

        delete tr;

        float startpos[3];
        GetClientEyePosition( client, startpos );

        TE_SetupBeamPoints( startpos, pos, g_LaserBeam, g_HaloSprite, 0, 0, 20.0, 1.0, 1.0, 0, 0.0, { 255, 0, 0, 255 }, 0 );
        TE_SendToAll( 0.0 );

        return false;
    }

    return true;
}