commonMain.ro.dragossusi.lifecycle.rememberViewModel.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of navigation Show documentation

Navigation for compose multiplatform

The newest version!

package ro.dragossusi.lifecycle

import androidx.compose.runtime.Composable
import androidx.compose.runtime.remember
import ro.dragossusi.navigation.LocalNavArguments
import ro.dragossusi.navigation.LocalViewModelStore
import ro.dragossusi.navigation.NavArguments
import ro.dragossusi.navigation.emptyArguments

@Suppress("UNCHECKED_CAST")
@Composable
fun  rememberViewModel(
    builder: (NavArguments) -> VM
): VM {
    val store = LocalViewModelStore.current
        ?: throw IllegalArgumentException("No ViewModelStore")
    val arguments = LocalNavArguments.current ?: emptyArguments()
    return remember(store, arguments) {
        store.getOrPut("nav_view_model") {
            builder(arguments)
        } as VM
    }
}