implemented toBytes

src/Data/CSV/Slurp.hs

@@ -23,7 +23,7 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 -}
 
-{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE LambdaCase, OverloadedStrings #-}
 
 module Data.CSV.Slurp (
   decodeRows,
@@ -32,7 +32,7 @@ module Data.CSV.Slurp (
   toBytes,
 ) where
 
-import Conduit (ConduitT, mapC, (.|))
+import Conduit (ConduitT, await, mapC, yield, (.|))
 import Control.Monad.Trans.State (StateT, evalStateT)
 import qualified Data.ByteString as BS
 import Data.Maybe (fromMaybe)
@@ -56,7 +56,12 @@ decodeUTF8 bs = case decodeUtf8' bs of
 
 -- | convert a stream to ByteStrings to a string of bytes
 toBytes :: Monad m => ConduitT BS.ByteString Word8 m ()
-toBytes = return ()
+toBytes = await >>= \case
+  Just bs -> do
+    let bytes = BS.unpack bs
+    mapM_ yield bytes
+    toBytes
+  Nothing -> return ()
 
 data DecodeState = DecodeState
   { isQuoted  :: Bool

test/Data/CSV/SlurpSpec.hs

@@ -23,6 +23,7 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 module Data.CSV.SlurpSpec (spec) where
 
 import Conduit (runConduit, (.|))
+import Data.Char (ord)
 import Data.Conduit.List (consume, sourceList)
 import Test.Hspec (Spec, context, describe, it, shouldBe, xit)
 
@@ -33,6 +34,7 @@ spec = describe "Data.CSV.Slurp" $ do
   decodeRowsSpec
   decodeRawRowsSpec
   decodeUTF8Spec
+  toBytesSpec
 
 decodeRowsSpec :: Spec
 decodeRowsSpec = describe "decodeRows" $ mapM_
@@ -199,4 +201,12 @@ decodeUTF8Spec = describe "decodeUTF8" $ mapM_
   , ( "blank",        "",         Just ""      )
   ]
 
+toBytesSpec :: Spec
+toBytesSpec = describe "toBytes" $ let
+  input    = ["ab", "cd"]
+  expected = map (fromIntegral . ord) "abcd"
+  in it ("should be " ++ show expected) $ do
+    result <- runConduit $ sourceList input .| toBytes .| consume
+    result `shouldBe` expected
+
 --jl