From 38e3ba3d97c381cf13fd237c2873dec4cb4fc7b6 Mon Sep 17 00:00:00 2001
From: Aria Moradi <aria.moradi007@gmail.com>
Date: Thu, 24 Dec 2020 01:22:01 +0330
Subject: [PATCH] a little refactor

---
 .../main/kotlin/ir/armor/tachidesk/Main.kt    | 109 ++++++++++--------
 1 file changed, 61 insertions(+), 48 deletions(-)

diff --git a/app/src/main/kotlin/ir/armor/tachidesk/Main.kt b/app/src/main/kotlin/ir/armor/tachidesk/Main.kt
index 0450978..6d9e3a1 100644
--- a/app/src/main/kotlin/ir/armor/tachidesk/Main.kt
+++ b/app/src/main/kotlin/ir/armor/tachidesk/Main.kt
@@ -3,63 +3,76 @@ package ir.armor.tachidesk
 import com.googlecode.d2j.dex.Dex2jar
 import com.googlecode.d2j.reader.DexFileReader
 import com.googlecode.dex2jar.tools.Dex2jarCmd
+import eu.kanade.tachiyomi.extension.api.ExtensionGithubApi
+import eu.kanade.tachiyomi.network.NetworkHelper
+import eu.kanade.tachiyomi.source.CatalogueSource
+import eu.kanade.tachiyomi.source.model.MangasPage
+import eu.kanade.tachiyomi.source.online.HttpSource
+import kotlinx.coroutines.runBlocking
+import okhttp3.Request
+import okio.BufferedSink
+import okio.buffer
+import okio.sink
+import java.io.File
+import java.net.URL
+import java.net.URLClassLoader
+import kotlin.system.exitProcess
 
 class Main {
     companion object {
+        const val contentRoot = "/tmp/tachidesk"
+
+        @JvmStatic
+        fun downloadAPK(url: String, apkPath: String){
+            val request = Request.Builder().url(url).build()
+            val response = NetworkHelper().client.newCall(request).execute();
+
+            val downloadedFile = File(apkPath)
+            val sink = downloadedFile.sink().buffer()
+            sink.writeAll(response.body!!.source())
+            sink.close()
+        }
+
+
         @JvmStatic
         fun main(args: Array<String>) {
-//            val contentRoot = "/tmp/tachidesk"
-//            File(contentRoot).mkdirs()
-//            var sourcePkg = ""
-//
-//            // get list of extensions
-//            var apkToDownload: String = ""
-//            runBlocking {
-//                val api = ExtensionGithubApi()
-//                val source = api.findExtensions().first {
-//                    api.getApkUrl(it).endsWith("killsixbilliondemons-v1.2.3.apk")
-//                }
-//                apkToDownload = api.getApkUrl(source)
-//                sourcePkg = source.pkgName
-//            }
-//            apkToDownload = "https://raw.githubusercontent.com/inorichi/tachiyomi-extensions/repo/apk/tachiyomi-en.killsixbilliondemons-v1.2.3.apk"
-//            println(apkToDownload)
-//
-//            val apkFileName = apkToDownload.split("/").last()
-//            val apkFilePath = "$contentRoot/$apkFileName"
-//            val zipDirPath = apkFilePath.substringBefore(".apk")
-//            val jarFilePath = "$zipDirPath.jar"
-//
-//            val request = Request.Builder().url(apkToDownload).build()
-//            val response = NetworkHelper().client.newCall(request).execute();
-//            println(response.code)
-//
-//            val downloadedFile = File(apkFilePath)
-//            val sink: BufferedSink = downloadedFile.sink().buffer()
-//            sink.writeAll(response.body!!.source())
-//            sink.close()
-//
-//            Runtime.getRuntime().exec("unzip ${downloadedFile.absolutePath} -d $zipDirPath").waitFor()
-//            Runtime.getRuntime().exec("dex2jar $zipDirPath/classes.dex -o $jarFilePath").waitFor()
-//
-//            val child = URLClassLoader(arrayOf<URL>(URL("file:$jarFilePath")), this.javaClass.classLoader)
-//            val classToLoad = Class.forName("eu.kanade.tachiyomi.extension.en.killsixbilliondemons.KillSixBillionDemons", true, child)
-//            val instance = classToLoad.newInstance() as CatalogueSource
-//            val result = instance.fetchPopularManga(1)
-//            val mangasPage = result.toBlocking().first() as MangasPage
-//            mangasPage.mangas.forEach{
-//                println(it.title)
-//            }
-//            exitProcess(0)
+            File(contentRoot).mkdirs()
+            var sourcePkg = ""
 
-            val apk = "/tmp/tachidesk/tachiyomi-en.killsixbilliondemons-v1.2.3.apk"
-            val dex = "/tmp/tachidesk/tachiyomi-en.killsixbilliondemons-v1.2.3.dex"
-            val jar = "/tmp/tachidesk/tachiyomi-en.killsixbilliondemons-v1.2.3.jar"
-            val pkg = APKExtractor.extract_dex_and_read_className(apk, dex)
+            // get list of extensions
+            var apkToDownload: String = ""
+            runBlocking {
+                val api = ExtensionGithubApi()
+                val source = api.findExtensions().first {
+                    api.getApkUrl(it).endsWith("killsixbilliondemons-v1.2.3.apk")
+                }
+                apkToDownload = api.getApkUrl(source)
+                sourcePkg = source.pkgName
+            }
 
-            Dex2jarCmd.main(dex, "-o", jar, "--force")
+            val apkFileName = apkToDownload.split("/").last()
+            val apkFilePath = "$contentRoot/$apkFileName"
+            val zipDirPath = apkFilePath.substringBefore(".apk")
+            val jarFilePath = "$zipDirPath.jar"
+            val dexFilePath = "$zipDirPath.dex"
+
+            // download apk file
+            downloadAPK(apkToDownload, apkFilePath)
 
 
+            val className = APKExtractor.extract_dex_and_read_className(apkFilePath, dexFilePath)
+            // dex -> jar
+            Dex2jarCmd.main(dexFilePath, "-o", jarFilePath, "--force")
+
+            val child = URLClassLoader(arrayOf<URL>(URL("file:$jarFilePath")), this.javaClass.classLoader)
+            val classToLoad = Class.forName(className, true, child)
+            val instance = classToLoad.newInstance() as HttpSource
+            val result = instance.fetchPopularManga(1)
+            val mangasPage = result.toBlocking().first() as MangasPage
+            mangasPage.mangas.forEach {
+                println(it.title)
+            }
+            exitProcess(0)
         }
     }
 }