ESX = nil
local Weapons = {}
local Loaded = false
-----------------------------------------------------------
-----------------------------------------------------------
Citizen.CreateThread(function()
    while ESX == nil do
        TriggerEvent('esx:getSharedObject', function(obj) ESX = obj end)
        Wait(0)
    end

    Citizen.SetTimeout(8000, function()
      if not Loaded then
        print("Force Load")
        Loaded=true
        RemoveGears()
        SetGears()
      end
    end)
    while not Loaded do
        Wait(250)
    end

    while true do
        local playerPed = GetPlayerPed(-1)
        for i = 1, #Config.RealWeapons, 1 do

            local weaponHash = GetHashKey(Config.RealWeapons[i].name)
            if HasPedGotWeapon(playerPed, weaponHash, false) then
                local onPlayer = false
                for k, entity in pairs(Weapons) do
                    if entity then
                        if k == Config.RealWeapons[i].name then
                            onPlayer = true
                            break
                        end
                    end
                end
                local hash = GetSelectedPedWeapon(playerPed)
                if not onPlayer and weaponHash ~= hash then
                    SetGear(Config.RealWeapons[i].name)
                elseif onPlayer and weaponHash == hash then
                    RemoveGear(Config.RealWeapons[i].name)
                end
            else
                RemoveGear(Config.RealWeapons[i].name)
            end
        end
        Citizen.Wait(1000)
    end
end)
-----------------------------------------------------------
-----------------------------------------------------------
AddEventHandler('skinchanger:modelLoaded', function()
    RemoveGears()
    SetGears()
    Loaded = true
end)
-----------------------------------------------------------
-----------------------------------------------------------
RegisterNetEvent('esx:removeWeapon')
AddEventHandler('esx:removeWeapon', function(weaponName)
    RemoveGear(weaponName)
end)
-----------------------------------------------------------
-----------------------------------------------------------
-- Remove only one weapon that's on the ped
function RemoveGear(weapon)
    local _Weapons = {}
    for weaponName, entity in pairs(Weapons) do
        if weaponName ~= weapon then
            _Weapons[weaponName] = entity
        else
            ESX.Game.DeleteObject(entity)
        end
    end

    Weapons = _Weapons
end

-----------------------------------------------------------
-----------------------------------------------------------
-- Remove all weapons that are on the ped
function RemoveGears()
    for weaponName, entity in pairs(Weapons) do
        ESX.Game.DeleteObject(entity)
    end
    Weapons = {}
end

-----------------------------------------------------------
-----------------------------------------------------------
-- Add one weapon on the ped
function SetGear(weapon)
    local bone = nil
    local boneX = 0.0
    local boneY = 0.0
    local boneZ = 0.0
    local boneXRot = 0.0
    local boneYRot = 0.0
    local boneZRot = 0.0
    local playerPed = GetPlayerPed(-1)
    local model = nil
    local playerData = ESX.GetPlayerData()

    for i = 1, #Config.RealWeapons, 1 do
        if Config.RealWeapons[i].name == weapon then
            bone = Config.RealWeapons[i].bone
            boneX = Config.RealWeapons[i].x
            boneY = Config.RealWeapons[i].y
            boneZ = Config.RealWeapons[i].z
            boneXRot = Config.RealWeapons[i].xRot
            boneYRot = Config.RealWeapons[i].yRot
            boneZRot = Config.RealWeapons[i].zRot
            model = Config.RealWeapons[i].model
            break
        end
    end

    ESX.Game.SpawnObject(model, {
        x = x,
        y = y,
        z = z
    }, function(obj)
        local playerPed = GetPlayerPed(-1)
        local boneIndex = GetPedBoneIndex(playerPed, bone)
        local bonePos = GetWorldPositionOfEntityBone(playerPed, boneIndex)
        AttachEntityToEntity(obj, playerPed, boneIndex, boneX, boneY, boneZ, boneXRot, boneYRot, boneZRot, false, false, false, false, 2, true)
        Weapons[weapon] = obj
    end)
end

-----------------------------------------------------------
-----------------------------------------------------------
-- Add all the weapons in the xPlayer's loadout
-- on the ped
function SetGears()
    local bone = nil
    local boneX = 0.0
    local boneY = 0.0
    local boneZ = 0.0
    local boneXRot = 0.0
    local boneYRot = 0.0
    local boneZRot = 0.0
    local playerPed = GetPlayerPed(-1)
    local model = nil
    local playerData = ESX.GetPlayerData()
    local weapon = nil

    for i = 1, #playerData.loadout, 1 do

        for j = 1, #Config.RealWeapons, 1 do
            if Config.RealWeapons[j].name == playerData.loadout[i].name then

                bone = Config.RealWeapons[j].bone
                boneX = Config.RealWeapons[j].x
                boneY = Config.RealWeapons[j].y
                boneZ = Config.RealWeapons[j].z
                boneXRot = Config.RealWeapons[j].xRot
                boneYRot = Config.RealWeapons[j].yRot
                boneZRot = Config.RealWeapons[j].zRot
                model = Config.RealWeapons[j].model
                weapon = Config.RealWeapons[j].name

                break
            end
        end

        local _wait = true

        ESX.Game.SpawnObject(model, {
            x = x,
            y = y,
            z = z
        }, function(obj)

            local playerPed = GetPlayerPed(-1)
            local boneIndex = GetPedBoneIndex(playerPed, bone)
            local bonePos = GetWorldPositionOfEntityBone(playerPed, boneIndex)

            AttachEntityToEntity(obj, playerPed, boneIndex, boneX, boneY, boneZ, boneXRot, boneYRot, boneZRot, false, false, false, false, 2, true)

            Weapons[weapon] = obj
            _wait = false
        end)

        while _wait do
            Wait(0)
        end
    end
end