{-# LANGUAGE
    OverloadedStrings
  , UnicodeSyntax
  #-}
{-# OPTIONS_HADDOCK prune #-}

-- |Manipulation of MIME Types.
module Network.HTTP.Lucu.MIMEType
    ( MIMEType(..)
    , parseMIMEType
    , printMIMEType

    , mimeTypeP
    , mimeTypeListP
    )
    where
import Control.Applicative
import Data.Ascii (Ascii, AsciiBuilder, CIAscii)
import qualified Data.Ascii as A
import Data.Attoparsec.Char8 as P
import qualified Data.ByteString.Char8 as C8
import Data.Map (Map)
import Data.Monoid.Unicode
import Data.Text (Text)
import Network.HTTP.Lucu.Parser.Http
import Network.HTTP.Lucu.RFC2231
import Prelude hiding (min)
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, Show)

-- |Convert a 'MIMEType' to 'AsciiBuilder'.
printMIMEType ∷ MIMEType → AsciiBuilder
printMIMEType (MIMEType maj min params)
    = A.toAsciiBuilder (A.fromCIAscii maj) ⊕
      A.toAsciiBuilder "/" ⊕
      A.toAsciiBuilder (A.fromCIAscii min) ⊕
      printParams params

-- |Parse 'MIMEType' from an 'Ascii'. This function throws an
-- exception for parse error.
parseMIMEType ∷ Ascii → MIMEType
parseMIMEType str
    = let p  = do t ← mimeTypeP
                  endOfInput
                  return t
          bs = A.toByteString str
      in
        case parseOnly p bs of
          Right  t → t
          Left err → error ("unparsable MIME Type: " ⧺ C8.unpack bs ⧺ ": " ⧺ err)

mimeTypeP ∷ Parser MIMEType
mimeTypeP = do maj    ← A.toCIAscii <$> token
               _      ← char '/'
               min    ← A.toCIAscii <$> token
               params ← paramsP
               return $ MIMEType maj min params

mimeTypeListP ∷ Parser [MIMEType]
mimeTypeListP = listOf mimeTypeP