From d126d8fe4159b68feadd4145f0e2a1f90d7ca502 Mon Sep 17 00:00:00 2001
From: huangyi <yi.codeplayer@gmail.com>
Date: Sun, 23 Oct 2011 13:22:18 +0800
Subject: make encoder and decoder invertable, and tests to ensure that.

---
 Network/DNS/Response.hs | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

(limited to 'Network/DNS/Response.hs')

diff --git a/Network/DNS/Response.hs b/Network/DNS/Response.hs
index 36f6203..abcb8ba 100644
--- a/Network/DNS/Response.hs
+++ b/Network/DNS/Response.hs
@@ -1,6 +1,6 @@
 {-# LANGUAGE OverloadedStrings #-}
 
-module Network.DNS.Response (responseIter, parseResponse) where
+module Network.DNS.Response (responseIter, parseResponse, runDNSFormat, runDNSFormat_) where
 
 import Control.Applicative
 import Control.Monad
@@ -12,13 +12,21 @@ import Network.DNS.Internal
 import Network.DNS.StateBinary
 import Data.Enumerator (Enumerator, Iteratee, run_, ($$))
 import Data.ByteString (ByteString)
+import qualified Data.ByteString.Lazy as BL
 
-responseIter :: Iteratee ByteString IO (DNSFormat, PState)
-responseIter = runSGet decodeResponse
+runDNSFormat :: BL.ByteString -> Either String (DNSFormat, PState)
+runDNSFormat bs = runSGet decodeResponse bs
 
-parseResponse :: Enumerator ByteString IO (a,b)
-              -> Iteratee ByteString IO (a,b)
-              -> IO a
+runDNSFormat_ :: BL.ByteString -> Either String DNSFormat
+runDNSFormat_ bs = fst <$> runDNSFormat bs
+
+responseIter :: Monad m => Iteratee ByteString m (DNSFormat, PState)
+responseIter = iterSGet decodeResponse
+
+parseResponse :: (Functor m, Monad m)
+              => Enumerator ByteString m (a,b)
+              -> Iteratee ByteString m (a,b)
+              -> m a
 parseResponse enum iter = fst <$> run_ (enum $$ iter)
 
 ----------------------------------------------------------------
-- 
cgit v1.2.3