优化

mgz0227 · Sep 9, 2024 · 71038ab · 71038ab
1 parent 0d7690b
commit 71038ab
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/app/src/main/java/io/legado/app/ui/browser/WebViewModel.kt b/app/src/main/java/io/legado/app/ui/browser/WebViewModel.kt
@@ -11,7 +11,6 @@ import io.legado.app.base.BaseViewModel
 import io.legado.app.constant.AppConst
 import io.legado.app.data.appDb
 import io.legado.app.exception.NoStackTraceException
-import io.legado.app.help.CacheManager
 import io.legado.app.help.IntentData
 import io.legado.app.help.http.newCallResponseBody
 import io.legado.app.help.http.okHttpClient
@@ -97,7 +96,7 @@ class WebViewModel(application: Application) : BaseViewModel(application) {
 
     fun saveVerificationResult(webView: WebView, success: () -> Unit) {
         if (!sourceVerificationEnable) {
-            execute { success.invoke() }
+            return success.invoke()
         }
         if (refetchAfterSuccess) {
             execute {
@@ -109,13 +108,15 @@ class WebViewModel(application: Application) : BaseViewModel(application) {
                     source = source
                 ).getStrResponseAwait(useWebView = false).body
                 SourceVerificationHelp.setResult(sourceOrigin, html ?: "")
+            }.onSuccess {
                 success.invoke()
             }
         } else {
             webView.evaluateJavascript("document.documentElement.outerHTML") {
                 execute {
                     html = StringEscapeUtils.unescapeJson(it).trim('"')
                     SourceVerificationHelp.setResult(sourceOrigin, html ?: "")
+                }.onSuccess {
                     success.invoke()
                 }
             }