From 74ef573bd33ba24ec0923d6d30d7570660ad3b46 Mon Sep 17 00:00:00 2001
From: jopejoe1 <34899572+jopejoe1@users.noreply.github.com>
Date: Mon, 20 Dec 2021 01:04:44 +1300
Subject: [PATCH] Added Support for Bato.to Mirrors (#10137)

* Update AndroidManifest.xml

* Update BatoToUrlActivity.kt

* Added Mirror Support

* Update build.gradle

* Added missing url path
---
 src/all/batoto/AndroidManifest.xml            | 17 ++++++-
 src/all/batoto/build.gradle                   |  2 +-
 .../tachiyomi/extension/all/batoto/BatoTo.kt  | 44 ++++++++++++++++++-
 .../extension/all/batoto/BatoToUrlActivity.kt |  4 +-
 4 files changed, 60 insertions(+), 7 deletions(-)

diff --git a/src/all/batoto/AndroidManifest.xml b/src/all/batoto/AndroidManifest.xml
index d9ff10fa5..c6052e048 100644
--- a/src/all/batoto/AndroidManifest.xml
+++ b/src/all/batoto/AndroidManifest.xml
@@ -14,11 +14,24 @@ package="eu.kanade.tachiyomi.extension">
             <category android:name="android.intent.category.DEFAULT" />
             <category android:name="android.intent.category.BROWSABLE" />
 
+            <data android:host="*.bato.to" />
+            <data android:host="bato.to" />
+            <data android:host="*.batotoo.com" />
+            <data android:host="batotoo.com" />
+            <data android:host="*.comiko.net" />
+            <data android:host="comiko.net" />
+            <data android:host="*.battwo.com" />
+            <data android:host="battwo.com" />
+            <data android:host="*.mangatoto.com" />
+            <data android:host="mangatoto.com" />
+
             <data
-                android:host="bato.to"
                 android:pathPattern="/series/..*"
                 android:scheme="https" />
+            <data
+                android:pathPattern="/subject-overview/..*"
+                android:scheme="https" />
         </intent-filter>
     </activity>
 </application>
-</manifest>
\ No newline at end of file
+</manifest>
diff --git a/src/all/batoto/build.gradle b/src/all/batoto/build.gradle
index c85897759..dae102522 100644
--- a/src/all/batoto/build.gradle
+++ b/src/all/batoto/build.gradle
@@ -6,7 +6,7 @@ ext {
     extName = 'Bato.to'
     pkgNameSuffix = 'all.batoto'
     extClass = '.BatoToFactory'
-    extVersionCode = 15
+    extVersionCode = 16
     isNsfw = true
 }
 
diff --git a/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoTo.kt b/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoTo.kt
index c5e5fc262..883978678 100644
--- a/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoTo.kt
+++ b/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoTo.kt
@@ -1,8 +1,13 @@
 package eu.kanade.tachiyomi.extension.all.batoto
 
+import android.app.Application
+import android.content.SharedPreferences
+import androidx.preference.ListPreference
+import androidx.preference.PreferenceScreen
 import com.squareup.duktape.Duktape
 import eu.kanade.tachiyomi.network.GET
 import eu.kanade.tachiyomi.network.asObservableSuccess
+import eu.kanade.tachiyomi.source.ConfigurableSource
 import eu.kanade.tachiyomi.source.model.Filter
 import eu.kanade.tachiyomi.source.model.FilterList
 import eu.kanade.tachiyomi.source.model.MangasPage
@@ -22,6 +27,8 @@ import org.jsoup.nodes.Document
 import org.jsoup.nodes.Element
 import org.jsoup.parser.Parser
 import rx.Observable
+import uy.kohesive.injekt.Injekt
+import uy.kohesive.injekt.api.get
 import uy.kohesive.injekt.injectLazy
 import java.util.Calendar
 import java.util.concurrent.TimeUnit
@@ -29,10 +36,43 @@ import java.util.concurrent.TimeUnit
 open class BatoTo(
     override val lang: String,
     private val siteLang: String
-) : ParsedHttpSource() {
+) : ConfigurableSource, ParsedHttpSource() {
+
+    private val preferences: SharedPreferences by lazy {
+        Injekt.get<Application>().getSharedPreferences("source_$id", 0x0000)
+    }
 
     override val name: String = "Bato.to"
-    override val baseUrl: String = "https://bato.to"
+    override val baseUrl: String = getResolutionPref()!!
+
+    override fun setupPreferenceScreen(screen: PreferenceScreen) {
+        val mirrorPref = ListPreference(screen.context).apply {
+            key = "${MIRROR_PREF_KEY}_$lang"
+            title = MIRROR_PREF_TITLE
+            entries = MIRROR_PREF_ENTRIES
+            entryValues = MIRROR_PREF_ENTRY_VALUES
+            setDefaultValue(MIRROR_PREF_DEFAULT_VALUE)
+            summary = "%s"
+
+            setOnPreferenceChangeListener { _, newValue ->
+                val selected = newValue as String
+                val index = findIndexOfValue(selected)
+                val entry = entryValues[index] as String
+                preferences.edit().putString("${MIRROR_PREF_KEY}_$lang", entry).commit()
+            }
+        }
+        screen.addPreference(mirrorPref)
+    }
+
+    private fun getResolutionPref(): String? = preferences.getString("${MIRROR_PREF_KEY}_$lang", MIRROR_PREF_DEFAULT_VALUE)
+
+    companion object {
+        private const val MIRROR_PREF_KEY = "MIRROR"
+        private const val MIRROR_PREF_TITLE = "Mirror"
+        private val MIRROR_PREF_ENTRIES = arrayOf("Bato.to", "Batotoo.com", "Comiko.net", "Battwo.com", "Mangatoto.com")
+        private val MIRROR_PREF_ENTRY_VALUES = arrayOf("https://bato.to", "https://batotoo.com", "https://comiko.net", "https://battwo.com", "https://mangatoto.com")
+        private val MIRROR_PREF_DEFAULT_VALUE = MIRROR_PREF_ENTRY_VALUES[0]
+    }
 
     override val supportsLatest = true
     private val json: Json by injectLazy()
diff --git a/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoToUrlActivity.kt b/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoToUrlActivity.kt
index e0a44f174..4f1bd140e 100644
--- a/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoToUrlActivity.kt
+++ b/src/all/batoto/src/eu/kanade/tachiyomi/extension/all/batoto/BatoToUrlActivity.kt
@@ -15,7 +15,7 @@ class BatoToUrlActivity : Activity() {
         val pathSegments = intent?.data?.pathSegments
 
         if (host != null && pathSegments != null) {
-            val query = fromGuya(pathSegments)
+            val query = fromBatoTo(pathSegments)
 
             if (query == null) {
                 Log.e("BatoToUrlActivity", "Unable to parse URI from intent $intent")
@@ -40,7 +40,7 @@ class BatoToUrlActivity : Activity() {
         exitProcess(0)
     }
 
-    private fun fromGuya(pathSegments: MutableList<String>): String? {
+    private fun fromBatoTo(pathSegments: MutableList<String>): String? {
         return if (pathSegments.size >= 2) {
             val id = pathSegments[1]
             "ID:$id"