cl_rkidnap

local LangTbl = RKidnapConfig.Language[RKidnapConfig.LanguageToUse]


local RKS_BoneManipulations = {
    ["Restrained"] = {
        ["ValveBiped.Bip01_R_UpperArm"] = Angle(-28,18,-21),
        ["ValveBiped.Bip01_L_Hand"] = Angle(0,0,119),
        ["ValveBiped.Bip01_L_Forearm"] = Angle(15,20,40),
        ["ValveBiped.Bip01_L_UpperArm"] = Angle(15, 26, 0),
        ["ValveBiped.Bip01_R_Forearm"] = Angle(0,50,0),
        ["ValveBiped.Bip01_R_Hand"] = Angle(45,34,-15),
        ["ValveBiped.Bip01_L_Finger01"] = Angle(0,50,0),
        ["ValveBiped.Bip01_R_Finger0"] = Angle(10,2,0),
        ["ValveBiped.Bip01_R_Finger1"] = Angle(-10,0,0),
        ["ValveBiped.Bip01_R_Finger11"] = Angle(0,-40,0),
        ["ValveBiped.Bip01_R_Finger12"] = Angle(0,-30,0)
    },
    ["HandsUp"] = {
        ["ValveBiped.Bip01_R_UpperArm"] = Angle(73,35,128),
        ["ValveBiped.Bip01_L_Hand"] = Angle(-12,12,90),
        ["ValveBiped.Bip01_L_Forearm"] = Angle(-28,-29,44),
        ["ValveBiped.Bip01_R_Forearm"] = Angle(-22,1,15),
        ["ValveBiped.Bip01_L_UpperArm"] = Angle(-77,-46,4),
        ["ValveBiped.Bip01_R_Hand"] = Angle(33,39,-21),
        ["ValveBiped.Bip01_L_Finger01"] = Angle(0,30,0),
        ["ValveBiped.Bip01_L_Finger1"] = Angle(0,45,0),
        ["ValveBiped.Bip01_L_Finger11"] = Angle(0,45,0),
        ["ValveBiped.Bip01_L_Finger2"] = Angle(0,45,0),
        ["ValveBiped.Bip01_L_Finger21"] = Angle(0,45,0),
        ["ValveBiped.Bip01_L_Finger3"] = Angle(0,45,0),
        ["ValveBiped.Bip01_L_Finger31"] = Angle(0,45,0),
        ["ValveBiped.Bip01_R_Finger0"] = Angle(-10,0,0),
        ["ValveBiped.Bip01_R_Finger11"] = Angle(0,30,0),
        ["ValveBiped.Bip01_R_Finger2"] = Angle(20,25,0),
        ["ValveBiped.Bip01_R_Finger21"] = Angle(0,45,0),
        ["ValveBiped.Bip01_R_Finger3"] = Angle(20,35,0),
        ["ValveBiped.Bip01_R_Finger31"] = Angle(0,45,0)
    }
}

net.Receive("rks_bonemanipulate", function()
    local Player, Type, Reset = net.ReadEntity(), net.ReadString(), net.ReadBool()

    if IsValid(Player) then
        for k,v in pairs(RKS_BoneManipulations[Type]) do
            local Bone = Player:LookupBone(k)
            if Bone then
                if Reset then
                    Player:ManipulateBoneAngles(Bone, Angle(0,0,0))
                else
                    Player:ManipulateBoneAngles(Bone, v)
                end
            end
        end
        if !Reset and Type == "Restrained" and table.HasValue(RKidnapConfig.FEMALE_MODELS, Player:GetModel()) then
            local LEFT_UP_ARM, RIGHT_UP_ARM, RIGHT_FORE_ARM = Player:LookupBone("ValveBiped.Bip01_L_UpperArm"), Player:LookupBone("ValveBiped.Bip01_R_UpperArm"), Player:LookupBone("ValveBiped.Bip01_R_Forearm")
            if LEFT_UP_ARM then
                Player:ManipulateBoneAngles(Player:LookupBone("ValveBiped.Bip01_L_UpperArm"), Angle(15, 23, 0))
            end
            if RIGHT_UP_ARM then
                Player:ManipulateBoneAngles(Player:LookupBone("ValveBiped.Bip01_R_UpperArm"), Angle(-28, 5, -21))
            end
            if RIGHT_FORE_ARM then
                Player:ManipulateBoneAngles(Player:LookupBone("ValveBiped.Bip01_R_Forearm"), Angle(0, 60, 10))
            end
        end
        if RKidnapConfig.DisablePlayerShadow then
            Player:DrawShadow(false)
        end
    end
end)

net.Receive("tbfy_surr", function(Player, len)
    local SurrTime = net.ReadFloat()
    if SurrTime == 0 then
        LocalPlayer().Surrendering = false
    else
        LocalPlayer().Surrendering = SurrTime
    end
end)

hook.Add("HUDPaint", "TBFY_Surr", function()
    local ST = LocalPlayer().Surrendering
    if ST then
        local W,H = ScrW(), ScrH()
        local TimeLeft = math.Round(ST - CurTime(),1)
        draw.SimpleTextOutlined("Surrendering - " .. TimeLeft,"rks_ko_text",ScrW()/2,ScrH()/2,Color(255,255,255,255), TEXT_ALIGN_CENTER, TEXT_ALIGN_CENTER,2,Color(0,0,0,255))
    end
end)

local PLAYER = FindMetaTable("Player")
//Whacky way to add text without overriding the function completely
hook.Add("loadCustomDarkRPItems", "rks_set_drawPINFO", function()
    local OldDrawPlayerInfo = PLAYER.drawPlayerInfo
    function RKS_AddInCuffs(self)
        if RKidnapConfig.DisplayOverheadRestrained and self:GetNWBool("rks_restrained", false) then
            local pos = self:EyePos()

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

            draw.DrawNonParsedText("Restrained", "DarkRPHUD2", pos.x + 1, pos.y - 19, Color(0,0,0,255), 1)
            draw.DrawNonParsedText("Restrained", "DarkRPHUD2", pos.x, pos.y - 20, Color(255,255,255,255) , 1)
        end
        OldDrawPlayerInfo(self)
    end
    PLAYER.drawPlayerInfo = RKS_AddInCuffs
end)

net.Receive("rks_update_ragdollcolor", function()
    local Player, Ragdoll = net.ReadEntity(), net.ReadEntity()

    Ragdoll.GetPlayerColor = function() return Player:GetPlayerColor() end
end)

net.Receive("rks_knockout", function()
    local KnockedOut = net.ReadBool()
    LocalPlayer().RKSKOStart = CurTime() + RKidnapConfig.KnockoutTime
    LocalPlayer().RKSKO = KnockedOut
    souvenir = { "will remember everything that happened", "will not remember anything" }
end)

net.Receive("rks_update_blindfold", function()
    local Blindfolded = net.ReadBool()
    LocalPlayer().RKSBlindfolded = Blindfolded
end)

net.Receive("rks_send_inspect_information", function()
    local Player, WepAmount = net.ReadEntity(), net.ReadFloat()

    local WepTbl = {}
    for i = 1, WepAmount do
        local TID = net.ReadFloat()
        local WepC = net.ReadString()

        if WepC and WepC != "" then
            WepTbl[TID] = WepC
        end
    end

    local InsMenu = vgui.Create("rks_inspect_menu")
    InsMenu:SetupInfo(Player, WepTbl)
    LocalPlayer().LastInspect = Player
end)

surface.CreateFont( "rks_ko_text", {
    font = "Verdana",
    size = 20,
    weight = 1000,
    antialias = true,
})

hook.Add("HUDPaintBackground", "RKS_PaintHUD", function()
    if LocalPlayer().RKSBlindfolded then
        local W,H = ScrW(), ScrH()
        draw.RoundedBox( 0, 0, 0, W, H, Color( 0, 0, 0, 255 ) )
    elseif LocalPlayer().RKSKO then
        local W,H = ScrW(), ScrH()
        local TimeLeft = math.Round(LocalPlayer().RKSKOStart - CurTime())
        draw.RoundedBox( 0, 0, 0, W, H, Color( 0, 0, 0, 255 ) )
        draw.SimpleText( "You" .. table.Random( souvenir ) .. ".", "rks_ko_text", 150, 150, color_white )
        draw.SimpleTextOutlined(string.format(LangTbl.KnockedOut, TimeLeft),"rks_ko_text",ScrW()/2,ScrH()/12,Color(255,255,255,255), TEXT_ALIGN_CENTER, TEXT_ALIGN_BOTTOM,2,Color(0,0,0,255))
    end
end)

local KeyToCheck = RKidnapConfig.KEY
hook.Add("KeyPress", "RKS_DragKeyPress", function(Player, Key)
    if Key == KeyToCheck then
        if !Player:InVehicle() and !Player:GetNWBool("rks_restrained", false) then
            local Trace = {}
            Trace.start = Player:GetShootPos();
            Trace.endpos = Trace.start + Player:GetAimVector() * 100;
            Trace.filter = Player;

            local Tr = util.TraceLine(Trace);
            local TEnt = Tr.Entity
            if IsValid(TEnt) and TEnt:IsPlayer() and TEnt:GetNWBool("rks_restrained", false) then
                local UnRestrainM = vgui.Create("DMenu")
                UnRestrainM:AddOption(LangTbl.UnRestrain, function() net.Start("rks_unrestrain") net.WriteEntity(TEnt) net.SendToServer() end)
                UnRestrainM:AddOption(LangTbl.Drag, function() net.Start("rks_drag") net.WriteEntity(TEnt) net.SendToServer() end)
                UnRestrainM:Open()

                UnRestrainM:SetPos(ScrW()/2,ScrH()/2)
            end
        end
    end
end)

surface.CreateFont( "rks_inspect_headline", {
    font = "Arial",
    size = 20,
    weight = 1000,
    antialias = true,
})

surface.CreateFont( "rks_inspect_information", {
    font = "Arial",
    size = 20,
    weight = 100,
    antialias = true,
})

surface.CreateFont( "rks_inspect_stealweapon", {
    font = "Arial",
    size = 15,
    weight = 1000,
    antialias = true,
})

surface.CreateFont( "rks_inspect_stealmoney", {
    font = "Arial",
    size = 14,
    weight = 1000,
    antialias = true,
})

surface.CreateFont( "rks_inspect_weaponname", {
    font = "Arial",
    size = 14,
    weight = 100,
    antialias = true,
})

local MainPanelColor = Color(255,255,255,200)
local HeaderColor = Color(50,50,50,255)
local SecondPanelColor = Color(215,215,220,255)
local ButtonColor = Color(50,50,50,255)
local ButtonColorHovering = Color(75,75,75,200)
local ButtonColorPressed = Color(150,150,150,200)
local ButtonOutline = Color(0,0,0,200)

local PANEL = {}

function PANEL:Init()
    self.ButtonText = ""
    self.BColor = ButtonColor
    self.Font = "rks_inspect_headline"
    self:SetText("")
end

function PANEL:UpdateColours()

    if self:IsDown() or self.m_bSelected then self.BColor = ButtonColorPressed return end
    if self.Hovered and !self:GetDisabled() then self.BColor = ButtonColorHovering return end

    self.BColor = ButtonColor
    return
end

function PANEL:SetBText(Text)
    self.ButtonText = Text
end

function PANEL:SetBFont(Font)
    self.Font = Font
end

function PANEL:Paint(W,H)
    draw.RoundedBox(4, 0, 0, W, H, self.BColor)
    draw.SimpleText(self.ButtonText, self.Font, W/2, H/2, Color(255,255,255,255), TEXT_ALIGN_CENTER, TEXT_ALIGN_CENTER )
end
vgui.Register( "rks_button", PANEL, "DButton")

local PANEL = {}

function PANEL:Init()
    self.Name = ""
    self.WID = 0

    self.StealItem = vgui.Create("rks_button", self)
    self.StealItem:SetBText("STEAL")
    self.StealItem:SetBFont("rks_inspect_stealweapon")
    self.StealItem.DoClick = function() if !RKidnapConfig.AllowStealingWeapons then LocalPlayer():ChatPrint("The server owner has disabled stealing weapons.") return end net.Start("rks_stealweapon") net.WriteEntity(LocalPlayer().LastInspect) net.WriteFloat(self.WID) net.SendToServer() self:Remove() end

    self.WModel = vgui.Create( "ModelImage", self )
end

function PANEL:Paint(W,H)
    draw.SimpleText(self.Name, "rks_inspect_weaponname", W/2, 0, Color(0,0,0,255), TEXT_ALIGN_CENTER, TEXT_ALIGN_TOP )
end

function PANEL:PerformLayout(W,H)
    self.StealItem:SetPos(2.5,H-20)
    self.StealItem:SetSize(W-5,15)

    self.WModel:SetPos(2.5,10)
    self.WModel:SetSize(W-15,W-15)
end

function PANEL:SetInfo(Wep, ID)
    local SWEPTable = weapons.GetStored(Wep)
    if SWEPTable then
        if SWEPTable.WorldModel then
            self.WModel:SetModel(SWEPTable.WorldModel)
        end
        self.Name = SWEPTable.PrintName
    else
        self.Name = ""
    end
    self.WID = ID
end
vgui.Register( "rks_weapon", PANEL)

local PANEL = {}

function PANEL:Init()
    self:ShowCloseButton(false)
    self:SetTitle("")
    self:MakePopup()

    self.Name = "INVALID"
    self.Job = "INVALID"
    self.SteamID = "INVALID"
    self.Wallet = 0
    self.WepItems = {}

    self.TopDPanel = vgui.Create("DPanel", self)
    self.TopDPanel.Paint = function(selfp, W,H)
        draw.RoundedBoxEx(8, 0, 0, W, H, HeaderColor, true, true, false, false)
        draw.SimpleText("Inspecting: " .. self.Name, "rks_inspect_headline", W/2, H/2, Color( 255, 255, 255, 255 ), TEXT_ALIGN_CENTER, TEXT_ALIGN_CENTER )
    end

    self.InfoDPanel = vgui.Create("DPanel", self)
    self.InfoDPanel.Paint = function(selfp, W,H)
        draw.RoundedBoxEx(8, 0, 0, W, H, SecondPanelColor, false, false, true, true)
        local TW, TH = surface.GetTextSize("Name: ")
        draw.SimpleText("Name:", "rks_inspect_headline", 5, 10, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
        draw.SimpleText(self.Name, "rks_inspect_information", 5+TW, 10, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
        TW, TH = surface.GetTextSize("SteamID: ")
        draw.SimpleText("SteamID:", "rks_inspect_headline", 5, 25, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
        draw.SimpleText(self.SteamID, "rks_inspect_information", 5 + TW, 25, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
        TW, TH = surface.GetTextSize("Job: ")
        draw.SimpleText("Job:", "rks_inspect_headline", 5, 40, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
        draw.SimpleText(self.Job, "rks_inspect_information", 5 + TW, 40, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
        TW, TH = surface.GetTextSize("Wallet: ")
        draw.SimpleText("Wallet:", "rks_inspect_headline", 5, 55, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
        draw.SimpleText(self.Wallet, "rks_inspect_information", 5 + TW, 55, Color(0,0,0,255), TEXT_ALIGN_LEFT, TEXT_ALIGN_CENTER )
    end

    self.WeaponHeader = vgui.Create("DPanel", self)
    self.WeaponHeader.Paint = function(selfp, W,H)
        draw.RoundedBoxEx(8, 0, 0, W, H, HeaderColor, true, true, false, false)
        draw.SimpleText("Weapon List", "rks_inspect_headline", W/2, H/2, Color( 255, 255, 255, 255 ), TEXT_ALIGN_CENTER, TEXT_ALIGN_CENTER )
    end

    self.WeaponList = vgui.Create("DScrollPanel", self)
    self.WeaponList.Paint = function(selfp, W, H)
        draw.RoundedBoxEx(4, 0, 0, W-15, H, SecondPanelColor, false, false, true, true)
    end

    self.WeaponList.VBar.Paint = function() end
    self.WeaponList.VBar.btnUp.Paint = function() end
    self.WeaponList.VBar.btnDown.Paint = function() end
    self.WeaponList.VBar.btnGrip.Paint = function() end

    if DarkRP then
        self.StealMoney = vgui.Create("rks_button", self.InfoDPanel)
        self.StealMoney:SetBText("STEAL")
        self.StealMoney:SetBFont("rks_inspect_stealmoney")
        self.StealMoney.DoClick = function()
            if IsValid(LocalPlayer().LastInspect) then
                net.Start("rks_stealcash")
                    net.WriteEntity(LocalPlayer().LastInspect)
                    if !RKidnapConfig.MoneyStealRandomAmount then
                        net.WriteFloat(self.StealMoneyAmount:GetValue())
                    end
                net.SendToServer()

                timer.Simple(1, function()
                    if IsValid(self) then
                        self.Wallet = DarkRP.formatMoney(LocalPlayer().LastInspect:getDarkRPVar("money"))
                        self:PerformLayout()
                    end
                end)
            end
        end

        if !RKidnapConfig.MoneyStealRandomAmount then
            self.StealMoneyAmount = vgui.Create("DTextEntry", self.InfoDPanel)
            self.StealMoneyAmount:SetNumeric(true)
        end
    end

    self.CloseButton = vgui.Create("rks_button", self)
    self.CloseButton:SetBText("X")
    self.CloseButton.DoClick = function() self:Remove() end
end

local function IsJobRanksLoadout(Player, Wep)
    local Rank = Player:GetJobRank()
    local Job = Player:Team()
    local WMatched = false

    if JobRanks and JobRanks[Job] then
        local JobTbl = JobRanks[Job]
        if JobTbl.ExtraLoadoutSingleRank and JobTbl.ExtraLoadoutSingleRank[Rank] then
            local SLoadout = JobTbl.ExtraLoadoutSingleRank[Rank]
            for k,v in pairs(SLoadout) do
                if v == Wep then
                    WMatched = true
                    break
                end
            end
        end
        if !WMatched and JobTbl.ExtraLoadout then
            local RLoadout = JobTbl.ExtraLoadout
            for k,v in pairs(RLoadout) do
                if v <= Rank and k == Wep then
                    WMatched = true
                    break
                end
            end
        end
    end
    return WMatched
end

function PANEL:SetupInfo(Player, WepTbl)
    self.Name = Player:Nick()
    self.SteamID = Player:SteamID()

    local jobTable = {}
    if DarkRP then
        self.Job = Player:getDarkRPVar("job")
        self.Wallet = DarkRP.formatMoney(Player:getDarkRPVar("money"))
        jobTable = Player:getJobTable()
    end

    for k,v in pairs(WepTbl) do
        if !RKidnapConfig.BlackListedWeapons[v] then
            if RKidnapConfig.AllowStealingJobWeapons or (jobTable.weapons and !table.HasValue(jobTable.weapons, v) and (!JobRanksConfig or !IsJobRanksLoadout(Player, v))) then
                local Wep = vgui.Create("rks_weapon", self.WeaponList)
                Wep:SetInfo(v, k)

                self.WepItems[k] = Wep
            end
        end
    end
end

local TopH = 25
local InfoH =75
local WeaponH = 25
local WeaponListH = 180
function PANEL:Paint(W,H)
    draw.RoundedBoxEx(8, 0, TopH, W, H-TopH, MainPanelColor,false,false,true,true)
end

local WepsPerLine = 4
local Width, Height, Padding = 300, 330, 5
function PANEL:PerformLayout()
    self:SetPos(ScrW()/2-Width/2, ScrH()/2-Height/2)
    self:SetSize(Width, TopH+InfoH+WeaponH+WeaponListH)

    self.TopDPanel:SetPos(0,0)
    self.TopDPanel:SetSize(Width,TopH)

    self.InfoDPanel:SetPos(Padding,TopH+Padding)
    self.InfoDPanel:SetSize(Width-Padding*2,InfoH-Padding*2)

    self.WeaponHeader:SetPos(Padding,TopH+InfoH)
    self.WeaponHeader:SetSize(Width-Padding*2,WeaponH)

    self.WeaponList:SetPos(Padding,TopH+InfoH+WeaponH)
    self.WeaponList:SetSize(Width+Padding, WeaponListH-Padding)

    local WAvailable = self.WeaponList:GetWide()-15
    local WepWSize = WAvailable/WepsPerLine

    local NumSlots = 0
    local CRow = 0
    for k,v in pairs(self.WepItems) do
        if IsValid(v) then
            if NumSlots >= WepsPerLine then
                NumSlots = 0
                CRow = CRow + 1
            end
            v:SetPos(WepWSize*(NumSlots),CRow*(WepWSize+15))
            v:SetSize(WepWSize,WepWSize+15)
            NumSlots = NumSlots + 1
        end
    end

    if self.StealMoney then
        local TW, TH = surface.GetTextSize("Wallet:  " .. self.Wallet)
        if !RKidnapConfig.MoneyStealRandomAmount then
            if self.StealMoneyAmount then
                self.StealMoneyAmount:SetPos(TW, 49)
                self.StealMoneyAmount:SetSize(45,13)
            end

            self.StealMoney:SetPos(TW+50, 49)
            self.StealMoney:SetSize(45,13)
        else
            self.StealMoney:SetPos(TW, 49)
            self.StealMoney:SetSize(45,13)
        end
    end

    self.CloseButton:SetPos(Width-TopH,TopH/2-9)
    self.CloseButton:SetSize(20, 20)
end
vgui.Register("rks_inspect_menu", PANEL, "DFrame")