2 files changed, 15 insertions, 0 deletions

diff --git a/bencoding.cabal b/bencoding.cabal
index cfad921..b01d188 100644
--- a/bencoding.cabal
+++ b/bencoding.cabal
@@ -21,14 +21,18 @@ source-repository head
 library
   exposed-modules:     Data.BEncode
   other-modules:
+
+-- WARN package text is used only for instance declaration
   build-depends:       base       == 4.5.*
                      , containers >= 0.4.0
                      , bytestring >= 0.10.2.0
                      , attoparsec >= 0.10.4.0
+                     , text       >= 0.11.2.3
                      , ansi-wl-pprint
 
   hs-source-dirs:      src
 
+  extensions:          PatternGuards
   ghc-options:         -Wall -fno-warn-unused-do-bind
 -- -Werror
 
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