Advanced global search
parent
4ee52e149e
commit
74900970e1
GPG Key ID:
676DEE768C17A9D7
@ -0,0 +1,13 @@
|
|||||||
|
package org.koitharu.kotatsu.core.nav
|
||||||
|
|
||||||
|
import dagger.hilt.EntryPoint
|
||||||
|
import dagger.hilt.InstallIn
|
||||||
|
import dagger.hilt.components.SingletonComponent
|
||||||
|
import org.koitharu.kotatsu.core.prefs.AppSettings
|
||||||
|
|
||||||
|
@EntryPoint
|
||||||
|
@InstallIn(SingletonComponent::class)
|
||||||
|
interface AppRouterEntryPoint {
|
||||||
|
|
||||||
|
val settings: AppSettings
|
||||||
|
}
|
||||||
@ -1,58 +0,0 @@
|
|||||||
package org.koitharu.kotatsu.core.ui.dialog
|
|
||||||
|
|
||||||
import android.content.Context
|
|
||||||
import android.content.DialogInterface
|
|
||||||
import androidx.annotation.UiContext
|
|
||||||
import androidx.core.net.ConnectivityManagerCompat
|
|
||||||
import dagger.Lazy
|
|
||||||
import org.koitharu.kotatsu.R
|
|
||||||
import org.koitharu.kotatsu.core.prefs.AppSettings
|
|
||||||
import org.koitharu.kotatsu.core.prefs.TriStateOption
|
|
||||||
import org.koitharu.kotatsu.core.util.ext.connectivityManager
|
|
||||||
import javax.inject.Inject
|
|
||||||
|
|
||||||
class CommonAlertDialogs @Inject constructor(
|
|
||||||
private val settings: Lazy<AppSettings>,
|
|
||||||
) {
|
|
||||||
|
|
||||||
fun askForDownloadOverMeteredNetwork(
|
|
||||||
@UiContext context: Context,
|
|
||||||
onConfirmed: (allow: Boolean) -> Unit
|
|
||||||
) {
|
|
||||||
when (settings.get().allowDownloadOnMeteredNetwork) {
|
|
||||||
TriStateOption.ENABLED -> onConfirmed(true)
|
|
||||||
TriStateOption.DISABLED -> onConfirmed(false)
|
|
||||||
TriStateOption.ASK -> {
|
|
||||||
if (!ConnectivityManagerCompat.isActiveNetworkMetered(context.connectivityManager)) {
|
|
||||||
onConfirmed(true)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
val listener = DialogInterface.OnClickListener { _, which ->
|
|
||||||
when (which) {
|
|
||||||
DialogInterface.BUTTON_POSITIVE -> {
|
|
||||||
settings.get().allowDownloadOnMeteredNetwork = TriStateOption.ENABLED
|
|
||||||
onConfirmed(true)
|
|
||||||
}
|
|
||||||
|
|
||||||
DialogInterface.BUTTON_NEUTRAL -> {
|
|
||||||
onConfirmed(true)
|
|
||||||
}
|
|
||||||
|
|
||||||
DialogInterface.BUTTON_NEGATIVE -> {
|
|
||||||
settings.get().allowDownloadOnMeteredNetwork = TriStateOption.DISABLED
|
|
||||||
onConfirmed(false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
BigButtonsAlertDialog.Builder(context)
|
|
||||||
.setIcon(R.drawable.ic_network_cellular)
|
|
||||||
.setTitle(R.string.download_cellular_confirm)
|
|
||||||
.setPositiveButton(R.string.allow_always, listener)
|
|
||||||
.setNeutralButton(R.string.allow_once, listener)
|
|
||||||
.setNegativeButton(R.string.dont_allow, listener)
|
|
||||||
.create()
|
|
||||||
.show()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
package org.koitharu.kotatsu.search.domain
|
||||||
|
|
||||||
|
enum class SearchKind {
|
||||||
|
|
||||||
|
SIMPLE, TITLE, AUTHOR, TAG
|
||||||
|
}
|
||||||
@ -0,0 +1,11 @@
|
|||||||
|
package org.koitharu.kotatsu.search.domain
|
||||||
|
|
||||||
|
import org.koitharu.kotatsu.parsers.model.Manga
|
||||||
|
import org.koitharu.kotatsu.parsers.model.MangaListFilter
|
||||||
|
import org.koitharu.kotatsu.parsers.model.SortOrder
|
||||||
|
|
||||||
|
data class SearchResults(
|
||||||
|
val listFilter: MangaListFilter,
|
||||||
|
val sortOrder: SortOrder,
|
||||||
|
val manga: List<Manga>,
|
||||||
|
)
|
||||||
@ -0,0 +1,124 @@
|
|||||||
|
package org.koitharu.kotatsu.search.domain
|
||||||
|
|
||||||
|
import dagger.assisted.Assisted
|
||||||
|
import dagger.assisted.AssistedFactory
|
||||||
|
import dagger.assisted.AssistedInject
|
||||||
|
import org.koitharu.kotatsu.core.parser.MangaDataRepository
|
||||||
|
import org.koitharu.kotatsu.core.parser.MangaRepository
|
||||||
|
import org.koitharu.kotatsu.core.util.ext.printStackTraceDebug
|
||||||
|
import org.koitharu.kotatsu.parsers.model.Manga
|
||||||
|
import org.koitharu.kotatsu.parsers.model.MangaListFilter
|
||||||
|
import org.koitharu.kotatsu.parsers.model.MangaSource
|
||||||
|
import org.koitharu.kotatsu.parsers.model.SortOrder
|
||||||
|
import org.koitharu.kotatsu.parsers.util.almostEquals
|
||||||
|
import org.koitharu.kotatsu.parsers.util.levenshteinDistance
|
||||||
|
import org.koitharu.kotatsu.parsers.util.runCatchingCancellable
|
||||||
|
|
||||||
|
private const val MATCH_THRESHOLD_DEFAULT = 0.2f
|
||||||
|
|
||||||
|
class SearchV2Helper @AssistedInject constructor(
|
||||||
|
@Assisted private val source: MangaSource,
|
||||||
|
private val mangaRepositoryFactory: MangaRepository.Factory,
|
||||||
|
private val dataRepository: MangaDataRepository,
|
||||||
|
) {
|
||||||
|
|
||||||
|
suspend operator fun invoke(query: String, kind: SearchKind): SearchResults? {
|
||||||
|
val repository = mangaRepositoryFactory.create(source)
|
||||||
|
val listFilter = repository.getFilter(query, kind) ?: return null
|
||||||
|
val sortOrder = repository.getSortOrder(kind)
|
||||||
|
val list = repository.getList(0, sortOrder, listFilter)
|
||||||
|
if (list.isEmpty()) {
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
val result = list.toMutableList()
|
||||||
|
result.postFilter(query, kind)
|
||||||
|
result.sortByRelevance(query, kind)
|
||||||
|
return SearchResults(listFilter = listFilter, sortOrder = sortOrder, manga = result)
|
||||||
|
}
|
||||||
|
|
||||||
|
private suspend fun MangaRepository.getFilter(query: String, kind: SearchKind): MangaListFilter? = when (kind) {
|
||||||
|
SearchKind.SIMPLE,
|
||||||
|
SearchKind.TITLE,
|
||||||
|
SearchKind.AUTHOR -> if (filterCapabilities.isSearchSupported) { // TODO author support
|
||||||
|
MangaListFilter(query = query)
|
||||||
|
} else {
|
||||||
|
null
|
||||||
|
}
|
||||||
|
|
||||||
|
SearchKind.TAG -> {
|
||||||
|
val tags = this@SearchV2Helper.dataRepository.findTags(this.source) + runCatchingCancellable {
|
||||||
|
this@getFilter.getFilterOptions().availableTags
|
||||||
|
}.onFailure { e ->
|
||||||
|
e.printStackTraceDebug()
|
||||||
|
}.getOrDefault(emptySet())
|
||||||
|
val tag = tags.find { x -> x.title.equals(query, ignoreCase = true) }
|
||||||
|
if (tag != null) {
|
||||||
|
MangaListFilter(tags = setOf(tag))
|
||||||
|
} else {
|
||||||
|
null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun MutableList<Manga>.postFilter(query: String, kind: SearchKind) {
|
||||||
|
when (kind) {
|
||||||
|
SearchKind.TITLE -> retainAll { m ->
|
||||||
|
m.matches(query, MATCH_THRESHOLD_DEFAULT)
|
||||||
|
}
|
||||||
|
|
||||||
|
SearchKind.AUTHOR -> retainAll { m ->
|
||||||
|
m.author.isNullOrEmpty() || m.author.equals(query, ignoreCase = true)
|
||||||
|
}
|
||||||
|
|
||||||
|
SearchKind.SIMPLE, // no filtering expected
|
||||||
|
SearchKind.TAG -> Unit
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun MutableList<Manga>.sortByRelevance(query: String, kind: SearchKind) {
|
||||||
|
when (kind) {
|
||||||
|
SearchKind.SIMPLE,
|
||||||
|
SearchKind.TITLE -> sortBy { m ->
|
||||||
|
minOf(m.title.levenshteinDistance(query), m.altTitle?.levenshteinDistance(query) ?: Int.MAX_VALUE)
|
||||||
|
}
|
||||||
|
|
||||||
|
SearchKind.AUTHOR -> sortByDescending { m ->
|
||||||
|
m.author?.equals(query, ignoreCase = true) == true
|
||||||
|
}
|
||||||
|
|
||||||
|
SearchKind.TAG -> sortByDescending { m ->
|
||||||
|
m.tags.any { tag -> tag.title.equals(query, ignoreCase = true) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun MangaRepository.getSortOrder(kind: SearchKind): SortOrder {
|
||||||
|
val preferred: SortOrder = when (kind) {
|
||||||
|
SearchKind.SIMPLE,
|
||||||
|
SearchKind.TITLE,
|
||||||
|
SearchKind.AUTHOR -> SortOrder.RELEVANCE
|
||||||
|
|
||||||
|
SearchKind.TAG -> SortOrder.POPULARITY
|
||||||
|
}
|
||||||
|
return if (preferred in sortOrders) {
|
||||||
|
preferred
|
||||||
|
} else {
|
||||||
|
defaultSortOrder
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
private fun Manga.matches(query: String, threshold: Float): Boolean {
|
||||||
|
return matchesTitles(title, query, threshold) || matchesTitles(altTitle, query, threshold)
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun matchesTitles(a: String?, b: String?, threshold: Float): Boolean {
|
||||||
|
return !a.isNullOrEmpty() && !b.isNullOrEmpty() && a.almostEquals(b, threshold)
|
||||||
|
}
|
||||||
|
|
||||||
|
@AssistedFactory
|
||||||
|
interface Factory {
|
||||||
|
|
||||||
|
fun create(source: MangaSource): SearchV2Helper
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue