/*---------------------------------------------------------------------------
HUD ConVars
---------------------------------------------------------------------------*/
local ConVars = {}
local HUDWidth = 300
local HUDHeight = 115
local Color = Color
local CurTime = CurTime
local cvars = cvars
local draw = draw
local GetConVar = GetConVar
local hook = hook
local IsValid = IsValid
local Lerp = Lerp
local LocalPlayer = LocalPlayer
local math = math
local pairs = pairs
local ScrW, ScrH = ScrW, ScrH
local SortedPairs = SortedPairs
local string = string
local surface = surface
local table = table
local timer = timer
local tostring = tostring
local plyMeta = FindMetaTable("Player")

surface.CreateFont("aHUD28", {font = "Open Sans", size = 28, weight = 500})
surface.CreateFont("aHUD26", {font = "Open Sans", size = 26, weight = 500})
surface.CreateFont("aHUD24", {font = "Open Sans", size = 24, weight = 500})
surface.CreateFont("aHUD22", {font = "Open Sans", size = 22, weight = 500})
surface.CreateFont("aHUD20", {font = "Open Sans", size = 20, weight = 500})

include("ahud_config.lua")

include("vgui/cl_circleavatar.lua")
if aHUD.CustomNotifications then
include("vgui/cl_notification.lua")
end

if aHUD.CustomVoicepanels then
include("vgui/cl_voice.lua")
end


aHUD.Cur = aHUD.Cur or {}

aHUD.Cur.Health = nil
aHUD.Cur.Wallet = nil
aHUD.Cur.Salary = nil
aHUD.Cur.Job = nil

aHUD.Cur.Agenda = nil

aHUD.GetLongestTextLength = function(strings, font)
local longest = 0

surface.SetFont(font)
for k, v in pairs(strings) do
local w, h = surface.GetTextSize(v)
if w > longest then
longest = w
end
end
return longest
end

aHUD.Arrested = function() end
usermessage.Hook("GotArrested", function(msg)
local StartArrested = CurTime()
local ArrestedUntil = msg:ReadFloat()

Arrested = function()
if CurTime() - StartArrested <= ArrestedUntil and LocalPlayer():getDarkRPVar("Arrested") then
draw.RoundedBox(4, Scrw/2 - 175, Scrh - Scrh / 12, 350, 32, Color(62, 62, 62, 255))
draw.RoundedBox(2, Scrw/2 - 175, Scrh - Scrh / 12, 350, 2, Color(215, 85, 80))
draw.SimpleText(DarkRP.getPhrase("youre_arrested", math.ceil(ArrestedUntil - (CurTime() - StartArrested))), "aHUD24", Scrw/2, Scrh - Scrh / 12 + 4, Color(230, 230, 230, 255), TEXT_ALIGN_CENTER, TEXT_ALIGN_TOP)
elseif not LocalPlayer():getDarkRPVar("Arrested") then
Arrested = function() end
end
end
end)

aHUD.AdminTell = function() end
usermessage.Hook("AdminTell", function(msg)
timer.Remove("DarkRP_AdminTell")
local Message = msg:ReadString()

AdminTell = function()
draw.RoundedBox(4, 10, 10, Scrw - 20, 110, colors.darkblack)
draw.DrawNonParsedText(DarkRP.getPhrase("listen_up"), "GModToolName", Scrw / 2 + 10, 10, colors.white, 1)
draw.DrawNonParsedText(Message, "ChatFont", Scrw / 2 + 10, 90, colors.brightred, 1)
end

timer.Create("DarkRP_AdminTell", 10, 1, function()
AdminTell = function() end
end)
end)

hook.Add("Initialize", "aHUDInit", function()

MsgC( Color(66, 139, 202), "[aHUD] ", Color(210, 210, 210), "Loaded aHUD by ", Color(66, 139, 202) ,"arie ", Color(210, 210, 210),"(STEAM_0:0:22593800)\n" )

hook.Add("Think", "aHUDThink", function()
if aHUD.Cur and aHUD.Cur.Health then
if aHUD.Cur.Health != LocalPlayer():Health() then
aHUD.Cur.Health = Lerp(7 * FrameTime(), aHUD.Cur.Health, (LocalPlayer():Health() or 0))
end

if aHUD.Cur.Wallet != LocalPlayer():getDarkRPVar("money") then
aHUD.Cur.Wallet = Lerp(7 * FrameTime(), aHUD.Cur.Wallet, (LocalPlayer():getDarkRPVar("money") or 0))
end

if aHUD.Cur.Salary != LocalPlayer():getDarkRPVar("salary") then
aHUD.Cur.Salary = Lerp(7 * FrameTime(), aHUD.Cur.Salary, (LocalPlayer():getDarkRPVar("salary") or 0))
end

if aHUD.Cur.Job != LocalPlayer():getDarkRPVar("job") then
aHUD.Cur.Job = LocalPlayer():getDarkRPVar("job")
end
end
end)
end)

hook.Add("InitPostEntity", "aHUDInitPostEntity", function()
aHUD.Model = vgui.Create("AvatarCircleMask")
aHUD.Model:ParentToHUD()
aHUD.Model:SetSize(64, 64)
aHUD.Model:SetPos(16, ScrH() - HUDHeight+17)
aHUD.Model:SetPlayer(LocalPlayer(), 64)
aHUD.Model:SetMaskSize(64 / 2)
end)

hook.Add("DarkRPVarChanged", "agendaHUD", function(ply, var, _, new)
if ply ~= LocalPlayer() then return end
if var == "agenda" and new then
aHUD.Cur.Agenda = DarkRP.textWrap(new:gsub("//", "\n"):gsub("\\n", "\n"), "DarkRPHUD1", 440)
end

if var == "MAAS" then
aHUD.Cur.Salary = new
elseif var == "MESLEK" then
aHUD.Cur.Job = new
elseif var == "PARA" then
aHUD.Cur.Wallet = new
end
end)

local Scrw, Scrh, RelativeX, RelativeY
local Page = Material("icon16/page_white_text.png")

local agendaText
aHUD.DarkRPFuncs = function()

local chbxX, chboxY = chat.GetChatBoxPos()
if GetGlobalBool("DarkRP_LockDown") then
local cin = (math.sin(CurTime()) + 1) / 2
local chatBoxSize = math.floor(Scrh / 4)
draw.SimpleText(DarkRP.getPhrase("lockdown_started"), "aHUD24",chbxX, chboxY + chatBoxSize, Color(230, cin * 230,cin * 230, 255), TEXT_ALIGN_LEFT, TEXT_ALIGN_TOP)
end

local agenda = LocalPlayer():getAgendaTable()
if not agenda then return end
aHUD.Cur.Agenda = aHUD.Cur.Agenda or DarkRP.textWrap((LocalPlayer():getDarkRPVar("agenda") or ""):gsub("//", "\n"):gsub("\\n", "\n"), "aHUD20", 440)

draw.RoundedBox(4, 10, 10, 460, 130, Color(62, 62, 62, 255))
draw.RoundedBox(4, 10, 10, 460, 4, aHUD.Color)
draw.RoundedBox(0, 10, 14, 460, 18, aHUD.Color)

draw.SimpleText(agenda.Title, "aHUD22", 20, 9, Color(230, 230, 230), TEXT_ALIGN_LEFT, TEXT_ALIGN_TOP)

draw.DrawText(aHUD.Cur.Agenda , "aHUD20", 20, 35, Color(230, 230, 230), TEXT_ALIGN_LEFT)
end

local salaryText, walletText, jobText
local function DrawHUD()
Scrw, Scrh = ScrW(), ScrH()
RelativeX, RelativeY = 5, Scrh

local expand = aHUD.GetLongestTextLength({LocalPlayer():Name()}, "aHUD28")

HUDWidth = math.max(300, expand + 95)

draw.RoundedBox(0, 10, Scrh - HUDHeight - 6, HUDWidth, HUDHeight, Color(41, 41, 41, 255))
draw.RoundedBox(4, 13, Scrh - HUDHeight + 13 , HUDWidth - 6, HUDHeight-22, Color(62, 62, 62, 255))
draw.RoundedBox(2, 88, Scrh - HUDHeight + 38, HUDWidth - 87, 2, aHUD.Color)
draw.RoundedBox(0, 10, Scrh - HUDHeight - 10, HUDWidth, 20, aHUD.Color)

draw.RoundedBox(2, RelativeX + 12, Scrh - HUDHeight + 85, HUDWidth - 14, 10, Color(45, 45, 45, 255))
draw.RoundedBox(2, RelativeX + 12, Scrh - HUDHeight + 85, math.Clamp(aHUD.Cur.Health, 2, 100)/100*(HUDWidth-14), 10, aHUD.Color)

draw.RoundedBox(2, RelativeX + 12, Scrh - HUDHeight + 98, HUDWidth - 14, 5, Color(45, 45, 45, 255))
if LocalPlayer():Armor() and LocalPlayer():Armor() > 2 then
draw.RoundedBox(2, RelativeX + 12, Scrh - HUDHeight + 98, math.Clamp(LocalPlayer():Armor() or 0, 0, 100)/100*(HUDWidth-14), 5, aHUD.ArmorBarColor)
end

if not DarkRP.disabledDefaults["modules"]["hungermod"] then
HUDHeight = 123
aHUD.Model:SetPos(16, ScrH() - HUDHeight + 15)
if LocalPlayer():getDarkRPVar("Energy") and LocalPlayer():getDarkRPVar("Energy") > 2 then
draw.RoundedBox(2, RelativeX + 12, Scrh - HUDHeight + 106, HUDWidth - 14, 5, Color(45, 45, 45, 255))
draw.RoundedBox(2, RelativeX + 12, Scrh - HUDHeight + 106, math.Clamp(LocalPlayer():getDarkRPVar("Energy") or 0, 0, 100)/100*(HUDWidth-14), 5, aHUD.HungerBarColor)
end
end

draw.SimpleText(LocalPlayer():Name(), "aHUD28", 90, Scrh - HUDHeight+12, aHUD.Color, TEXT_ALIGN_LEFT, TEXT_ALIGN_TOP)
draw.SimpleText(aHUD.Cur.Job, "aHUD24", 90, Scrh - HUDHeight + 39, Color(200, 200, 200), TEXT_ALIGN_LEFT, TEXT_ALIGN_TOP)
draw.SimpleText(DarkRP.formatMoney(math.Round(aHUD.Cur.Wallet)), "aHUD22", 90, Scrh - HUDHeight + 61, Color(140, 140, 140), TEXT_ALIGN_LEFT, TEXT_ALIGN_TOP)
draw.SimpleText(DarkRP.formatMoney(math.Round(aHUD.Cur.Salary)), "aHUD22", HUDWidth -1, Scrh - HUDHeight + 61, Color(140, 140, 140), TEXT_ALIGN_RIGHT, TEXT_ALIGN_TOP)

if IsValid(LocalPlayer():GetActiveWeapon()) then
local mag_extra = LocalPlayer():GetAmmoCount(LocalPlayer():GetActiveWeapon():GetPrimaryAmmoType())
local ammo = (math.max((LocalPlayer():GetActiveWeapon():Clip1()), 0)) .. " / " .. mag_extra

if ammo != "0 / 0" then

draw.RoundedBox(0, Scrw - 110, Scrh - 50, 100, 40, Color(41, 41, 41, 255))
draw.RoundedBox(2, Scrw - 110, Scrh - 50, 100, 10, aHUD.Color)
draw.RoundedBox(4, Scrw - 108, Scrh - 38 , 96, 26, Color(62, 62, 62, 255))

draw.SimpleText(ammo, "aHUD28", (Scrw - 110) + 100/2, Scrh - 40, Color(190, 190, 190), TEXT_ALIGN_CENTER, TEXT_ALIGN_TOP)
end
end

aHUD.DarkRPFuncs()

aHUD.Arrested()
aHUD.AdminTell()
end


plyMeta.drawPlayerInfo = function(self)
if not IsValid(self) then return end

local pos = self:EyePos()
pos.z = pos.z + 10
pos = pos:ToScreen()

if not self:getDarkRPVar("wanted") then
pos.y = pos.y - 50
end

local strings = {self:Nick(), RPExtraTeams[self:Team()].name or "Unknown", self:Health(), self:getDarkRPVar("job") or team.GetName(self:Team())}
self.aHUDwidth = math.max(aHUD.GetLongestTextLength(strings, "aHUD22"), 100) + 52
self.aHUDcolour = Color(RPExtraTeams[self:Team()].color.r, RPExtraTeams[self:Team()].color.g, RPExtraTeams[self:Team()].color.b)

if not self.HUDPanel then

self.HUDPanel = vgui.Create("DPanel")
self.HUDPanel:ParentToHUD()

self.HUDPanel.Paint = function(this, w, h)
if not IsValid(self) then
this:Remove()
--self.HUDPanel = nil
return
end

if GAMEMODE.Config.showname then
local data = {
text = self:Nick(),
pos = {46, 10},
font = "aHUD22",
xalign = TEXT_ALIGN_LEFT,
yalign = TEXT_ALIGN_TOP,
color = color_black
}

draw.TextShadow(data, 1, 230)
draw.SimpleText(self:Nick(), "aHUD22", 46, 10, self.aHUDcolour, TEXT_ALIGN_LEFT, TEXT_ALIGN_TOP)
end

if GAMEMODE.Config.showhealth then
draw.RoundedBox(0, 45, 31, math.Clamp(self:Health(), 0, 100)/100*(w - 50), 2, self.aHUDcolour)
end

if GAMEMODE.Config.showjob then
local data = {
text = self:getDarkRPVar("job") or team.GetName(self:Team()),
pos = {46, 32},
font = "aHUD22",
xalign = TEXT_ALIGN_LEFT,
yalign = TEXT_ALIGN_TOP,
color = color_black
}
draw.TextShadow(data, 1, 230)
draw.SimpleText(self:getDarkRPVar("job") or team.GetName(self:Team()), "aHUD22", 46, 32, Color(170, 170, 170), TEXT_ALIGN_LEFT, TEXT_ALIGN_TOP)
end
end

self.HUDPanel.HUDAvatar = vgui.Create("AvatarCircleMask", self.HUDPanel)
self.HUDPanel.HUDAvatar:SetPlayer(self, 32)
self.HUDPanel.HUDAvatar:SetSize(38, 38)
self.HUDPanel.HUDAvatar:SetPos(4, 13)
self.HUDPanel.HUDAvatar:SetMaskSize(38 / 2)
end

self.HUDPanel:SetSize(self.aHUDwidth, 56)
self.HUDPanel:SetPos(pos.x - self.aHUDwidth/2, pos.y)

if self:getDarkRPVar("HasGunlicense") then
surface.SetMaterial(Page)
surface.SetDrawColor(255,255,255,255)
surface.DrawTexturedRect(pos.x-16, pos.y - 36, 32, 32)
end
end

plyMeta.drawWantedInfo = function(self)

if not self:Alive() then return end

local pos = self:EyePos()
if not pos:isInSight({LocalPlayer(), self}) then return end

pos.z = pos.z + 10
pos = pos:ToScreen()

if GAMEMODE.Config.showname then
local nick, plyTeam = self:Nick(), self:Team()
local data = {
text = nick,
pos = {pos.x, pos.y},
font = "aHUD22",
xalign = TEXT_ALIGN_CENTER,
yalign = TEXT_ALIGN_CENTER,
color = color_black
}
draw.TextShadow(data, 1, 200)
draw.SimpleText(nick, "aHUD22", pos.x, pos.y, RPExtraTeams[plyTeam] and RPExtraTeams[plyTeam].color or team.GetColor(plyTeam), TEXT_ALIGN_CENTER, TEXT_ALIGN_CENTER)
end

local wantedText = DarkRP.getPhrase("wanted", tostring(self:getDarkRPVar("wantedReason")))
local data = {
text = wantedText,
pos = {pos.x - 25, pos.y - 20},
font = "aHUD22",
xalign = TEXT_ALIGN_CENTER,
yalign = TEXT_ALIGN_CENTER,
color = color_black
}
draw.TextShadow(data, 1, 200)
draw.SimpleText(wantedText, "aHUD22", pos.x - 25, pos.y - 20, Color(215, 85, 80), TEXT_ALIGN_CENTER, TEXT_ALIGN_CENTER)
end

local function DrawEntityDisplay()
local shootPos = LocalPlayer():GetShootPos()
local aimVec = LocalPlayer():GetAimVector()


for k, ply in pairs(players or player.GetAll()) do
if ply == LocalPlayer() or not ply:Alive() or ply:GetNoDraw() then
if ply.HUDPanel then
ply.HUDPanel:Remove()
ply.HUDPanel = nil
end
continue
end
local hisPos = ply:GetShootPos()
if ply:getDarkRPVar("wanted") then
if ply.HUDPanel then
ply.HUDPanel:Remove()
ply.HUDPanel = nil
end
ply:drawWantedInfo()
continue
end

if GAMEMODE.Config.globalshow then
ply:drawPlayerInfo()
elseif hisPos:DistToSqr(shootPos) < 160000 then
local pos = hisPos - shootPos
local unitPos = pos:GetNormalized()
if unitPos:Dot(aimVec) > 0.95 then
local trace = util.QuickTrace(shootPos, pos, LocalPlayer())
if trace.Hit and trace.Entity ~= ply then
if ply.HUDPanel then
ply.HUDPanel:Remove()
ply.HUDPanel = nil
end
break
end
ply:drawPlayerInfo()
else
if ply.HUDPanel then
ply.HUDPanel:Remove()
ply.HUDPanel = nil
end
end
else
if ply.HUDPanel then
ply.HUDPanel:Remove()
ply.HUDPanel = nil
end
end
end

local tr = LocalPlayer():GetEyeTrace()

if IsValid(tr.Entity) and tr.Entity:isKeysOwnable() and tr.Entity:GetPos():DistToSqr(LocalPlayer():GetPos()) < 40000 then
tr.Entity:drawOwnableInfo()
end
end

local function DisplayNotify(msg)
local txt = msg:ReadString()
GAMEMODE:AddNotify(txt, msg:ReadShort(), msg:ReadLong())
surface.PlaySound("buttons/lightswitch2.wav")

MsgC(Color(255, 20, 20, 255), "[DarkRP] ", Color(200, 200, 200, 255), txt, "\n")
end
usermessage.Hook("_Notify", DisplayNotify)

hook.Add("HUDPaint", "aHUDPaint", function()
if not IsValid(LocalPlayer()) then return end

if aHUD.Cur.Health then
aHUD.Cur.Health = aHUD.Cur.Health or LocalPlayer():Health()
aHUD.Cur.Wallet = aHUD.Cur.Wallet or LocalPlayer():getDarkRPVar("money")
aHUD.Cur.Salary = aHUD.Cur.Salary or LocalPlayer():getDarkRPVar("salary")
aHUD.Cur.Job = aHUD.Cur.Job or LocalPlayer():getDarkRPVar("job")
else
aHUD.Cur.Health = LocalPlayer():Health()
aHUD.Cur.Wallet = LocalPlayer():getDarkRPVar("money")
aHUD.Cur.Salary = LocalPlayer():getDarkRPVar("salary")
aHUD.Cur.Job = LocalPlayer():getDarkRPVar("job")
end


DrawHUD()
DrawEntityDisplay()
end)

for k, ply in pairs(player.GetAll()) do
if ply.HUDPanel then
ply.HUDPanel:Remove()
ply.HUDPanel = nil
end
end

local hideHUDElements = {
["DarkRP_HUD"] = true,
["DarkRP_EntityDisplay"] = true,
["DarkRP_LocalPlayerHUD"] = true,
["DarkRP_Hungermod"] = true,
["DarkRP_Agenda"] = true,
CHudAmmo = true
}

hook.Add("HUDShouldDraw", "aHUDHideHUD", function(name)
if hideHUDElements[name] then return false end
end)

concommand.Add("aHUD_TestNotifications", function()
notification.AddLegacy( "Testing generic notification", 0, 10 )
notification.AddLegacy( "Testing error notification", 1, 10 )
notification.AddLegacy( "Testing undo notification", 2, 10 )
notification.AddLegacy( "Testing hint notification", 3, 10 )
notification.AddLegacy( "Testing cleanup notification", 4, 10 )
end)