pbprateek · August 16, 2024 13:24 · Oct 23, 2023 · Aug 17, 2023
diff --git a/ComposeScopedViewModelProvider.kt b/ComposeScopedViewModelProvider.kt
@@ -66,13 +66,14 @@ class CompositionScopedViewModelStoreOwner(
 
     override val defaultViewModelCreationExtras: CreationExtras =
         MutableCreationExtras(originalFactoryProvider.defaultViewModelCreationExtras).apply {
+            val bundle = Bundle()
+            get(DEFAULT_ARGS_KEY)?.let {
+                bundle.putAll(it)
+            }
             extraParams?.let {
-                val savedStateHandlerNew: Bundle? = get(DEFAULT_ARGS_KEY)
-                savedStateHandlerNew?.putAll(it)
-                if (savedStateHandlerNew != null)
-                    set(DEFAULT_ARGS_KEY, savedStateHandlerNew)
+                bundle.putAll(it)
             }
-
+            set(DEFAULT_ARGS_KEY, bundle)
         }
 
     override val defaultViewModelProviderFactory: ViewModelProvider.Factory

diff --git a/ComposeScopedViewModelProvider.kt b/ComposeScopedViewModelProvider.kt
@@ -0,0 +1,82 @@
+import android.os.Bundle
+import androidx.compose.runtime.Composable
+import androidx.compose.runtime.RememberObserver
+import androidx.compose.runtime.remember
+import androidx.lifecycle.*
+import androidx.lifecycle.viewmodel.CreationExtras
+import androidx.lifecycle.viewmodel.MutableCreationExtras
+import androidx.lifecycle.viewmodel.compose.LocalViewModelStoreOwner
+import androidx.lifecycle.viewmodel.compose.viewModel
+
+
+//This is creating using help from,Thanks for that
+//https://gist.github.com/manuelvicnt/a2e4c4812243ac1b218b24d0ac8d22bb#file-provideviewmodels-kt
+
+//This works with hilt
+
+// This is a way to scope ViewModels to the Composition.
+// However, this doesn't survive configuration changes or procress death on its own.
+// You can handle all config changes in compose by making the activity handle those in the Manifest file
+// e.g. android:configChanges="colorMode|density|fontScale|keyboard|keyboardHidden|layoutDirection|locale|mcc|mnc|navigation|orientation|screenLayout|screenSize|smallestScreenSize|touchscreen|uiMode">
+
+
+@Composable
+public inline fun <reified VM : ViewModel> ComposeScopedViewModelProvider(
+    key: String? = null,
+    extraParams: Bundle? = null
+): VM {
+    val originalViewModelStoreOwner = checkNotNull(LocalViewModelStoreOwner.current) {
+        "No ViewModelStoreOwner was provided via LocalViewModelStoreOwner"
+    }
+    val composeViewModelStoreOwner =
+        remember() {
+            CompositionScopedViewModelStoreOwner(
+                originalViewModelStoreOwner as HasDefaultViewModelProviderFactory,
+                extraParams
+            )
+        }
+    return viewModel(viewModelStoreOwner = composeViewModelStoreOwner, key = key)
+}
+
+
+class CompositionScopedViewModelStoreOwner(
+    private val originalFactoryProvider: HasDefaultViewModelProviderFactory,
+    private val extraParams: Bundle?
+) :
+    ViewModelStoreOwner, RememberObserver, HasDefaultViewModelProviderFactory {
+
+
+    private val viewModelStoreNew = ViewModelStore()
+
+
+    override val viewModelStore: ViewModelStore = viewModelStoreNew
+
+    override fun onAbandoned() {
+        viewModelStore.clear()
+    }
+
+    override fun onForgotten() {
+        viewModelStore.clear()
+    }
+
+    override fun onRemembered() {
+        // Nothing to do here
+    }
+
+
+    override val defaultViewModelCreationExtras: CreationExtras =
+        MutableCreationExtras(originalFactoryProvider.defaultViewModelCreationExtras).apply {
+            extraParams?.let {
+                val savedStateHandlerNew: Bundle? = get(DEFAULT_ARGS_KEY)
+                savedStateHandlerNew?.putAll(it)
+                if (savedStateHandlerNew != null)
+                    set(DEFAULT_ARGS_KEY, savedStateHandlerNew)
+            }
+
+        }
+
+    override val defaultViewModelProviderFactory: ViewModelProvider.Factory
+        get() = originalFactoryProvider.defaultViewModelProviderFactory
+
+
+}
diff --git a/Example.kt b/Example.kt
@@ -0,0 +1,15 @@
+@Composable
+fun ExampleCompose(
+    modifier: Modifier = Modifier
+) {
+    val mcqViewModel: TestViewModel = ComposeScopedViewModelProvider()
+
+    //or
+
+     val viewModel:  TestViewModel = ComposeScopedViewModelProvider(
+        key = id,
+        extraParams = Bundle().apply {
+            putString(EXTRAS_ID, id)
+        }
+    )
+}