+ instance for text

1 files changed, 11 insertions, 0 deletions

diff --git a/src/Data/BEncode.hs b/src/Data/BEncode.hs
index 7539ede..357f63f 100644
--- a/src/Data/BEncode.hs
+++ b/src/Data/BEncode.hs
@@ -42,6 +42,8 @@ import qualified Data.ByteString.Lazy as Lazy
 import           Data.ByteString.Internal as B (c2w, w2c)
 import qualified Data.ByteString.Builder as B
 import qualified Data.ByteString.Builder.Prim as BP ()
+import           Data.Text (Text)
+import qualified Data.Text.Encoding as T
 import           Text.PrettyPrint.ANSI.Leijen (Pretty, Doc, pretty, (<+>), (</>))
 import qualified Text.PrettyPrint.ANSI.Leijen as PP
 
@@ -117,6 +119,15 @@ instance BEncodable ByteString where
   fromBEncode _           = decodingError "string"
   {-# INLINE fromBEncode #-}
 
+
+instance BEncodable Text where
+  toBEncode = toBEncode . T.encodeUtf8
+  {-# INLINE toBEncode #-}
+
+  fromBEncode b = T.decodeUtf8 <$> fromBEncode b
+  {-# INLINE fromBEncode #-}
+
+
 {-
 instance BEncodable String where
   toBEncode = BString . BC.pack