local isActive = false
local isDisabled = false

exports('disableTargeting', function(state)
    if state then
        isActive = false
    end

    isDisabled = state
end)

local hasFocus = false

local function setNuiFocus(state, cursor)
    if state then SetCursorLocation(0.5, 0.5) end

    hasFocus = state
    SetNuiFocus(state, cursor or false)
    SetNuiFocusKeepInput(state)
end

local SendNuiMessage = SendNuiMessage
local GetEntityCoords = GetEntityCoords
local RaycastFromCamera = RaycastFromCamera
local GetEntityType = GetEntityType
local HasEntityClearLosToEntity = HasEntityClearLosToEntity
local GetCurrentZone = GetCurrentZone
local PlayerHasGroups = PlayerHasGroups or function() return true end
local PlayerHasItems = PlayerHasItems
local GetEntityBoneIndexByName = GetEntityBoneIndexByName
local GetEntityBonePosition_2 = GetEntityBonePosition_2
local next = next
local GetEntityModel = GetEntityModel
local GetEntityOptions = GetEntityOptions
local IsDisabledControlJustPressed = IsDisabledControlJustPressed
local DisableControlAction = DisableControlAction
local DisablePlayerFiring = DisablePlayerFiring
local options = {}
local currentTarget = {}

-- Toggle ox_target, instead of holding the hotkey
local toggleHotkey = GetConvarInt('ox_target:toggleHotkey', 0) == 1
local mouseButton = GetConvarInt('ox_target:leftClick', 1) == 1 and 24 or 25
local debug = GetConvarInt('ox_target:debug', 0) == 1

local function enableTargeting()
    local sleep = 1000
    if isDisabled or isActive or IsNuiFocused() or IsPauseMenuActive() then return end
    SendNuiMessage('{"event": "visible", "state": true}')

    isActive = true
    local flag, hit, entityHit, endCoords, distance, currentZone, nearbyZones, lastEntity, entityType, entityModel, hasTick = 511
    local getNearbyZones, drawSprites = DrawSprites()

    while isActive do
        local playerCoords = GetEntityCoords(cache.ped)
        hit, entityHit, endCoords = RaycastFromCamera(flag)
        entityType = entityHit ~= 0 and GetEntityType(entityHit) or 0
        distance = #(playerCoords - endCoords)

        if entityType == 0 then
            local _flag = flag == 511 and 26 or 511
            local _hit, _entityHit, _endCoords = RaycastFromCamera(_flag)
            local _distance = #(playerCoords - _endCoords)

            if _distance < distance then
                flag, hit, entityHit, endCoords, distance = _flag, _hit, _entityHit, _endCoords, _distance
                entityType = entityHit ~= 0 and GetEntityType(entityHit) or 0
            end
        end

        if hit and distance < 7 then
            local newOptions
            local lastZone = currentZone

            if getNearbyZones then
                ---@type CZone[]?, CZone?
                nearbyZones, currentZone = getNearbyZones(endCoords)
            else
                ---@type CZone?
                currentZone = GetCurrentZone(endCoords)
            end

            if lastZone ~= currentZone or entityHit ~= lastEntity then
                if next(options) then
                    table.wipe(options)
                    SendNuiMessage('{"event": "leftTarget"}')
                end

                if flag ~= 511 then
                    entityHit = HasEntityClearLosToEntity(entityHit, cache.ped, 7) and entityHit or 0
                end

                if lastEntity ~= entityHit and debug then
                    if lastEntity then
                        SetEntityDrawOutline(lastEntity, false)
                    end

                    if entityType ~= 1 then
                        SetEntityDrawOutline(entityHit, true)
                    end
                end

                if entityHit ~= 0 then
                    local success, result = pcall(GetEntityModel, entityHit)
                    entityModel = success and result

                    if entityModel then
                        newOptions = GetEntityOptions(entityHit, entityType, entityModel)
                    end

                end
            end

            options = newOptions or options or {}

            if currentZone then
                if (not newOptions and currentZone.id ~= currentTarget?.zone) or (lastEntity ~= entityHit) then
                    newOptions = options
                end

                currentTarget.zone = currentZone.id
                options.zone = currentZone.options
            elseif currentTarget.zone then
                currentTarget.zone = nil
                options.zone = nil
            end

            lastEntity = entityHit
            currentTarget.entity = entityHit
            currentTarget.coords = endCoords
            currentTarget.distance = distance
            local hidden = 0
            local totalOptions = 0

            for _, v in pairs(options) do
                local optionCount = #v
                totalOptions += optionCount

                for i = 1, optionCount do
                    local option = v[i]
                    local hide

                    if option.distance and distance > option.distance then
                        hide = true
                    end

                    if option.groups and not PlayerHasGroups(option.groups) then
                        hide = true
                    end

                    if option.items and not PlayerHasItems(option.items, option.anyItem) then
                        hide = true
                    end

                    local bone = option.bones

                    if bone then
                        local _type = type(bone)

                        if _type == 'string' then
                            local boneId = GetEntityBoneIndexByName(entityHit, bone)

                            if boneId ~= -1 and #(endCoords - GetEntityBonePosition_2(entityHit, boneId)) <= 2 then
                                bone = boneId
                            else
                                hide = true
                            end
                        elseif _type == 'table' then
                            local closestBone, boneDistance

                            for j = 1, #bone do
                                local boneId = GetEntityBoneIndexByName(entityHit, bone[j])

                                if boneId ~= -1 then
                                    local dist = #(endCoords - GetEntityBonePosition_2(entityHit, boneId))

                                    if dist <= (boneDistance or 1) then
                                        closestBone = boneId
                                        boneDistance = dist
                                    end
                                end
                            end

                            if closestBone then
                                bone = closestBone
                            else
                                hide = true
                            end
                        end
                    end

                    if not hide and option.canInteract then
                        local success, resp = pcall(option.canInteract, entityHit, distance, endCoords, option.name, bone)
                        hide = not success or not resp
                    end

                    if not newOptions and v[i].hide ~= hide then
                        newOptions = options
                    end

                    v[i].hide = hide

                    if hide then hidden += 1 end
                end
            end

            if newOptions and next(newOptions) then
                options = newOptions

                if hidden == totalOptions then
                    SendNuiMessage('{"event": "leftTarget"}')
                else
                    SendNuiMessage(json.encode({
                        event = 'setTarget',
                        options = options
                    }, { sort_keys=true }))
                end
            end
        elseif lastEntity then
            if debug then SetEntityDrawOutline(lastEntity, false) end
            if options then table.wipe(options) end
            SendNuiMessage('{"event": "leftTarget"}')
            lastEntity = nil
        else Wait(50) end

        if toggleHotkey and IsPauseMenuActive() then
            isActive = false
        end

        if not hasTick then
            hasTick = true

            CreateThread(function()
                while isActive do
                    if debug then
                        ---@diagnostic disable-next-line: param-type-mismatch
                        DrawMarker(28, endCoords.x, endCoords.y, endCoords.z, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.2, 0.2, 0.2, 255, 42, 24, 100, false, false, 0, true, false, false, false)
                    end

                    if nearbyZones then
                        drawSprites(endCoords)
                    end

                    DisablePlayerFiring(cache.playerId, true)
                    DisableControlAction(0, 25, true)

                    if hasFocus then
                        DisableControlAction(0, 1, true)
                        DisableControlAction(0, 2, true)

                        if options and IsDisabledControlJustPressed(0, 25) then
                            setNuiFocus(false, false)
                        end
                    elseif options and IsDisabledControlJustPressed(0, mouseButton) then
                        setNuiFocus(true, true)
                    end

                    Wait(0)
                end
            end)
        end

        if not next(options) then
            flag = flag == 511 and 26 or 511
        end
        Wait(sleep)
    end

    if lastEntity and debug then
        SetEntityDrawOutline(lastEntity, false)
    end
end

local function disableTargeting()
    isActive = false
    setNuiFocus(false)
    SendNuiMessage('{"event": "visible", "state": false}')
    table.wipe(currentTarget)
    table.wipe(options)
end

-- Default keybind to toggle targeting (https://docs.fivem.net/docs/game-references/input-mapper-parameter-ids/keyboard)
local hotkey = GetConvar('ox_target:defaultHotkey', 'LMENU')

if toggleHotkey then
    RegisterCommand('ox_target', function()
        if isActive then
            return disableTargeting()
        end

        return enableTargeting()
    end, false)

    RegisterKeyMapping('ox_target', locale('toggle_targeting'), 'keyboard', hotkey)
else
    RegisterCommand('+ox_target', function() CreateThread(enableTargeting) end, false)
    RegisterCommand('-ox_target', disableTargeting, false)
    RegisterKeyMapping('+ox_target', locale('toggle_targeting'), 'keyboard', hotkey)
end

local function getResponse(option, server)
    local response = table.clone(option)
    response.entity = currentTarget.entity
    response.zone = currentTarget.zone
    response.coords = currentTarget.coords
    response.distance = currentTarget.distance

    if server then
        response.entity = response.entity ~= 0 and NetworkGetEntityIsNetworked(response.entity) and NetworkGetNetworkIdFromEntity(response.entity) or 0
    end

    response.icon = nil
    response.groups = nil
    response.items = nil
    response.canInteract = nil
    response.onSelect = nil
    response.export = nil
    response.event = nil
    response.serverEvent = nil
    response.command = nil

    return response
end

RegisterNUICallback('select', function(data, cb)
    cb(1)
    setNuiFocus(false)

    local option = options?[data[1]][data[2]]

    if option then
        if option.onSelect then
            option.onSelect(option.qtarget and currentTarget.entity or getResponse(option))
        elseif option.export then
            exports[option.resource][option.export](nil, getResponse(option))
        elseif option.event then
            TriggerEvent(option.event, getResponse(option))
        elseif option.serverEvent then
            TriggerServerEvent(option.serverEvent, getResponse(option, true))
        elseif option.command then
            ExecuteCommand(option.command)
        end
    end

    if IsNuiFocused() then
        isActive = false
    end
end)