优化

app/src/main/java/io/legado/app/help/source/SourceVerificationHelp.kt

@@ -8,55 +8,73 @@ import io.legado.app.help.IntentData
 import io.legado.app.ui.association.VerificationCodeActivity
 import io.legado.app.ui.browser.WebViewActivity
 import io.legado.app.utils.startActivity
-import kotlinx.coroutines.runBlocking
 import splitties.init.appCtx
+import java.util.concurrent.locks.LockSupport
+import kotlin.time.Duration.Companion.minutes
 
 
 object SourceVerificationHelp {
 
-    private var key: String = ""
+    private val waitTime = 1.minutes.inWholeNanoseconds
-    /** 
+
+    private fun getKey(source: BaseSource) = getKey(source.getKey())
+    fun getKey(sourceKey: String) = "${sourceKey}_verificationResult"
+
+    /**
      * 获取书源验证结果
      * 图片验证码 防爬 滑动验证码 点击字符 等等
      */
-    fun getVerificationResult(source: BaseSource?, url: String, title: String, useBrowser: Boolean): String {
+    fun getVerificationResult(
-        source ?: throw NoStackTraceException("getVerificationResult parameter source cannot be null")
+        source: BaseSource?,
-        return runBlocking {
+        url: String,
-            key = "${source.getKey()}_verificationResult"
+        title: String,
-            CacheManager.delete(key)
+        useBrowser: Boolean
+    ): String {
+        source
+            ?: throw NoStackTraceException("getVerificationResult parameter source cannot be null")
 
-            if (!useBrowser) {
+        val key = getKey(source)
-                appCtx.startActivity<VerificationCodeActivity> {
+        CacheManager.delete(key)
-                    putExtra("imageUrl", url)
-                    putExtra("sourceOrigin", source.getKey())
-                    putExtra("sourceName", source.getTag())
-                }
-            } else {
-                startBrowser(source, url, title, true)
-            }
 
-            var waitUserInput = false
+        if (!useBrowser) {
-            while(CacheManager.get(key) == null) {
+            appCtx.startActivity<VerificationCodeActivity> {
-                if (!waitUserInput) {
+                putExtra("imageUrl", url)
-                    AppLog.putDebug("等待返回验证结果...")
+                putExtra("sourceOrigin", source.getKey())
-                    waitUserInput = true
+                putExtra("sourceName", source.getTag())
-                }
+                IntentData.put(key, Thread.currentThread())
             }
-            CacheManager.get(key)!!.let {
+        } else {
-                it.ifBlank {
+            startBrowser(source, url, title, true)
-                    throw NoStackTraceException("验证结果为空")
+        }
-                }
+
+        var waitUserInput = false
+        while (CacheManager.get(key) == null) {
+            if (!waitUserInput) {
+                AppLog.putDebug("等待返回验证结果...")
+                waitUserInput = true
             }
-       }
+            LockSupport.parkNanos(this, waitTime)
+        }
+
+        return CacheManager.get(key)!!.let {
+            it.ifBlank {
+                throw NoStackTraceException("验证结果为空")
+            }
+        }
     }
 
     /**
      * 启动内置浏览器
-      @param saveResult 保存网页源代码到数据库
+     * @param saveResult 保存网页源代码到数据库
      */
-    fun startBrowser(source: BaseSource?, url: String, title: String, saveResult: Boolean? = false) {
+    fun startBrowser(
+        source: BaseSource?,
+        url: String,
+        title: String,
+        saveResult: Boolean? = false
+    ) {
         source ?: throw NoStackTraceException("startBrowser parameter source cannot be null")
-        key = "${source.getKey()}_verificationResult"
+        val key = getKey(source)
         appCtx.startActivity<WebViewActivity> {
             putExtra("title", title)
             putExtra("url", url)
@@ -64,10 +82,14 @@ object SourceVerificationHelp {
             putExtra("sourceName", source.getTag())
             putExtra("sourceVerificationEnable", saveResult)
             IntentData.put(url, source.getHeaderMap(true))
+            IntentData.put(key, Thread.currentThread())
         }
     }
 
-    fun checkResult() {
+
+    fun checkResult(key: String) {
         CacheManager.get(key) ?: CacheManager.putMemory(key, "")
+        val thread = IntentData.get<Thread>(key)
+        LockSupport.unpark(thread)
     }
-}
+}

app/src/main/java/io/legado/app/ui/association/VerificationCodeDialog.kt

@@ -56,20 +56,20 @@ class VerificationCodeDialog() : BaseDialogFragment(R.layout.dialog_verification
         setLayout(1f, ViewGroup.LayoutParams.WRAP_CONTENT)
     }
 
-    override fun onFragmentCreated(view: View, savedInstanceState: Bundle?) {
+    private var sourceOrigin: String? = null
+    private var key = ""
+
+    override fun onFragmentCreated(view: View, savedInstanceState: Bundle?): Unit = binding.run {
         initMenu()
-        binding.run {
+        val arguments = arguments ?: return@run
-            toolBar.setBackgroundColor(primaryColor)
+        toolBar.setBackgroundColor(primaryColor)
-            arguments?.let { arguments ->
+        toolBar.subtitle = arguments.getString("sourceName")
-                toolBar.subtitle = arguments.getString("sourceName")
+        sourceOrigin = arguments.getString("sourceOrigin")
-                val sourceOrigin = arguments.getString("sourceOrigin")
+        key = SourceVerificationHelp.getKey(sourceOrigin!!)
-                arguments.getString("imageUrl")?.let { imageUrl ->
+        val imageUrl = arguments.getString("imageUrl") ?: return@run
-                    loadImage(imageUrl, sourceOrigin)
+        loadImage(imageUrl, sourceOrigin)
-                    verificationCodeImageView.setOnClickListener {
+        verificationCodeImageView.setOnClickListener {
-                        showDialogFragment(PhotoDialog(imageUrl, sourceOrigin))
+            showDialogFragment(PhotoDialog(imageUrl, sourceOrigin))
-                    }
-                }
-            }
         }
     }
 
@@ -118,20 +118,16 @@ class VerificationCodeDialog() : BaseDialogFragment(R.layout.dialog_verification
     override fun onMenuItemClick(item: MenuItem): Boolean {
         when (item.itemId) {
             R.id.menu_ok -> {
-                val sourceOrigin = arguments?.getString("sourceOrigin")
-                val key = "${sourceOrigin}_verificationResult"
                 val verificationCode = binding.verificationCode.text.toString()
-                verificationCode.let {
+                CacheManager.putMemory(key, verificationCode)
-                    CacheManager.putMemory(key, it)
+                dismiss()
-                    dismiss()
-                }
             }
         }
         return false
     }
 
     override fun onDestroy() {
-        SourceVerificationHelp.checkResult()
+        SourceVerificationHelp.checkResult(key)
         super.onDestroy()
         activity?.finish()
     }

app/src/main/java/io/legado/app/ui/browser/WebViewActivity.kt

@@ -172,8 +172,12 @@ class WebViewActivity : VMBaseActivity<ActivityWebViewBinding, WebViewModel>() {
         return super.onKeyUp(keyCode, event)
     }
 
+    override fun finish() {
+        SourceVerificationHelp.checkResult(viewModel.key)
+        super.finish()
+    }
+
     override fun onDestroy() {
-        SourceVerificationHelp.checkResult()
         super.onDestroy()
         binding.webView.destroy()
     }

app/src/main/java/io/legado/app/ui/browser/WebViewModel.kt

@@ -14,6 +14,7 @@ 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
+import io.legado.app.help.source.SourceVerificationHelp
 import io.legado.app.model.analyzeRule.AnalyzeUrl
 import io.legado.app.utils.DocumentUtils
 import io.legado.app.utils.FileUtils
@@ -30,6 +31,7 @@ class WebViewModel(application: Application) : BaseViewModel(application) {
     val headerMap: HashMap<String, String> = hashMapOf()
     var sourceVerificationEnable: Boolean = false
     var sourceOrigin: String = ""
+    var key = ""
 
     fun initData(
         intent: Intent,
@@ -39,6 +41,7 @@ class WebViewModel(application: Application) : BaseViewModel(application) {
             val url = intent.getStringExtra("url")
                 ?: throw NoStackTraceException("url不能为空")
             sourceOrigin = intent.getStringExtra("sourceOrigin") ?: ""
+            key = SourceVerificationHelp.getKey(sourceOrigin)
             sourceVerificationEnable = intent.getBooleanExtra("sourceVerificationEnable", false)
             val headerMapF = IntentData.get<Map<String, String>>(url)
             val analyzeUrl = AnalyzeUrl(url, headerMapF = headerMapF)