From 61a2ae943173defa6e44fbea3a7c9fbd07a21bd7 Mon Sep 17 00:00:00 2001
From: tarsin <yuanqingxiang233@163.com>
Date: Tue, 6 Jun 2023 22:23:36 +0800
Subject: [PATCH] Implement UConfigScreen

---
 .../ehviewer/ui/settings/UConfigScreen.kt     | 114 ++++++++++++++++++
 1 file changed, 114 insertions(+)

diff --git a/app/src/main/java/com/hippo/ehviewer/ui/settings/UConfigScreen.kt b/app/src/main/java/com/hippo/ehviewer/ui/settings/UConfigScreen.kt
index b539a8594..77f862230 100644
--- a/app/src/main/java/com/hippo/ehviewer/ui/settings/UConfigScreen.kt
+++ b/app/src/main/java/com/hippo/ehviewer/ui/settings/UConfigScreen.kt
@@ -1,7 +1,121 @@
 package com.hippo.ehviewer.ui.settings
 
+import android.annotation.SuppressLint
+import android.webkit.CookieManager
+import android.webkit.WebResourceRequest
+import android.webkit.WebView
+import androidx.compose.foundation.layout.padding
+import androidx.compose.material.icons.Icons
+import androidx.compose.material.icons.filled.ArrowBack
+import androidx.compose.material.icons.filled.Check
+import androidx.compose.material3.Icon
+import androidx.compose.material3.IconButton
+import androidx.compose.material3.Scaffold
+import androidx.compose.material3.Text
+import androidx.compose.material3.TopAppBar
 import androidx.compose.runtime.Composable
+import androidx.compose.runtime.DisposableEffect
+import androidx.compose.runtime.remember
+import androidx.compose.ui.Modifier
+import androidx.compose.ui.res.stringResource
+import arrow.atomic.Atomic
+import com.google.accompanist.web.AccompanistWebViewClient
+import com.google.accompanist.web.WebView
+import com.google.accompanist.web.rememberWebViewState
+import com.hippo.ehviewer.R
+import com.hippo.ehviewer.client.EhCookieStore
+import com.hippo.ehviewer.client.EhCookieStore.KEY_SETTINGS_PROFILE
+import com.hippo.ehviewer.client.EhUrl
+import com.hippo.ehviewer.ui.LocalNavController
+import eu.kanade.tachiyomi.util.lang.launchIO
+import okhttp3.Cookie
+import okhttp3.HttpUrl.Companion.toHttpUrl
 
+@SuppressLint("SetJavaScriptEnabled")
 @Composable
 fun UConfigScreen() {
+    val navController = LocalNavController.current
+    val url = EhUrl.uConfigUrl
+    val webview = remember { Atomic<WebView?>(null) }
+    Scaffold(
+        topBar = {
+            TopAppBar(
+                title = { Text(text = stringResource(id = R.string.u_config)) },
+                navigationIcon = {
+                    IconButton(onClick = { navController.popBackStack() }) {
+                        Icon(imageVector = Icons.Default.ArrowBack, contentDescription = null)
+                    }
+                },
+                actions = {
+                    IconButton(
+                        onClick = {
+                            webview.get()?.loadUrl(
+                                """
+                                    javascript:(function() {
+                                    var apply = document.getElementById("apply").children[0];
+                                    apply.click();
+                                    })();
+                                    """,
+                            )
+                        },
+                    ) {
+                        Icon(imageVector = Icons.Default.Check, contentDescription = null)
+                    }
+                },
+            )
+        },
+    ) { paddingValues ->
+        val state = rememberWebViewState(url = url)
+        val webViewClient = remember {
+            object : AccompanistWebViewClient() {
+                override fun shouldOverrideUrlLoading(view: WebView, request: WebResourceRequest): Boolean {
+                    // Never load other urls
+                    return true
+                }
+            }
+        }
+        WebView(
+            state = state,
+            modifier = Modifier.padding(paddingValues),
+            onCreated = {
+                it.settings.run {
+                    builtInZoomControls = true
+                    displayZoomControls = false
+                    javaScriptEnabled = true
+                }
+            },
+            client = webViewClient,
+            factory = { WebView(it).apply { webview.set(this) } },
+        )
+        DisposableEffect(Unit) {
+            CookieManager.getInstance().apply {
+                flush()
+                removeAllCookies(null)
+                removeSessionCookies(null)
+                // Copy cookies from okhttp cookie store to CookieManager
+                EhCookieStore.getCookies(url.toHttpUrl()).forEach {
+                    setCookie(url, it.toString())
+                }
+            }
+            onDispose {
+                // Put cookies back to okhttp cookie store
+                val cookieManager = CookieManager.getInstance()
+                val cookiesString = cookieManager.getCookie(url)
+                if (!cookiesString.isNullOrEmpty()) {
+                    val hostUrl = EhUrl.host.toHttpUrl()
+                    launchIO {
+                        EhCookieStore.deleteCookie(hostUrl, KEY_SETTINGS_PROFILE)
+                        // The cookies saved in the uconfig page should not be shared between e and ex
+                        for (header in cookiesString.split(";".toRegex()).dropLastWhile { it.isEmpty() }) {
+                            Cookie.parse(hostUrl, header)?.let {
+                                EhCookieStore.addCookie(longLive(it))
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
 }
+
+private fun longLive(cookie: Cookie) = Cookie.Builder().name(cookie.name).value(cookie.value).domain(cookie.domain).path(cookie.path).expiresAt(Long.MAX_VALUE).build()