io.datalbry.jetbrains.space.client.profile.ProfilesClientImpl.kt Maven / Gradle / Ivy

Go to download
package io.datalbry.jetbrains.space.client.profile

import io.datalbry.jetbrains.space.client.PaginationIterator
import io.datalbry.jetbrains.space.models.profile.Profile
import io.datalbry.jetbrains.space.models.profile.ProfileIdentifier
import kotlinx.coroutines.runBlocking
import kotlinx.datetime.TimeZone
import kotlinx.datetime.toJavaLocalDate
import kotlinx.datetime.toJavaLocalDateTime
import kotlinx.datetime.toLocalDateTime
import space.jetbrains.api.runtime.Batch
import space.jetbrains.api.runtime.BatchInfo
import space.jetbrains.api.runtime.SpaceHttpClientWithCallContext
import space.jetbrains.api.runtime.resources.teamDirectory
import space.jetbrains.api.runtime.types.TD_MemberProfile

class ProfilesClientImpl(private val spaceClient: SpaceHttpClientWithCallContext) : ProfilesClient {

    override fun getProfile(profileIdentifier: ProfileIdentifier): Profile {
        val profile: TD_MemberProfile = runBlocking {
            spaceClient.teamDirectory.profiles.getProfile(
                profile = space.jetbrains.api.runtime.types.ProfileIdentifier.Id(profileIdentifier.key)
            )
        }

        return Profile(
            id = profile.id,
            archived = profile.archived,
            avatar = profile.avatar,
            joined = profile.joined?.toJavaLocalDate()?.atStartOfDay(),
            left = profile.left?.toJavaLocalDate()?.atStartOfDay(),
            leftAt = profile.leftAt?.toLocalDateTime(TimeZone.UTC)?.toJavaLocalDateTime(),
            firstName = profile.name.firstName,
            lastName = profile.name.lastName,
            notAMember = profile.notAMember,
            profilePicture = profile.profilePicture,
            smallAvatar = profile.smallAvatar,
            speaksEnglish = profile.speaksEnglish,
            username = profile.username,
        )
    }

    override fun getProfileIdentifier(): Iterator {
        return PaginationIterator(
            { getNextBatch(it) },
            { ProfileIdentifier(it.id) }
        )
    }

    private fun getNextBatch(batchInfo: BatchInfo): Batch {
        return runBlocking {
            spaceClient.teamDirectory.profiles.getAllProfiles(batchInfo = batchInfo) {
                id()
            }
        }
    }
}