-- Database Verification

local readyFunction = MySQL.ready
local databaseConnected, bansTableExists = readyFunction == nil, readyFunction == nil
if readyFunction ~= nil then
    MySQL.ready(function()
        databaseConnected = true
        local DatabaseInfo = QBCore.Functions.GetDatabaseInfo()
        if not DatabaseInfo or not DatabaseInfo.exists then return end
        local result = MySQL.query.await('SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ? AND TABLE_NAME = "bans";', { DatabaseInfo.database })
        if result and result[1] then
            bansTableExists = true
        end
    end)
end

-- Handle player connect

AddEventHandler('playerConnecting', function(name, _, deferrals)
    local src = source
    deferrals.defer()
    if QBCore.Config.Server.Closed and not IsPlayerAceAllowed(src, 'qbadmin.join') then
        return deferrals.done(QBCore.Config.Server.ClosedReason)
    end
    if not databaseConnected then
        return deferrals.done(Lang:t('error.connecting_database_error'))
    end
    if QBCore.Config.Server.Whitelist then
        Wait(0)
        deferrals.update(string.format(Lang:t('info.checking_whitelisted'), name))
        if not QBCore.Functions.IsWhitelisted(src) then
            return deferrals.done(Lang:t('error.not_whitelisted'))
        end
    end
    Wait(0)
    deferrals.update(string.format('Hello %s. Your license is being checked', name))
    local license = QBCore.Functions.GetIdentifier(src, 'license')
    if not license then
        return deferrals.done(Lang:t('error.no_valid_license'))
    elseif QBCore.Config.Server.CheckDuplicateLicense and QBCore.Functions.IsLicenseInUse(license) then
        return deferrals.done(Lang:t('error.duplicate_license'))
    end
    Wait(0)
    deferrals.update(string.format(Lang:t('info.checking_ban'), name))
    if not bansTableExists then
        return deferrals.done(Lang:t('error.ban_table_not_found'))
    end
    local success, isBanned, reason = pcall(QBCore.Functions.IsPlayerBanned, src)
    if not success then return deferrals.done(Lang:t('error.connecting_database_error')) end
    if isBanned then return deferrals.done(reason) end
    Wait(0)
    deferrals.update(string.format(Lang:t('info.join_server'), name))
    deferrals.done()
end)

-- Handle player disconnect

AddEventHandler('playerDropped', function(reason)
    local src = source
    local state = Player(src).state
    if not state['citizenid'] then return end
    local playerName = GetPlayerName(src) or 'Unknown'
    local license = state['license'] or 'Unknown License'
    TriggerEvent('qb-log:server:CreateLog', 'joinleave', 'Dropped', 'red', '**' .. playerName .. '** (' .. license .. ') left..\n **Reason:** ' .. reason)
    TriggerEvent('QBCore:Server:PlayerDropped', src)
    QBCore.Player.Save(src)
    if GetResourceState('qb-inventory') ~= 'missing' then
        exports['qb-inventory']:SaveInventory(src)
    end
    for _, key in pairs(GetStateBagKeys('player:' .. src)) do
        state:set(key, nil, true)
    end
    QBCore.Player_Buckets[license] = nil
end)

-- Config Editor

RegisterNetEvent('qb-core:server:configEditor', function(fileName, fileData)
    local src = source
    local Player = QBCore.Functions.GetPlayer(src)
    if not Player then return end
    if not QBCore.Functions.HasPermission('god') then
        local filePaths = {
            ['config'] = 'shared/json/config.json',
            ['gangs'] = 'shared/json/gangs.json',
            ['items'] = 'shared/json/items.json',
            ['jobs'] = 'shared/json/jobs.json',
            ['playerdata'] = 'shared/json/player_defaults.json',
            ['vehicles'] = 'shared/json/vehicles.json',
            ['weapons'] = 'shared/json/weapons.json'
        }
        local filePath = filePaths[fileName]
        if not filePath then
            print('^1[QBCore] Invalid file name: ' .. fileName)
            return
        end
        local encodedData = json.encode(fileData, { indent = true })
        local saved = SaveResourceFile('qb-core', filePath, encodedData, -1)
        if saved then
            print('^2[QBCore] Player ' .. GetPlayerName(src) .. ' (ID: ' .. src .. ') successfully saved ' .. fileName .. ' configuration')
            TriggerClientEvent('QBCore:Notify', src, 'Configuration saved successfully', 'success')
        else
            print('^1[QBCore] Failed to save ' .. fileName .. ' configuration')
            TriggerClientEvent('QBCore:Notify', src, 'Failed to save configuration', 'error')
        end
    else
        print('^1[QBCore] Player ' .. GetPlayerName(src) .. ' (ID: ' .. src .. ') attempted to save ' .. fileName .. ' configuration without permission')
        TriggerClientEvent('QBCore:Notify', src, 'You don\'t have permission to edit configs', 'error')
    end
end)