Anyone in here using KMM ViewModel? Trying to a cr...

# kotlin-inject

[brandonmcansh]

Anyone in here using KMM ViewModel? Trying to a create an extension function similar to hiltViewModel() that caches vm instances

[brandonmcansh]

was able to get this to work via:

@Composable
inline fun <reified T : ScreenModel> Screen.screenViewModel(
    crossinline factory: @DisallowComposableCalls () -> T = {
        with (ViewModelComponent::class.create()) {
            getViewModel<T>()
        } as T
    }
): T = this.rememberScreenModel(tag = T::class.qualifiedName) { factory() }

with ViewModelComponent being:

@Component
abstract class ViewModelComponent: AppComponent() {
    abstract val loginViewModel: LoginViewModel
    abstract val homeViewModel: HomeViewModel

    inline fun <reified T: ScreenModel> getViewModel(): ScreenModel {
        return when (val clazz = T::class) {
            LoginViewModel::class -> loginViewModel
            HomeViewModel::class -> homeViewModel
            else -> throw NoSuchElementException("No viewModel found for ${clazz.simpleName}")
        }
    }
}

[brandonmcansh]

now I can

[brandonmcansh]

@Composable
private fun Screens.Home.Content(
    modifier: Modifier = Modifier,
    viewModel: HomeViewModel = screenViewModel()
) {

[dorche]

What does rememberScreenModel do? I've done something similar but no remember calls (mainly because Hilt didn't seem to use remember)

@Composable
inline fun <reified VM : ViewModel, reified F: ViewModelFactory<VM>> kotlinInjectViewModel(
    factory: (ApplicationComponent) -> F,
    viewModelStoreOwner: ViewModelStoreOwner = checkNotNull(LocalViewModelStoreOwner.current) {
        "No ViewModelStoreOwner was provided via LocalViewModelStoreOwner"
    },
): VM {
    val context = LocalContext.current
    val component = factory(context.applicationComponent)
    return viewModel(
        viewModelStoreOwner = viewModelStoreOwner,
        factory = component.factory()
    )
}

[brandonmcansh]

rememberScreenModel is available from voyager navigation which is:

@Composable
public inline fun <reified T : ScreenModel> Screen.rememberScreenModel(
    tag: String? = null,
    crossinline factory: @DisallowComposableCalls () -> T
): T =
    remember(ScreenModelStore.getKey<T>(this, tag)) {
        ScreenModelStore.getOrPut(this, tag, factory)
    }

[brandonmcansh]

https://github.com/adrielcafe/voyager

[brandonmcansh]

Similar concept to what you have setup when boiled down