From adfab4425273fe376aa049237cd730442b1339d9 Mon Sep 17 00:00:00 2001
From: kunfei <gekunfei@live.com>
Date: Fri, 31 Mar 2023 12:43:12 +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

---
 .../app/ui/document/FileManageActivity.kt     | 22 ++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/io/legado/app/ui/document/FileManageActivity.kt b/app/src/main/java/io/legado/app/ui/document/FileManageActivity.kt
index c166e16b2..0fe82388e 100644
--- a/app/src/main/java/io/legado/app/ui/document/FileManageActivity.kt
+++ b/app/src/main/java/io/legado/app/ui/document/FileManageActivity.kt
@@ -29,6 +29,7 @@ class FileManageActivity : VMBaseActivity<ActivityFileManageBinding, FileManageV
 
     override val binding by viewBinding(ActivityFileManageBinding::inflate)
     override val viewModel by viewModels<FileManageViewModel>()
+    private val dirParent = ".."
     private val searchView: SearchView by lazy {
         binding.titleBar.findViewById(R.id.search_view)
     }
@@ -38,6 +39,7 @@ class FileManageActivity : VMBaseActivity<ActivityFileManageBinding, FileManageV
     private val fileAdapter by lazy {
         FileAdapter()
     }
+    private val currentFiles = arrayListOf<File>()
 
     override fun onActivityCreated(savedInstanceState: Bundle?) {
         initView()
@@ -65,15 +67,30 @@ class FileManageActivity : VMBaseActivity<ActivityFileManageBinding, FileManageV
             }
 
             override fun onQueryTextChange(newText: String?): Boolean {
-
+                updateFiles()
                 return false
             }
         })
     }
 
+    private fun updateFiles() {
+        if (searchView.query.isNotEmpty()) {
+            currentFiles.filter {
+                it.name == dirParent || it.name.contains(searchView.query)
+            }.let {
+                fileAdapter.setItems(it)
+            }
+        } else {
+            fileAdapter.setItems(currentFiles)
+        }
+    }
+
     override fun observeLiveBus() {
         viewModel.filesLiveData.observe(this) {
-            fileAdapter.setItems(it)
+            searchView.setQuery("", false)
+            currentFiles.clear()
+            currentFiles.addAll(it)
+            updateFiles()
         }
     }
 
@@ -126,7 +143,6 @@ class FileManageActivity : VMBaseActivity<ActivityFileManageBinding, FileManageV
         private val upIcon = ConvertUtils.toDrawable(FilePickerIcon.getUpDir())!!
         private val folderIcon = ConvertUtils.toDrawable(FilePickerIcon.getFolder())!!
         private val fileIcon = ConvertUtils.toDrawable(FilePickerIcon.getFile())!!
-        private val dirParent = ".."
 
         override fun getViewBinding(parent: ViewGroup): ItemFileBinding {
             return ItemFileBinding.inflate(inflater, parent, false)