Updated code for zlib-bindings >= 0.1

maltem · Mar 24, 2012 · 82e68b0 · 82e68b0
1 parent aaa1eb1
commit 82e68b0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Codec/Zlib/Enum.hs b/Codec/Zlib/Enum.hs
@@ -41,7 +41,7 @@ decompress' inf (Continue k) = do
             chunk <- liftIO $ finishInflate inf
             lift $ runIteratee $ k $ Chunks [chunk]
         Just bs -> do
-            chunks <- liftIO $ withInflateInput inf bs $ callback
+            chunks <- liftIO $ (feedInflate inf bs >>= callback)
             step <- lift $ runIteratee $ k $ Chunks chunks
             decompress' inf step
 decompress' _ step = return step
@@ -64,10 +64,10 @@ compress' def (Continue k) = do
     x <- EL.head
     case x of
         Nothing -> do
-            chunks <- liftIO $ finishDeflate def $ callback
+            chunks <- liftIO $ callback (finishDeflate def)
             lift $ runIteratee $ k $ Chunks chunks
         Just bs -> do
-            chunks <- liftIO $ withDeflateInput def bs $ callback
+            chunks <- liftIO $ (feedDeflate def bs >>= callback)
             step <- lift $ runIteratee $ k $ Chunks chunks
             compress' def step
 compress' _ step = return step