From b8017b6736407bae9e690a909cb558e85eda3dff Mon Sep 17 00:00:00 2001
From: Xwite <1797350009@qq.com>
Date: Fri, 17 Mar 2023 09:48:11 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ui/book/import/BaseImportBookActivity.kt  |  2 +-
 .../main/java/io/legado/app/utils/UrlUtil.kt  | 41 ++++++++++---------
 2 files changed, 23 insertions(+), 20 deletions(-)

diff --git a/app/src/main/java/io/legado/app/ui/book/import/BaseImportBookActivity.kt b/app/src/main/java/io/legado/app/ui/book/import/BaseImportBookActivity.kt
index 6e31646a1..06fddad74 100644
--- a/app/src/main/java/io/legado/app/ui/book/import/BaseImportBookActivity.kt
+++ b/app/src/main/java/io/legado/app/ui/book/import/BaseImportBookActivity.kt
@@ -93,7 +93,7 @@ abstract class BaseImportBookActivity<VM : ViewModel> : VMBaseActivity<ActivityI
         }
     }
 
-    protected fun onArchiveFileClick(fileDoc: FileDoc) {
+    open fun onArchiveFileClick(fileDoc: FileDoc) {
         val fileNames = ArchiveUtils.getArchiveFilesName(fileDoc) {
             it.matches(AppPattern.bookFileRegex)
         }
diff --git a/app/src/main/java/io/legado/app/utils/UrlUtil.kt b/app/src/main/java/io/legado/app/utils/UrlUtil.kt
index a633be2f3..346f669b8 100644
--- a/app/src/main/java/io/legado/app/utils/UrlUtil.kt
+++ b/app/src/main/java/io/legado/app/utils/UrlUtil.kt
@@ -1,5 +1,7 @@
 package io.legado.app.utils
 
+import io.legado.app.BuildConfig
+import io.legado.app.help.config.AppConfig
 import io.legado.app.constant.AppLog
 import io.legado.app.model.analyzeRule.AnalyzeUrl
 import java.net.HttpURLConnection
@@ -75,11 +77,24 @@ object UrlUtil {
         conn.instanceFollowRedirects = false
         conn.connect()
 
+        if (AppConfig.recordLog || BuildConfig.DEBUG) {
+            val headers = conn.headerFields
+            val headersString = buildString {
+                headers.forEach { (key, value) ->
+                    value.forEach {
+                        append(key)
+                        append(": ")
+                        append(it)
+                        append("\n")
+                    }
+                }
+            }
+        }
+
         // val fileSize = conn.getContentLengthLong() / 1024
-        /** Content-Disposition 存在三种情况
+        /** Content-Disposition 存在三种情况 文件名应该用引号 有些用空格
          * filename="filename"
-         * filename=filename
-         * filename*=charset''filename
+         * filename*="charset''filename"
          */
         val raw: String? = conn.getHeaderField("Content-Disposition")
         // Location跳转到实际链接
@@ -90,13 +105,13 @@ object UrlUtil {
             val names = hashSetOf<String>()
             fileNames.forEach {
                 var fileName = it.substringAfter("=")
+                    .trim()
+                    .replace("^\"".toRegex(), "")
+                    .replace("\"$".toRegex(), "")
                 if (it.contains("filename*")) {
                     val data = fileName.split("''")
                     names.add(URLDecoder.decode(data[1], data[0]))
                 } else {
-                    fileName = fileName
-                        .replace("^\"".toRegex(), "")
-                        .replace("\"$".toRegex(), "")
                     names.add(
                             String(
                             fileName.toByteArray(StandardCharsets.ISO_8859_1),
@@ -110,19 +125,7 @@ object UrlUtil {
             val newUrl= URL(URLDecoder.decode(redirectUrl, "UTF-8"))
             getFileNameFromPath(newUrl)
         } else {
-            // 其余情况 返回响应头
-            val headers = conn.headerFields
-            val headersString = buildString {
-                headers.forEach { (key, value) ->
-                    value.forEach {
-                        append(key)
-                        append(": ")
-                        append(it)
-                        append("\n")
-                    }
-                }
-            }
-            AppLog.put("Cannot obtain URL file name:\n$headersString")
+            AppLog.put("Cannot obtain URL file name, enable recordLog fo rdetail")
             null
         }
     }