package com.bitmovin.player.reactnative

import com.bitmovin.player.api.buffer.BufferLevel
import com.bitmovin.player.api.media.MediaType
import com.bitmovin.player.reactnative.converter.toBufferType
import com.bitmovin.player.reactnative.converter.toJson
import com.facebook.react.bridge.*
import com.facebook.react.module.annotations.ReactModule

private const val MODULE_NAME = "BufferModule"
private const val INVALID_BUFFER_TYPE = "Invalid buffer type"

@ReactModule(name = MODULE_NAME)
class BufferModule(context: ReactApplicationContext) : BitmovinBaseModule(context) {
    override fun getName() = MODULE_NAME

    /**
     * Gets the [BufferLevel] from the Player
     * @param nativeId Target player id.
     * @param type The [type of buffer][toBufferType] to return the level for.
     * @param promise JS promise object.
     */
    @ReactMethod
    fun getLevel(nativeId: NativeId, type: String, promise: Promise) {
        promise.map.resolveOnUiThread {
            val player = getPlayer(nativeId)
            val bufferType = type.toBufferTypeOrThrow()
            RNBufferLevels(
                audio = player.buffer.getLevel(bufferType, MediaType.Audio),
                video = player.buffer.getLevel(bufferType, MediaType.Video),
            ).toJson()
        }
    }

    /**
     * Sets the target buffer level for the chosen buffer type across all media types.
     * @param nativeId Target player id.
     * @param type The [type of buffer][toBufferType] to set the target level for.
     * @param value The value to set.
     */
    @ReactMethod
    fun setTargetLevel(nativeId: NativeId, type: String, value: Double, promise: Promise) {
        promise.unit.resolveOnUiThread {
            getPlayer(nativeId).buffer.setTargetLevel(type.toBufferTypeOrThrow(), value)
        }
    }

    private fun String.toBufferTypeOrThrow() = toBufferType() ?: throw IllegalArgumentException(INVALID_BUFFER_TYPE)
}

/**
 * Representation of the React Native API `BufferLevels` object.
 * This is necessary as we need a unified representation of the different APIs from both Android and iOS.
 */
data class RNBufferLevels(val audio: BufferLevel, val video: BufferLevel)