X.Prelude: Add takeS

xmonad · Oct 26, 2023 · d668e4c · d668e4c
1 parent f654082
commit d668e4c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/XMonad/Prelude.hs b/XMonad/Prelude.hs
@@ -44,6 +44,7 @@ module XMonad.Prelude (
     Stream(..),
     (+~),
     cycleS,
+    takeS,
     toList,
     fromList,
 ) where
@@ -504,3 +505,8 @@ infixr 5 +~
 -- | Absorb a non-empty list into an infinite stream.
 cycleS :: NonEmpty a -> Stream a
 cycleS (x :| xs) = s where s = x :~ xs +~ s
+
+-- | @takeS n stream@ returns the first @n@ elements of @stream@; if @n < 0@,
+-- this returns the empty list.
+takeS :: Int -> Stream a -> [a]
+takeS n = take n . toList