More precise types for tNodes torrent file field

1 files changed, 8 insertions, 1 deletions

diff --git a/src/Data/Torrent.hs b/src/Data/Torrent.hs
index 7833c261..e76ea57e 100644
--- a/src/Data/Torrent.hs
+++ b/src/Data/Torrent.hs
@@ -18,6 +18,7 @@
 --
 {-# LANGUAGE CPP                        #-}
 {-# LANGUAGE FlexibleInstances          #-}
+{-# LANGUAGE OverlappingInstances       #-}
 {-# LANGUAGE BangPatterns               #-}
 {-# LANGUAGE GeneralizedNewtypeDeriving #-}
 {-# LANGUAGE DeriveDataTypeable         #-}
@@ -85,6 +86,7 @@ import           Data.Text as T
 import Data.Time
 import Data.Time.Clock.POSIX
 import Data.Typeable
+import Network (HostName)
 import Network.URI
 import Text.PrettyPrint as PP
 import Text.PrettyPrint.Class
@@ -219,7 +221,7 @@ data Torrent = Torrent
   , tInfoDict     :: !InfoDict
     -- ^ Info about each content file.
 
-  , tNodes        :: !(Maybe [NodeAddr ByteString])
+  , tNodes        :: !(Maybe [NodeAddr HostName])
     -- ^ This key should be set to the /K closest/ nodes in the
     -- torrent generating client's routing table. Alternatively, the
     -- key could be set to a known good 'Network.BitTorrent.Core.Node'
@@ -286,6 +288,11 @@ instance BEncode POSIXTime where
   fromBEncode (BInteger i) = return $ fromIntegral i
   fromBEncode _            = decodingError $ "POSIXTime"
 
+-- TODO to bencoding package
+instance BEncode String where
+  toBEncode = toBEncode . T.pack
+  fromBEncode v = T.unpack <$> fromBEncode v
+
 instance BEncode Torrent where
   toBEncode Torrent {..} = toDict $
        "announce"      .=? tAnnounce