]> gitweb @ CieloNegro.org - Lucu.git/blobdiff - Network/HTTP/Lucu/MIMEType.hs
Make use of mimeType quasi-quoter.
[Lucu.git] / Network / HTTP / Lucu / MIMEType.hs
index 36cdf0f82797b491783e9fd8ad1cac48fc2848ff..d60b70e5d5f21d4704ee6bb269f271692d35e58e 100644 (file)
@@ -1,81 +1,82 @@
 {-# LANGUAGE
-    OverloadedStrings
+    DeriveDataTypeable
+  , OverloadedStrings
+  , RecordWildCards
+  , TemplateHaskell
   , UnicodeSyntax
   #-}
-
--- |Manipulation of MIME Types.
+-- |Parsing and printing MIME Media Types
+-- (<http://tools.ietf.org/html/rfc2046>).
 module Network.HTTP.Lucu.MIMEType
     ( MIMEType(..)
-    , mkMIMEType
 
     , parseMIMEType
     , printMIMEType
 
-    , mimeTypeP
-    , mimeTypeListP
+    , mimeType
+    , mimeTypeList
     )
     where
 import Control.Applicative
 import Data.Ascii (Ascii, AsciiBuilder, CIAscii)
 import qualified Data.Ascii as A
 import Data.Attoparsec.Char8 as P
-import Data.Map (Map)
 import Data.Monoid.Unicode
-import Data.Text (Text)
+import Data.Typeable
+import Language.Haskell.TH.Syntax
+import Network.HTTP.Lucu.MIMEParams
+import Network.HTTP.Lucu.Parser
 import Network.HTTP.Lucu.Parser.Http
-import Network.HTTP.Lucu.RFC2231
-import Prelude hiding (min)
+import Network.HTTP.Lucu.Utils
 import Prelude.Unicode
 
--- |@'MIMEType' \"major\" \"minor\" [(\"name\", \"value\")]@
--- represents \"major\/minor; name=value\".
-data MIMEType = MIMEType {
-      mtMajor  ∷ !CIAscii
-    , mtMinor  ∷ !CIAscii
-    , mtParams ∷ !(Map CIAscii Text)
-    } deriving (Eq)
-
-instance Show MIMEType where
-    show = A.toString ∘ A.fromAsciiBuilder ∘ printMIMEType
+-- |A media type, subtype, and parameters.
+data MIMEType
+    = MIMEType {
+        mtMedia  ∷ !CIAscii
+      , mtSub    ∷ !CIAscii
+      , mtParams ∷ !MIMEParams
+      }
+    deriving (Eq, Show, Read, Typeable)
 
--- |Construct a 'MIMEType' without any parameters.
-mkMIMEType ∷ CIAscii → CIAscii → MIMEType
-{-# INLINE mkMIMEType #-}
-mkMIMEType maj min
-    = MIMEType maj min (∅)
+instance Lift MIMEType where
+    lift (MIMEType {..})
+        = [| MIMEType {
+               mtMedia  = $(liftCIAscii mtMedia)
+             , mtSub    = $(liftCIAscii mtSub)
+             , mtParams = $(lift mtParams)
+             }
+           |]
 
--- |Convert a 'MIMEType' to 'AsciiBuilder'.
+-- |Convert a 'MIMEType' to an 'AsciiBuilder'.
 printMIMEType ∷ MIMEType → AsciiBuilder
 {-# INLINEABLE printMIMEType #-}
-printMIMEType (MIMEType maj min params)
-    = A.toAsciiBuilder (A.fromCIAscii maj) ⊕
+printMIMEType (MIMEType {..})
+    = A.toAsciiBuilder (A.fromCIAscii mtMedia) ⊕
       A.toAsciiBuilder "/" ⊕
-      A.toAsciiBuilder (A.fromCIAscii min) ⊕
-      printParams params
+      A.toAsciiBuilder (A.fromCIAscii mtSub) ⊕
+      printMIMEParams mtParams
 
 -- |Parse 'MIMEType' from an 'Ascii'. This function throws an
--- exception for parse error.
+-- exception for parse error. For literals consider using
+-- 'Network.HTTP.Lucu.MIMEType.TH.mimeType'.
 parseMIMEType ∷ Ascii → MIMEType
 {-# INLINEABLE parseMIMEType #-}
 parseMIMEType str
-    = case parseOnly p $ A.toByteString str of
+    = case parseOnly (finishOff mimeType) $ A.toByteString str of
         Right  t → t
         Left err → error ("unparsable MIME Type: " ⧺ A.toString str ⧺ ": " ⧺ err)
-    where
-      p ∷ Parser MIMEType
-      {-# INLINE p #-}
-      p = do t ← mimeTypeP
-             endOfInput
-             return t
 
-mimeTypeP ∷ Parser MIMEType
-{-# INLINEABLE mimeTypeP #-}
-mimeTypeP = do maj    ← A.toCIAscii <$> token
-               _      ← char '/'
-               min    ← A.toCIAscii <$> token
-               params ← paramsP
-               return $ MIMEType maj min params
+-- |'Parser' for an 'MIMEType'.
+mimeType ∷ Parser MIMEType
+{-# INLINEABLE mimeType #-}
+mimeType = do media  ← A.toCIAscii <$> token
+              _      ← char '/'
+              sub    ← A.toCIAscii <$> token
+              params ← mimeParams
+              return $ MIMEType media sub params
 
-mimeTypeListP ∷ Parser [MIMEType]
-{-# INLINE mimeTypeListP #-}
-mimeTypeListP = listOf mimeTypeP
+-- |'Parser' for a list of 'MIMEType's.
+mimeTypeList ∷ Parser [MIMEType]
+{-# INLINE mimeTypeList #-}
+mimeTypeList = listOf mimeType