update search_multi

obgnail · Dec 14, 2024 · e67e508 · e67e508
1 parent 6798e56
commit e67e508
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/plugin/search_multi.js b/plugin/search_multi.js
@@ -221,16 +221,21 @@ class searchMultiKeywordPlugin extends BasePlugin {
 
         async function traverse(dir) {
             const files = await readdir(dir)
-            await Promise.all(files.map(async file => {
+            const promises = files.map(async file => {
                 const path = Path.join(dir, file)
                 const stats = await stat(path)
-                if (stats.isFile() && (!fileFilter || fileFilter(path, stats))) {
-                    const buffer = await readFile(path)
-                    callback({ path, file, stats, buffer })
-                } else if (stats.isDirectory() && (!dirFilter || dirFilter(file))) {
-                    await traverse(path)
+                if (stats.isFile()) {
+                    if (fileFilter(path, stats)) {
+                        const buffer = await readFile(path)
+                        callback({ path, file, stats, buffer })
+                    }
+                } else if (stats.isDirectory()) {
+                    if (dirFilter(file)) {
+                        await traverse(path)
+                    }
                 }
-            }))
+            })
+            await Promise.all(promises)
         }
 
         await traverse(dir)
@@ -576,7 +581,7 @@ class SearchHelper {
                 const content = source.buffer.toString()
                 const ast = parser(content)
                 const nodes = preorder(ast, filter)
-                return nodes.map(transformer).filter(Boolean)
+                return nodes.flatMap(transformer).filter(Boolean)
             }
         }
         const buildQualifier = (scope, name, parser, filter, transformer) => {