1 files changed, 21 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/Address.hs b/src/Network/BitTorrent/Address.hs
index a8e12b35..2132f8f9 100644
--- a/src/Network/BitTorrent/Address.hs
+++ b/src/Network/BitTorrent/Address.hs
@@ -91,8 +91,10 @@ module Network.BitTorrent.Address
 import Control.Applicative
 import Control.Monad
 import Control.Exception (onException)
+#ifdef VERSION_bencoding
 import Data.BEncode as BE
 import Data.BEncode.BDict (BKey)
+#endif
 import Data.Bits
 import qualified Data.ByteString as BS
 import qualified Data.ByteString.Internal as BS
@@ -204,7 +206,11 @@ instance Address a => Address (PeerAddr a) where
 
 -- | Peer identifier is exactly 20 bytes long bytestring.
 newtype PeerId = PeerId { getPeerId :: ByteString }
-  deriving (Show, Eq, Ord, BEncode, Typeable)
+  deriving ( Show, Eq, Ord, Typeable
+#ifdef VERSION_bencoding
+           , BEncode
+#endif
+           )
 
 peerIdLen :: Int
 peerIdLen = 20
@@ -369,6 +375,7 @@ genPeerId = azureusStyle defaultClientId defaultVersionNumber <$> timestamp
 --  Port number
 -----------------------------------------------------------------------}
 
+#ifdef VERSION_bencoding
 instance BEncode PortNumber where
   toBEncode   = toBEncode    .  fromEnum
   fromBEncode = fromBEncode >=> portNumber
@@ -378,6 +385,7 @@ instance BEncode PortNumber where
         | 0 <= n && n <= fromIntegral (maxBound :: Word16)
         = pure $ fromIntegral n
         | otherwise = decodingError $ "PortNumber: " ++ show n
+#endif
 
 instance Serialize PortNumber where
   get = fromIntegral <$> getWord16be
@@ -417,6 +425,7 @@ deriving instance Typeable IP
 deriving instance Typeable IPv4
 deriving instance Typeable IPv6
 
+#ifdef VERSION_bencoding
 ipToBEncode :: Show i => i -> BValue
 ipToBEncode ip = BString $ BS8.pack $ show ip
 {-# INLINE ipToBEncode #-}
@@ -444,6 +453,7 @@ instance BEncode IPv6 where
   {-# INLINE toBEncode #-}
   fromBEncode = ipFromBEncode
   {-# INLINE fromBEncode #-}
+#endif
 
 -- | When 'get'ing an IP it must be 'isolate'd to the appropriate
 -- number of bytes since we have no other way of telling which
@@ -508,6 +518,7 @@ data PeerAddr a = PeerAddr
   , peerPort :: {-# UNPACK #-} !PortNumber
   } deriving (Show, Eq, Ord, Typeable, Functor)
 
+#ifdef VERSION_bencoding
 peer_ip_key, peer_id_key, peer_port_key :: BKey
 peer_ip_key   = "ip"
 peer_id_key   = "peer id"
@@ -527,6 +538,7 @@ instance (Typeable a, BEncode a) => BEncode (PeerAddr a) where
              <*>! peer_port_key
     where
       peerAddr = flip PeerAddr
+#endif
 
 -- | The tracker's 'compact peer list' compatible encoding. The
 -- 'peerId' is always 'Nothing'.
@@ -642,7 +654,12 @@ peerSocket socketType pa = do
 --   Normally, /this/ node id should be saved between invocations
 --   of the client software.
 newtype NodeId = NodeId ByteString
-  deriving (Show, Eq, Ord, BEncode, Typeable)
+  deriving (Show, Eq, Ord, Typeable
+#ifdef VERSION_bencoding
+           , BEncode
+#endif
+           )
+
 
 nodeIdSize :: Int
 nodeIdSize = 20
@@ -771,12 +788,14 @@ instance Serialize a => Serialize (NodeAddr a) where
   put NodeAddr {..} = put nodeHost >> put nodePort
   {-# INLINE put #-}
 
+#ifdef VERSION_bencoding
 -- | Torrent file compatible encoding.
 instance BEncode a => BEncode (NodeAddr a) where
   toBEncode NodeAddr {..} = toBEncode (nodeHost, nodePort)
   {-# INLINE toBEncode #-}
   fromBEncode b = uncurry NodeAddr <$> fromBEncode b
   {-# INLINE fromBEncode #-}
+#endif
 
 instance Hashable a => Hashable (NodeAddr a) where
   hashWithSalt s NodeAddr {..} = hashWithSalt s (nodeHost, nodePort)