MicroUtils/repos/ktor/client/src/commonMain/kotlin/dev/inmo/micro_utils/repos/ktor/client/key_value/KtorWriteStandardKeyValueRepo.kt

package dev.inmo.micro_utils.repos.ktor.client.key_value

import dev.inmo.micro_utils.ktor.client.*
import dev.inmo.micro_utils.ktor.common.*
import dev.inmo.micro_utils.repos.WriteStandardKeyValueRepo
import dev.inmo.micro_utils.repos.ktor.common.key_value.*
import io.ktor.client.HttpClient
import kotlinx.coroutines.flow.Flow
import kotlinx.serialization.KSerializer
import kotlinx.serialization.builtins.*

class KtorWriteStandardKeyValueRepo<K, V> (
    private var baseUrl: String,
    private var unifiedRequester: UnifiedRequester,
    private var keySerializer: KSerializer<K>,
    private var valueSerializer: KSerializer<V>,
) : WriteStandardKeyValueRepo<K, V> {
    private val keyValueMapSerializer = MapSerializer(keySerializer, valueSerializer)
    private val keysListSerializer = ListSerializer(keySerializer)

    constructor(
        baseUrl: String,
        client: HttpClient,
        keySerializer: KSerializer<K>,
        valueSerializer: KSerializer<V>,
        serialFormat: StandardKtorSerialFormat = standardKtorSerialFormat
    ) : this (
        baseUrl, UnifiedRequester(client, serialFormat), keySerializer, valueSerializer
    )

    override val onNewValue: Flow<Pair<K, V>> = unifiedRequester.createStandardWebsocketFlow(
        buildStandardUrl(baseUrl, onNewValueRoute),
        deserializer = PairSerializer(keySerializer, valueSerializer)
    )

    override val onValueRemoved: Flow<K> = unifiedRequester.createStandardWebsocketFlow(
        buildStandardUrl(baseUrl, onValueRemovedRoute),
        deserializer = keySerializer
    )

    override suspend fun set(toSet: Map<K, V>) = unifiedRequester.unipost(
        buildStandardUrl(
            baseUrl,
            setRoute
        ),
        BodyPair(keyValueMapSerializer, toSet),
        Unit.serializer()
    )

    override suspend fun unset(toUnset: List<K>) = unifiedRequester.unipost(
        buildStandardUrl(
            baseUrl,
            unsetRoute,
        ),
        BodyPair(keysListSerializer, toUnset),
        Unit.serializer()
    )
}