client.lua

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

-- ─────────────────────────────────────────────────────────────────────────────
--  Start Raffle — multi-step prize selection flow
--  Called after server sends back vehicle + item lists via openStartDialog
-- ─────────────────────────────────────────────────────────────────────────────

local function startRaffleFlow(payload)
    local raffleName  = payload.raffleName  or 'Mystery Raffle'
    local vehicleList = payload.vehicleList or {}
    local itemList    = payload.itemList    or {}
    local presets     = payload.presets     or {}

    -- ── Step 1: Raffle name + prize type ────────────────────────────────────
    local step1 = lib.inputDialog('🎰 New Raffle — Step 1 of 2', {
        {
            type        = 'input',
            label       = 'Raffle Name',
            default     = raffleName,
            max         = 50,
            required    = true,
        },
        {
            type     = 'select',
            label    = 'Prize Type',
            description = 'Choose how the prize will be delivered',
            options  = {
                { label = '🎁 Custom / Manual',   value = 'custom'  },
                { label = '🚗 Vehicle to Garage', value = 'vehicle' },
                { label = '💰 Cash or Bank',       value = 'cash'    },
                { label = '📦 Inventory Item',     value = 'item'    },
            },
            required = true,
        },
    })

    if not step1 or not step1[1] or not step1[2] then return end

    local name      = step1[1]:match('^%s*(.-)%s*$')
    if name == '' then name = 'Mystery Raffle' end
    local prizeType = step1[2]

    -- ── Step 2: Prize details (depends on type) ──────────────────────────────

    if prizeType == 'custom' then
        -- Custom: text input with optional preset quick-pick
        local presetOpts = {}
        for i, p in ipairs(presets) do
            presetOpts[i] = { label = p, value = p }
        end

        local step2 = lib.inputDialog('🎁 Custom Prize — Step 2 of 2', {
            {
                type        = 'input',
                label       = 'Prize Description',
                description = 'Staff delivers this manually after /endraffle',
                placeholder = Config.DefaultPrizeHint,
                max         = 100,
            },
            {
                type        = 'select',
                label       = '— OR pick a preset —',
                description = 'Leave the text field blank to use a preset instead',
                options     = presetOpts,
                required    = false,
            },
        })

        if not step2 then return end

        local typed     = step2[1] and step2[1]:match('^%s*(.-)%s*$') or ''
        local preset    = step2[2] or ''
        local prizeDesc = (typed ~= '') and typed or (preset ~= '' and preset or nil)

        if not prizeDesc then
            lib.notify({ title = 'Raffle', description = 'A prize description is required.', type = 'error' })
            return
        end

        TriggerServerEvent('dynamic-raffle:server:startRaffle', name, 'custom', { description = prizeDesc })

    elseif prizeType == 'vehicle' then
        -- Vehicle: searchable select from QBCore.Shared.Vehicles
        local vehOpts = {}
        for i, v in ipairs(vehicleList) do
            vehOpts[i] = { label = v.label, value = v.model }
        end

        local step2 = lib.inputDialog(
            '🚗 Vehicle Prize — Step 2 of 2  (' .. #vehicleList .. ' vehicles)', {
            {
                type     = 'select',
                label    = 'Select Prize Vehicle',
                description = 'Vehicle will be inserted directly into the winner\'s garage',
                options  = vehOpts,
                required = true,
            },
        })

        if not step2 or not step2[1] then return end

        TriggerServerEvent('dynamic-raffle:server:startRaffle', name, 'vehicle', {
            model = step2[1],
        })

    elseif prizeType == 'cash' then
        -- Cash: amount + bank or cash-on-hand
        local step2 = lib.inputDialog('💰 Cash Prize — Step 2 of 2', {
            {
                type        = 'number',
                label       = 'Prize Amount ($)',
                description = 'Delivered automatically when /endraffle is called',
                default     = 50000,
                min         = 1,
                max         = 10000000,
                required    = true,
            },
            {
                type    = 'select',
                label   = 'Deliver To',
                options = {
                    { label = '🏦 Bank Account', value = 'bank' },
                    { label = '💵 Cash on Hand', value = 'cash' },
                },
                required = true,
            },
        })

        if not step2 or not step2[1] then return end

        TriggerServerEvent('dynamic-raffle:server:startRaffle', name, 'cash', {
            amount  = tonumber(step2[1]) or 50000,
            account = step2[2] or 'bank',
        })

    elseif prizeType == 'item' then
        -- Item: ox_inventory item list + quantity
        local itemOpts = {}
        for i, item in ipairs(itemList) do
            itemOpts[i] = { label = item.label .. '  [' .. item.name .. ']', value = item.name }
        end

        local step2 = lib.inputDialog(
            '📦 Item Prize — Step 2 of 2  (' .. #itemList .. ' items)', {
            {
                type     = 'select',
                label    = 'Select Prize Item',
                description = 'Item will be added directly to the winner\'s ox_inventory',
                options  = itemOpts,
                required = true,
            },
            {
                type     = 'number',
                label    = 'Quantity',
                default  = 1,
                min      = 1,
                max      = 100,
                required = true,
            },
        })

        if not step2 or not step2[1] then return end

        TriggerServerEvent('dynamic-raffle:server:startRaffle', name, 'item', {
            name  = step2[1],
            count = math.floor(tonumber(step2[2]) or 1),
        })
    end
end

-- Server fires this back after /startraffle or "Start New Raffle" in manage menu
RegisterNetEvent('dynamic-raffle:client:openStartDialog', function(payload)
    startRaffleFlow(payload)
end)

-- ─────────────────────────────────────────────────────────────────────────────
--  Commands
-- ─────────────────────────────────────────────────────────────────────────────

-- /startraffle [name] — quick shortcut, asks server for lists then opens flow
RegisterCommand('startraffle', function(_, args)
    local raffleName = table.concat(args, ' ')
    if raffleName == '' then raffleName = 'Mystery Raffle' end
    TriggerServerEvent('dynamic-raffle:server:requestStartDialog', raffleName)
end, false)

RegisterCommand('endraffle', function()
    TriggerServerEvent('dynamic-raffle:server:endRaffle')
end, false)

RegisterCommand('joinraffle', function()
    TriggerServerEvent('dynamic-raffle:server:joinRaffle')
end, false)

RegisterCommand('raffleping', function()
    TriggerServerEvent('dynamic-raffle:server:ping')
end, false)

RegisterCommand('rafflestatus', function()
    TriggerServerEvent('dynamic-raffle:server:status')
end, false)

-- /rafflemanage — unified admin hub
RegisterCommand('rafflemanage', function()
    TriggerServerEvent('dynamic-raffle:server:getManageData')
end, false)

-- ─────────────────────────────────────────────────────────────────────────────
--  Manage Menu Builder
--  Builds the full ox_lib context menu tree from server data and opens it.
--  Also re-invoked automatically after any preset change.
-- ─────────────────────────────────────────────────────────────────────────────

local function GetPrizeTypeIcon(prizeType)
    if prizeType == 'vehicle' then return '🚗'
    elseif prizeType == 'cash' then return '💰'
    elseif prizeType == 'item' then return '📦'
    else return '🎁' end
end

local function BuildManageMenus(data)
    local mainOptions = {}

    -- ── Section 1: Start / Current Raffle ────────────────────────────────────
    if data.active and data.raffle then
        -- Active raffle — show participants + quick controls
        local r           = data.raffle
        local typeIcon    = GetPrizeTypeIcon(r.prizeType)
        local currentOpts = {
            { title = typeIcon .. ' Prize',    description = r.prize,      disabled = true },
            { title = '🕐 Started by',          description = r.startedBy,  disabled = true },
            { title = '👥 Entries',             description = tostring(#data.entries) .. ' entered', disabled = true },
            { title = '────────────────────────────────────', disabled = true },
            {
                title       = '📣 Send Reminder Now',
                description = 'Blast the reminder notification to all players',
                event       = 'dynamic-raffle:client:menuPing',
            },
            {
                title       = '🏁 End Raffle & Draw Winner',
                description = 'Picks a winner from ' .. #data.entries .. ' eligible entries',
                event       = 'dynamic-raffle:client:menuEndRaffle',
            },
            { title = '────────────────────────────────────', disabled = true },
        }

        if #data.entries > 0 then
            table.insert(currentOpts, { title = '👥 Participants', disabled = true })
            for _, pname in ipairs(data.entries) do
                table.insert(currentOpts, { title = '  👤 ' .. pname, disabled = true })
            end
        else
            table.insert(currentOpts, { title = 'No entries yet — raffle is open', disabled = true })
        end

        lib.registerContext({
            id      = 'raffle_manage_current',
            title   = '📋 ' .. r.name,
            menu    = 'raffle_manage_main',
            options = currentOpts,
        })

        table.insert(mainOptions, {
            title       = '📋 Active: ' .. r.name,
            description = r.prize .. '  ·  ' .. #data.entries .. ' entries',
            arrow       = true,
            menu        = 'raffle_manage_current',
        })

    else
        -- No active raffle — show Start button
        table.insert(mainOptions, {
            title       = '🚀 Start New Raffle',
            description = 'Pick prize type: vehicle, cash, item, or custom',
            event       = 'dynamic-raffle:client:menuStartRaffle',
        })
    end

    -- ── Section 2: Raffle History ─────────────────────────────────────────────
    local historySubOpts = {}

    if #data.history == 0 then
        table.insert(historySubOpts, { title = 'No completed raffles yet', disabled = true })
    else
        for i = #data.history, 1, -1 do   -- newest first
            local h     = data.history[i]
            local subId = 'raffle_manage_history_' .. i
            local icon  = GetPrizeTypeIcon(h.prizeType)

            local detailOpts = {
                { title = '🏆 Winner',        description = h.winner,             disabled = true },
                { title = icon .. ' Prize',   description = h.prize,              disabled = true },
                { title = '👥 Total Entries', description = tostring(#h.entries), disabled = true },
                { title = '🕐 Completed',     description = h.date,               disabled = true },
                { title = '────────────────────────────────────', disabled = true },
            }

            for _, pname in ipairs(h.entries) do
                table.insert(detailOpts, {
                    title       = (pname == h.winner and '🏆 ' or '  👤 ') .. pname,
                    description = pname == h.winner and 'Winner' or nil,
                    disabled    = true,
                })
            end

            lib.registerContext({
                id      = subId,
                title   = h.name,
                menu    = 'raffle_manage_history',
                options = detailOpts,
            })

            table.insert(historySubOpts, {
                title       = h.name,
                description = '🏆 ' .. h.winner .. '  ·  ' .. h.prize .. '  ·  ' .. h.date,
                arrow       = true,
                menu        = subId,
            })
        end
    end

    lib.registerContext({
        id      = 'raffle_manage_history',
        title   = '🏆 Raffle History',
        menu    = 'raffle_manage_main',
        options = historySubOpts,
    })

    table.insert(mainOptions, {
        title       = '🏆 Raffle History',
        description = #data.history .. ' completed raffle(s)',
        arrow       = true,
        menu        = 'raffle_manage_history',
    })

    -- ── Section 3: Winner Cooldowns ─────────────────────────────────────────────
    if data.cooldownLimit and data.cooldownLimit > 0 then
        local cooldownOpts = {}

        if not data.cooldowns or #data.cooldowns == 0 then
            table.insert(cooldownOpts, {
                title    = 'No active cooldowns',
                description = 'Everyone is eligible to win',
                disabled = true,
            })
        else
            for _, cd in ipairs(data.cooldowns) do
                table.insert(cooldownOpts, {
                    title       = '⏳ ' .. cd.name,
                    description = cd.remaining .. ' raffle(s) remaining on cooldown',
                    disabled    = true,
                })
            end
        end

        lib.registerContext({
            id      = 'raffle_manage_cooldowns',
            title   = '⏳ Winner Cooldowns',
            menu    = 'raffle_manage_main',
            options = cooldownOpts,
        })

        table.insert(mainOptions, {
            title       = '⏳ Winner Cooldowns',
            description = (#(data.cooldowns or {})) .. ' player(s) sitting out  ·  ' .. data.cooldownLimit .. ' raffle cool-off',
            arrow       = true,
            menu        = 'raffle_manage_cooldowns',
        })
    end

    -- ── Section 3: Preset Prizes ──────────────────────────────────────────────
    local presetOpts = {}

    for i, preset in ipairs(data.presets) do
        local subId = 'raffle_preset_' .. i

        lib.registerContext({
            id      = subId,
            title   = '🎁 ' .. preset,
            menu    = 'raffle_manage_presets',
            options = {
                {
                    title       = '✏️ Edit',
                    description = 'Update this preset description',
                    event       = 'dynamic-raffle:client:editPreset',
                    args        = { index = i, current = preset },
                },
                {
                    title       = '🗑️ Remove',
                    description = 'Permanently delete this preset',
                    event       = 'dynamic-raffle:client:removePreset',
                    args        = { index = i, label = preset },
                },
            },
        })

        table.insert(presetOpts, {
            title       = preset,
            description = 'Edit or remove',
            arrow       = true,
            menu        = subId,
        })
    end

    table.insert(presetOpts, {
        title       = '➕ Add New Preset',
        description = 'Add a new quick-select prize option',
        event       = 'dynamic-raffle:client:addPreset',
    })

    lib.registerContext({
        id      = 'raffle_manage_presets',
        title   = '🎁 Preset Prizes',
        menu    = 'raffle_manage_main',
        options = presetOpts,
    })

    table.insert(mainOptions, {
        title       = '🎁 Preset Prizes',
        description = #data.presets .. ' preset(s)  ·  Edit quick-select options',
        arrow       = true,
        menu        = 'raffle_manage_presets',
    })

    -- ── Register & open main ──────────────────────────────────────────────────
    lib.registerContext({
        id      = 'raffle_manage_main',
        title   = '🎰 Raffle Manager',
        options = mainOptions,
    })

    lib.showContext('raffle_manage_main')
end

-- ─────────────────────────────────────────────────────────────────────────────
--  Manage Menu Action Handlers
-- ─────────────────────────────────────────────────────────────────────────────

-- "Start New Raffle" inside /rafflemanage
AddEventHandler('dynamic-raffle:client:menuStartRaffle', function()
    TriggerServerEvent('dynamic-raffle:server:requestStartDialog', 'Mystery Raffle')
end)

-- "End Raffle & Draw Winner" inside manage menu
AddEventHandler('dynamic-raffle:client:menuEndRaffle', function()
    local confirm = lib.alertDialog({
        header   = '🏁 End Raffle',
        content  = 'Draw a winner now and deliver the prize automatically?',
        centered = true,
        cancel   = true,
    })
    if confirm ~= 'confirm' then return end
    TriggerServerEvent('dynamic-raffle:server:endRaffle')
end)

-- "Send Reminder Now" inside manage menu
AddEventHandler('dynamic-raffle:client:menuPing', function()
    TriggerServerEvent('dynamic-raffle:server:ping')
end)

-- ─────────────────────────────────────────────────────────────────────────────
--  Preset CRUD Handlers
-- ─────────────────────────────────────────────────────────────────────────────

AddEventHandler('dynamic-raffle:client:editPreset', function(args)
    local result = lib.inputDialog('✏️ Edit Preset', {
        {
            type     = 'input',
            label    = 'Prize Description',
            default  = args.current,
            max      = 100,
            required = true,
        },
    })
    if not result or not result[1] or result[1]:match('^%s*$') then return end
    TriggerServerEvent('dynamic-raffle:server:editPreset', args.index, result[1])
end)

AddEventHandler('dynamic-raffle:client:removePreset', function(args)
    local confirm = lib.alertDialog({
        header   = '🗑️ Remove Preset',
        content  = 'Remove **' .. (args.label or 'this preset') .. '**? This cannot be undone.',
        centered = true,
        cancel   = true,
    })
    if confirm ~= 'confirm' then return end
    TriggerServerEvent('dynamic-raffle:server:removePreset', args.index)
end)

AddEventHandler('dynamic-raffle:client:addPreset', function()
    local result = lib.inputDialog('➕ Add New Preset', {
        {
            type        = 'input',
            label       = 'Prize Description',
            placeholder = 'e.g. 🎮 Custom Controller',
            max         = 100,
            required    = true,
        },
    })
    if not result or not result[1] or result[1]:match('^%s*$') then return end
    TriggerServerEvent('dynamic-raffle:server:addPreset', result[1])
end)

-- ─────────────────────────────────────────────────────────────────────────────
--  Server → Client net events
-- ─────────────────────────────────────────────────────────────────────────────

-- Open / refresh the manage menu (also fires after preset changes)
RegisterNetEvent('dynamic-raffle:client:openManage', function(data)
    BuildManageMenus(data)
end)

-- Generic notification
RegisterNetEvent('dynamic-raffle:notify', function(msg, msgType)
    lib.notify({
        title       = '🎰 Raffle',
        description = msg,
        type        = msgType or 'inform',
        duration    = 6000,
    })
end)

-- Server-wide winner announcement
RegisterNetEvent('dynamic-raffle:announce', function(winnerName, prize, raffleName)
    lib.notify({
        title       = '🏆 We Have a Winner!',
        description = winnerName .. ' won the ' .. raffleName .. '!\n🎁 Prize: ' .. prize,
        type        = 'success',
        duration    = 10000,
    })
end)

-- Winner-only golden NUI overlay
RegisterNetEvent('dynamic-raffle:youWon', function(prize, raffleName)
    SetNuiFocus(false, false)  -- keep game control
    SendNUIMessage({
        action   = 'showWinner',
        prize    = prize,
        raffle   = raffleName,
        duration = Config.WinnerDisplayTime,
    })

    SetTimeout(Config.WinnerDisplayTime + 500, function()
        SendNUIMessage({ action = 'hide' })
    end)
end)