sv_lyc_handler.lua

util.AddNetworkString("ttt2_lyc_transform")

local function ResetLycanthrope(ply)
  ply:SetNWBool("LycTransformed", false)
  ply.lycRegen = nil
end

local function ResetAllLycanthropes()
  for _, ply in ipairs(player.GetAll()) do
    ResetLycanthrope(ply)
  end
end

local function HasTeammates(ply)
  if not ply:Alive() or ply:IsSpec() then return false end

  local count = 0
  for _, pl in ipairs(player.GetAll()) do
    if pl == ply then continue end

    if pl:HasTeam(ply:GetTeam()) and pl:Alive() and not pl:IsSpec() then
      count = count + 1
    end
  end

  return count > 0
end

local function UnleashLycanthrope(ply)
  if HasTeammates(ply) then return end
  print("[TTT2 Lycanthrope] " .. ply:Nick() .. " is transforming!")
  ply:SetNWBool("LycTransformed", true)
  ply:SetMaxHealth(GetConVar("ttt2_lyc_maxhealth"):GetInt())
  ply:GiveArmor(GetConVar("ttt2_lyc_armor"):GetInt())
  SendFullStateUpdate()


  net.Start("ttt2_lyc_transform")
  net.Broadcast()
  ply.lycRegen = CurTime()
end

hook.Add("Think", "LycRegenThink", function()
  for _, ply in ipairs(player.GetAll()) do
    if not ply:Alive() or ply:IsSpec() or ply:GetSubRole() ~= ROLE_LYCANTHROPE then continue end

    if not ply:GetNWBool("LycTransformed") then continue end

    if ply.lycRegen <= CurTime() then
      ply:SetHealth(math.Clamp(ply:Health() + GetConVar("ttt2_lyc_health_regen"):GetInt(), 0, ply:GetMaxHealth()))
      ply.lycRegen = CurTime() + GetConVar("ttt2_lyc_regen_rate"):GetFloat()
    end
  end
end)

hook.Add("EntityTakeDamage", "LycTakeDamage", function(ply, dmginfo)
  if not IsValid(ply) or not ply:IsPlayer() or ply:IsSpec() or not ply:Alive() then return end
  if ply:GetSubRole() ~= ROLE_LYCANTHROPE or not ply:GetNWBool("LycTransformed") then return end
  if dmginfo:GetDamage() > ply:Health() then return end

  ply.lycRegen = CurTime() + GetConVar("ttt2_lyc_health_delay"):GetInt()
end)

hook.Add("EntityTakeDamage", "LycDealDamage", function(ply, dmginfo)
  if not IsValid(ply) or not ply:IsPlayer() or ply:IsSpec() or not ply:Alive() then return end
  local attacker = dmginfo:GetAttacker()

  if not IsValid(attacker) or not attacker:IsPlayer() then return end
  if attacker:GetSubRole() ~= ROLE_LYCANTHROPE or not attacker:GetNWBool("LycTransformed") then return end

  local dmg_mult = GetConVar("ttt2_lyc_dmg"):GetFloat()

  dmginfo:ScaleDamage(dmg_mult)
end)

hook.Add("TTT2PostPlayerDeath", "OnLycTeamDeath", function(victim, _, attacker)
  for _, ply in ipairs(player.GetAll()) do
    if ply:GetSubRole() ~= ROLE_LYCANTHROPE  or not ply:Alive() or ply:IsSpec() then continue end
    if victim:HasTeam(ply:GetTeam()) then
      UnleashLycanthrope(ply)
    end
  end
end)

hook.Add("PlayerSpawn", "LycSpawned", function(ply)
  if not IsValid(ply) or not ply:Alive() or ply:IsSpec() then return end

  if ply:GetSubRole() == ROLE_LYCANTHROPE then
    UnleashLycanthrope(ply)
  end
end)

hook.Add("PlayerDisconnected", "LycDisconnectTeam", function(ply)
  for _, pl in ipairs(player.GetAll()) do
    if pl:GetSubRole() ~= ROLE_LYCANTHROPE or not pl:Alive() or pl:IsSpec() then continue end
    if pl:HasTeam(ply:GetTeam()) then
      UnleashLycanthrope(pl)
    end
  end
end)

hook.Add("TTT2SpecialRoleSyncing", "TTT2RoleLycHide", function(ply, tbl)
  if GetConVar("ttt2_lyc_know_role"):GetBool() then return end

  for lyc in pairs(tbl) do
    if lyc:GetSubRole() == ROLE_LYCANTHROPE and not lyc:GetNWBool("LycTransformed") then
      tbl[lyc] = {ROLE_INNOCENT, TEAM_INNOCENT}
    end
  end
end)

hook.Add("TTTonCorpseCreated", "ModifyLycRag", function(rag, ply)
  if not IsValid(ply) or not IsValid(rag) or ply:GetSubRole() ~= ROLE_LYCANTHROPE or ply:GetNWBool("LycTransformed") then return end

  rag.was_role = ROLE_INNOCENT
  rag.role_color = INNOCENT.ltcolor
end)

hook.Add("TTTEndRound", "ResetLycanthrope", ResetAllLycanthropes)
hook.Add("TTTPrepareRound", "ResetLycanthrope", ResetAllLycanthropes)
hook.Add("TTTBeginRound", "ResetLycanthrope", ResetAllLycanthropes)