suspend_complete.kt

suspend fun View.awaitLayout() = suspendCancellableCoroutine<Unit>{ cont ->
    if (isLaidOut) {
        // if view is laid out, just resume coroutine
        cont.resume(Unit)
        return@suspendCancellableCoroutine
    }
    val listener = object : View.OnLayoutChangeListener {
        override fun onLayoutChange(
            view: View,
            left: Int,
            top: Int,
            right: Int,
            bottom: Int,
            oldLeft: Int,
            oldTop: Int,
            oldRight: Int,
            oldBottom: Int
        ) {
            view.removeOnLayoutChangeListener(this)
            cont.resume(Unit)
        }
    }
    
    // remove listener when the job is cancelled
    cont.invokeOnCancellation { removeOnLayoutChangeListener(listener) }
    addOnLayoutChangeListener(listener)
}