Restartable.kt

/**
* Inspired by Jaewoong Eum's flow-operators library (https://github.com/skydoves/flow-operators).
*/

class Restartable {

    private val restartFlow = MutableSharedFlow<SharingCommand>(extraBufferCapacity = 2)

    fun restart() {
        restartFlow.tryEmit(SharingCommand.STOP_AND_RESET_REPLAY_CACHE)
        restartFlow.tryEmit(SharingCommand.START)
    }

    companion object {

        @JvmStatic
        fun SharingStarted.with(restartable: Restartable) = SharingStarted { subscriptionCount ->
            merge(restartable.restartFlow, command(subscriptionCount))
        }
    }
}