Add spanM

mokus0 · Jul 7, 2016 · 9b9bfa5 · 9b9bfa5
1 parent f2fd736
commit 9b9bfa5
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/Control/Monad/Loops.hs b/src/Control/Monad/Loops.hs
@@ -385,6 +385,17 @@ dropWhileM p (x:xs) = do
                 then dropWhileM p xs
                 else return (x:xs)
 
+-- | Monad 'span'.
+spanM :: (Monad m) => (a -> m Bool) -> [a] -> m ([a], [a])
+spanM _ [] = return []
+spanM p xs@(x:xs') = do
+        q <- p x
+        if q
+                then do
+                  (ys, zs) <- spanM p xs'
+                  return (x:ys, zs)
+                else return ([], xs)
+
 -- |like 'dropWhileM' but trims both ends of the list.
 trimM :: (Monad m) => (a -> m Bool) -> [a] -> m [a]
 trimM p xs = do