local QBCore = exports['qb-core']:GetCoreObject()
local trackedVehicles = {}

-- Debug Funktion
local function Debug(msg)
    if Config.Debug then
        print("[AntiDespawn] " .. msg)
    end
end

-- Funktion um zu prüfen ob Fahrzeugklasse erlaubt ist
local function IsVehicleClassAllowed(vehicle)
    local vehicleClass = GetVehicleClass(vehicle)
    
    -- Prüfe Blacklist
    for _, blacklistedClass in pairs(Config.BlacklistedVehicleClasses) do
        if vehicleClass == blacklistedClass then
            return false
        end
    end
    
    -- Prüfe Whitelist
    for _, allowedClass in pairs(Config.AllowedVehicleClasses) do
        if vehicleClass == allowedClass then
            return true
        end
    end
    
    return false
end

-- Starke Anti-Despawn Funktion
local function PreventDespawn(vehicle)
    if DoesEntityExist(vehicle) then
        -- Grundlegende Persistenz
        SetEntityAsMissionEntity(vehicle, true, true)
        SetVehicleHasBeenOwnedByPlayer(vehicle, true)
        SetVehicleNeedsToBeHotwired(vehicle, false)
        
        -- Zusätzliche Flags
        SetEntityLoadCollisionFlag(vehicle, true)
        SetVehicleIsStolen(vehicle, false)
        SetVehicleIsWanted(vehicle, false)
        
        -- Verhindere dass das Fahrzeug als "abandoned" markiert wird
        if DecorExistOn(vehicle, "IgnoredByQuickSave") then
            DecorSetBool(vehicle, "IgnoredByQuickSave", false)
        end
        
        return true
    end
    return false
end

-- Event Handler für Fahrzeug betreten (nur Fahrersitz)
CreateThread(function()
    local lastVehicle = 0
    
    while true do
        Wait(1000)
        
        local playerPed = PlayerPedId()
        local currentVehicle = GetVehiclePedIsIn(playerPed, false)
        
        -- Spieler ist als Fahrer in ein Fahrzeug eingestiegen
        if currentVehicle ~= 0 and currentVehicle ~= lastVehicle then
            -- Prüfe ob Spieler auf Fahrersitz ist
            local driver = GetPedInVehicleSeat(currentVehicle, -1)
            
            -- Nur wenn Spieler der Fahrer ist (Seat -1)
            if driver == playerPed and IsVehicleClassAllowed(currentVehicle) then
                local plate = QBCore.Functions.GetPlate(currentVehicle)
                trackedVehicles[plate] = currentVehicle
                
                -- Sofort starke Despawn-Verhinderung
                PreventDespawn(currentVehicle)
                
                Debug("Fahrzeug wird nun getrackt: " .. plate)
                
                -- Registriere Fahrzeug beim Server
                local vehicleCoords = GetEntityCoords(currentVehicle)
                local vehicleHeading = GetEntityHeading(currentVehicle)
                local vehicleModel = GetEntityModel(currentVehicle)
                
                TriggerServerEvent('antidespawn:server:registerVehicle', plate, vehicleModel, vehicleCoords, vehicleHeading)
            end
        end
        
        lastVehicle = currentVehicle
    end
end)

-- Kontinuierliche Despawn-Verhinderung für alle getrackten Fahrzeuge
CreateThread(function()
    while true do
        Wait(10000) -- Alle 10 Sekunden
        
        for plate, vehicle in pairs(trackedVehicles) do
            if DoesEntityExist(vehicle) then
                PreventDespawn(vehicle)
                
                -- Aktualisiere Position
                local vehicleCoords = GetEntityCoords(vehicle)
                local vehicleHeading = GetEntityHeading(vehicle)
                
                TriggerServerEvent('antidespawn:server:updateVehicle', plate, vehicleCoords, vehicleHeading)
                
                Debug("Aktualisiere Fahrzeug: " .. plate)
            else
                trackedVehicles[plate] = nil
                Debug("Fahrzeug existiert nicht mehr: " .. plate)
            end
        end
    end
end)

-- Lade Fahrzeuge beim Spawn
RegisterNetEvent('QBCore:Client:OnPlayerLoaded', function()
    Debug("Spieler geladen, lade Fahrzeuge...")
    Wait(10000) -- Warte bis alles geladen ist
    TriggerServerEvent('antidespawn:server:loadVehicles')
end)

-- Spawne ein Fahrzeug
RegisterNetEvent('antidespawn:client:spawnVehicle', function(data)
    Debug("Spawne Fahrzeug: " .. data.plate)
    
    -- Prüfe ob Fahrzeug bereits existiert
    local existingVehicle = GetVehicleByPlate(data.plate)
    if existingVehicle then
        Debug("Fahrzeug existiert bereits: " .. data.plate)
        trackedVehicles[data.plate] = existingVehicle
        PreventDespawn(existingVehicle)
        return
    end
    
    -- Spawne Fahrzeug
    local modelHash = data.model
    
    RequestModel(modelHash)
    local timeout = 0
    while not HasModelLoaded(modelHash) and timeout < 100 do
        Wait(100)
        timeout = timeout + 1
    end
    
    if HasModelLoaded(modelHash) then
        local vehicle = CreateVehicle(modelHash, data.coords.x, data.coords.y, data.coords.z, data.heading, true, false)
        
        if DoesEntityExist(vehicle) then
            -- Setze Kennzeichen
            SetVehicleNumberPlateText(vehicle, data.plate)
            
            -- Setze Fuel
            if GetResourceState(Config.FuelSystem) == 'started' then
                exports[Config.FuelSystem]:SetFuel(vehicle, data.fuel or 100)
            end
            
            -- Verhindere Despawn
            PreventDespawn(vehicle)
            
            -- Füge zu getrackten Fahrzeugen hinzu
            trackedVehicles[data.plate] = vehicle
            
            Debug("Fahrzeug erfolgreich gespawnt: " .. data.plate)
        else
            Debug("Fehler beim Spawnen des Fahrzeugs: " .. data.plate)
        end
        
        SetModelAsNoLongerNeeded(modelHash)
    else
        Debug("Modell konnte nicht geladen werden: " .. data.plate)
    end
end)

-- Hilfsfunktion um Fahrzeug anhand Kennzeichen zu finden
function GetVehicleByPlate(plate)
    local vehicles = GetGamePool('CVehicle')
    for _, vehicle in pairs(vehicles) do
        if QBCore.Functions.GetPlate(vehicle) == plate then
            return vehicle
        end
    end
    return nil
end

-- jg-advanced-garage Events
RegisterNetEvent('jg-advancedgarages:client:vehicle-stored', function(data)
    if data and data.plate and trackedVehicles[data.plate] then
        trackedVehicles[data.plate] = nil
        TriggerServerEvent('antidespawn:server:removeVehicle', data.plate)
        Debug("Fahrzeug in Garage gespeichert, entferne aus Tracking: " .. data.plate)
    end
end)

-- Cleanup beim Disconnect
AddEventHandler('onResourceStop', function(resourceName)
    if resourceName == GetCurrentResourceName() then
        trackedVehicles = {}
    end
end)