GM.Name = "Feedc0de's coop Mode"
GM.Author = "0xFEEDC0DE64"
GM.Email = "daniel@brunner.ninja"
GM.Website = "brunner.ninja"

DeriveGamemode("sandbox")

include( "levels.lua" )
include( "progression_shared.lua" )

if (SERVER) then
    function ChangeRandomLevel()
        local randomLevel = GetRandomLevel()
        if not randomLevel then
            ErrorNoHalt("[custom_coop] No other installed coop map was found in levels.lua\n")
            return false
        end

        print("[custom_coop] Changing to random level " .. randomLevel)
        RunConsoleCommand("changelevel", randomLevel)
        return true
    end

    function NextRandomLevel()
        local randomLevel = GetRandomLevel()
        if not randomLevel then
            ErrorNoHalt("[custom_coop] Cannot set nextlevel: no other installed coop map was found\n")
            return nil
        end

        print("[custom_coop] Setting nextlevel to " .. randomLevel)
        RunConsoleCommand("nextlevel", randomLevel)
        return randomLevel
    end

    concommand.Add("randomlevel", ChangeRandomLevel)
    concommand.Add("randomnextlevel", NextRandomLevel)

    hook.Add("Initialize", "SetupRandomLevels", function()
        timer.Simple(0, NextRandomLevel)
    end)
end

--    timer.Simple(
--        1,
--        function()
--            function HideThings(name)
--                if (name == "CHudDamageIndicator" or name == "CHudHintDisplay") then
--                    return false
--                end
--            end
--            hook.Add("HUDShouldDraw", "HideThings", HideThings)
--        end
--    )

function GM:IsSpawnpointSuitable(ply, spawnpointent, bMakeSuitable)
    local Pos = spawnpointent:GetPos()

    -- Note that we're searching the default hull size here for a player in the way of our spawning.
    -- This seems pretty rough, seeing as our player's hull could be different.. but it should do the job
    -- ( HL2DM kills everything within a 128 unit radius )
    local Ents = ents.FindInBox(Pos + Vector(-16, -16, 0), Pos + Vector(16, 16, 72))

    if (ply:Team() == TEAM_SPECTATOR or ply:Team() == TEAM_UNASSIGNED) then
        return true
    end

    local Blockers = 0

    for k, v in pairs(Ents) do
        if (IsValid(v) and v:GetClass() == "player" and v:Alive()) then
            Blockers = Blockers + 1

            if (bMakeSuitable) then
                v:SetVelocity(v:GetPos() + Vector(100, 100, 0))
            end
        end
    end

    if (bMakeSuitable) then
        return true
    end
    if (Blockers > 0) then
        return false
    end
    return true
end

local hl2dmLoadout
local automaticLoadout
local BuildAutomaticLoadout
local ammoWeaponPairs = {
    { weapon = "weapon_pistol", ammo = { "item_ammo_pistol", "item_ammo_pistol_large" } },
    { weapon = "weapon_smg1", ammo = { "item_ammo_smg1", "item_ammo_smg1_large", "item_ammo_smg1_grenade" } },
    { weapon = "weapon_357", ammo = { "item_ammo_357", "item_ammo_357_large" } },
    { weapon = "weapon_ar2", ammo = { "item_ammo_ar2", "item_ammo_ar2_large", "item_ammo_ar2_altfire" } },
    { weapon = "weapon_shotgun", ammo = { "item_box_buckshot" } },
    { weapon = "weapon_crossbow", ammo = { "item_ammo_crossbow" } },
    { weapon = "weapon_rpg", ammo = { "item_rpg_round" } }
}

if SERVER then
    hl2dmLoadout = CreateConVar(
        "gc_hl2dmloadout",
        "1",
        FCVAR_ARCHIVE,
        "Legacy map loadout: 0 off, 1 supply weapons missing for placed ammo, 2 full HL2DM loadout"
    )

    local function MapProvidesWeapon(className)
        if #ents.FindByClass(className) > 0 then return true end

        for _, equip in ipairs(ents.FindByClass("game_player_equip")) do
            local weaponNames = equip:GetSaveTable().m_weaponNames or {}
            for _, equippedClass in pairs(weaponNames) do
                if string.lower(tostring(equippedClass)) == className then return true end
            end
        end

        return false
    end

    BuildAutomaticLoadout = function()
        automaticLoadout = {}

        for _, pair in ipairs(ammoWeaponPairs) do
            local hasAmmoPickup = false
            for _, ammoClass in ipairs(pair.ammo) do
                if #ents.FindByClass(ammoClass) > 0 then
                    hasAmmoPickup = true
                    break
                end
            end

            if hasAmmoPickup and not MapProvidesWeapon(pair.weapon) then
                automaticLoadout[#automaticLoadout + 1] = pair.weapon
            end
        end
    end

    hook.Add("InitPostEntity", "CustomCoopDetectMissingAmmoWeapons", BuildAutomaticLoadout)
end

function GM:PlayerLoadout(ply)
    if striponspawn then return end

    ply:Give("weapon_fists")
    local loadoutMode = hl2dmLoadout and hl2dmLoadout:GetInt() or 0
    if loadoutMode <= 0 then return end

    if loadoutMode == 1 then
        -- InitPostEntity normally prepares this before anybody spawns. The
        -- fallback also covers script auto-refresh during a running map.
        if not automaticLoadout and BuildAutomaticLoadout then BuildAutomaticLoadout() end
        for _, weaponClass in ipairs(automaticLoadout or {}) do
            ply:Give(weaponClass)
        end
        return
    end

    -- Forced compatibility mode mirrors CHL2MP_Player::GiveDefaultItems.
    ply:Give("weapon_crowbar")
    ply:Give("weapon_pistol")
    ply:Give("weapon_smg1")
    ply:Give("weapon_frag")
    ply:Give("weapon_physcannon")

    ply:GiveAmmo(255, "Pistol", true)
    ply:GiveAmmo(45, "SMG1", true)
    ply:GiveAmmo(1, "Grenade", true)
    ply:GiveAmmo(6, "Buckshot", true)
    ply:GiveAmmo(6, "357", true)
end

hook.Add( "SpawnMenuOpen", "DisallowSpawnMenu", function ()
    return LocalPlayer():IsAdmin()
end)

hook.Add( "ContextMenuOpen", "DisallowSpawnMenu", function ()
    return LocalPlayer():IsAdmin()
end)

function GM:PlayerNoClip(ply, desiredState)
    if desiredState then
        return ply:IsAdmin()
    else
        return true
    end
end

function GM:PlayerDeathSound()
    return true
end

function GM:DrawDeathNotice(x, y)
    return false
end

function GM:PlayerSpawnVehicle(ply, model, name, table)
    return ply:IsAdmin()
end

function GM:PlayerSpawnSWEP(ply, weapon, info)
    return ply:IsAdmin()
end

function GM:PlayerSpawnSENT(ply, class)
    return ply:IsAdmin()
end

function GM:PlayerSpawnRagdoll(ply, model)
    return ply:IsAdmin()
end

function GM:PlayerSpawnProp(ply, model)
    return ply:IsAdmin()
end

function GM:PlayerSpawnObject(ply, model, skin)
    return ply:IsAdmin()
end

function GM:PlayerSpawnNPC(ply, npc_type, weapon)
    return ply:IsAdmin()
end

function GM:PlayerSpawnEffect(ply, model)
    return ply:IsAdmin()
end

function GM:PlayerGiveSWEP(ply, weapon, swep)
    return ply:IsAdmin()
end

--function GM:HUDAmmoPickedUp( item, amount )
--    return false;
--end