Add possibility of having VM key

Author: mlykotom

Fruitties/iosApp/iosApp/ViewModelStoreProvider.swift

@@ -34,13 +34,15 @@ class IOSViewModelStoreOwner: ObservableObject {
         Lifecycle_viewmodelViewModelStore()
 
     func viewModel<T: AnyObject>(
+        key: String? = nil,
         factory: Lifecycle_viewmodelViewModelProviderFactory,
         extras: Lifecycle_viewmodelCreationExtras,
     ) -> T {
         let vm =
             viewModelStore.getViewModel(
                 modelClass: T.self,
                 factory: factory,
+                key: key,
                 extras: extras
             ) as! T
 

Fruitties/shared/src/commonMain/kotlin/com/example/fruitties/viewmodel/CartViewModel.kt

@@ -30,15 +30,6 @@ class CartViewModel(
     private val repository: DataRepository,
 ) : ViewModel() {
 
-    init {
-        println("hello from CartViewmodel")
-    }
-
-    override fun onCleared() {
-        super.onCleared()
-        println("clearing CartViewModel")
-    }
-
     val cartUiState: StateFlow<CartUiState> =
         repository.cartDetails
             .map { details ->

Fruitties/shared/src/commonMain/kotlin/com/example/fruitties/viewmodel/ViewModelFactory.kt

@@ -25,6 +25,15 @@ fun creationExtras(appContainer: AppContainer): CreationExtras =
         set(APP_CONTAINER_KEY, appContainer)
     }
 
+fun creationExtras(
+    appContainer: AppContainer,
+    additional: MutableCreationExtras.() -> Unit
+): CreationExtras =
+    MutableCreationExtras().apply {
+        set(APP_CONTAINER_KEY, appContainer)
+        additional()
+    }
+
 inline fun <reified T : ViewModel> vmFactory(
     crossinline initializer: CreationExtras.(AppContainer) -> T
 ) =

Fruitties/shared/src/iosMain/kotlin/com/example/fruitties/di/viewmodel/IOSViewModelStoreOwner.kt

@@ -14,11 +14,12 @@ import kotlin.reflect.KClass
 fun ViewModelStore.getViewModel(
     modelClass: ObjCClass,
     factory: ViewModelProvider.Factory,
+    key: String?,
     extras: CreationExtras,
 ): ViewModel {
     @Suppress("UNCHECKED_CAST")
-    val vmClass = (getOriginalKotlinClass(modelClass) as? KClass<ViewModel>)
+    val vmClass = getOriginalKotlinClass(modelClass) as? KClass<ViewModel>
         ?: error("modelClass isn't a ViewModel type")
     val provider = ViewModelProvider.create(this, factory, extras)
-    return provider[vmClass]
-}
+    return key?.let { provider[key, vmClass] } ?: provider[vmClass]
+}