From a40408122733f474a5a6f7b5f3dc55df8a6e6d6f Mon Sep 17 00:00:00 2001
From: "Alkenso (Vladimir Vashurkin)" <alkensox@gmail.com>
Date: Sun, 29 Sep 2024 13:46:08 +0300
Subject: [PATCH] Improve FileEnumerator

---
 .../Filesystem & Bundle/FileEnumerator.swift  | 22 +++++++++++--------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/Sources/SpellbookFoundation/Filesystem & Bundle/FileEnumerator.swift b/Sources/SpellbookFoundation/Filesystem & Bundle/FileEnumerator.swift
index b7ed7d6..fcec637 100644
--- a/Sources/SpellbookFoundation/Filesystem & Bundle/FileEnumerator.swift	
+++ b/Sources/SpellbookFoundation/Filesystem & Bundle/FileEnumerator.swift	
@@ -34,7 +34,7 @@ public final class FileEnumerator {
     public var locationFilter: ((URL) -> FilterVerdict)?
     
     /// Forward flags to underlying `FileManager.DirectoryEnumerator`.
-    public var options: FileManager.DirectoryEnumerationOptions = []
+    public var options: (URL) -> FileManager.DirectoryEnumerationOptions = { _ in [] }
     
     /// Creates `FileEnumerator` that enumerates given locations recursively
     /// - Parameters:
@@ -99,7 +99,7 @@ extension FileEnumerator: Sequence, IteratorProtocol {
             let type = try? next.resourceValues(forKeys: [.fileResourceTypeKey]).fileResourceType
             var filterVerdict: FilterVerdict?
             
-            //  If directory is not interested, skip whole content.
+            // If directory is not interested, skip whole content.
             if type == .directory {
                 filterVerdict = locationFilter?(next)
                 if filterVerdict?.children == false {
@@ -107,13 +107,13 @@ extension FileEnumerator: Sequence, IteratorProtocol {
                 }
             }
             
-            //  Check if `next` is interested according to its type.
+            // Check if `next` is interested according to its type.
             if !types.isEmpty {
                 guard let type else { continue }
                 guard types.contains(type) else { continue }
             }
             
-            //  Check if `next` is interested according to its URL.
+            // Check if `next` is interested according to its URL.
             if let verdict = filterVerdict ?? locationFilter?(next), !verdict.current {
                 continue
             }
@@ -125,25 +125,29 @@ extension FileEnumerator: Sequence, IteratorProtocol {
     }
     
     private func nextUnfiltered() -> URL? {
-        //  If next file exists, just return it.
+        // If next file exists, just return it.
         if let next = enumerator?.nextObject() as? URL {
             return next
         }
         
-        //  All files/locations enumerated. 'nil' means the end of the sequence.
+        // All files/locations enumerated. 'nil' means the end of the sequence.
         guard let nextLocation = locations.popLast() else {
             return nil
         }
         
-        //  If location doesn't exists, just skip it.
+        // If location doesn't exists, just skip it.
         var isDirectory = false
         guard FileManager.default.fileExists(at: nextLocation, isDirectory: &isDirectory) else {
             return nextUnfiltered()
         }
         
-        //  If location is directory, update enumerator.
+        // If location is directory, update enumerator.
         if isDirectory {
-            enumerator = FileManager.default.enumerator(at: nextLocation, includingPropertiesForKeys: [.fileResourceTypeKey], options: options)
+            enumerator = FileManager.default.enumerator(
+                at: nextLocation,
+                includingPropertiesForKeys: [.fileResourceTypeKey],
+                options: options(nextLocation)
+            )
         }
         
         return nextLocation