SharedViewModel (Compose Navigation)

SharedViewModel.kt

import androidx.compose.runtime.Composable
import androidx.compose.runtime.LaunchedEffect
import androidx.lifecycle.ViewModel
import org.koin.androidx.compose.getViewModel
import java.lang.ref.WeakReference

class SharedViewModelHolder<T: ViewModel> {
    var sharedViewModelReference = WeakReference<T>(null)
}

@Composable
inline fun <reified T: ViewModel> SharedViewModelHolder<T>.createNewSharedViewModel(): T {
    val viewModel: T = getViewModel()
    LaunchedEffect(Unit) {
        if (sharedViewModelReference.get() !== viewModel) {
            sharedViewModelReference = WeakReference(viewModel)
        }
    }
    return viewModel
}

@Composable
inline fun <reified T : ViewModel> SharedViewModelHolder<T>.getSharedViewModel(): T {
    return sharedViewModelReference.get() ?: getViewModel()
}