Some more useful APIs for avoiding state production when inactive.

molecule-runtime/src/commonMain/kotlin/app/cash/molecule/moleculeActive.kt

@@ -18,14 +18,18 @@ package app.cash.molecule
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.LaunchedEffect
 import androidx.compose.runtime.State
+import androidx.compose.runtime.compositionLocalOf
 import androidx.compose.runtime.mutableStateOf
 import androidx.compose.runtime.remember
 import kotlin.coroutines.AbstractCoroutineContextElement
 import kotlin.coroutines.CoroutineContext
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.collectLatest
+import kotlinx.coroutines.flow.first
+import kotlinx.coroutines.suspendCancellableCoroutine
 
 internal class MoleculeActiveContextElement(
   val isMoleculeActive: StateFlow<Boolean>,
@@ -78,6 +82,13 @@ public suspend fun CoroutineScope.repeatWhileMoleculeActive(block: suspend () ->
   runBlockWhenActive(moleculeActiveFlow, block)
 }
 
+/**
+ * Suspends until the Molecule is active.
+ */
+public suspend fun CoroutineScope.awaitMoleculeActive() {
+  awaitActive(moleculeActiveFlow)
+}
+
 private suspend fun runBlockWhenActive(
   moleculeActiveFlow: StateFlow<Boolean>,
   block: suspend () -> Unit,
@@ -87,3 +98,7 @@ private suspend fun runBlockWhenActive(
     block()
   }
 }
+
+private suspend fun awaitActive(moleculeActiveFlow: StateFlow<Boolean>) {
+  moleculeActiveFlow.first { it }
+}