Extend supported prefersKeyPath functions

Author: nicklockwood

Sources/Examples.swift

@@ -1128,8 +1128,10 @@ private struct Examples {
     let preferKeyPath = """
     ```diff
     - let barArray = fooArray.map { $0.bar }
-
     + let barArray = fooArray.map(\\.bar)
+
+    - let barArray = fooArray.compactMap { $0.optionalBar }
+    + let barArray = fooArray.compactMap(\\.optionalBar)
     ```
     """
 

Sources/Rules.swift

@@ -5127,13 +5127,13 @@ public struct _FormatRules {
             if parenthesized {
                 prevToken = formatter.last(.nonSpaceOrLinebreak, before: prevIndex) ?? prevToken
             }
-            guard [.identifier("map"), .identifier("flatMap"),
-                   .identifier("compactMap")].contains(prevToken),
-                let nextIndex = formatter.index(of: .nonSpaceOrLinebreak, after: i, if: {
-                    $0 == .identifier("$0")
-                }),
-                let endIndex = formatter.endOfScope(at: i),
-                let lastIndex = formatter.index(of: .nonSpaceOrLinebreak, before: endIndex)
+            guard case let .identifier(name) = prevToken,
+                  ["map", "flatMap", "compactMap", "allSatisfy", "filter"].contains(name),
+                  let nextIndex = formatter.index(of: .nonSpaceOrLinebreak, after: i, if: {
+                      $0 == .identifier("$0")
+                  }),
+                  let endIndex = formatter.endOfScope(at: i),
+                  let lastIndex = formatter.index(of: .nonSpaceOrLinebreak, before: endIndex)
             else {
                 return
             }