use VmStoreOwner instead of VmStore directly

Author: mlykotom

Fruitties/iosApp/iosApp/IOSViewModelStoreOwner.swift

@@ -14,7 +14,7 @@ class IOSViewModelStoreOwner: ObservableObject, ViewModelStoreOwner {
         extras: CreationExtras? = nil
     ) -> T {
         do {
-            return try viewModelStore.getViewModel(
+            return try viewModel(
                 modelClass: T.self,
                 factory: factory,
                 key: key,

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

@@ -10,6 +10,7 @@ import kotlinx.cinterop.ObjCClass
 import kotlinx.cinterop.getOriginalKotlinClass
 import kotlin.reflect.KClass
 
+
 /**
  * This function allows retrieving any ViewModel from Swift Code with generics.
  * We only get [ObjCClass] type for the [modelClass], because the interop between Kotlin and Swift
@@ -18,7 +19,7 @@ import kotlin.reflect.KClass
 @Suppress("unused") // Android Studio is not aware of iOS usage.
 @OptIn(BetaInteropApi::class)
 @Throws(IllegalStateException::class)
-fun ViewModelStore.getViewModel(
+fun ViewModelStoreOwner.viewModel(
     modelClass: ObjCClass,
     factory: ViewModelProvider.Factory,
     key: String?,
@@ -28,6 +29,5 @@ fun ViewModelStore.getViewModel(
     val vmClass = getOriginalKotlinClass(modelClass) as? KClass<ViewModel>
         ?: error("modelClass isn't a ViewModel type")
     val provider = ViewModelProvider.create(this, factory, extras ?: CreationExtras.Empty)
-    return key?.let { provider[key, vmClass] } ?: provider[vmClass]
-}
-
+    return key?.let { provider.get(key, vmClass) } ?: provider.get(vmClass)
+}