From 6a191b063866eef47b13530f1e1e9338f7c7b9db Mon Sep 17 00:00:00 2001
From: Samfun75 <38332931+Samfun75@users.noreply.github.com>
Date: Sat, 25 Feb 2023 13:11:20 +0300
Subject: [PATCH] UHDMovies: Add size sorting pref, fix quality extractor
 (#1330)

---
 src/en/uhdmovies/build.gradle                 |  2 +-
 .../animeextension/en/uhdmovies/UHDMovies.kt  | 54 +++++++++++++------
 2 files changed, 39 insertions(+), 17 deletions(-)

diff --git a/src/en/uhdmovies/build.gradle b/src/en/uhdmovies/build.gradle
index 0babdbf94..ac972314f 100644
--- a/src/en/uhdmovies/build.gradle
+++ b/src/en/uhdmovies/build.gradle
@@ -6,7 +6,7 @@ ext {
     extName = 'UHD Movies'
     pkgNameSuffix = 'en.uhdmovies'
     extClass = '.UHDMovies'
-    extVersionCode = 7
+    extVersionCode = 8
     libVersion = '13'
 }
 
diff --git a/src/en/uhdmovies/src/eu/kanade/tachiyomi/animeextension/en/uhdmovies/UHDMovies.kt b/src/en/uhdmovies/src/eu/kanade/tachiyomi/animeextension/en/uhdmovies/UHDMovies.kt
index c98735207..678410bcc 100644
--- a/src/en/uhdmovies/src/eu/kanade/tachiyomi/animeextension/en/uhdmovies/UHDMovies.kt
+++ b/src/en/uhdmovies/src/eu/kanade/tachiyomi/animeextension/en/uhdmovies/UHDMovies.kt
@@ -134,7 +134,10 @@ class UHDMovies : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
                     ).replaceFirst("^0+(?!$)".toRegex(), "")
 
                 val qualityMatch = qualityRegex.find(prevP.text())
-                val quality = qualityMatch?.value ?: "HD"
+                val quality = qualityMatch?.value ?: let {
+                    val qualityMatchOwn = qualityRegex.find(row.text())
+                    qualityMatchOwn?.value ?: "HD"
+                }
 
                 row.select("a").filter {
                     !it.text().contains("Zip", true) &&
@@ -173,7 +176,10 @@ class UHDMovies : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
             }.map { row ->
                 val prevP = row.previousElementSibling()
                 val qualityMatch = qualityRegex.find(prevP.text())
-                val quality = qualityMatch?.value ?: "HD"
+                val quality = qualityMatch?.value ?: let {
+                    val qualityMatchOwn = qualityRegex.find(row.text())
+                    qualityMatchOwn?.value ?: "HD"
+                }
 
                 val collectionName = row.previousElementSiblings().prev("h1,h2,h3,pre").first().text()
                     .replace("Download", "", true).trim()
@@ -314,25 +320,24 @@ class UHDMovies : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
     }
 
     override fun List<Video>.sort(): List<Video> {
-        val quality = preferences.getString("preferred_quality", null)
+        val quality = preferences.getString("preferred_quality", "1080")!!
+        val ascSort = preferences.getString("preferred_size_sort", "asc")!! == "asc"
 
-        val newList = mutableListOf<Video>()
-        if (quality != null) {
-            var preferred = 0
-            for (video in this) {
-                if (video.quality.contains(quality)) {
-                    newList.add(preferred, video)
-                    preferred++
-                } else {
-                    newList.add(video)
-                }
+        val comparator = compareByDescending<Video> { it.quality.contains(quality) }.let { cmp ->
+            if (ascSort) {
+                cmp.thenBy { it.quality.fixQuality() }
+            } else {
+                cmp.thenByDescending { it.quality.fixQuality() }
             }
-
-            return newList
         }
-        return this
+        return this.sortedWith(comparator)
     }
 
+    private fun String.fixQuality(): Float = this.substringAfterLast("-").trim()
+        .replace("GB", "", true)
+        .replace("MB", "", true)
+        .toFloat()
+
     override fun setupPreferenceScreen(screen: PreferenceScreen) {
         val videoQualityPref = ListPreference(screen.context).apply {
             key = "preferred_quality"
@@ -349,7 +354,24 @@ class UHDMovies : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
                 preferences.edit().putString(key, entry).commit()
             }
         }
+        val sizeSortPref = ListPreference(screen.context).apply {
+            key = "preferred_size_sort"
+            title = "Preferred Size Sort"
+            entries = arrayOf("Ascending", "Descending")
+            entryValues = arrayOf("asc", "dec")
+            setDefaultValue("asc")
+            summary = "%s -  Sort order to be used after the videos are sorted by their quality."
+
+            setOnPreferenceChangeListener { _, newValue ->
+                val selected = newValue as String
+                val index = findIndexOfValue(selected)
+                val entry = entryValues[index] as String
+                preferences.edit().putString(key, entry).commit()
+            }
+        }
+
         screen.addPreference(videoQualityPref)
+        screen.addPreference(sizeSortPref)
     }
 
     @Serializable