refactored, more layering

1 files changed, 12 insertions, 4 deletions

diff --git a/Presence/XMLToByteStrings.hs b/Presence/XMLToByteStrings.hs
index d8d2e965..92ac05cd 100644
--- a/Presence/XMLToByteStrings.hs
+++ b/Presence/XMLToByteStrings.hs
@@ -1,5 +1,7 @@
 {-# LANGUAGE CPP #-}
-module XMLToByteStrings (xmlToByteStrings) where
+module XMLToByteStrings 
+    ( xmlToByteStrings
+    , prettyPrint ) where
 
 import Control.Monad.IO.Class
 import Control.Monad.Trans.Class
@@ -8,13 +10,15 @@ import Control.Monad.Fix
 import Control.Monad (when)
 import           Data.Conduit
 import qualified Data.Conduit.List as CL
+import qualified Data.Conduit.Binary as CB
 import Data.XML.Types as XML (Event)
-import qualified Data.ByteString as S (ByteString)
+import qualified Data.ByteString as S (ByteString,append)
+import qualified Data.ByteString.Char8 as S (putStrLn)
 #ifdef RENDERFLUSH
 import Data.Conduit.Blaze (builderToByteStringFlush)
-import Text.XML.Stream.Render (def,renderBuilderFlush)
+import Text.XML.Stream.Render (def,renderBuilderFlush,renderBytes,rsPretty)
 #else
-import Text.XML.Stream.Render (def,renderBytes)
+import Text.XML.Stream.Render (def,renderBytes,rsPretty)
 #endif
 
 fixMaybeT f = (>> return ()) . runMaybeT . fix $ f
@@ -62,3 +66,7 @@ renderChunks = fixMaybeT $  \loop -> do
     loop
 #endif
 
+
+prettyPrint prefix xs = 
+    liftIO $ do
+        CL.sourceList xs $= renderBytes (def { rsPretty=True }) =$= CB.lines $$ CL.mapM_ (S.putStrLn . (prefix `S.append`))