From c15f4c7fd04ff7a3d62f49610689e14185da4aef Mon Sep 17 00:00:00 2001
From: Jobobby04 <jobobby04@users.noreply.github.com>
Date: Sat, 16 May 2020 19:44:12 -0400
Subject: [PATCH] Update debug functions

---
 app/src/main/java/exh/debug/DebugFunctions.kt | 50 ++++++++++++++++++-
 .../main/java/exh/eh/EHentaiUpdateWorker.kt   |  4 +-
 2 files changed, 51 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/exh/debug/DebugFunctions.kt b/app/src/main/java/exh/debug/DebugFunctions.kt
index cc22c0ba3..af84d6577 100644
--- a/app/src/main/java/exh/debug/DebugFunctions.kt
+++ b/app/src/main/java/exh/debug/DebugFunctions.kt
@@ -55,6 +55,52 @@ object DebugFunctions {
         }
     }
 
+    fun getEHMangaListForEHUpdater(): String {
+        val galleries = mutableListOf(String())
+        runBlocking {
+            val metadataManga = db.getFavoriteMangaWithMetadata().await()
+
+            val allManga = metadataManga.asFlow().cancellable().mapNotNull { manga ->
+                if (manga.source != EH_SOURCE_ID && manga.source != EXH_SOURCE_ID) {
+                    return@mapNotNull null
+                }
+                manga
+            }.toList()
+
+            for (manga in allManga) {
+                val meta = db.getFlatMetadataForManga(manga.id!!).await()?.raise<EHentaiSearchMetadata>()
+                if (meta != null) {
+                    // remove age flag
+                    galleries += "Aged: ${meta.aged}\t Title: ${manga.title}"
+                }
+            }
+        }
+        return galleries.joinToString(",\n")
+    }
+
+    fun countAgedFlagInEXHManga(): Int {
+        var agedAmount = 0
+        runBlocking {
+            val metadataManga = db.getFavoriteMangaWithMetadata().await()
+
+            val allManga = metadataManga.asFlow().cancellable().mapNotNull { manga ->
+                if (manga.source != EH_SOURCE_ID && manga.source != EXH_SOURCE_ID) {
+                    return@mapNotNull null
+                }
+                manga
+            }.toList()
+
+            for (manga in allManga) {
+                val meta = db.getFlatMetadataForManga(manga.id!!).await()?.raise<EHentaiSearchMetadata>()
+                if (meta != null && meta.aged) {
+                    // remove age flag
+                    agedAmount++
+                }
+            }
+        }
+        return agedAmount
+    }
+
     fun addAllMangaInDatabaseToLibrary() {
         db.inTransaction {
             db.lowLevel().executeSQL(
@@ -93,8 +139,8 @@ object DebugFunctions {
 
     fun convertAllExhentaiGalleriesToEhentai() = convertSources(EXH_SOURCE_ID, EH_SOURCE_ID)
 
-    fun testLaunchEhentaiBackgroundUpdater() {
-        EHentaiUpdateWorker.launchBackgroundTest(app)
+    fun testLaunchEhentaiBackgroundUpdater(): String {
+        return EHentaiUpdateWorker.launchBackgroundTest(app)
     }
 
     fun rescheduleEhentaiBackgroundUpdater() {
diff --git a/app/src/main/java/exh/eh/EHentaiUpdateWorker.kt b/app/src/main/java/exh/eh/EHentaiUpdateWorker.kt
index fbf12bf79..4588370dc 100644
--- a/app/src/main/java/exh/eh/EHentaiUpdateWorker.kt
+++ b/app/src/main/java/exh/eh/EHentaiUpdateWorker.kt
@@ -337,12 +337,14 @@ class EHentaiUpdateWorker : JobService(), CoroutineScope {
                 .build()
         }
 
-        fun launchBackgroundTest(context: Context) {
+        fun launchBackgroundTest(context: Context): String {
             val jobScheduler = context.jobScheduler
             if (jobScheduler.schedule(context.testBackgroundJobInfo()) == JobScheduler.RESULT_FAILURE) {
                 logger.e("Failed to schedule background test job!")
+                return "Failed"
             } else {
                 logger.d("Successfully scheduled background test job!")
+                return "Success"
             }
         }