local playerIdentity = {}
local alreadyRegistered = {}
local multichar = ESX.GetConfig().Multichar

local function deleteIdentityFromDatabase(xPlayer)
    MySQL.query.await("UPDATE users SET firstname = ?, lastname = ?, dateofbirth = ?, sex = ?, height = ?, skin = ? WHERE identifier = ?", { nil, nil, nil, nil, nil, nil, xPlayer.identifier })

    if Config.FullCharDelete then
        MySQL.update.await("UPDATE addon_account_data SET money = 0 WHERE account_name IN (?) AND owner = ?", { { "bank_savings", "caution" }, xPlayer.identifier })

        MySQL.prepare.await("UPDATE datastore_data SET data = ? WHERE name IN (?) AND owner = ?", { "'{}'", { "user_ears", "user_glasses", "user_helmet", "user_mask" }, xPlayer.identifier })
    end
end

function SetPlayerData(xPlayer, data)
    local name = ("%s %s"):format(data.firstName, data.lastName)
    xPlayer.setName(name)
    xPlayer.set("firstName", data.firstName)
    xPlayer.set("lastName", data.lastName)
    xPlayer.set("dateofbirth", data.dateOfBirth)
    xPlayer.set("sex", data.sex)
    xPlayer.set("height", data.height)

    local state = Player(xPlayer.source).state
    state:set("name", name, true)
    state:set("firstName", data.firstName, true)
    state:set("lastName", data.lastName, true)
    state:set("dateofbirth", data.dateOfBirth, true)
    state:set("sex", data.sex, true)
    state:set("height", data.height, true)
end

local function deleteIdentity(xPlayer)
    if not alreadyRegistered[xPlayer.identifier] then
        return
    end

    SetPlayerData(xPlayer, {firstName = nil, lastName = nil, dateOfBirth = nil, sex = nil, height = nil})
    deleteIdentityFromDatabase(xPlayer)
end


local function saveIdentityToDatabase(identifier, identity)
    MySQL.update.await("UPDATE users SET firstname = ?, lastname = ?, dateofbirth = ?, sex = ?, height = ? WHERE identifier = ?", { identity.firstName, identity.lastName, identity.dateOfBirth, identity.sex, identity.height, identifier })
end

local function checkDOBFormat(str)
    str = tostring(str)
    if not string.match(str, "(%d%d)/(%d%d)/(%d%d%d%d)") then
        return false
    end

    local d, m, y = string.match(str, "(%d+)/(%d+)/(%d+)")

    m = tonumber(m)
    d = tonumber(d)
    y = tonumber(y)

    if ((d <= 0) or (d > 31)) or ((m <= 0) or (m > 12)) or ((y <= Config.LowestYear) or (y > Config.HighestYear)) then
        return false
    elseif m == 4 or m == 6 or m == 9 or m == 11 then
        return d <= 30
    elseif m == 2 then
        if y % 400 == 0 or (y % 100 ~= 0 and y % 4 == 0) then
            return d <= 29
        else
            return d <= 28
        end
    else
        return d <= 31
    end
end

local function formatDate(str)
    local d, m, y = string.match(str, "(%d+)/(%d+)/(%d+)")
    local date = str

    if Config.DateFormat == "MM/DD/YYYY" then
        date = m .. "/" .. d .. "/" .. y
    elseif Config.DateFormat == "YYYY/MM/DD" then
        date = y .. "/" .. m .. "/" .. d
    end

    return date
end

local function checkAlphanumeric(str)
    return (string.match(str, "%W"))
end

local function checkForNumbers(str)
    return (string.match(str, "%d"))
end

local function checkNameFormat(name)
    if not checkAlphanumeric(name) and not checkForNumbers(name) then
        local stringLength = string.len(name)
        return stringLength > 0 and stringLength < Config.MaxNameLength
    end

    return false
end

local function checkSexFormat(sex)
    if not sex then
        return false
    end
    return sex == "m" or sex == "M" or sex == "f" or sex == "F"
end

local function checkHeightFormat(height)
    local numHeight = tonumber(height) or 0
    return numHeight >= Config.MinHeight and numHeight <= Config.MaxHeight
end

local function convertToLowerCase(str)
    return string.lower(str)
end

local function convertFirstLetterToUpper(str)
    return str:gsub("^%l", string.upper)
end

local function formatName(name)
    local loweredName = convertToLowerCase(name)
    return convertFirstLetterToUpper(loweredName)
end

if Config.UseDeferrals then
    AddEventHandler("playerConnecting", function(_, _, deferrals)
        deferrals.defer()
        local _, identifier = source, ESX.GetIdentifier(source)
        Wait(100)

        if identifier then
            MySQL.single("SELECT firstname, lastname, dateofbirth, sex, height FROM users WHERE identifier = ?", { identifier }, function(result)
                if result then
                    if result.firstname then
                        playerIdentity[identifier] = {
                            firstName = result.firstname,
                            lastName = result.lastname,
                            dateOfBirth = result.dateofbirth,
                            sex = result.sex,
                            height = result.height,
                        }

                        deferrals.done()
                    else
                        deferrals.presentCard(
                            [==[{"type": "AdaptiveCard","body":[{"type":"Container","items":[{"type":"ColumnSet",
                                "columns":[{"type":"Column","items":[{"type":"Input.Text","placeholder":"First Name",
                                "id":"firstname","maxLength":15},{"type":"Input.Text","placeholder":"Date of Birth (MM/DD/YYYY)",
                                "id":"dateofbirth","maxLength":10}],"width":"stretch"},{"type":"Column","width":"stretch",
                                "items":[{"type":"Input.Text","placeholder":"Last Name","id":"lastname","maxLength":15},
                                {"type":"Input.Text","placeholder":"Height (48-96 inches)","id":"height","maxLength":2}]}]},
                                {"type":"Input.ChoiceSet","placeholder":"Sex","choices":[{"title":"Male","value":"m"},
                                {"title":"Female","value":"f"}],"style":"expanded","id":"sex"}]},{"type": "ActionSet",
                                "actions": [{"type":"Action.Submit","title":"Submit"}]}],
                                "$schema": "http://adaptivecards.io/schemas/adaptive-card.json","version":"1.0"}]==],
                            function(data)
                                if data.firstname == "" or data.lastname == "" or data.dateofbirth == "" or data.sex == "" or data.height == "" then
                                    deferrals.done(TranslateCap("data_incorrect"))
                                else
                                    if checkNameFormat(data.firstname) and checkNameFormat(data.lastname) and checkDOBFormat(data.dateofbirth) and checkSexFormat(data.sex) and checkHeightFormat(data.height) then
                                        playerIdentity[identifier] = {
                                            firstName = formatName(data.firstname),
                                            lastName = formatName(data.lastname),
                                            dateOfBirth = data.dateofbirth,
                                            sex = data.sex,
                                            height = tonumber(data.height),
                                            saveToDatabase = true,
                                        }

                                        deferrals.done()
                                    else
                                        deferrals.done(TranslateCap("invalid_format"))
                                    end
                                end
                            end
                        )
                    end
                else
                    deferrals.presentCard(
                        [==[{"type": "AdaptiveCard","body":[{"type":"Container","items":[{"type":"ColumnSet","columns":[{
                            "type":"Column","items":[{"type":"Input.Text","placeholder":"First Name","id":"firstname",
                            "maxLength":15},{"type":"Input.Text","placeholder":"Date of Birth (MM/DD/YYYY)","id":"dateofbirth",
                            "maxLength":10}],"width":"stretch"},{"type":"Column","width":"stretch","items":[{"type":"Input.Text",
                            "placeholder":"Last Name","id":"lastname","maxLength":15},{"type":"Input.Text",
                            "placeholder":"Height (48-96 inches)","id":"height","maxLength":2}]}]},{"type":"Input.ChoiceSet",
                            "placeholder":"Sex","choices":[{"title":"Male","value":"m"},{"title":"Female","value":"f"}],
                            "style":"expanded","id":"sex"}]},{"type": "ActionSet","actions": [{"type":"Action.Submit",
                            "title":"Submit"}]}],"$schema": "http://adaptivecards.io/schemas/adaptive-card.json","version":"1.0"}]==],
                        function(data)
                            if data.firstname == "" or data.lastname == "" or data.dateofbirth == "" or data.sex == "" or data.height == "" then
                                return deferrals.done(TranslateCap("data_incorrect"))
                            end
                            if not checkNameFormat(data.firstname) then
                                return deferrals.done(TranslateCap("invalid_firstname_format"))
                            end
                            if not checkNameFormat(data.lastname) then
                                return deferrals.done(TranslateCap("invalid_lastname_format"))
                            end
                            if not checkDOBFormat(data.dateofbirth) then
                                return deferrals.done(TranslateCap("invalid_dob_format"))
                            end
                            if not checkSexFormat(data.sex) then
                                return deferrals.done(TranslateCap("invalid_sex_format"))
                            end
                            if not checkHeightFormat(data.height) then
                                return deferrals.done(TranslateCap("invalid_height_format"))
                            end

                            playerIdentity[identifier] = {
                                firstName = formatName(data.firstname),
                                lastName = formatName(data.lastname),
                                dateOfBirth = formatDate(data.dateofbirth),
                                sex = data.sex,
                                height = tonumber(data.height),
                                saveToDatabase = true,
                            }

                            deferrals.done()
                        end
                    )
                end
            end)
        else
            deferrals.done(TranslateCap("no_identifier"))
        end
    end)

    RegisterNetEvent("esx:playerLoaded")
    AddEventHandler("esx:playerLoaded", function(_, xPlayer)
        if not playerIdentity[xPlayer.identifier] then
            return xPlayer.kick(_("missing_identity"))
        end

        local currentIdentity = playerIdentity[xPlayer.identifier]
        SetPlayerData(xPlayer, currentIdentity)

        if currentIdentity.saveToDatabase then
            saveIdentityToDatabase(xPlayer.identifier, currentIdentity)
        end

        Wait(0)
        alreadyRegistered[xPlayer.identifier] = true
        TriggerClientEvent("ui_identity:alreadyRegistered", xPlayer.source)
        playerIdentity[xPlayer.identifier] = nil
    end)
else
    local function setIdentity(xPlayer)
        if not alreadyRegistered[xPlayer.identifier] then
            return
        end
        local currentIdentity = playerIdentity[xPlayer.identifier]
        SetPlayerData(xPlayer, currentIdentity)

        TriggerClientEvent("ui_identity:setPlayerData", xPlayer.source, currentIdentity)
        if currentIdentity.saveToDatabase then
            saveIdentityToDatabase(xPlayer.identifier, currentIdentity)
        end

        playerIdentity[xPlayer.identifier] = nil
    end

    local function checkIdentity(xPlayer)
        MySQL.single("SELECT firstname, lastname, dateofbirth, sex, height FROM users WHERE identifier = ?", { xPlayer.identifier }, function(result)
            if not result then
                return TriggerClientEvent("ui_identity:showRegisterIdentity", xPlayer.source)
            end
            if not result.firstname then
                playerIdentity[xPlayer.identifier] = nil
                alreadyRegistered[xPlayer.identifier] = false
                return TriggerClientEvent("ui_identity:showRegisterIdentity", xPlayer.source)
            end

            playerIdentity[xPlayer.identifier] = {
                firstName = result.firstname,
                lastName = result.lastname,
                dateOfBirth = result.dateofbirth,
                sex = result.sex,
                height = result.height,
            }

            alreadyRegistered[xPlayer.identifier] = true
            setIdentity(xPlayer)
        end)
    end

    if not multichar then
        AddEventHandler("playerConnecting", function(_, _, deferrals)
            deferrals.defer()
            local _, identifier = source, ESX.GetIdentifier(source)
            Wait(40)

            if not identifier then
                return deferrals.done(TranslateCap("no_identifier"))
            end
            MySQL.single("SELECT firstname, lastname, dateofbirth, sex, height FROM users WHERE identifier = ?", { identifier }, function(result)
                if not result then
                    playerIdentity[identifier] = nil
                    alreadyRegistered[identifier] = false
                    return deferrals.done()
                end
                if not result.firstname then
                    playerIdentity[identifier] = nil
                    alreadyRegistered[identifier] = false
                    return deferrals.done()
                end

                playerIdentity[identifier] = {
                    firstName = result.firstname,
                    lastName = result.lastname,
                    dateOfBirth = result.dateofbirth,
                    sex = result.sex,
                    height = result.height,
                }

                alreadyRegistered[identifier] = true

                deferrals.done()
            end)
        end)

        AddEventHandler("onResourceStart", function(resource)
            if resource ~= GetCurrentResourceName() then
                return
            end
            Wait(300)

            while not ESX do
                Wait(0)
            end

            local xPlayers = ESX.GetExtendedPlayers()

            for i = 1, #xPlayers do
                if xPlayers[i] then
                    checkIdentity(xPlayers[i])
                end
            end
        end)

        RegisterNetEvent("esx:playerLoaded", function(_, xPlayer)
            local currentIdentity = playerIdentity[xPlayer.identifier]

            if currentIdentity and alreadyRegistered[xPlayer.identifier] then
                SetPlayerData(xPlayer, currentIdentity)

                TriggerClientEvent("ui_identity:setPlayerData", xPlayer.source, currentIdentity)
                if currentIdentity.saveToDatabase then
                    saveIdentityToDatabase(xPlayer.identifier, currentIdentity)
                end

                Wait(0)

                TriggerClientEvent("ui_identity:alreadyRegistered", xPlayer.source)

                playerIdentity[xPlayer.identifier] = nil
            else
                TriggerClientEvent("ui_identity:showRegisterIdentity", xPlayer.source)
            end
        end)
    end

    ESX.RegisterServerCallback("ui_identity:registerIdentity", function(source, cb, data)
        local xPlayer = ESX.GetPlayerFromId(source)
        if not checkNameFormat(data.firstname) then
            TriggerClientEvent("esx:showNotification", source, TranslateCap("invalid_firstname_format"), "error")
            return cb(false)
        end
        if not checkNameFormat(data.lastname) then
            TriggerClientEvent("esx:showNotification", source, TranslateCap("invalid_lastname_format"), "error")
            return cb(false)
        end
        if not checkSexFormat(data.sex) then
            TriggerClientEvent("esx:showNotification", source, TranslateCap("invalid_sex_format"), "error")
            return cb(false)
        end
        if not checkDOBFormat(data.dateofbirth) then
            TriggerClientEvent("esx:showNotification", source, TranslateCap("invalid_dob_format"), "error")
            return cb(false)
        end
        if not checkHeightFormat(data.height) then
            TriggerClientEvent("esx:showNotification", source, TranslateCap("invalid_height_format"), "error")
            return cb(false)
        end
        if xPlayer then
            if alreadyRegistered[xPlayer.identifier] then
                xPlayer.showNotification(TranslateCap("already_registered"), "error")
                return cb(false)
            end

            playerIdentity[xPlayer.identifier] = {
                firstName = formatName(data.firstname),
                lastName = formatName(data.lastname),
                dateOfBirth = formatDate(data.dateofbirth),
                sex = data.sex,
                height = data.height,
            }

            local currentIdentity = playerIdentity[xPlayer.identifier]

            SetPlayerData(xPlayer, currentIdentity)

            TriggerClientEvent("ui_identity:setPlayerData", xPlayer.source, currentIdentity)
            saveIdentityToDatabase(xPlayer.identifier, currentIdentity)
            alreadyRegistered[xPlayer.identifier] = true
            playerIdentity[xPlayer.identifier] = nil
            return cb(true)
        end

        if not multichar then
            TriggerClientEvent("esx:showNotification", source, TranslateCap("data_incorrect"), "error")
            return cb(false)
        end

        local formattedFirstName = formatName(data.firstname)
        local formattedLastName = formatName(data.lastname)
        local formattedDate = formatDate(data.dateofbirth)

        data.firstname = formattedFirstName
        data.lastname = formattedLastName
        data.dateofbirth = formattedDate
        local Identity = {
            firstName = formattedFirstName,
            lastName = formattedLastName,
            dateOfBirth = formattedDate,
            sex = data.sex,
            height = data.height,
        }

        TriggerEvent("ui_identity:completedRegistration", source, data)
        TriggerClientEvent("ui_identity:setPlayerData", source, Identity)
        cb(true)
    end)
end

if Config.EnableCommands then
    ESX.RegisterCommand("char", "user", function(xPlayer)
        if xPlayer and xPlayer.getName() then
            xPlayer.showNotification(TranslateCap("active_character", xPlayer.getName()))
        else
            xPlayer.showNotification(TranslateCap("error_active_character"))
        end
    end, false, { help = TranslateCap("show_active_character") })

    ESX.RegisterCommand("chardel", "user", function(xPlayer)
        if xPlayer and xPlayer.getName() then
            if Config.UseDeferrals then
                xPlayer.kick(TranslateCap("deleted_identity"))
                Wait(1500)
                deleteIdentity(xPlayer)
                xPlayer.showNotification(TranslateCap("deleted_character"))
                playerIdentity[xPlayer.identifier] = nil
                alreadyRegistered[xPlayer.identifier] = false
            else
                deleteIdentity(xPlayer)
                xPlayer.showNotification(TranslateCap("deleted_character"))
                playerIdentity[xPlayer.identifier] = nil
                alreadyRegistered[xPlayer.identifier] = false
                TriggerClientEvent("ui_identity:showRegisterIdentity", xPlayer.source)
            end
        else
            xPlayer.showNotification(TranslateCap("error_delete_character"))
        end
    end, false, { help = TranslateCap("delete_character") })
end

if Config.EnableDebugging then
    ESX.RegisterCommand("xPlayerGetFirstName", "user", function(xPlayer)
        if xPlayer and xPlayer.get("firstName") then
            xPlayer.showNotification(TranslateCap("return_debug_xPlayer_get_first_name", xPlayer.get("firstName")))
        else
            xPlayer.showNotification(TranslateCap("error_debug_xPlayer_get_first_name"))
        end
    end, false, { help = TranslateCap("debug_xPlayer_get_first_name") })

    ESX.RegisterCommand("xPlayerGetLastName", "user", function(xPlayer)
        if xPlayer and xPlayer.get("lastName") then
            xPlayer.showNotification(TranslateCap("return_debug_xPlayer_get_last_name", xPlayer.get("lastName")))
        else
            xPlayer.showNotification(TranslateCap("error_debug_xPlayer_get_last_name"))
        end
    end, false, { help = TranslateCap("debug_xPlayer_get_last_name") })

    ESX.RegisterCommand("xPlayerGetFullName", "user", function(xPlayer)
        if xPlayer and xPlayer.getName() then
            xPlayer.showNotification(TranslateCap("return_debug_xPlayer_get_full_name", xPlayer.getName()))
        else
            xPlayer.showNotification(TranslateCap("error_debug_xPlayer_get_full_name"))
        end
    end, false, { help = TranslateCap("debug_xPlayer_get_full_name") })

    ESX.RegisterCommand("xPlayerGetSex", "user", function(xPlayer)
        if xPlayer and xPlayer.get("sex") then
            xPlayer.showNotification(TranslateCap("return_debug_xPlayer_get_sex", xPlayer.get("sex")))
        else
            xPlayer.showNotification(TranslateCap("error_debug_xPlayer_get_sex"))
        end
    end, false, { help = TranslateCap("debug_xPlayer_get_sex") })

    ESX.RegisterCommand("xPlayerGetDOB", "user", function(xPlayer)
        if xPlayer and xPlayer.get("dateofbirth") then
            xPlayer.showNotification(TranslateCap("return_debug_xPlayer_get_dob", xPlayer.get("dateofbirth")))
        else
            xPlayer.showNotification(TranslateCap("error_debug_xPlayer_get_dob"))
        end
    end, false, { help = TranslateCap("debug_xPlayer_get_dob") })

    ESX.RegisterCommand("xPlayerGetHeight", "user", function(xPlayer)
        if xPlayer and xPlayer.get("height") then
            xPlayer.showNotification(TranslateCap("return_debug_xPlayer_get_height", xPlayer.get("height")))
        else
            xPlayer.showNotification(TranslateCap("error_debug_xPlayer_get_height"))
        end
    end, false, { help = TranslateCap("debug_xPlayer_get_height") })
end