MicroUtils/coroutines/src/commonMain/kotlin/dev/inmo/micro_utils/coroutines/AwaitFirst.kt

package dev.inmo.micro_utils.coroutines

import kotlinx.coroutines.*
import kotlin.coroutines.*

suspend fun <T> Iterable<Deferred<T>>.awaitFirstWithDeferred(
    scope: CoroutineScope,
    cancelOnResult: Boolean = true
): Pair<Deferred<T>, T> = suspendCoroutine<Pair<Deferred<T>, T>> { continuation ->
    scope.launch(SupervisorJob()) {
        val scope = this
        forEach {
            scope.launch {
                continuation.resume(it to it.await())
                scope.cancel()
            }
        }
    }
}.also {
    if (cancelOnResult) {
        forEach {
            try {
                it.cancel()
            } catch (e: IllegalStateException) {
                e.printStackTrace()
            }
        }
    }
}

suspend fun <T> Iterable<Deferred<T>>.awaitFirst(
    scope: CoroutineScope,
    cancelOnResult: Boolean = true
): T = awaitFirstWithDeferred(scope, cancelOnResult).second
suspend fun <T> Iterable<Deferred<T>>.awaitFirst(
    cancelOthers: Boolean = true
): T = awaitFirst(CoroutineScope(coroutineContext), cancelOthers)