jvmMain.io.dyte.webrtc.MediaStream.kt Maven / Gradle / Ivy

Go to download
package io.dyte.webrtc

import dev.onvoid.webrtc.media.MediaStream as NativeMediaStream
import dev.onvoid.webrtc.media.audio.AudioTrack
import dev.onvoid.webrtc.media.video.VideoTrack
import java.util.UUID

actual class MediaStream
internal constructor(
  val native: NativeMediaStream? = null,
  actual val id: String = native?.id() ?: UUID.randomUUID().toString(),
) {

  private val _tracks = mutableListOf()
  actual val tracks: List = _tracks

  actual open fun addTrack(track: MediaStreamTrack) {

    native?.let {
      when (track.native) {
        is AudioTrack -> it.addTrack(track.native)
        is VideoTrack -> it.addTrack(track.native)
        else -> error("Unknown MediaStreamTrack kind: ${track.kind}")
      }
    }
    _tracks += track
  }

  actual open fun getTrackById(id: String): MediaStreamTrack? {
    return tracks.firstOrNull { it.id == id }
  }

  actual open fun removeTrack(track: MediaStreamTrack) {

    native?.let {
      when (track.native) {
        is AudioTrack -> it.removeTrack(track.native)
        is VideoTrack -> it.removeTrack(track.native)
        else -> error("Unknown MediaStreamTrack kind: ${track.kind}")
      }
    }
    _tracks -= track
  }

  actual open fun release() {
    tracks.forEach(MediaStreamTrack::stop)
    native?.dispose()
  }
}