com.pubnub.internal.endpoints.objects.member.ManageChannelMembersEndpoint.kt Maven / Gradle / Ivy

Go to download
package com.pubnub.internal.endpoints.objects.member

import com.pubnub.api.endpoints.objects.member.ManageChannelMembers
import com.pubnub.api.enums.PNOperationType
import com.pubnub.api.models.consumer.objects.member.MemberInput
import com.pubnub.api.models.consumer.objects.member.PNMember
import com.pubnub.api.models.consumer.objects.member.PNMemberArrayResult
import com.pubnub.api.retry.RetryableEndpointGroup
import com.pubnub.internal.EndpointCore
import com.pubnub.internal.PubNubImpl
import com.pubnub.internal.endpoints.objects.internal.CollectionQueryParameters
import com.pubnub.internal.endpoints.objects.internal.IncludeQueryParam
import com.pubnub.internal.extension.toPNMemberArrayResult
import com.pubnub.internal.models.server.objects_api.ChangeMemberInput
import com.pubnub.internal.models.server.objects_api.EntityArrayEnvelope
import com.pubnub.internal.models.server.objects_api.ServerMemberInput
import com.pubnub.internal.models.server.objects_api.UUIDId
import retrofit2.Call
import retrofit2.Response

/**
 * @see [PubNubImpl.manageChannelMembers]
 */
class ManageChannelMembersEndpoint(
    pubnub: PubNubImpl,
    private val uuidsToSet: Collection,
    private val uuidsToRemove: Collection,
    private val channel: String,
    private val collectionQueryParameters: CollectionQueryParameters,
    private val includeQueryParam: IncludeQueryParam,
) : EndpointCore, PNMemberArrayResult>(pubnub), ManageChannelMembers {
    override fun doWork(queryParams: HashMap): Call> {
        val params =
            queryParams + collectionQueryParameters.createCollectionQueryParams() + includeQueryParam.createIncludeQueryParams()

        return retrofitManager.objectsService.patchChannelMembers(
            channel = channel,
            subKey = configuration.subscribeKey,
            options = params,
            body =
                ChangeMemberInput(
                    delete = uuidsToRemove.map { ServerMemberInput(UUIDId(id = it)) },
                    set =
                        uuidsToSet.map {
                            ServerMemberInput(
                                uuid = UUIDId(id = it.uuid),
                                custom = it.custom,
                                status = it.status,
                            )
                        },
                ),
        )
    }

    override fun createResponse(input: Response>): PNMemberArrayResult = input.toPNMemberArrayResult()

    override fun operationType(): PNOperationType = PNOperationType.ObjectsOperation()

    override fun getEndpointGroupName(): RetryableEndpointGroup = RetryableEndpointGroup.APP_CONTEXT
}