com.pubnub.internal.endpoints.presence.HereNowEndpoint.kt Maven / Gradle / Ivy

Go to download
package com.pubnub.internal.endpoints.presence

import com.google.gson.JsonElement
import com.pubnub.api.endpoints.presence.HereNow
import com.pubnub.api.enums.PNOperationType
import com.pubnub.api.models.consumer.presence.PNHereNowChannelData
import com.pubnub.api.models.consumer.presence.PNHereNowOccupantData
import com.pubnub.api.models.consumer.presence.PNHereNowResult
import com.pubnub.api.retry.RetryableEndpointGroup
import com.pubnub.internal.EndpointCore
import com.pubnub.internal.PubNubImpl
import com.pubnub.internal.models.server.Envelope
import com.pubnub.internal.toCsv
import retrofit2.Call
import retrofit2.Response

/**
 * @see [PubNubImpl.hereNow]
 */
class HereNowEndpoint internal constructor(
    pubnub: PubNubImpl,
    override val channels: List = emptyList(),
    override val channelGroups: List = emptyList(),
    override val includeState: Boolean = false,
    override val includeUUIDs: Boolean = true,
) : EndpointCore, PNHereNowResult>(pubnub), HereNow {
    private fun isGlobalHereNow() = channels.isEmpty() && channelGroups.isEmpty()

    override fun getAffectedChannels() = channels

    override fun getAffectedChannelGroups() = channelGroups

    override fun doWork(queryParams: HashMap): Call> {
        addQueryParams(queryParams)

        return if (!isGlobalHereNow()) {
            retrofitManager.presenceService.hereNow(
                configuration.subscribeKey,
                channels.toCsv(),
                queryParams,
            )
        } else {
            retrofitManager.presenceService.globalHereNow(
                configuration.subscribeKey,
                queryParams,
            )
        }
    }

    override fun createResponse(input: Response>): PNHereNowResult {
        return if (isGlobalHereNow() || (channels.size > 1 || channelGroups.isNotEmpty())) {
            parseMultipleChannelResponse(input.body()?.payload!!)
        } else {
            parseSingleChannelResponse(input.body()!!)
        }
    }

    override fun operationType() = PNOperationType.PNHereNowOperation

    override fun getEndpointGroupName(): RetryableEndpointGroup = RetryableEndpointGroup.PRESENCE

    private fun parseSingleChannelResponse(input: Envelope): PNHereNowResult {
        val pnHereNowResult =
            PNHereNowResult(
                totalChannels = 1,
                totalOccupancy = input.occupancy,
            )

        val pnHereNowChannelData =
            PNHereNowChannelData(
                channelName = channels[0],
                occupancy = input.occupancy,
            )

        if (includeUUIDs) {
            pnHereNowChannelData.occupants = prepareOccupantData(input.uuids!!)
            pnHereNowResult.channels[channels[0]] = pnHereNowChannelData
        }

        return pnHereNowResult
    }

    private fun parseMultipleChannelResponse(input: JsonElement): PNHereNowResult {
        val pnHereNowResult =
            PNHereNowResult(
                totalChannels = pubnub.mapper.elementToInt(input, "total_channels"),
                totalOccupancy = pubnub.mapper.elementToInt(input, "total_occupancy"),
            )

        val it = pubnub.mapper.getObjectIterator(input, "channels")

        while (it.hasNext()) {
            val entry = it.next()
            val pnHereNowChannelData =
                PNHereNowChannelData(
                    channelName = entry.key,
                    occupancy = pubnub.mapper.elementToInt(entry.value, "occupancy"),
                )
            if (includeUUIDs) {
                pnHereNowChannelData.occupants = prepareOccupantData(pubnub.mapper.getField(entry.value, "uuids")!!)
            }
            pnHereNowResult.channels[entry.key] = pnHereNowChannelData
        }

        return pnHereNowResult
    }

    private fun prepareOccupantData(input: JsonElement): MutableList {
        val occupantsResults = mutableListOf()

        val it = pubnub.mapper.getArrayIterator(input)
        while (it?.hasNext()!!) {
            val occupant = it.next()
            occupantsResults.add(
                if (includeState) {
                    PNHereNowOccupantData(
                        uuid = pubnub.mapper.elementToString(occupant, "uuid")!!,
                        state = pubnub.mapper.getField(occupant, "state"),
                    )
                } else {
                    PNHereNowOccupantData(
                        uuid = pubnub.mapper.elementToString(occupant)!!,
                    )
                },
            )
        }

        return occupantsResults
    }

    private fun addQueryParams(queryParams: MutableMap) {
        if (includeState) {
            queryParams["state"] = "1"
        }
        if (!includeUUIDs) {
            queryParams["disable_uuids"] = "1"
        }
        if (channelGroups.isNotEmpty()) {
            queryParams["channel-group"] = channelGroups.toCsv()
        }
    }
}