ContentType is a String
This commit is contained in:
parent
2ba7dc6780
commit
8a20a416ba
142
Yesod/Content.hs
142
Yesod/Content.hs
@ -12,9 +12,20 @@ module Yesod.Content
|
|||||||
, toContent
|
, toContent
|
||||||
-- * Mime types
|
-- * Mime types
|
||||||
-- ** Data type
|
-- ** Data type
|
||||||
, ContentType (..)
|
, ContentType
|
||||||
, contentTypeFromString
|
, typeHtml
|
||||||
, contentTypeToString
|
, typePlain
|
||||||
|
, typeJson
|
||||||
|
, typeXml
|
||||||
|
, typeAtom
|
||||||
|
, typeJpeg
|
||||||
|
, typePng
|
||||||
|
, typeGif
|
||||||
|
, typeJavascript
|
||||||
|
, typeCss
|
||||||
|
, typeFlv
|
||||||
|
, typeOgv
|
||||||
|
, typeOctet
|
||||||
-- ** File extensions
|
-- ** File extensions
|
||||||
, typeByExt
|
, typeByExt
|
||||||
, ext
|
, ext
|
||||||
@ -124,7 +135,7 @@ instance HasReps ChooseRep where
|
|||||||
chooseRep = id
|
chooseRep = id
|
||||||
|
|
||||||
instance HasReps () where
|
instance HasReps () where
|
||||||
chooseRep = defChooseRep [(TypePlain, const $ return $ cs "")]
|
chooseRep = defChooseRep [(typePlain, const $ return $ cs "")]
|
||||||
|
|
||||||
instance HasReps [(ContentType, Content)] where
|
instance HasReps [(ContentType, Content)] where
|
||||||
chooseRep a cts = return $
|
chooseRep a cts = return $
|
||||||
@ -134,73 +145,67 @@ instance HasReps [(ContentType, Content)] where
|
|||||||
(x:_) -> x
|
(x:_) -> x
|
||||||
_ -> error "chooseRep [(ContentType, Content)] of empty"
|
_ -> error "chooseRep [(ContentType, Content)] of empty"
|
||||||
where
|
where
|
||||||
go = simpleContentType . contentTypeToString
|
go = simpleContentType
|
||||||
|
|
||||||
newtype RepHtml = RepHtml Content
|
newtype RepHtml = RepHtml Content
|
||||||
instance HasReps RepHtml where
|
instance HasReps RepHtml where
|
||||||
chooseRep (RepHtml c) _ = return (TypeHtml, c)
|
chooseRep (RepHtml c) _ = return (typeHtml, c)
|
||||||
newtype RepJson = RepJson Content
|
newtype RepJson = RepJson Content
|
||||||
instance HasReps RepJson where
|
instance HasReps RepJson where
|
||||||
chooseRep (RepJson c) _ = return (TypeJson, c)
|
chooseRep (RepJson c) _ = return (typeJson, c)
|
||||||
data RepHtmlJson = RepHtmlJson Content Content
|
data RepHtmlJson = RepHtmlJson Content Content
|
||||||
instance HasReps RepHtmlJson where
|
instance HasReps RepHtmlJson where
|
||||||
chooseRep (RepHtmlJson html json) = chooseRep
|
chooseRep (RepHtmlJson html json) = chooseRep
|
||||||
[ (TypeHtml, html)
|
[ (typeHtml, html)
|
||||||
, (TypeJson, json)
|
, (typeJson, json)
|
||||||
]
|
]
|
||||||
newtype RepPlain = RepPlain Content
|
newtype RepPlain = RepPlain Content
|
||||||
instance HasReps RepPlain where
|
instance HasReps RepPlain where
|
||||||
chooseRep (RepPlain c) _ = return (TypePlain, c)
|
chooseRep (RepPlain c) _ = return (typePlain, c)
|
||||||
newtype RepXml = RepXml Content
|
newtype RepXml = RepXml Content
|
||||||
instance HasReps RepXml where
|
instance HasReps RepXml where
|
||||||
chooseRep (RepXml c) _ = return (TypeXml, c)
|
chooseRep (RepXml c) _ = return (typeXml, c)
|
||||||
|
|
||||||
-- | Equality is determined by converting to a 'String' via
|
type ContentType = String
|
||||||
-- 'contentTypeToString'. This ensures that, for example, 'TypeJpeg' is the
|
|
||||||
-- same as 'TypeOther' \"image/jpeg\". However, note that 'TypeHtml' is *not*
|
|
||||||
-- the same as 'TypeOther' \"text/html\", since 'TypeHtml' is defined as UTF-8
|
|
||||||
-- encoded. See 'contentTypeToString'.
|
|
||||||
data ContentType =
|
|
||||||
TypeHtml
|
|
||||||
| TypePlain
|
|
||||||
| TypeJson
|
|
||||||
| TypeXml
|
|
||||||
| TypeAtom
|
|
||||||
| TypeJpeg
|
|
||||||
| TypePng
|
|
||||||
| TypeGif
|
|
||||||
| TypeJavascript
|
|
||||||
| TypeCss
|
|
||||||
| TypeFlv
|
|
||||||
| TypeOgv
|
|
||||||
| TypeOctet
|
|
||||||
| TypeOther String
|
|
||||||
deriving (Show)
|
|
||||||
|
|
||||||
-- | This is simply a synonym for 'TypeOther'. However, equality works as
|
typeHtml :: ContentType
|
||||||
-- expected; see 'ContentType'.
|
typeHtml = "text/html; charset=utf-8"
|
||||||
contentTypeFromString :: String -> ContentType
|
|
||||||
contentTypeFromString = TypeOther
|
|
||||||
|
|
||||||
-- | This works as expected, with one caveat: the builtin textual content types
|
typePlain :: ContentType
|
||||||
-- ('TypeHtml', 'TypePlain', etc) all include \"; charset=utf-8\" at the end of
|
typePlain = "text/plain; charset=utf-8"
|
||||||
-- their basic content-type. If another encoding is desired, please use
|
|
||||||
-- 'TypeOther'.
|
typeJson :: ContentType
|
||||||
contentTypeToString :: ContentType -> String
|
typeJson = "application/json; charset=utf-8"
|
||||||
contentTypeToString TypeHtml = "text/html; charset=utf-8"
|
|
||||||
contentTypeToString TypePlain = "text/plain; charset=utf-8"
|
typeXml :: ContentType
|
||||||
contentTypeToString TypeJson = "application/json; charset=utf-8"
|
typeXml = "text/xml"
|
||||||
contentTypeToString TypeXml = "text/xml"
|
|
||||||
contentTypeToString TypeAtom = "application/atom+xml"
|
typeAtom :: ContentType
|
||||||
contentTypeToString TypeJpeg = "image/jpeg"
|
typeAtom = "application/atom+xml"
|
||||||
contentTypeToString TypePng = "image/png"
|
|
||||||
contentTypeToString TypeGif = "image/gif"
|
typeJpeg :: ContentType
|
||||||
contentTypeToString TypeJavascript = "text/javascript; charset=utf-8"
|
typeJpeg = "image/jpeg"
|
||||||
contentTypeToString TypeCss = "text/css; charset=utf-8"
|
|
||||||
contentTypeToString TypeFlv = "video/x-flv"
|
typePng :: ContentType
|
||||||
contentTypeToString TypeOgv = "video/ogg"
|
typePng = "image/png"
|
||||||
contentTypeToString TypeOctet = "application/octet-stream"
|
|
||||||
contentTypeToString (TypeOther s) = s
|
typeGif :: ContentType
|
||||||
|
typeGif = "image/gif"
|
||||||
|
|
||||||
|
typeJavascript :: ContentType
|
||||||
|
typeJavascript = "text/javascript; charset=utf-8"
|
||||||
|
|
||||||
|
typeCss :: ContentType
|
||||||
|
typeCss = "text/css; charset=utf-8"
|
||||||
|
|
||||||
|
typeFlv :: ContentType
|
||||||
|
typeFlv = "video/x-flv"
|
||||||
|
|
||||||
|
typeOgv :: ContentType
|
||||||
|
typeOgv = "video/ogg"
|
||||||
|
|
||||||
|
typeOctet :: ContentType
|
||||||
|
typeOctet = "application/octet-stream"
|
||||||
|
|
||||||
-- | Removes \"extra\" information at the end of a content type string. In
|
-- | Removes \"extra\" information at the end of a content type string. In
|
||||||
-- particular, removes everything after the semicolon, if present.
|
-- particular, removes everything after the semicolon, if present.
|
||||||
@ -210,22 +215,19 @@ contentTypeToString (TypeOther s) = s
|
|||||||
simpleContentType :: String -> String
|
simpleContentType :: String -> String
|
||||||
simpleContentType = fst . span (/= ';')
|
simpleContentType = fst . span (/= ';')
|
||||||
|
|
||||||
instance Eq ContentType where
|
|
||||||
(==) = (==) `on` contentTypeToString
|
|
||||||
|
|
||||||
-- | Determine a mime-type based on the file extension.
|
-- | Determine a mime-type based on the file extension.
|
||||||
typeByExt :: String -> ContentType
|
typeByExt :: String -> ContentType
|
||||||
typeByExt "jpg" = TypeJpeg
|
typeByExt "jpg" = typeJpeg
|
||||||
typeByExt "jpeg" = TypeJpeg
|
typeByExt "jpeg" = typeJpeg
|
||||||
typeByExt "js" = TypeJavascript
|
typeByExt "js" = typeJavascript
|
||||||
typeByExt "css" = TypeCss
|
typeByExt "css" = typeCss
|
||||||
typeByExt "html" = TypeHtml
|
typeByExt "html" = typeHtml
|
||||||
typeByExt "png" = TypePng
|
typeByExt "png" = typePng
|
||||||
typeByExt "gif" = TypeGif
|
typeByExt "gif" = typeGif
|
||||||
typeByExt "txt" = TypePlain
|
typeByExt "txt" = typePlain
|
||||||
typeByExt "flv" = TypeFlv
|
typeByExt "flv" = typeFlv
|
||||||
typeByExt "ogv" = TypeOgv
|
typeByExt "ogv" = typeOgv
|
||||||
typeByExt _ = TypeOctet
|
typeByExt _ = typeOctet
|
||||||
|
|
||||||
-- | Get a file extension (everything after last period).
|
-- | Get a file extension (everything after last period).
|
||||||
ext :: String -> String
|
ext :: String -> String
|
||||||
@ -247,7 +249,7 @@ propExt s =
|
|||||||
caseTypeByExt :: Assertion
|
caseTypeByExt :: Assertion
|
||||||
caseTypeByExt = do
|
caseTypeByExt = do
|
||||||
TypeJavascript @=? typeByExt (ext "foo.js")
|
TypeJavascript @=? typeByExt (ext "foo.js")
|
||||||
TypeHtml @=? typeByExt (ext "foo.html")
|
typeHtml @=? typeByExt (ext "foo.html")
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
-- | Format a 'UTCTime' in W3 format; useful for setting cookies.
|
-- | Format a 'UTCTime' in W3 format; useful for setting cookies.
|
||||||
|
|||||||
@ -233,7 +233,7 @@ toWaiApp' y segments env = do
|
|||||||
(cs sessionVal)
|
(cs sessionVal)
|
||||||
: hs
|
: hs
|
||||||
hs'' = map (headerToPair getExpires) hs'
|
hs'' = map (headerToPair getExpires) hs'
|
||||||
hs''' = (W.ContentType, cs $ contentTypeToString ct) : hs''
|
hs''' = (W.ContentType, cs ct) : hs''
|
||||||
return $ W.Response s hs''' $ case c of
|
return $ W.Response s hs''' $ case c of
|
||||||
ContentFile fp -> Left fp
|
ContentFile fp -> Left fp
|
||||||
ContentEnum e -> Right $ W.buffer
|
ContentEnum e -> Right $ W.buffer
|
||||||
@ -253,7 +253,7 @@ fullRender ar render route =
|
|||||||
ar ++ '/' : encodePathInfo (fixSegs $ render route)
|
ar ++ '/' : encodePathInfo (fixSegs $ render route)
|
||||||
|
|
||||||
httpAccept :: W.Request -> [ContentType]
|
httpAccept :: W.Request -> [ContentType]
|
||||||
httpAccept = map (contentTypeFromString . B.unpack)
|
httpAccept = map B.unpack
|
||||||
. parseHttpAccept
|
. parseHttpAccept
|
||||||
. fromMaybe B.empty
|
. fromMaybe B.empty
|
||||||
. lookup W.Accept
|
. lookup W.Accept
|
||||||
|
|||||||
@ -234,7 +234,7 @@ runHandler handler mrender sroute tomr ma tosa = YesodApp $ \eh rr cts -> do
|
|||||||
HCError e -> handleError e
|
HCError e -> handleError e
|
||||||
HCRedirect rt loc -> do
|
HCRedirect rt loc -> do
|
||||||
let hs = Header "Location" loc : headers
|
let hs = Header "Location" loc : headers
|
||||||
return (getRedirectStatus rt, hs, TypePlain, cs "", finalSession)
|
return (getRedirectStatus rt, hs, typePlain, cs "", finalSession)
|
||||||
HCSendFile ct fp -> E.catch
|
HCSendFile ct fp -> E.catch
|
||||||
(sendFile' ct fp)
|
(sendFile' ct fp)
|
||||||
(handleError . toErrorHandler)
|
(handleError . toErrorHandler)
|
||||||
@ -242,7 +242,7 @@ runHandler handler mrender sroute tomr ma tosa = YesodApp $ \eh rr cts -> do
|
|||||||
safeEh :: ErrorResponse -> YesodApp
|
safeEh :: ErrorResponse -> YesodApp
|
||||||
safeEh er = YesodApp $ \_ _ _ -> do
|
safeEh er = YesodApp $ \_ _ _ -> do
|
||||||
liftIO $ hPutStrLn stderr $ "Error handler errored out: " ++ show er
|
liftIO $ hPutStrLn stderr $ "Error handler errored out: " ++ show er
|
||||||
return (W.Status500, [], TypePlain, cs "Internal Server Error", [])
|
return (W.Status500, [], typePlain, cs "Internal Server Error", [])
|
||||||
|
|
||||||
-- | Redirect to the given route.
|
-- | Redirect to the given route.
|
||||||
redirect :: RedirectType -> Routes master -> GHandler sub master a
|
redirect :: RedirectType -> Routes master -> GHandler sub master a
|
||||||
|
|||||||
@ -27,7 +27,7 @@ import Data.Time.Clock (UTCTime)
|
|||||||
|
|
||||||
newtype RepAtom = RepAtom Content
|
newtype RepAtom = RepAtom Content
|
||||||
instance HasReps RepAtom where
|
instance HasReps RepAtom where
|
||||||
chooseRep (RepAtom c) _ = return (TypeAtom, c)
|
chooseRep (RepAtom c) _ = return (typeAtom, c)
|
||||||
|
|
||||||
atomFeed :: AtomFeed (Routes master) -> GHandler sub master RepAtom
|
atomFeed :: AtomFeed (Routes master) -> GHandler sub master RepAtom
|
||||||
atomFeed = fmap RepAtom . hamletToContent . template
|
atomFeed = fmap RepAtom . hamletToContent . template
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user