Kotlin flow which observes internet connectivity on Android

ConnectivityStateListener.kt

import android.content.Context
import android.net.ConnectivityManager
import android.net.Network
import android.net.NetworkCapabilities
import android.net.NetworkRequest
import android.util.Log
import androidx.core.content.getSystemService
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.channels.awaitClose
import kotlinx.coroutines.channels.onFailure
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.callbackFlow

/**
 * Listens for network callbacks using connectivity manager and returns a flow
 * which emits true when internet is available and false when not
 *
 * @param context the context
 * @return A cold flow which emits boolean values indicating internet connectivity
 */
@OptIn(ExperimentalCoroutinesApi::class)
fun connectivityStateFlow(context: Context): Flow<Boolean> = callbackFlow {
    val connectivityManager: ConnectivityManager? = context.getSystemService()

    val callbacks = object : ConnectivityManager.NetworkCallback() {
        override fun onAvailable(network: Network) {
            super.onAvailable(network)
            this@callbackFlow.trySend(true).onFailure {
                Log.e("connectivityStateFlow", "Error while sending connectivity state to flow", it)
            }

        }

        override fun onLost(network: Network) {
            super.onLost(network)
            this@callbackFlow.trySend(false).onFailure {
                Log.e("connectivityStateFlow", "Error while sending connectivity state to flow", it)
            }
        }
    }

    val networkRequest = NetworkRequest.Builder()
        .addCapability(NetworkCapabilities.NET_CAPABILITY_INTERNET)
        .build()

    connectivityManager?.registerNetworkCallback(networkRequest, callbacks)

    awaitClose {
        connectivityManager?.unregisterNetworkCallback(callbacks)
    }
}