|
1 | 1 | package com.shifthackz.aisdv1.presentation.screen.setup |
2 | 2 |
|
3 | | -import com.shifthackz.aisdv1.core.common.log.debugLog |
4 | 3 | import com.shifthackz.aisdv1.core.common.log.errorLog |
5 | 4 | import com.shifthackz.aisdv1.core.common.schedulers.SchedulersProvider |
6 | 5 | import com.shifthackz.aisdv1.core.common.schedulers.subscribeOnMainThread |
@@ -94,7 +93,22 @@ class ServerSetupViewModel( |
94 | 93 |
|
95 | 94 | ServerSetupIntent.DismissDialog -> setScreenModal(Modal.None) |
96 | 95 |
|
97 | | - is ServerSetupIntent.DownloadCardButtonClick -> localModelDownloadClickReducer(intent.model) |
| 96 | + is ServerSetupIntent.LocalModel.ClickReduce -> localModelDownloadClickReducer(intent.model) |
| 97 | + |
| 98 | + is ServerSetupIntent.LocalModel.DeleteConfirm -> updateState { |
| 99 | + !deleteModelUseCase(intent.model.id) |
| 100 | + .subscribeOnMainThread(schedulersProvider) |
| 101 | + .subscribeBy(::errorLog) |
| 102 | + it.copy( |
| 103 | + screenModal = Modal.None, |
| 104 | + localModels = currentState.localModels.withNewState( |
| 105 | + intent.model.copy( |
| 106 | + downloadState = DownloadState.Unknown, |
| 107 | + downloaded = false, |
| 108 | + ), |
| 109 | + ), |
| 110 | + ) |
| 111 | + } |
98 | 112 |
|
99 | 113 | is ServerSetupIntent.SelectLocalModel -> { |
100 | 114 | if (currentState.localModels.any { it.downloadState is DownloadState.Downloading }) { |
@@ -319,19 +333,8 @@ class ServerSetupViewModel( |
319 | 333 | } |
320 | 334 | } |
321 | 335 | // User deletes local model |
322 | | - localModel.downloaded -> { |
323 | | - updateState { |
324 | | - it.copy( |
325 | | - localModels = currentState.localModels.withNewState( |
326 | | - localModel.copy( |
327 | | - downloadState = DownloadState.Unknown, |
328 | | - downloaded = false, |
329 | | - ), |
330 | | - ), |
331 | | - ) |
332 | | - } |
333 | | - !deleteModelUseCase(localModel.id).subscribeOnMainThread(schedulersProvider) |
334 | | - .subscribeBy(::errorLog) |
| 336 | + localModel.downloaded -> updateState { |
| 337 | + it.copy(screenModal = Modal.DeleteLocalModelConfirm(localModel)) |
335 | 338 | } |
336 | 339 | // User requested new download operation |
337 | 340 | else -> { |
@@ -364,7 +367,6 @@ class ServerSetupViewModel( |
364 | 367 | setScreenModal(Modal.Error(message.asUiText())) |
365 | 368 | }, |
366 | 369 | onNext = { downloadState -> |
367 | | - debugLog("DOWNLOAD STATE : $downloadState") |
368 | 370 | updateState { |
369 | 371 | when (downloadState) { |
370 | 372 | is DownloadState.Complete -> it.copy( |
|
0 commit comments