package expo.modules.transparentvideo.records

import androidx.media3.common.MediaItem
import expo.modules.kotlin.records.Field
import expo.modules.kotlin.records.Record
import expo.modules.transparentvideo.enums.DRMType
import java.io.Serializable

internal class DRMOptions(
  @Field var type: DRMType = DRMType.WIDEVINE,
  @Field var licenseServer: String? = null,
  @Field var headers: Map<String, String>? = null,
  @Field var multiKey: Boolean = false
) : Record, Serializable {

  fun toDRMConfiguration(): MediaItem.DrmConfiguration {
    val drmConfiguration = MediaItem.DrmConfiguration.Builder(type.toUUID())
    licenseServer?.let { drmConfiguration.setLicenseUri(it) }
    headers?.let { drmConfiguration.setLicenseRequestHeaders(it) }
    drmConfiguration.setMultiSession(multiKey)
    return drmConfiguration.build()
  }
}