优化

app/src/main/java/io/legado/app/ui/book/import/BaseImportBookActivity.kt

@@ -21,8 +21,8 @@ import io.legado.app.utils.applyTint
 import io.legado.app.utils.startActivityForBook
 import io.legado.app.utils.toastOnUi
 import io.legado.app.utils.viewbindingdelegate.viewBinding
+import kotlinx.coroutines.suspendCancellableCoroutine
 import kotlin.coroutines.resume
-import kotlin.coroutines.suspendCoroutine
 
 abstract class BaseImportBookActivity<VM : ViewModel> :
     VMBaseActivity<ActivityImportBookBinding, VM>() {
@@ -49,7 +49,7 @@ abstract class BaseImportBookActivity<VM : ViewModel> :
     /**
      * 设置书籍保存位置
      */
-    protected suspend fun setBookStorage() = suspendCoroutine { block ->
+    protected suspend fun setBookStorage() = suspendCancellableCoroutine sc@{ block ->
         localBookTreeSelectListener = {
             localBookTreeSelectListener = null
             block.resume(it)
@@ -58,7 +58,7 @@ abstract class BaseImportBookActivity<VM : ViewModel> :
         if (!AppConfig.defaultBookTreeUri.isNullOrBlank()) {
             localBookTreeSelectListener = null
             block.resume(true)
-            return@suspendCoroutine
+            return@sc
         }
         //测试读写??
         val storageHelp = String(assets.open("storageHelp.md").readBytes())

app/src/main/java/io/legado/app/ui/book/import/local/ImportBookActivity.kt

@@ -8,6 +8,7 @@ import android.view.MenuItem
 import androidx.activity.addCallback
 import androidx.activity.viewModels
 import androidx.appcompat.widget.PopupMenu
+import androidx.core.net.toUri
 import androidx.documentfile.provider.DocumentFile
 import androidx.lifecycle.lifecycleScope
 import androidx.recyclerview.widget.LinearLayoutManager
@@ -166,12 +167,19 @@ class ImportBookActivity : BaseImportBookActivity<ImportBookViewModel>(),
                 selectFolder.launch()
             } else {
                 val rootUri = if (lastPath.isUri()) {
-                    Uri.parse(lastPath)
+                    lastPath.toUri()
                 } else {
                     Uri.fromFile(File(lastPath))
                 }
                 when {
-                    rootUri.isContentScheme() -> {
+                    rootUri.isContentScheme() -> initRootPath(rootUri)
+                    else -> initRootPath(rootUri.path!!)
+                }
+            }
+        }
+    }
+
+    private fun initRootPath(rootUri: Uri) {
         kotlin.runCatching {
             val doc = DocumentFile.fromTreeUri(this, rootUri)
             if (doc == null || doc.name.isNullOrEmpty()) {
@@ -180,7 +188,7 @@ class ImportBookActivity : BaseImportBookActivity<ImportBookViewModel>(),
             } else {
                 viewModel.subDocs.clear()
                 viewModel.rootDoc = FileDoc.fromDocumentFile(doc)
-                                upDocs(viewModel.rootDoc!!)
+                upPath()
             }
         }.onFailure {
             binding.tvEmptyMsg.visible()
@@ -188,12 +196,6 @@ class ImportBookActivity : BaseImportBookActivity<ImportBookViewModel>(),
         }
     }
 
-                    else -> initRootPath(rootUri.path!!)
-                }
-            }
-        }
-    }
-
     private fun initRootPath(path: String) {
         binding.tvEmptyMsg.visible()
         PermissionsCompat.Builder()
@@ -201,9 +203,15 @@ class ImportBookActivity : BaseImportBookActivity<ImportBookViewModel>(),
             .rationale(R.string.tip_perm_request_storage)
             .onGranted {
                 kotlin.runCatching {
-                    viewModel.rootDoc = FileDoc.fromFile(File(path))
+                    val file = File(path)
+                    if (!file.isDirectory) {
+                        binding.tvEmptyMsg.visible()
+                        selectFolder.launch()
+                    } else {
                         viewModel.subDocs.clear()
+                        viewModel.rootDoc = FileDoc.fromFile(file)
                         upPath()
+                    }
                 }.onFailure {
                     binding.tvEmptyMsg.visible()
                     selectFolder.launch()