Lifecycle aware extensions.

LifecycleExtension.kt

fun Activity.doOnceOnLifecycle(state: Lifecycle.State, action: () -> Unit) {
    (this as? LifecycleOwner)?.lifecycleScope?.launch {
        when (state) {
            Lifecycle.State.CREATED -> withCreated {
                action.invoke()
            }

            Lifecycle.State.STARTED -> withStarted {
                action.invoke()
            }

            Lifecycle.State.RESUMED -> withResumed {
                action.invoke()
            }

            else -> {
                action.invoke()
            }
        }
    }
}

fun Activity.doRepeatOnLifecycle(state: Lifecycle.State, action: () -> Unit) {
    (this as? LifecycleOwner)?.lifecycleScope?.launch {
        repeatOnLifecycle(state) {
            action.invoke()
        }
    }
}