main.lua

-- ============================================================
--  Entity Inspector | client/main.lua
--  Ace permission: motion_entityselector.use
-- ============================================================

local isAdmin     = false
local panelOpen   = false
local selectedEnt = nil

local scanning    = false
local pressedAt   = 0
local wasScanning = false

-- -- Key binding -----------------------------------------------
-- RegisterKeyMapping is the only correct way to detect MMB in FiveM.
-- Control 96 = INPUT_VEH_DUCK, NOT middle mouse.

RegisterCommand('+mmb_scan', function()
    if panelOpen then return end
    scanning  = true
    pressedAt = GetGameTimer()
end, false)

RegisterCommand('-mmb_scan', function()
    scanning = false
end, false)

RegisterKeyMapping('+mmb_scan', 'Entity Inspector Scan', 'MOUSE_BUTTON', 'MOUSE_MIDDLE')

-- -- Resource lifecycle ----------------------------------------

AddEventHandler('onClientResourceStart', function(res)
    if res ~= GetCurrentResourceName() then return end
    SetNuiFocus(false, false)
    SendNUIMessage({ type = 'close', data = {} })
    panelOpen   = false
    selectedEnt = nil
    scanning    = false
    wasScanning = false
    TriggerServerEvent('motion_entityselector:checkAdmin')
end)

AddEventHandler('onClientResourceStop', function(res)
    if res ~= GetCurrentResourceName() then return end
    SetNuiFocus(false, false)
    SendNUIMessage({ type = 'close', data = {} })
end)

RegisterNetEvent('motion_entityselector:setAdmin', function(state)
    isAdmin = state
    print(isAdmin
        and "^2[EntitySelector] Admin access granted.^7"
        or  "^3[EntitySelector] No admin access.^7")
end)

-- -- Helpers --------------------------------------------------

local function getTypeName(entity)
    local t = GetEntityType(entity)
    if t == 1 then return "Ped"
    elseif t == 2 then return "Vehicle"
    elseif t == 3 then return "Object"
    else return "Unknown" end
end

local function f2(n) return math.floor((n or 0) * 100) / 100 end
local function f1(n) return math.floor((n or 0) * 10)  / 10  end

-- -- Raycast ---------------------------------------------------

local function raycast()
    local cam  = GetGameplayCamCoords()
    local rot  = GetGameplayCamRot(2)
    local fwd  = vector3(
        math.sin(-math.rad(rot.z)) * math.cos(math.rad(rot.x)),
        math.cos(-math.rad(rot.z)) * math.cos(math.rad(rot.x)),
        math.sin( math.rad(rot.x))
    )
    local dest = cam + fwd * 50.0
    local ray  = StartShapeTestRay(
        cam.x, cam.y, cam.z,
        dest.x, dest.y, dest.z,
        -1, PlayerPedId(), 4
    )
    local _, hit, _, _, entity = GetShapeTestResult(ray)
    if hit == 1 and entity and entity ~= 0
        and DoesEntityExist(entity)
        and entity ~= PlayerPedId()
    then
        return entity
    end
    return nil
end

-- -- Bounding box ---------------------------------------------
-- Confirmed correct signature from FiveM docs:
-- GetEntityMatrix returns: forwardVector, rightVector, upVector, position
-- All four are proper vector3 values with .x .y .z

local function drawBoundingBox(entity)
    if not (entity and DoesEntityExist(entity)) then return end

    -- Guard the whole thing   streaming objects can go invalid mid-call
    local ok, err = pcall(function()
        local model = GetEntityModel(entity)
        if not model or model == 0 then return end

        local min, max = GetModelDimensions(model)

        -- FiveM confirmed order: forward, right, up, pos
        local fwd, right, up, pos = GetEntityMatrix(entity)

        local function corner(lx, ly, lz)
            -- lx = right axis, ly = forward axis, lz = up axis
            return vector3(
                pos.x + right.x*lx + fwd.x*ly + up.x*lz,
                pos.y + right.y*lx + fwd.y*ly + up.y*lz,
                pos.z + right.z*lx + fwd.z*ly + up.z*lz
            )
        end

        local c1 = corner(min.x, min.y, min.z)
        local c2 = corner(max.x, min.y, min.z)
        local c3 = corner(max.x, max.y, min.z)
        local c4 = corner(min.x, max.y, min.z)
        local c5 = corner(min.x, min.y, max.z)
        local c6 = corner(max.x, min.y, max.z)
        local c7 = corner(max.x, max.y, max.z)
        local c8 = corner(min.x, max.y, max.z)

        local r, g, b, a = 0, 210, 255, 255

        -- Bottom face
        DrawLine(c1.x,c1.y,c1.z, c2.x,c2.y,c2.z, r,g,b,a)
        DrawLine(c2.x,c2.y,c2.z, c3.x,c3.y,c3.z, r,g,b,a)
        DrawLine(c3.x,c3.y,c3.z, c4.x,c4.y,c4.z, r,g,b,a)
        DrawLine(c4.x,c4.y,c4.z, c1.x,c1.y,c1.z, r,g,b,a)
        -- Top face
        DrawLine(c5.x,c5.y,c5.z, c6.x,c6.y,c6.z, r,g,b,a)
        DrawLine(c6.x,c6.y,c6.z, c7.x,c7.y,c7.z, r,g,b,a)
        DrawLine(c7.x,c7.y,c7.z, c8.x,c8.y,c8.z, r,g,b,a)
        DrawLine(c8.x,c8.y,c8.z, c5.x,c5.y,c5.z, r,g,b,a)
        -- Vertical edges
        DrawLine(c1.x,c1.y,c1.z, c5.x,c5.y,c5.z, r,g,b,a)
        DrawLine(c2.x,c2.y,c2.z, c6.x,c6.y,c6.z, r,g,b,a)
        DrawLine(c3.x,c3.y,c3.z, c7.x,c7.y,c7.z, r,g,b,a)
        DrawLine(c4.x,c4.y,c4.z, c8.x,c8.y,c8.z, r,g,b,a)
    end)

    if not ok then
        -- Entity went invalid mid-draw, silently skip
    end
end

-- -- Floating label --------------------------------------------

local function drawLabel(entity)
    if not (entity and DoesEntityExist(entity)) then return end

    local pos = GetEntityCoords(entity, false)

    -- World3dToScreen2d returns (bool, float, float) but in some FiveM builds
    -- the bool is 0/1 not true/false, so test both.
    local onScreen, sx, sy = World3dToScreen2d(pos.x, pos.y, pos.z + 1.5)
    if not onScreen or onScreen == 0 or not sx or not sy then return end
    if sx ~= sx or sy ~= sy then return end  -- NaN guard

    local cam  = GetGameplayCamCoords()
    local dist = #(pos - cam)
    if dist < 0.01 then return end
    local scale = math.min(0.5, math.max(0.22, (1.0 / dist) * 2.5))

    -- Resolve the name fully before touching any native text-pipeline state.
    local name = "unknown"
    pcall(function()
        local n = GetEntityArchetypeName(entity)
        if type(n) == "string" and n ~= "" then
            name = n
        else
            local h = GetEntityModel(entity)
            if h and h ~= 0 then
                name = string.format("0x%08X", h)
            end
        end
    end)

    -- Wrap the entire Begin/Add/End chain so that if any native crashes it
    -- doesn't leave the text command pipeline in a broken state.
    pcall(function()
        SetTextScale(scale, scale)
        SetTextFont(4)
        SetTextColour(0, 210, 255, 255)
        SetTextOutline()
        SetTextCentre(true)
        BeginTextCommandDisplayText("STRING")
        AddTextComponentSubstringPlayerName(name)
        EndTextCommandDisplayText(sx, sy)
    end)
end

-- -- Collect data ----------------------------------------------

local function collectData(entity)
    if not DoesEntityExist(entity) then return nil end

    local pos     = GetEntityCoords(entity, true)
    local entType = getTypeName(entity)

    local model = "unknown"
    pcall(function()
        local n = GetEntityArchetypeName(entity)
        if n and n ~= "" then model = n
        else model = string.format("0x%08X", GetEntityModel(entity)) end
    end)

    local health, maxHp, heading = 0, 200, 0.0
    pcall(function()
        health  = GetEntityHealth(entity)
        maxHp   = GetEntityMaxHealth(entity)
        heading = GetEntityHeading(entity)
    end)

    local isNet = false
    local netIdStr, ownerStr = nil, "N/A"
    pcall(function()
        isNet = NetworkGetEntityIsNetworked(entity)
        if isNet then
            local ok, nid = pcall(NetworkGetNetworkIdFromEntity, entity)
            if ok and nid and nid > 0 then netIdStr = tostring(nid) end
            local ok2, oid = pcall(NetworkGetEntityOwner, entity)
            if ok2 and oid and oid >= 0 then
                ownerStr = GetPlayerName(oid) or ("Player " .. tostring(oid))
            end
        end
    end)

    local modelHash = " "
    pcall(function() modelHash = tostring(GetEntityModel(entity)) end)

    local d = {
        modelName  = model,
        entityType = entType,
        handle     = entity,
        netId      = netIdStr,
        modelHash  = modelHash,
        health     = health,
        maxHealth  = maxHp,
        heading    = f1(heading),
        pos        = { x = f2(pos.x), y = f2(pos.y), z = f2(pos.z) },
        networked  = isNet,
        owner      = ownerStr,
        extras     = {},
    }

    if entType == "Vehicle" then
        pcall(function()
            d.extras = {
                body   = math.floor(GetVehicleBodyHealth(entity)),
                engine = math.floor(GetVehicleEngineHealth(entity)),
                tank   = math.floor(GetVehiclePetrolTankHealth(entity)),
            }
        end)
    elseif entType == "Ped" then
        pcall(function()
            d.extras = {
                armour   = GetPedArmour(entity),
                isPlayer = IsPedAPlayer(entity),
            }
        end)
    end

    return d
end

-- -- Panel -----------------------------------------------------

local function openPanel(ent)
    panelOpen   = true
    selectedEnt = ent
    SetNuiFocus(true, true)
    local d = collectData(ent)
    if d then SendNUIMessage({ type = 'open', data = d }) end
end

local function closePanel()
    panelOpen   = false
    selectedEnt = nil
    SetNuiFocus(false, false)
    SendNUIMessage({ type = 'close', data = {} })
end

-- -- SCAN LOOP -------------------------------------------------
-- Wait(50) = 20 ticks/sec. Hold MMB to see box. Release to open panel.

CreateThread(function()
    local hover = nil

    while true do
        Wait(50)

        if not isAdmin or panelOpen then
            hover       = nil
            wasScanning = false
            goto continue
        end

        if scanning then
            wasScanning = true
            hover = raycast()
            if hover and DoesEntityExist(hover) then
                drawBoundingBox(hover)
                drawLabel(hover)
            else
                hover = nil
            end
        else
            if wasScanning then
                wasScanning = false
                if hover and DoesEntityExist(hover) then
                    openPanel(hover)
                end
                hover = nil
            end
        end

        ::continue::
    end
end)

-- -- Live refresh ----------------------------------------------

CreateThread(function()
    while true do
        Wait(500)
        if not panelOpen then goto skip end
        if selectedEnt and DoesEntityExist(selectedEnt) then
            local d = collectData(selectedEnt)
            if d then SendNUIMessage({ type = 'update', data = d }) end
        elseif selectedEnt then
            closePanel()
        end
        ::skip::
    end
end)

-- -- ESC close -------------------------------------------------

CreateThread(function()
    while true do
        Wait(0)
        if panelOpen and IsControlJustReleased(0, 200) then
            closePanel()
        end
    end
end)

-- -- NUI Callbacks ---------------------------------------------

RegisterNUICallback('close', function(_, cb)
    closePanel()
    cb('ok')
end)

print("^2[EntitySelector] Loaded   Hold MMB to highlight | Release to inspect | Ace: motion_entityselector.use^7")