Better name-rewriting engine

[Lucu.git] / Network / HTTP / Lucu / HandleLike.hs

diff --git a/Network/HTTP/Lucu/HandleLike.hs b/Network/HTTP/Lucu/HandleLike.hs
index f58264d9c6abd2b3fd5761c7fc0bd6cdfd5782e1..a45ce6cbf066f3d0a6bac21335b4b16dbef6cca1 100644 (file)
--- a/Network/HTTP/Lucu/HandleLike.hs
+++ b/Network/HTTP/Lucu/HandleLike.hs
@@ -1,19 +1,24 @@
 {-# LANGUAGE
-    UnicodeSyntax
+    DoAndIfThenElse
+  , UnicodeSyntax
   #-}
+-- |Type class for things behaves like a 'I.Handle'.
 module Network.HTTP.Lucu.HandleLike
     ( HandleLike(..)
+    , hPutBuilder
     )
     where
+import Blaze.ByteString.Builder (Builder)
+import qualified Blaze.ByteString.Builder as BB
 import qualified Data.ByteString as B
 import qualified Data.ByteString.Lazy.Char8 as L
 import qualified OpenSSL.Session as SSL
 import OpenSSL.X509
+import Prelude.Unicode
 import qualified System.IO as I
 
 class HandleLike h where
     hGetLBS ∷ h → IO L.ByteString
-    hPutLBS ∷ h → L.ByteString → IO ()
 
     hGetBS  ∷ h → Int → IO B.ByteString
     hPutBS  ∷ h → B.ByteString → IO ()
@@ -24,10 +29,8 @@ class HandleLike h where
     hFlush  ∷ h → IO ()
     hClose  ∷ h → IO ()
 
-
 instance HandleLike I.Handle where
     hGetLBS = L.hGetContents
-    hPutLBS = L.hPut
 
     hGetBS  = B.hGet
     hPutBS  = B.hPut
@@ -35,20 +38,22 @@ instance HandleLike I.Handle where
     hFlush  = I.hFlush
     hClose  = I.hClose
 
-
 instance HandleLike SSL.SSL where
-    hGetLBS   = SSL.lazyRead
-    hPutLBS   = SSL.lazyWrite
+    hGetLBS = SSL.lazyRead
 
-    hGetBS    = SSL.read
-    hPutBS    = SSL.write
+    hGetBS  = SSL.read
+    hPutBS  = SSL.write
 
     hGetPeerCert s
         = do isValid <- SSL.getVerifyResult s
              if isValid then
                  SSL.getPeerCertificate s
-               else
+             else
                  return Nothing
 
-    hFlush _  = return () -- unneeded
-    hClose s  = SSL.shutdown s SSL.Bidirectional
+    hFlush _ = return () -- No need to do anything.
+    hClose s = SSL.shutdown s SSL.Bidirectional
+
+hPutBuilder ∷ HandleLike h ⇒ h → Builder → IO ()
+{-# INLINE hPutBuilder #-}
+hPutBuilder = BB.toByteStringIO ∘ hPutBS