1 files changed, 24 insertions, 2 deletions

diff --git a/src/Network/KRPC/Method.hs b/src/Network/KRPC/Method.hs
index 916b38a8..2a791924 100644
--- a/src/Network/KRPC/Method.hs
+++ b/src/Network/KRPC/Method.hs
@@ -7,6 +7,7 @@
 --
 --   Normally, you don't need to import this module.
 --
+{-# LANGUAGE CPP                        #-}
 {-# LANGUAGE RankNTypes                 #-}
 {-# LANGUAGE MultiParamTypeClasses      #-}
 {-# LANGUAGE GeneralizedNewtypeDeriving #-}
@@ -17,7 +18,11 @@ module Network.KRPC.Method
        , KRPC (..)
        ) where
 
+#ifdef VERSION_bencoding
 import Data.BEncode (BEncode)
+#else
+import Data.Serialize
+#endif
 import Data.ByteString.Char8 as BC
 import Data.Char
 import Data.Monoid
@@ -38,7 +43,12 @@ import Network.KRPC.Message
 --     * result: Type of return value of the method.
 --
 newtype Method param result = Method { methodName :: MethodName }
-  deriving (Eq, Ord, IsString, BEncode)
+  deriving ( Eq, Ord
+#ifdef VERSION_bencoding
+           , IsString
+           , BEncode
+#endif
+           )
 
 -- | Example:
 --
@@ -49,7 +59,11 @@ instance (Typeable a, Typeable b) => Show (Method a b) where
 
 showsMethod :: forall a b. ( Typeable a , Typeable b ) => Method a b -> ShowS
 showsMethod (Method name) =
+#ifdef VERSION_bencoding
     showString (BC.unpack name) <>
+#else
+    shows (show name) <>
+#endif
     showString " :: " <>
     shows paramsTy <>
     showString " -> " <>
@@ -72,7 +86,13 @@ showsMethod (Method name) =
 --     method = \"ping\"
 --   @
 --
-class (Typeable req, BEncode req, Typeable resp, BEncode resp)
+class ( Typeable req, Typeable resp
+#ifdef VERSION_bencoding
+      , BEncode req, BEncode resp
+#else
+      , Serialize req, Serialize resp
+#endif
+      )
     => KRPC req resp where
 
   -- | Method name. Default implementation uses lowercased @req@
@@ -80,8 +100,10 @@ class (Typeable req, BEncode req, Typeable resp, BEncode resp)
   --
   method :: Method req resp
 
+#ifdef VERSION_bencoding
   -- TODO add underscores
   default method :: Typeable req => Method req resp
   method = Method $ fromString $ L.map toLower $ show $ typeOf hole
     where
       hole = error "krpc.method: impossible" :: req
+#endif