Local Diffusion models parallel downloads (#231)

* [Configuration] Local Diffusion models parallel downloads

* [Configuration] Unit test coverage

Author: ShiftHackZ

presentation/src/main/java/com/shifthackz/aisdv1/presentation/screen/setup/ServerSetupViewModel.kt

@@ -62,7 +62,7 @@ class ServerSetupViewModel(
             else -> AuthorizationCredentials.None
         }
 
-    private var downloadDisposable: Disposable? = null
+    private val downloadDisposables: MutableList<Pair<String, Disposable>> = mutableListOf()
 
     init {
         !Single.zip(
@@ -94,6 +94,13 @@ class ServerSetupViewModel(
             }
     }
 
+    override fun onCleared() {
+        downloadDisposables.forEach { (_, disposable) ->
+            disposable.dispose()
+        }
+        super.onCleared()
+    }
+
     override fun processIntent(intent: ServerSetupIntent) = when (intent) {
         is ServerSetupIntent.AllowLocalCustomModel -> updateState {
             it.copy(
@@ -383,8 +390,14 @@ class ServerSetupViewModel(
         when {
             // User cancels download
             localModel.downloadState is DownloadState.Downloading -> {
-                downloadDisposable?.dispose()
-                downloadDisposable = null
+                val index = downloadDisposables.indexOfFirst { it.first == localModel.id }
+                if (index != -1) {
+                    downloadDisposables[index].second.dispose()
+                    downloadDisposables.removeAt(index)
+                }
+                !deleteModelUseCase(localModel.id)
+                    .subscribeOnMainThread(schedulersProvider)
+                    .subscribeBy(::errorLog)
                 updateState {
                     it.copy(
                         localModels = currentState.localModels.withNewState(
@@ -408,14 +421,13 @@ class ServerSetupViewModel(
                         ),
                     )
                 }
-                downloadDisposable?.dispose()
-                downloadDisposable = null
-                downloadDisposable = downloadModelUseCase(localModel.id)
+                !downloadModelUseCase(localModel.id)
                     .distinctUntilChanged()
                     .doOnSubscribe { wakeLockInterActor.acquireWakelockUseCase() }
                     .doFinally { wakeLockInterActor.releaseWakeLockUseCase() }
                     .subscribeOnMainThread(schedulersProvider).subscribeBy(
                         onError = { t ->
+                            errorLog(t)
                             val message = t.localizedMessage ?: "Error"
                             updateState {
                                 it.copy(
@@ -448,7 +460,8 @@ class ServerSetupViewModel(
                                 }
                             }
                         },
-                    ).addToDisposable()
+                    )
+                    .also { downloadDisposables.add(localModel.id to it) }
             }
         }
     }

presentation/src/main/res/values-ru/strings.xml

@@ -20,7 +20,7 @@
     <string name="version">Версия %1$s</string>
 
     <string name="splash_status_initializing">Инициализация…</string>
-    <string name="splash_status_fetching">Загрузка занных…</string>
+    <string name="splash_status_fetching">Загрузка данных…</string>
     <string name="splash_status_launching">Запуск приложения!</string>
 
     <string name="local_no_img2img_support_title">Не доступно</string>

presentation/src/test/java/com/shifthackz/aisdv1/presentation/screen/setup/ServerSetupViewModelTest.kt

@@ -182,6 +182,10 @@ class ServerSetupViewModelTest : CoreViewModelTest<ServerSetupViewModel>() {
 
     @Test
     fun `given received LocalModel ClickReduce intent, model is downloading, expected UI state is Unknown`() {
+        every {
+            stubDeleteModelUseCase.invoke(any())
+        } returns Completable.complete()
+
         val localModel = mockServerSetupStateLocalModel.copy(
             downloadState = DownloadState.Downloading(22),
         )