From a8a65e953f8be92a3fa022ecc935547e130fd4c8 Mon Sep 17 00:00:00 2001
From: Koitharu <nvasya95@gmail.com>
Date: Thu, 10 Mar 2022 19:52:30 +0200
Subject: [PATCH] Support authorization for MangaLib and HentaiLib #122

---
 app/src/main/AndroidManifest.xml              |  1 +
 .../core/parser/site/MangaLibRepository.kt    | 10 +++++++--
 .../sources/auth/SourceAuthActivity.kt        | 22 ++++++++++++++++++-
 3 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index d42998bc7..cb22f131e 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -68,6 +68,7 @@
 			android:windowSoftInputMode="adjustResize" />
 		<activity
 			android:name="org.koitharu.kotatsu.settings.sources.auth.SourceAuthActivity"
+			android:configChanges="orientation|screenSize|screenLayout|keyboardHidden"
 			android:windowSoftInputMode="adjustResize" />
 		<activity
 			android:name="org.koitharu.kotatsu.core.ui.CrashActivity"
diff --git a/app/src/main/java/org/koitharu/kotatsu/core/parser/site/MangaLibRepository.kt b/app/src/main/java/org/koitharu/kotatsu/core/parser/site/MangaLibRepository.kt
index e279356ce..87868b327 100644
--- a/app/src/main/java/org/koitharu/kotatsu/core/parser/site/MangaLibRepository.kt
+++ b/app/src/main/java/org/koitharu/kotatsu/core/parser/site/MangaLibRepository.kt
@@ -217,11 +217,17 @@ open class MangaLibRepository(loaderContext: MangaLoaderContext) :
 	}
 
 	override fun isAuthorized(): Boolean {
-		return false
+		return loaderContext.cookieJar.getCookies(getDomain()).any {
+			it.name.startsWith("remember_web_")
+		}
 	}
 
 	override suspend fun getUsername(): String {
-		TODO("Not yet implemented")
+		val body = loaderContext.httpGet("https://${getDomain()}/messages").parseHtml().body()
+		if (body.baseUri().endsWith("/login")) {
+			throw AuthRequiredException(source)
+		}
+		return body.selectFirst(".profile-user__username")?.text() ?: parseFailed("Cannot find username")
 	}
 
 	private fun getSortKey(sortOrder: SortOrder?) = when (sortOrder) {
diff --git a/app/src/main/java/org/koitharu/kotatsu/settings/sources/auth/SourceAuthActivity.kt b/app/src/main/java/org/koitharu/kotatsu/settings/sources/auth/SourceAuthActivity.kt
index 5065b7466..e8177e497 100644
--- a/app/src/main/java/org/koitharu/kotatsu/settings/sources/auth/SourceAuthActivity.kt
+++ b/app/src/main/java/org/koitharu/kotatsu/settings/sources/auth/SourceAuthActivity.kt
@@ -16,6 +16,7 @@ import org.koitharu.kotatsu.R
 import org.koitharu.kotatsu.base.ui.BaseActivity
 import org.koitharu.kotatsu.browser.BrowserCallback
 import org.koitharu.kotatsu.browser.BrowserClient
+import org.koitharu.kotatsu.browser.ProgressChromeClient
 import org.koitharu.kotatsu.core.model.MangaSource
 import org.koitharu.kotatsu.core.parser.MangaRepositoryAuthProvider
 import org.koitharu.kotatsu.databinding.ActivityBrowserBinding
@@ -33,7 +34,7 @@ class SourceAuthActivity : BaseActivity<ActivityBrowserBinding>(), BrowserCallba
 		setContentView(ActivityBrowserBinding.inflate(layoutInflater))
 		val source = intent?.getParcelableExtra<MangaSource>(EXTRA_SOURCE)
 		if (source == null) {
-			finish()
+			finishAfterTransition()
 			return
 		}
 		repository = mangaRepositoryOf(source) as? MangaRepositoryAuthProvider ?: run {
@@ -53,6 +54,10 @@ class SourceAuthActivity : BaseActivity<ActivityBrowserBinding>(), BrowserCallba
 			javaScriptEnabled = true
 		}
 		binding.webView.webViewClient = BrowserClient(this)
+		binding.webView.webChromeClient = ProgressChromeClient(binding.progressBar)
+		if (savedInstanceState != null) {
+			return
+		}
 		val url = repository.authUrl
 		onTitleChanged(
 			source.title,
@@ -61,6 +66,21 @@ class SourceAuthActivity : BaseActivity<ActivityBrowserBinding>(), BrowserCallba
 		binding.webView.loadUrl(url)
 	}
 
+	override fun onSaveInstanceState(outState: Bundle) {
+		super.onSaveInstanceState(outState)
+		binding.webView.saveState(outState)
+	}
+
+	override fun onRestoreInstanceState(savedInstanceState: Bundle) {
+		super.onRestoreInstanceState(savedInstanceState)
+		binding.webView.restoreState(savedInstanceState)
+	}
+
+	override fun onDestroy() {
+		super.onDestroy()
+		binding.webView.destroy()
+	}
+
 	override fun onOptionsItemSelected(item: MenuItem): Boolean = when (item.itemId) {
 		android.R.id.home -> {
 			binding.webView.stopLoading()