//Includes
#include <sourcemod>
#include <sdktools>   //need for adding files to download table

#include <overlays>


//Compiler Options   //Optional, but please keep your code clean!
#pragma semicolon 1
#pragma newdecls required


//use a define or if needed (for cvars) a string
#define OVERLAYPATH "overlays/test"   //Path to the overlay relative to materials/.. - no need for extentions like .vmt or .vft


//Start
public void OnPluginStart()
{
    RegConsoleCmd("sm_testoverlay", Command_TestOverlay, "Show overlay to client");
    RegConsoleCmd("sm_testoverlayall", Command_TestOverlayAll, "Show overlay to all clients for 5 seconds");
    RegConsoleCmd("sm_testremoveoverlay", Command_TestRemoveOverlay, "Remove overlay of all clients");
}


//MapStart
public void OnMapStart()
{
    PrecacheDecalAnyDownload(OVERLAYPATH);   //Stock for adding overlay to download table and precaching.
}


//Show overlay to client
public Action Command_TestOverlay(int client, int args)
{
    ShowOverlay(client, OVERLAYPATH, 0.0);   //Show the overlay to the client - 0.0 as lifetime will show the overlay constant.

    return Plugin_Handled;
}


//Show overlay to all clients
public Action Command_TestOverlayAll(int client, int args)
{
    ShowOverlayAll(OVERLAYPATH, 5.0);   //Show the overlay to the all clients - 5.0 as lifetime will delete the overlay after 5 seconds.

    return Plugin_Handled;
}


//Show overlay to client
public Action Command_TestRemoveOverlay(int client, int args)
{
    for (int i = 1; i <= MaxClients; i++)   //Loop through all clients. Client validation in stock function
    {
        CreateTimer(0.0, DeleteOverlay, GetClientUserId(i));   //We use a timer to remove the overlay.
    }

    return Plugin_Handled;
}

// Precache & prepare download for overlays & decals
stock void PrecacheDecalAnyDownload(char[] sOverlay)
{
    char sBuffer[256];
    Format(sBuffer, sizeof(sBuffer), "%s.vmt", sOverlay);
    PrecacheDecal(sBuffer, true);
    Format(sBuffer, sizeof(sBuffer), "materials/%s.vmt", sOverlay);
    AddFileToDownloadsTable(sBuffer);

    Format(sBuffer, sizeof(sBuffer), "%s.vtf", sOverlay);
    PrecacheDecal(sBuffer, true);
    Format(sBuffer, sizeof(sBuffer), "materials/%s.vtf", sOverlay);
    AddFileToDownloadsTable(sBuffer);
}

// Show overlay to a client with lifetime | 0.0 = no auto remove
stock void ShowOverlay(int client, char[] path, float lifetime)
{
    if (!IsClientInGame(client) || IsFakeClient(client) || IsClientSourceTV(client) || IsClientReplay(client))
        return;

    ClientCommand(client, "r_screenoverlay \"%s.vtf\"", path);

    if (lifetime != 0.0)
        CreateTimer(lifetime, DeleteOverlay, GetClientUserId(client));
}

// Show overlay to all clients with lifetime | 0.0 = no auto remove
stock void ShowOverlayAll(char[] path, float lifetime)
{
    for (int i = 1; i <= MaxClients; i++)
    {
        if (!IsClientInGame(i) || IsFakeClient(i) || IsClientSourceTV(i) || IsClientReplay(i))
            continue;

        ClientCommand(i, "r_screenoverlay \"%s.vtf\"", path);

        if (lifetime != 0.0)
            CreateTimer(lifetime, DeleteOverlay, GetClientUserId(i));
    }
}

// Remove overlay from a client - Timer!
stock Action DeleteOverlay(Handle timer, any userid) 
{
    int client = GetClientOfUserId(userid);
    if (client <= 0 || !IsClientInGame(client) || IsFakeClient(client) || IsClientSourceTV(client) || IsClientReplay(client))
        return;

    ClientCommand(client, "r_screenoverlay \"\"");
}