|
|
|
|
@ -1,29 +1,18 @@
|
|
|
|
|
package org.koitharu.kotatsu.core.util.ext
|
|
|
|
|
|
|
|
|
|
import androidx.lifecycle.Lifecycle
|
|
|
|
|
import androidx.lifecycle.LifecycleDestroyedException
|
|
|
|
|
import androidx.lifecycle.LifecycleEventObserver
|
|
|
|
|
import androidx.lifecycle.LifecycleObserver
|
|
|
|
|
import androidx.lifecycle.LifecycleOwner
|
|
|
|
|
import androidx.lifecycle.ProcessLifecycleOwner
|
|
|
|
|
import androidx.lifecycle.lifecycleScope
|
|
|
|
|
import dagger.hilt.android.lifecycle.RetainedLifecycle
|
|
|
|
|
import kotlinx.coroutines.CancellableContinuation
|
|
|
|
|
import kotlinx.coroutines.CoroutineScope
|
|
|
|
|
import kotlinx.coroutines.Deferred
|
|
|
|
|
import kotlinx.coroutines.Dispatchers
|
|
|
|
|
import kotlinx.coroutines.Job
|
|
|
|
|
import kotlinx.coroutines.joinAll
|
|
|
|
|
import kotlinx.coroutines.plus
|
|
|
|
|
import kotlinx.coroutines.suspendCancellableCoroutine
|
|
|
|
|
import org.koitharu.kotatsu.core.util.AcraCoroutineErrorHandler
|
|
|
|
|
import org.koitharu.kotatsu.core.util.RetainedLifecycleCoroutineScope
|
|
|
|
|
import org.koitharu.kotatsu.parsers.util.cancelAll
|
|
|
|
|
import org.koitharu.kotatsu.parsers.util.runCatchingCancellable
|
|
|
|
|
import kotlin.coroutines.EmptyCoroutineContext
|
|
|
|
|
import kotlin.coroutines.cancellation.CancellationException
|
|
|
|
|
import kotlin.coroutines.resume
|
|
|
|
|
import kotlin.coroutines.resumeWithException
|
|
|
|
|
|
|
|
|
|
val processLifecycleScope: CoroutineScope
|
|
|
|
|
get() = ProcessLifecycleOwner.get().lifecycleScope + AcraCoroutineErrorHandler()
|
|
|
|
|
@ -31,54 +20,6 @@ val processLifecycleScope: CoroutineScope
|
|
|
|
|
val RetainedLifecycle.lifecycleScope: RetainedLifecycleCoroutineScope
|
|
|
|
|
inline get() = RetainedLifecycleCoroutineScope(this)
|
|
|
|
|
|
|
|
|
|
suspend fun Lifecycle.awaitStateAtLeast(state: Lifecycle.State) {
|
|
|
|
|
if (currentState.isAtLeast(state)) {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
suspendCancellableCoroutine { cont ->
|
|
|
|
|
val observer = ContinuationLifecycleObserver(this, cont, state)
|
|
|
|
|
addObserverFromAnyThread(observer)
|
|
|
|
|
cont.invokeOnCancellation {
|
|
|
|
|
removeObserverFromAnyThread(observer)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private class ContinuationLifecycleObserver(
|
|
|
|
|
private val lifecycle: Lifecycle,
|
|
|
|
|
private val continuation: CancellableContinuation<Unit>,
|
|
|
|
|
private val targetState: Lifecycle.State,
|
|
|
|
|
) : LifecycleEventObserver {
|
|
|
|
|
|
|
|
|
|
override fun onStateChanged(source: LifecycleOwner, event: Lifecycle.Event) {
|
|
|
|
|
if (event == Lifecycle.Event.upTo(targetState)) {
|
|
|
|
|
lifecycle.removeObserver(this)
|
|
|
|
|
continuation.resume(Unit)
|
|
|
|
|
} else if (event == Lifecycle.Event.ON_DESTROY) {
|
|
|
|
|
lifecycle.removeObserver(this)
|
|
|
|
|
continuation.resumeWithException(LifecycleDestroyedException())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private fun Lifecycle.addObserverFromAnyThread(observer: LifecycleObserver) {
|
|
|
|
|
val dispatcher = Dispatchers.Main.immediate
|
|
|
|
|
if (dispatcher.isDispatchNeeded(EmptyCoroutineContext)) {
|
|
|
|
|
dispatcher.dispatch(EmptyCoroutineContext) { addObserver(observer) }
|
|
|
|
|
} else {
|
|
|
|
|
addObserver(observer)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private fun Lifecycle.removeObserverFromAnyThread(observer: LifecycleObserver) {
|
|
|
|
|
val dispatcher = Dispatchers.Main.immediate
|
|
|
|
|
if (dispatcher.isDispatchNeeded(EmptyCoroutineContext)) {
|
|
|
|
|
dispatcher.dispatch(EmptyCoroutineContext) { removeObserver(observer) }
|
|
|
|
|
} else {
|
|
|
|
|
removeObserver(observer)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fun <T> Deferred<T>.getCompletionResultOrNull(): Result<T>? = if (isCompleted) {
|
|
|
|
|
getCompletionExceptionOrNull()?.let { error ->
|
|
|
|
|
Result.failure(error)
|
|
|
|
|
|
