Implement DispatchersProvider (#275)

Author: ShiftHackZ

app/src/main/java/com/shifthackz/aisdv1/app/di/ProvidersModule.kt

@@ -8,6 +8,7 @@ import com.shifthackz.aisdv1.core.common.appbuild.BuildType
 import com.shifthackz.aisdv1.core.common.appbuild.BuildVersion
 import com.shifthackz.aisdv1.core.common.file.FileProviderDescriptor
 import com.shifthackz.aisdv1.core.common.links.LinksProvider
+import com.shifthackz.aisdv1.core.common.schedulers.DispatchersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.SchedulersProvider
 import com.shifthackz.aisdv1.core.common.time.TimeProvider
 import com.shifthackz.aisdv1.domain.entity.ServerSource
@@ -26,6 +27,8 @@ import com.shifthackz.aisdv1.presentation.activity.AiStableDiffusionActivity
 import io.reactivex.rxjava3.android.schedulers.AndroidSchedulers
 import io.reactivex.rxjava3.core.Scheduler
 import io.reactivex.rxjava3.schedulers.Schedulers
+import kotlinx.coroutines.CoroutineDispatcher
+import kotlinx.coroutines.Dispatchers
 import org.koin.android.ext.koin.androidApplication
 import org.koin.android.ext.koin.androidContext
 import org.koin.dsl.module
@@ -132,6 +135,14 @@ val providersModule = module {
         }
     }
 
+    single<DispatchersProvider> {
+        object : DispatchersProvider {
+            override val io: CoroutineDispatcher = Dispatchers.IO
+            override val ui: CoroutineDispatcher = Dispatchers.Main
+            override val immediate: CoroutineDispatcher = Dispatchers.Main.immediate
+        }
+    }
+
     single<SchedulersProvider> {
         object : SchedulersProvider {
             override val io: Scheduler = Schedulers.io()

core/common/src/main/java/com/shifthackz/aisdv1/core/common/schedulers/DispatchersProvider.kt

@@ -0,0 +1,9 @@
+package com.shifthackz.aisdv1.core.common.schedulers
+
+import kotlinx.coroutines.CoroutineDispatcher
+
+interface DispatchersProvider {
+    val io: CoroutineDispatcher
+    val ui: CoroutineDispatcher
+    val immediate: CoroutineDispatcher
+}

gradle/libs.versions.toml

@@ -39,7 +39,7 @@ exif = "1.3.7"
 apacheStringUtils = "3.16.0"
 composeGestures = "4.0.0"
 crop = "0.1.1"
-mvi = "1.0.1"
+mvi = "1.0.2"
 dayNightSwitch = "1.0.0"
 imagepicker = "v2.0.3"
 catppuccin = "0.1.2"

presentation/src/main/java/com/shifthackz/aisdv1/presentation/activity/AiStableDiffusionViewModel.kt

@@ -2,6 +2,7 @@ package com.shifthackz.aisdv1.presentation.activity
 
 import com.shifthackz.aisdv1.core.common.extensions.EmptyLambda
 import com.shifthackz.aisdv1.core.common.log.errorLog
+import com.shifthackz.aisdv1.core.common.schedulers.DispatchersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.SchedulersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.subscribeOnMainThread
 import com.shifthackz.aisdv1.core.viewmodel.MviRxViewModel
@@ -14,6 +15,7 @@ import com.shifthackz.aisdv1.presentation.navigation.router.main.MainRouter
 import io.reactivex.rxjava3.kotlin.subscribeBy
 
 class AiStableDiffusionViewModel(
+    dispatchersProvider: DispatchersProvider,
     schedulersProvider: SchedulersProvider,
     mainRouter: MainRouter,
     drawerRouter: DrawerRouter,
@@ -23,6 +25,8 @@ class AiStableDiffusionViewModel(
 
     override val initialState = AppState()
 
+    override val effectDispatcher = dispatchersProvider.immediate
+
     init {
         !mainRouter.observe()
             .subscribeOnMainThread(schedulersProvider)

presentation/src/main/java/com/shifthackz/aisdv1/presentation/di/ViewModelModule.kt

@@ -57,6 +57,7 @@ val viewModelModule = module {
     viewModel { parameters ->
         OnBoardingViewModel(
             launchSource = LaunchSource.fromKey(parameters.get()),
+            dispatchersProvider = get(),
             mainRouter = get(),
             splashNavigationUseCase = get(),
             preferenceManager = get(),
@@ -69,6 +70,7 @@ val viewModelModule = module {
         val launchSource = LaunchSource.fromKey(parameters.get())
         ServerSetupViewModel(
             launchSource = launchSource,
+            dispatchersProvider = get(),
             getConfigurationUseCase = get(),
             getLocalAiModelsUseCase = get(),
             fetchAndGetHuggingFaceModelsUseCase = get(),
@@ -88,6 +90,7 @@ val viewModelModule = module {
     viewModel { parameters ->
         GalleryDetailViewModel(
             itemId = parameters.get(),
+            dispatchersProvider = get(),
             getGenerationResultUseCase = get(),
             getLastResultFromCacheUseCase = get(),
             deleteGalleryItemUseCase = get(),

presentation/src/main/java/com/shifthackz/aisdv1/presentation/modal/embedding/EmbeddingViewModel.kt

@@ -1,6 +1,7 @@
 package com.shifthackz.aisdv1.presentation.modal.embedding
 
 import com.shifthackz.aisdv1.core.common.log.errorLog
+import com.shifthackz.aisdv1.core.common.schedulers.DispatchersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.SchedulersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.subscribeOnMainThread
 import com.shifthackz.aisdv1.core.viewmodel.MviRxViewModel
@@ -12,13 +13,16 @@ import com.shifthackz.aisdv1.presentation.utils.ExtrasFormatter
 import io.reactivex.rxjava3.kotlin.subscribeBy
 
 class EmbeddingViewModel(
+    dispatchersProvider: DispatchersProvider,
     private val fetchAndGetEmbeddingsUseCase: FetchAndGetEmbeddingsUseCase,
     private val preferenceManager: PreferenceManager,
     private val schedulersProvider: SchedulersProvider,
 ) : MviRxViewModel<EmbeddingState, EmbeddingIntent, ExtrasEffect>() {
 
     override val initialState = EmbeddingState()
 
+    override val effectDispatcher = dispatchersProvider.immediate
+
     init {
         updateState {
             it.copy(source = preferenceManager.source)

presentation/src/main/java/com/shifthackz/aisdv1/presentation/modal/extras/ExtrasViewModel.kt

@@ -1,6 +1,7 @@
 package com.shifthackz.aisdv1.presentation.modal.extras
 
 import com.shifthackz.aisdv1.core.common.log.errorLog
+import com.shifthackz.aisdv1.core.common.schedulers.DispatchersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.SchedulersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.subscribeOnMainThread
 import com.shifthackz.aisdv1.core.common.time.TimeProvider
@@ -16,6 +17,7 @@ import com.shifthackz.aisdv1.presentation.utils.ExtrasFormatter
 import io.reactivex.rxjava3.kotlin.subscribeBy
 
 class ExtrasViewModel(
+    dispatchersProvider: DispatchersProvider,
     private val fetchAndGetLorasUseCase: FetchAndGetLorasUseCase,
     private val fetchAndGetHyperNetworksUseCase: FetchAndGetHyperNetworksUseCase,
     private val schedulersProvider: SchedulersProvider,
@@ -25,6 +27,8 @@ class ExtrasViewModel(
 
     override val initialState = ExtrasState()
 
+    override val effectDispatcher = dispatchersProvider.immediate
+
     init {
         updateState {
             it.copy(source = preferenceManager.source)

presentation/src/main/java/com/shifthackz/aisdv1/presentation/modal/history/InputHistoryViewModel.kt

@@ -3,6 +3,7 @@ package com.shifthackz.aisdv1.presentation.modal.history
 import androidx.paging.Pager
 import androidx.paging.PagingConfig
 import androidx.paging.PagingData
+import com.shifthackz.aisdv1.core.common.schedulers.DispatchersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.SchedulersProvider
 import com.shifthackz.aisdv1.core.imageprocessing.Base64ToBitmapConverter
 import com.shifthackz.aisdv1.core.viewmodel.MviRxViewModel
@@ -15,13 +16,16 @@ import com.shifthackz.android.core.mvi.EmptyState
 import kotlinx.coroutines.flow.Flow
 
 class InputHistoryViewModel(
+    dispatchersProvider: DispatchersProvider,
     private val getGenerationResultPagedUseCase: GetGenerationResultPagedUseCase,
     private val base64ToBitmapConverter: Base64ToBitmapConverter,
     private val schedulersProvider: SchedulersProvider,
 ) : MviRxViewModel<EmptyState, EmptyIntent, EmptyEffect>() {
 
     override val initialState = EmptyState
 
+    override val effectDispatcher = dispatchersProvider.immediate
+
     private val config = PagingConfig(
         pageSize = Constants.PAGINATION_PAYLOAD_SIZE,
         initialLoadSize = Constants.PAGINATION_PAYLOAD_SIZE

presentation/src/main/java/com/shifthackz/aisdv1/presentation/modal/tag/EditTagViewModel.kt

@@ -1,13 +1,18 @@
 package com.shifthackz.aisdv1.presentation.modal.tag
 
+import com.shifthackz.aisdv1.core.common.schedulers.DispatchersProvider
 import com.shifthackz.aisdv1.core.viewmodel.MviRxViewModel
 import com.shifthackz.aisdv1.presentation.utils.Constants
 import com.shifthackz.aisdv1.presentation.utils.ExtrasFormatter
 
-class EditTagViewModel : MviRxViewModel<EditTagState, EditTagIntent, EditTagEffect>() {
+class EditTagViewModel(
+    dispatchersProvider: DispatchersProvider,
+) : MviRxViewModel<EditTagState, EditTagIntent, EditTagEffect>() {
 
     override val initialState = EditTagState()
 
+    override val effectDispatcher = dispatchersProvider.immediate
+
     override fun processIntent(intent: EditTagIntent) {
         when (intent) {
             is EditTagIntent.InitialData -> updateState {

presentation/src/main/java/com/shifthackz/aisdv1/presentation/screen/debug/DebugMenuViewModel.kt

@@ -3,6 +3,7 @@ package com.shifthackz.aisdv1.presentation.screen.debug
 import com.shifthackz.aisdv1.core.common.file.FileProviderDescriptor
 import com.shifthackz.aisdv1.core.common.log.FileLoggingTree
 import com.shifthackz.aisdv1.core.common.log.errorLog
+import com.shifthackz.aisdv1.core.common.schedulers.DispatchersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.SchedulersProvider
 import com.shifthackz.aisdv1.core.common.schedulers.subscribeOnMainThread
 import com.shifthackz.aisdv1.core.model.asUiText
@@ -16,6 +17,7 @@ import io.reactivex.rxjava3.kotlin.subscribeBy
 import com.shifthackz.aisdv1.core.localization.R as LocalizationR
 
 class DebugMenuViewModel(
+    dispatchersProvider: DispatchersProvider,
     private val preferenceManager: PreferenceManager,
     private val fileProviderDescriptor: FileProviderDescriptor,
     private val debugInsertBadBase64UseCase: DebugInsertBadBase64UseCase,
@@ -26,6 +28,8 @@ class DebugMenuViewModel(
 
     override val initialState = DebugMenuState()
 
+    override val effectDispatcher = dispatchersProvider.immediate
+
     init {
         !preferenceManager
             .observe()