commonMain.com.petersamokhin.notionsdk.NotionImpl.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of notionsdk-js Show documentation
Notion SDK Kotlin Multiplatform
There is a newer version: 0.0.5
package com.petersamokhin.notionsdk

import co.touchlab.stately.concurrency.AtomicReference
import com.petersamokhin.notionsdk.data.NotionApiVersion
import com.petersamokhin.notionsdk.data.mapper.toDomain
import com.petersamokhin.notionsdk.data.model.internal.request.QueryDatabaseRequest
import com.petersamokhin.notionsdk.data.model.internal.response.QueryDatabaseResponse
import com.petersamokhin.notionsdk.data.model.internal.response.RetrieveDatabaseResponse
import com.petersamokhin.notionsdk.data.model.result.NotionDatabaseSchema
import com.petersamokhin.notionsdk.data.model.result.NotionDatabase
import io.ktor.client.*
import io.ktor.client.features.*
import io.ktor.client.features.json.*
import io.ktor.client.features.json.serializer.*
import io.ktor.client.request.*
import io.ktor.http.*

internal class NotionImpl(
    token: String,
    private val version: NotionApiVersion,
    private var httpClient: HttpClient
) : Notion {
    private val tokenReference = AtomicReference(token)

    init {
        httpClient = httpClient.withTokenAndVersion()
    }

    override val token: String
        get() = tokenReference.get()

    override fun setToken(token: String) =
        tokenReference.set(token)

    override fun setHttpClient(newHttpClient: HttpClient) {
        httpClient = newHttpClient.withTokenAndVersion()
    }

    override fun close() =
        httpClient.close()

    private fun HttpClient.withTokenAndVersion(): HttpClient =
        config {
            defaultRequest {
                header(HttpHeaders.Authorization, "Bearer $token")
                header(Notion.HEADER_VERSION, version.stringValue)
            }
            install(JsonFeature) {
                serializer = KotlinxSerializer(
                    kotlinx.serialization.json.Json { encodeDefaults = false; ignoreUnknownKeys = true }
                )
            }
        }

    override suspend fun queryDatabase(
        databaseId: String,
        startCursor: String?,
        pageSize: Int?
    ): NotionDatabase =
        httpClient.post("${Notion.API_BASE_URL}/${ENDPOINT_DATABASES}/$databaseId/$PATH_QUERY") {
            contentType(ContentType.Application.Json)
            body = QueryDatabaseRequest(
                startCursor = startCursor,
                pageSize = pageSize,
            )
        }.toDomain()

    override suspend fun retrieveDatabase(
        databaseId: String
    ): NotionDatabaseSchema =
        httpClient.get("${Notion.API_BASE_URL}/${ENDPOINT_DATABASES}/$databaseId")
            .toDomain()

    companion object {
        private const val ENDPOINT_DATABASES = "databases"
        private const val PATH_QUERY = "query"
    }
}