komikku/app/src/main/java/exh/util/SearchOverride.kt

package exh.util

import android.content.Context
import eu.kanade.tachiyomi.source.model.MangasPage
import eu.kanade.tachiyomi.source.online.UrlImportableSource
import exh.GalleryAddEvent
import exh.GalleryAdder
import kotlinx.coroutines.flow.flow
import rx.Observable

private val galleryAdder by lazy {
    GalleryAdder()
}

/**
 * A version of fetchSearchManga that supports URL importing
 */
fun UrlImportableSource.urlImportFetchSearchManga(context: Context, query: String, fail: () -> Observable<MangasPage>): Observable<MangasPage> =
    when {
        query.startsWith("http://") || query.startsWith("https://") -> {
            flow {
                emit(
                    galleryAdder.addGallery(context, query, false, this@urlImportFetchSearchManga)
                )
            }
                .asObservable()
                .map { res ->
                    MangasPage(
                        (
                            if (res is GalleryAddEvent.Success) {
                                listOf(res.manga)
                            } else {
                                emptyList()
                            }
                            ),
                        false
                    )
                }
        }
        else -> fail()
    }