4 files changed, 25 insertions, 27 deletions

diff --git a/krpc.cabal b/krpc.cabal
index f382244c..0ac9faac 100644
--- a/krpc.cabal
+++ b/krpc.cabal
@@ -49,7 +49,7 @@ library
 
                      , bytestring   >= 0.10
                      , containers   >= 0.4
-                     , bencoding    == 0.2.2.*
+                     , bencoding    == 0.3.*
 
                      , network      >= 2.3
   ghc-options:         -Wall
diff --git a/src/Remote/KRPC.hs b/src/Remote/KRPC.hs
index 3659ec66..5c913daa 100644
--- a/src/Remote/KRPC.hs
+++ b/src/Remote/KRPC.hs
@@ -164,7 +164,7 @@ data Method param result = Method {
   , methodVals   :: [ValName]
   } deriving (Eq, Ord, Generic)
 
-instance BEncodable (Method a b)
+instance BEncode (Method a b)
 
 instance (Typeable a, Typeable b) => Show (Method a b) where
   showsPrec _ = showsMethod
@@ -224,16 +224,16 @@ method :: MethodName -> [ParamName] -> [ValName] -> Method param result
 method = Method
 {-# INLINE method #-}
 
-lookupKey :: ParamName -> Map ByteString BEncode -> Result BEncode
+lookupKey :: ParamName -> BDict -> Result BValue
 lookupKey x = maybe (Left ("not found key " ++ BC.unpack x)) Right . M.lookup x
 
-extractArgs :: [ParamName] -> Map ParamName BEncode -> Result BEncode
+extractArgs :: [ParamName] -> BDict -> Result BValue
 extractArgs []  d = Right $ if M.null d then  BList [] else BDict d
 extractArgs [x] d = lookupKey x d
 extractArgs xs  d = BList <$> mapM (`lookupKey` d) xs
 {-# INLINE extractArgs #-}
 
-injectVals :: [ParamName] -> BEncode -> [(ParamName, BEncode)]
+injectVals :: [ParamName] -> BValue -> [(ParamName, BValue)]
 injectVals []  (BList []) = []
 injectVals []  (BDict d ) = M.toList d
 injectVals []   be        = invalidParamList [] be
@@ -242,7 +242,7 @@ injectVals ps  (BList as) = L.zip ps as
 injectVals ps   be        = invalidParamList ps be
 {-# INLINE injectVals #-}
 
-invalidParamList :: [ParamName] -> BEncode -> a
+invalidParamList :: [ParamName] -> BValue -> a
 invalidParamList pl be
   = error $ "KRPC invalid parameter list: " ++ show pl ++ "\n" ++
             "while procedure args are: "    ++ show be
@@ -262,14 +262,14 @@ instance Exception RPCException
 -- | Address of remote can be called by client.
 type RemoteAddr = KRemoteAddr
 
-queryCall :: BEncodable param
+queryCall :: BEncode param
           => KRemote -> KRemoteAddr
           -> Method param result -> param -> IO ()
 queryCall sock addr m arg = sendMessage q addr sock
   where
     q = kquery (methodName m) (injectVals (methodParams m) (toBEncode arg))
 
-getResult :: BEncodable result
+getResult :: BEncode result
           => KRemote
           -> Method param result -> IO result
 getResult sock m = do
@@ -285,7 +285,7 @@ getResult sock m = do
 -- | Makes remote procedure call. Throws RPCException on any error
 -- occurred.
 call :: (MonadBaseControl IO host, MonadIO host)
-     => (BEncodable param, BEncodable result)
+     => (BEncode param, BEncode result)
      => RemoteAddr          -- ^ Address of callee.
      -> Method param result -- ^ Procedure to call.
      -> param               -- ^ Arguments passed by callee to procedure.
@@ -294,7 +294,7 @@ call addr m arg = liftIO $ withRemote $ \sock -> do call_ sock addr m arg
 
 -- | The same as 'call' but use already opened socket.
 call_ :: (MonadBaseControl IO host, MonadIO host)
-     => (BEncodable param, BEncodable result)
+     => (BEncode param, BEncode result)
      => Remote              -- ^ Socket to use
      -> RemoteAddr          -- ^ Address of callee.
      -> Method param result -- ^ Procedure to call.
@@ -313,7 +313,7 @@ type MethodHandler remote = (MethodName, HandlerBody remote)
 -- we can safely erase types in (==>)
 -- | Assign method implementation to the method signature.
 (==>) :: forall (remote :: * -> *) (param :: *) (result :: *).
-           (BEncodable param,  BEncodable result)
+           (BEncode param,  BEncode result)
         => Monad remote
         => Method param result      -- ^ Signature.
         -> (param -> remote result) -- ^ Implementation.
@@ -324,7 +324,7 @@ infix 1 ==>
 
 -- | Similar to '==>@' but additionally pass caller address.
 (==>@)  :: forall (remote :: * -> *) (param :: *) (result :: *).
-           (BEncodable param,  BEncodable result)
+           (BEncode param,  BEncode result)
         => Monad remote
         => Method param result      -- ^ Signature.
         -> (KRemoteAddr -> param -> remote result) -- ^ Implementation.
diff --git a/src/Remote/KRPC/Protocol.hs b/src/Remote/KRPC/Protocol.hs
index 06e54f78..d28fdbeb 100644
--- a/src/Remote/KRPC/Protocol.hs
+++ b/src/Remote/KRPC/Protocol.hs
@@ -74,8 +74,8 @@ data KError
   | MethodUnknown { errorMessage :: ByteString }
    deriving (Show, Read, Eq, Ord)
 
-instance BEncodable KError where
-  {-# SPECIALIZE instance BEncodable KError #-}
+instance BEncode KError where
+  {-# SPECIALIZE instance BEncode KError #-}
   {-# INLINE toBEncode #-}
   toBEncode e = fromAscAssocs -- WARN: keep keys sorted
     [ "e" --> (errorCode e, errorMessage e)
@@ -125,11 +125,11 @@ type ParamName  = ByteString
 --
 data KQuery = KQuery {
     queryMethod :: MethodName
-  , queryArgs   :: Map ParamName BEncode
+  , queryArgs   :: Map ParamName BValue
   } deriving (Show, Read, Eq, Ord)
 
-instance BEncodable KQuery where
-  {-# SPECIALIZE instance BEncodable KQuery #-}
+instance BEncode KQuery where
+  {-# SPECIALIZE instance BEncode KQuery #-}
   {-# INLINE toBEncode #-}
   toBEncode (KQuery m args) = fromAscAssocs -- WARN: keep keys sorted
     [ "a" --> BDict args
@@ -145,7 +145,7 @@ instance BEncodable KQuery where
 
   fromBEncode _ = decodingError "KQuery"
 
-kquery :: MethodName -> [(ParamName, BEncode)] -> KQuery
+kquery :: MethodName -> [(ParamName, BValue)] -> KQuery
 kquery name args = KQuery name (M.fromList args)
 {-# INLINE kquery #-}
 
@@ -163,12 +163,10 @@ type ValName = ByteString
 --
 --   > { "y" : "r", "r" : [<val1>, <val2>, ...] }
 --
-newtype KResponse = KResponse {
-    respVals :: Map ValName BEncode
-  } deriving (Show, Read, Eq, Ord)
+newtype KResponse = KResponse { respVals :: BDict }
+  deriving (Show, Read, Eq, Ord)
 
-instance BEncodable KResponse where
-  {-# SPECIALIZE instance BEncodable KResponse #-}
+instance BEncode KResponse where
   {-# INLINE toBEncode #-}
   toBEncode (KResponse vals) = fromAscAssocs  -- WARN: keep keys sorted
     [ "r" --> vals
@@ -183,7 +181,7 @@ instance BEncodable KResponse where
   fromBEncode _ = decodingError "KDict"
 
 
-kresponse :: [(ValName, BEncode)] -> KResponse
+kresponse :: [(ValName, BValue)] -> KResponse
 kresponse = KResponse . M.fromList
 {-# INLINE kresponse #-}
 
@@ -208,7 +206,7 @@ maxMsgSize = 64 * 1024 -- max udp size
 
 
 -- TODO eliminate toStrict
-sendMessage :: BEncodable msg => msg -> KRemoteAddr -> KRemote -> IO ()
+sendMessage :: BEncode msg => msg -> KRemoteAddr -> KRemote -> IO ()
 sendMessage msg (host, port) sock =
   sendAllTo sock (LB.toStrict (encoded msg)) (SockAddrInet port host)
 {-# INLINE sendMessage #-}
diff --git a/tests/Shared.hs b/tests/Shared.hs
index f64112da..1060cfc8 100644
--- a/tests/Shared.hs
+++ b/tests/Shared.hs
@@ -32,8 +32,8 @@ swapM = method "swap" ["x", "y"] ["b", "a"]
 shiftR :: Method ((), Int, [Int]) ([Int], (), Int)
 shiftR = method "shiftR" ["x", "y", "z"] ["a", "b", "c"]
 
-rawM :: Method BEncode BEncode
+rawM :: Method BValue BValue
 rawM = method "rawM" [""] [""]
 
-rawDictM :: Method BEncode BEncode
+rawDictM :: Method BValue BValue
 rawDictM = method "m" [] []
\ No newline at end of file