Use Voyager for a few screens

2022-11-26 13:36:06 -05:00
parent bf9b2ca2ff
commit 7df12c68fd
13 changed files with 333 additions and 371 deletions
@@ -0,0 +1,53 @@
+package exh.ui.smartsearch
+
+import cafe.adriel.voyager.core.model.StateScreenModel
+import cafe.adriel.voyager.core.model.coroutineScope
+import eu.kanade.domain.manga.interactor.NetworkToLocalManga
+import eu.kanade.domain.manga.model.Manga
+import eu.kanade.tachiyomi.source.CatalogueSource
+import eu.kanade.tachiyomi.source.SourceManager
+import eu.kanade.tachiyomi.ui.browse.source.SourcesController
+import eu.kanade.tachiyomi.util.lang.launchIO
+import exh.smartsearch.SmartSearchEngine
+import kotlinx.coroutines.CancellationException
+import uy.kohesive.injekt.Injekt
+import uy.kohesive.injekt.api.get
+
+class SmartSearchScreenModel(
+    private val sourceId: Long,
+    private val config: SourcesController.SmartSearchConfig,
+    private val networkToLocalManga: NetworkToLocalManga = Injekt.get(),
+    private val sourceManager: SourceManager = Injekt.get(),
+) : StateScreenModel<SmartSearchScreenModel.SearchResults?>(null) {
+    private val smartSearchEngine = SmartSearchEngine()
+
+    val source = sourceManager.get(sourceId) as CatalogueSource
+
+    init {
+        coroutineScope.launchIO {
+            val result = try {
+                val resultManga = smartSearchEngine.smartSearch(source, config.origTitle)
+                if (resultManga != null) {
+                    val localManga = networkToLocalManga.await(resultManga)
+                    SearchResults.Found(localManga)
+                } else {
+                    SearchResults.NotFound
+                }
+            } catch (e: Exception) {
+                if (e is CancellationException) {
+                    throw e
+                } else {
+                    SearchResults.Error
+                }
+            }
+
+            mutableState.value = result
+        }
+    }
+
+    sealed class SearchResults {
+        data class Found(val manga: Manga) : SearchResults()
+        object NotFound : SearchResults()
+        object Error : SearchResults()
+    }
+}