QBCore = exports['qb-core']:GetCoreObject()
local currentDealer = nil
local dealerIsHome = false
local waitingDelivery = nil
local activeDelivery = nil
local deliveryTimeout = 0
local waitingKeyPress = false
local dealerCombo = nil
local drugDeliveryZone

-- Handlers

AddStateBagChangeHandler('isLoggedIn', nil, function(_, _, value)
    if value then
        QBCore.Functions.TriggerCallback('md-drugs:server:RequestConfig', function(DealerConfig)
            QBConfig.Dealers = DealerConfig
        end)
        Wait(1000)
        InitZones()
    else
        if not Config.UseTarget and dealerCombo then dealerCombo:destroy() end
    end
end)

-- Functions

local function GetClosestDealer()
    local ped = PlayerPedId()
    local pCoords = GetEntityCoords(ped)
    for k, v in pairs(QBConfig.Dealers) do
        local dealerCoords = vector3(v.coords.x, v.coords.y, v.coords.z)
        if #(pCoords - dealerCoords) < 2.0 then
            currentDealer = k
            break
        end
    end
end
CreateThread(function()
    local dealermenu = {}
    if Config.StupidassNewQbItemName then
        for k, v in pairs (QBConfig.ProductsStupidNameRewrite) do 
            dealermenu[#dealermenu + 1] = {
                          icon = "nui://"..Config.imagelink..QBCore.Shared.Items[v.name].image,
                          title = v.label,
                          event = "md-drugs:client:travellingmerchantox",
                          args = {
                              item = v.name,
                              cost = v.price,
                             info = v.info,
                             amount = v.amount,
                            }
                        }
               
                   lib.registerContext({id = 'dealermenu',title = "Dealer Menu", options = dealermenu})
        end           
    else
         for k, v in pairs (QBConfig.Products) do 
            dealermenu[#dealermenu + 1] = {
                          icon = "nui://"..Config.imagelink..QBCore.Shared.Items[v.name].image,
                          title = v.label,
                          event = "md-drugs:client:travellingmerchantox",
                          args = {
                              item = v.name,
                              cost = v.price,
                             info = v.info,
                             amount = v.amount,
                            }
                        }

                   lib.registerContext({id = 'dealermenu',title = "Dealer Menu", options = dealermenu})
        end     
    end
end)

local function RandomDeliveryItemOnRep()
    local myRep = QBCore.Functions.GetPlayerData().metadata["dealerrep"]
    local availableItems = {}
    for k, _ in pairs(QBConfig.DeliveryItems) do
        if QBConfig.DeliveryItems[k]["minrep"] <= myRep then
            availableItems[#availableItems+1] = k
        end
    end
    return availableItems[math.random(1, #availableItems)]
end

local function RequestDelivery()
    if not waitingDelivery then
        GetClosestDealer()
        
        local amount = math.random(1, 3)
        local item = RandomDeliveryItemOnRep()

        QBCore.Functions.Notify("Currently searching for a delivery...", 'success')
        if Config.NearbyDeliveries == true then
            local playerPed = PlayerPedId()
            local playerCoords = GetEntityCoords(playerPed)
            local nearbyLocations = {}
        -- Filter out the nearby locations
            for _, location in ipairs(Config.DeliveryLocations) do
                local distance = #(playerCoords - location.coords)
                if distance <= Config.DeliveryWithin then
                    nearbyLocations[#nearbyLocations + 1] = location
                end
            end

            -- Select a random location from the nearby locations
            if #nearbyLocations > 0 then
                local selectedLocation = nearbyLocations[math.random(1, #nearbyLocations)]
                waitingDelivery = {
                    ["coords"] = selectedLocation.coords,
                    ["locationLabel"] = selectedLocation.label,
                    ["amount"] = amount,
                    ["dealer"] = currentDealer,
                    ["itemData"] = Config.DeliveryItems[item],
                    ["item"] = item
                }
            else
                QBCore.Functions.Notify("Nothing Close By", 'error')
                
            end
        else
            local location = math.random(1, #QBConfig.DeliveryLocations)

            waitingDelivery = {
                ["coords"] = QBConfig.DeliveryLocations[location]["coords"],
                ["locationLabel"] = QBConfig.DeliveryLocations[location]["label"],
                ["amount"] = amount,
                ["dealer"] = currentDealer,
                ["itemData"] = QBConfig.DeliveryItems[item],
                ["item"] = item
            }
            if QBConfig.Debug == true then
                print(QBConfig.DeliveryLocations[location]["coords"])
                print(QBConfig.DeliveryLocations[location]["label"])
            end
        end
		
        TriggerServerEvent('md-drugs:server:giveDeliveryItems', waitingDelivery)
        SetTimeout(2000, function()
           TriggerServerEvent('qb-phone:server:sendNewMail', {
               sender = QBConfig.Dealers[currentDealer]["name"],
               subject = "Delivery Location",
               message = "Deliver This For Me",
               button = {
                   enabled = true,
                   buttonEvent = "md-drugs:client:setLocation",
                   buttonData = waitingDelivery
               }
           })
        end)
    else
        QBCore.Functions.Notify("You Havent Even Finished The One I Told You Too", 'error')
    end
end

local function DeliveryTimer()
    CreateThread(function()
        while deliveryTimeout - 1 > 0 do
            deliveryTimeout = deliveryTimeout - 1
            Wait(1000)
        end
        deliveryTimeout = 0
    end)
end


local function PoliceCallDeliveries()
print " Tell your devs they didnt do something and to read the readme it isnt hard"
end

local function DeliverStuff()
    if deliveryTimeout > 0 then
        Wait(500)
        exports["rpemotes"]:EmoteCommandStart("uncuff", 0)
        PoliceCallDeliveries()
        QBCore.Functions.Progressbar("work_dropbox", "handing Off Package", 3500, false, true, {
            disableMovement = true,
            disableCarMovement = true,
            disableMouse = false,
            disableCombat = true,
        }, {}, {}, {}, function() -- Done
            TriggerServerEvent('md-drugs:server:successDelivery', activeDelivery, true)
            activeDelivery = nil
            exports['qb-target']:RemoveZone('drugDeliveryZone')
            ClearPedTasks(PlayerPedId())
        end)
    else
        TriggerServerEvent('md-drugs:server:successDelivery', activeDelivery, false)
    end
    deliveryTimeout = 0
end

local function SetMapBlip(x, y)
    SetNewWaypoint(x, y)
    QBCore.Functions.Notify("Marked On Your Map", 'success');
end

-- PolyZone specific functions


function InitZones()
    if next(QBConfig.Dealers) == nil then return end
   
     for k,v in pairs(QBConfig.Dealers) do
		local Ped = "g_m_y_famdnf_01"
         RequestModel(Ped)
         while not HasModelLoaded(Ped) do
             Wait(0)
         end
         
         testdealer = CreatePed(0, Ped,v.coords.x,v.coords.y,v.coords.z-1, 90.0, false, false)
          FreezeEntityPosition(testdealer, true)
         SetEntityInvincible(testdealer, true)
         exports["qb-target"]:AddBoxZone("dealer_"..k, vector3(v.coords.x, v.coords.y, v.coords.z), 1.5, 1.5, {
             name = "dealer_"..k,
             heading = v.heading,
             minZ = v.coords.z - 1,
             maxZ = v.coords.z + 1,
             debugPoly = false,
         }, {
             options = {
                 {
                     icon = 'fas fa-user-secret',
                     label = "Deliver",
                     action = function()
                         RequestDelivery()
                     end,
                     canInteract = function()
                         GetClosestDealer()
                         local hours = GetClockHours()
                         local min = QBConfig.Dealers[currentDealer]["time"]["min"]
                         local max = QBConfig.Dealers[currentDealer]["time"]["max"]
                         if max < min then
                             if hours <= max then
                                 if not waitingDelivery then
                                     return true
                                 end
                             elseif hours >= min then
                                 if not waitingDelivery then
                                     return true
                                 end
                             end
                         else
                             if hours >= min and hours <= max then
                                 if not waitingDelivery then
                                     return true
                                 end
                             end
                         end
                     end
                 },
                 {
                     icon = 'fas fa-user-secret',
                     label = "Open Shop",
                     action = function()
                        lib.showContext('dealermenu')
                     end,
                     canInteract = function()
                         GetClosestDealer()
                         local hours = GetClockHours()
                         local min = QBConfig.Dealers[currentDealer]["time"]["min"]
                         local max = QBConfig.Dealers[currentDealer]["time"]["max"]
                         if max < min then
                             if hours <= max then
                                 return true
                             elseif hours >= min then
                                 return true
                             end
                         else
                             if hours >= min and hours <= max then
                                 return true
                             end
                         end
                     end
                 }
             },
             distance = 1.5
         })
     end
   
end

-- Events

RegisterNetEvent('md-drugs:client:RefreshDealers', function(DealerData)
    if not Config.UseTarget and dealerCombo then dealerCombo:destroy() end
    QBConfig.Dealers = DealerData
    Wait(1000)
    InitZones()
end)

RegisterNetEvent('md-drugs:client:updateDealerItems', function(itemData, amount)
    TriggerServerEvent('md-drugs:server:updateDealerItems', itemData, amount, currentDealer)
end)

RegisterNetEvent('md-drugs:client:setDealerItems', function(itemData, amount, dealer)
    QBConfig.Dealers[dealer]["products"][itemData.slot].amount = QBConfig.Dealers[dealer]["products"][itemData.slot].amount - amount
end)

RegisterNetEvent('md-drugs:client:setLocation', function(locationData)
    if activeDelivery then
        SetMapBlip(activeDelivery["coords"]["x"], activeDelivery["coords"]["y"])
        QBCore.Functions.Notify("Already Delivering", 'error')
        return
    end
    activeDelivery = locationData
    deliveryTimeout = 300
    DeliveryTimer()
    SetMapBlip(activeDelivery["coords"].x, activeDelivery["coords"].y)
	local Ped = "g_m_y_famdnf_01"
     RequestModel(Ped)
     while not HasModelLoaded(Ped) do
         Wait(0)
     end
     
     Buyer = CreatePed(0, Ped,activeDelivery["coords"].x, activeDelivery["coords"].y, activeDelivery["coords"].z-1, 180.0, false, false)
     FreezeEntityPosition(Buyer, true)
    SetEntityInvincible(Buyer, true)
    if Config.UseTarget then
       exports['qb-target']:AddTargetEntity(Buyer, {
            options = {
                {
                    icon = 'fas fa-user-secret',
                    label = "Hand Off",
                    action = function()
                        DeliverStuff()
						Wait(4000)
						DeleteEntity(Buyer)
                        waitingDelivery = nil
                    end,
                    canInteract = function()
                        if waitingDelivery then
                            return true
                        end
                    end
                }
            },
            distance = 1.5
        })
    end
end)

RegisterNetEvent('md-drugs:client:sendDeliveryMail', function(type, deliveryData)
    if type == 'perfect' then
        TriggerServerEvent('qb-phone:server:sendNewMail', {
            sender = QBConfig.Dealers[deliveryData["dealer"]]["name"],
            subject = "Delivery",
            message = "You did it"
        })
    elseif type == 'bad' then
        TriggerServerEvent('qb-phone:server:sendNewMail', {
            sender = QBConfig.Dealers[deliveryData["dealer"]]["name"],
            subject = "Delivery",
            message = "You Fucked Me Over"
        })
    elseif type == 'late' then
        TriggerServerEvent('qb-phone:server:sendNewMail', {
            sender = QBConfig.Dealers[deliveryData["dealer"]]["name"],
            subject = "Delivery",
            message = "Better Late Than Dead"
        })
    end
end)

RegisterCommand('checkrep', function()
    local drugrep = QBCore.Functions.GetPlayerData().metadata["dealerrep"]
    QBCore.Functions.Notify("Current Rep: "..drugrep)
end)