local currentGear = {
    mask = 0,
    tank = 0,
    oxygen = 0,
    enabled = false
}

local oxygenLevel = 0

lib.callback.register('qbx_divegear:client:fillTank', function()
    if oxygenLevel ~= 0 then
        exports.qbx_core:Notify(Lang:t("error.oxygenlevel", {oxygenlevel = oxygenLevel}), 'error')
        return false
    end

    oxygenLevel = Config.OxygenLevel
    exports.qbx_core:Notify(Lang:t("success.tube_filled"), 'success')
    return true
end)

local function deleteGear()
	if currentGear.mask ~= 0 then
        DetachEntity(currentGear.mask, false, true)
        DeleteEntity(currentGear.mask)
		currentGear.mask = 0
    end

	if currentGear.tank ~= 0 then
        DetachEntity(currentGear.tank, false, true)
        DeleteEntity(currentGear.tank)
		currentGear.tank = 0
	end
end

local function takeOffSuit()
    if lib.progressBar({
        duration = 5000,
        label = Lang:t("info.pullout_suit"),
        useWhileDead = false,
        canCancel = true,
        anim = {
            dict = "clothingshirt",
            clip = "try_shirt_positive_d",
            blendIn = 8.0
        }
    }) then
        SetEnableScuba(cache.ped, false)
        SetPedMaxTimeUnderwater(cache.ped, 50.00)
        currentGear.enabled = false
        deleteGear()
        exports.qbx_core:Notify(Lang:t("success.took_out"))
        TriggerServerEvent("InteractSound_SV:PlayOnSource", nil, 0.25)
    end

    ClearPedTasks(cache.ped)
end

local function DrawText(text)
    ClearPedTasks(cache.ped)
	SetTextFont(4)
	SetTextProportional(true)
	SetTextScale(0.0, 0.45)
	SetTextDropshadow(1, 0, 0, 0, 255)
	SetTextEdge(1, 0, 0, 0, 255)
	SetTextDropShadow()
	SetTextOutline()
	BeginTextCommandDisplayText("STRING")
	AddTextComponentSubstringPlayerName(text)
    EndTextCommandDisplayText(0.45, 0.90)
end

local function putOnSuit()
    if oxygenLevel <= 0 then
        exports.qbx_core:Notify(Lang:t("error.need_otube"), 'error')
        return
    end

    if IsPedSwimming(cache.ped) or cache.vehicle then
        exports.qbx_core:Notify(Lang:t("error.not_standing_up"), 'error')
        return
    end

    if lib.progressBar({
        duration = 5000,
        label = Lang:t("info.put_suit"),
        useWhileDead = false,
        canCancel = true,
        anim = {
            dict = "clothingshirt",
            clip = "try_shirt_positive_d",
            blendIn = 8.0
        }
    }) then
        deleteGear()
        local maskModel = `p_d_scuba_mask_s`
        local tankModel = `p_s_scuba_tank_s`
        lib.requestModel(maskModel)
        lib.requestModel(tankModel)
        currentGear.tank = CreateObject(tankModel, 1.0, 1.0, 1.0, true, true, false)
        local bone1 = GetPedBoneIndex(cache.ped, 24818)
        AttachEntityToEntity(currentGear.tank, cache.ped, bone1, -0.25, -0.25, 0.0, 180.0, 90.0, 0.0, true, true, false, false, 2, true)
        currentGear.mask = CreateObject(maskModel, 1.0, 1.0, 1.0, true, true, false)
        local bone2 = GetPedBoneIndex(cache.ped, 12844)
        AttachEntityToEntity(currentGear.mask, cache.ped, bone2, 0.0, 0.0, 0.0, 180.0, 90.0, 0.0, true, true, false, false, 2, true)
        SetEnableScuba(cache.ped, true)
        SetPedMaxTimeUnderwater(cache.ped, 2000.00)
        currentGear.enabled = true
        TriggerServerEvent("InteractSound_SV:PlayOnSource", "breathdivingsuit", 0.25)
        CreateThread(function()
            while currentGear.enabled do
                if IsPedSwimmingUnderWater(cache.ped) and oxygenLevel > 0 then
                    oxygenLevel -= 1
                    if oxygenLevel % 10 == 0 and oxygenLevel ~= Config.OxygenLevel then
                        TriggerServerEvent("InteractSound_SV:PlayOnSource", "breathdivingsuit", 0.25)
                    elseif oxygenLevel == 0 then
                        SetEnableScuba(cache.ped, false)
                        SetPedMaxTimeUnderwater(cache.ped, 1.00)
                        TriggerServerEvent("InteractSound_SV:PlayOnSource", nil, 0.25)
                        break
                    end
                end
                Wait(1000)
            end
        end)
        CreateThread(function()
            while currentGear.enabled do
                if IsPedSwimmingUnderWater(cache.ped) then
                    DrawText(oxygenLevel..'⏱')
                end
                Wait(0)
            end
        end)
    end

    ClearPedTasks(cache.ped)
end

RegisterNetEvent('qbx_divegear:client:useGear', function()
    if currentGear.enabled then
        takeOffSuit()
    else
        putOnSuit()
    end
end)