Merge pull request #119 from srghma/master

feat: findMapWithIndex -> implement

Author: hdgarrood

src/Data/FoldableWithIndex.purs

@@ -11,6 +11,7 @@ module Data.FoldableWithIndex
   , allWithIndex
   , anyWithIndex
   , findWithIndex
+  , findMapWithIndex
   , foldrDefault
   , foldlDefault
   , foldMapDefault
@@ -279,6 +280,24 @@ findWithIndex p = foldlWithIndex go Nothing
     go i Nothing x | p i x = Just { index: i, value: x }
     go _ r _ = r
 
+-- | Try to find an element in a data structure which satisfies a predicate mapping
+-- | with access to the index.
+findMapWithIndex
+  :: forall i a b f
+   . FoldableWithIndex i f
+  => (i -> a -> Maybe b)
+  -> f a
+  -> Maybe b
+findMapWithIndex f = foldlWithIndex go Nothing
+  where
+    go
+      :: i
+      -> Maybe b
+      -> a
+      -> Maybe b
+    go i Nothing x = f i x
+    go _ r _ = r
+
 -- | A default implementation of `foldr` using `foldrWithIndex`
 foldrDefault
   :: forall i f a b

test/Main.purs

@@ -6,7 +6,7 @@ import Data.Bifoldable (class Bifoldable, bifoldl, bifoldr, bifoldMap, bifoldrDe
 import Data.Bifunctor (class Bifunctor, bimap)
 import Data.Bitraversable (class Bitraversable, bisequenceDefault, bitraverse, bisequence, bitraverseDefault)
 import Data.Foldable (class Foldable, find, findMap, fold, indexl, indexr, foldMap, foldMapDefaultL, foldMapDefaultR, foldl, foldlDefault, foldr, foldrDefault, length, maximum, maximumBy, minimum, minimumBy, null, surroundMap)
-import Data.FoldableWithIndex (class FoldableWithIndex, findWithIndex, foldMapWithIndex, foldMapWithIndexDefaultL, foldMapWithIndexDefaultR, foldlWithIndex, foldlWithIndexDefault, foldrWithIndex, foldrWithIndexDefault, surroundMapWithIndex)
+import Data.FoldableWithIndex (class FoldableWithIndex, findWithIndex, findMapWithIndex, foldMapWithIndex, foldMapWithIndexDefaultL, foldMapWithIndexDefaultR, foldlWithIndex, foldlWithIndexDefault, foldrWithIndex, foldrWithIndexDefault, surroundMapWithIndex)
 import Data.Function (on)
 import Data.FunctorWithIndex (class FunctorWithIndex, mapWithIndex)
 import Data.Int (toNumber, pow)
@@ -129,6 +129,10 @@ main = do
     let pred x = if x > 5 then Just (x * 100) else Nothing
     assert $ findMap pred [1, 5, 10, 20] == Just 1000
 
+  log "Test findMapWithIndex" *> do
+    let pred i x = if x >= 5 && i >= 3 then Just { i, x } else Nothing
+    assert $ findMapWithIndex pred [1, 5, 10, 20] == Just { i: 3, x: 20 }
+
   log "Test maximum"
   assert $ maximum (arrayFrom1UpTo 10) == Just 10