--- #hide, prune
+{-# LANGUAGE
+ OverloadedStrings
+ , RecordWildCards
+ , TypeOperators
+ , UnicodeSyntax
+ #-}
module Network.HTTP.Lucu.DefaultPage
( getDefaultPage
- , writeDefaultPage
+ , defaultPageContentType
, mkDefaultPage
)
where
-
-import Control.Arrow
-import Control.Arrow.ArrowList
-import Control.Concurrent.STM
-import Control.Monad
-import qualified Data.ByteString.Lazy.Char8 as B
-import Data.ByteString.Lazy.Char8 (ByteString)
-import Data.Maybe
-import Network
-import Network.HTTP.Lucu.Config
-import Network.HTTP.Lucu.Headers
-import Network.HTTP.Lucu.Interaction
-import Network.HTTP.Lucu.Request
-import Network.HTTP.Lucu.Response
-import Network.URI
-import System.IO.Unsafe
-import Text.Printf
-import Text.XML.HXT.Arrow.WriteDocument
-import Text.XML.HXT.Arrow.XmlArrow
-import Text.XML.HXT.Arrow.XmlIOStateArrow
-import Text.XML.HXT.DOM.TypeDefs
-import Text.XML.HXT.DOM.XmlKeywords
-
-
-getDefaultPage :: Config -> Maybe Request -> Response -> String
+import Blaze.ByteString.Builder (Builder)
+import qualified Blaze.ByteString.Builder.Char.Utf8 as BB
+import Control.Arrow
+import Control.Arrow.ArrowList
+import Control.Arrow.ListArrow
+import Control.Arrow.Unicode
+import Data.Ascii (Ascii)
+import qualified Data.Ascii as A
+import qualified Data.CaseInsensitive as CI
+import Data.Maybe
+import qualified Data.Text as T
+import Network.HTTP.Lucu.Config
+import Network.HTTP.Lucu.Headers
+import Network.HTTP.Lucu.Request
+import Network.HTTP.Lucu.Response
+import Network.URI hiding (path)
+import Prelude.Unicode
+import Text.XML.HXT.Arrow.WriteDocument
+import Text.XML.HXT.Arrow.XmlArrow
+import Text.XML.HXT.Arrow.XmlState
+import Text.XML.HXT.DOM.TypeDefs
+
+getDefaultPage ∷ Config → Maybe Request → Response → Builder
+{-# INLINEABLE getDefaultPage #-}
getDefaultPage conf req res
- = let msgA = getMsg req res
+ = let msgA = getMsg req res
+ [xmlStr] = runLA ( mkDefaultPage conf (resStatus res) msgA
+ ⋙
+ writeDocumentToString [ withIndent True ]
+ ) ()
in
- unsafePerformIO $
- do [xmlStr] <- runX ( mkDefaultPage conf (resStatus res) msgA
- >>>
- writeDocumentToString [ (a_indent, v_1) ]
- )
- return xmlStr
-
-
-writeDefaultPage :: Interaction -> STM ()
-writeDefaultPage itr
- = do wroteHeader <- readTVar (itrWroteHeader itr)
-
- -- Content-Type が正しくなければ補完できない。
- res <- readItr itr itrResponse id
- when (getHeader "Content-Type" res == Just defaultPageContentType)
- $ do reqM <- readItr itr itrRequest id
-
- let conf = itrConfig itr
- page = B.pack $ getDefaultPage conf reqM res
-
- writeTVar (itrBodyToSend itr)
- $ page
-
-
-mkDefaultPage :: (ArrowXml a) => Config -> StatusCode -> a b XmlTree -> a b XmlTree
+ BB.fromString xmlStr
+
+defaultPageContentType ∷ Ascii
+{-# INLINE defaultPageContentType #-}
+defaultPageContentType = "application/xhtml+xml"
+
+mkDefaultPage ∷ (ArrowXml (⇝), StatusCode sc)
+ ⇒ Config
+ → sc
+ → b ⇝ XmlTree
+ → b ⇝ XmlTree
+{-# INLINEABLE mkDefaultPage #-}
mkDefaultPage conf status msgA
- = let (sCode, sMsg) = statusCode status
- sig = cnfServerSoftware conf
- ++ " at "
- ++ cnfServerHost conf
- ++ ( case cnfServerPort conf of
- Service serv -> ", service " ++ serv
- PortNumber num -> ", port " ++ show num
- UnixSocket path -> ", unix socket " ++ show path
- )
+ = let sStr = A.toString $ A.fromAsciiBuilder $ printStatusCode status
+ sig = concat [ A.toString (cnfServerSoftware conf)
+ , " at "
+ , T.unpack $ CI.original $ cnfServerHost conf
+ ]
in ( eelem "/"
+= ( eelem "html"
+= sattr "xmlns" "http://www.w3.org/1999/xhtml"
+= ( eelem "head"
+= ( eelem "title"
- += txt (printf "%03d %s" sCode sMsg)
+ += txt sStr
))
+= ( eelem "body"
+= ( eelem "h1"
- += txt sMsg
+ += txt sStr
)
+= ( eelem "p" += msgA )
+= eelem "hr"
+= ( eelem "address" += txt sig ))))
-
-getMsg :: (ArrowXml a) => Maybe Request -> Response -> a b XmlTree
-getMsg req res
- = case resStatus res of
- -- 1xx は body を持たない
- -- 2xx の body は補完しない
-
- -- 3xx
- MovedPermanently
- -> txt (printf "The resource at %s has been moved to " path)
- <+>
- eelem "a" += sattr "href" loc
- += txt loc
- <+>
- txt " permanently."
-
- Found
- -> txt (printf "The resource at %s is currently located at " path)
- <+>
- eelem "a" += sattr "href" loc
- += txt loc
- <+>
- txt ". This is not a permanent relocation."
-
- SeeOther
- -> txt (printf "The resource at %s can be found at " path)
- <+>
- eelem "a" += sattr "href" loc
- += txt loc
- <+>
- txt "."
-
- TemporaryRedirect
- -> txt (printf "The resource at %s is temporarily located at " path)
- <+>
- eelem "a" += sattr "href" loc
- += txt loc
- <+>
- txt "."
-
- -- 4xx
- BadRequest
- -> txt "The server could not understand the request you sent."
-
- Unauthorized
- -> txt (printf "You need a valid authentication to access %s" path)
-
- Forbidden
- -> txt (printf "You don't have permission to access %s" path)
-
- NotFound
- -> txt (printf "The requested URL %s was not found on this server." path)
-
- Gone
- -> txt (printf "The resource at %s was here in past times, but has gone permanently." path)
-
- RequestEntityTooLarge
- -> txt (printf "The request entity you sent for %s was too big to accept." path)
-
- RequestURITooLarge
- -> txt "The request URI you sent was too big to accept."
-
- -- 5xx
- InternalServerError
- -> txt (printf "An internal server error has occured during the process of your request to %s" path)
-
- ServiceUnavailable
- -> txt "The service is temporarily unavailable. Try later."
-
- _ -> none
-
-
+getMsg ∷ (ArrowXml (⇝)) ⇒ Maybe Request → Response → b ⇝ XmlTree
+{-# INLINEABLE getMsg #-}
+getMsg req res@(Response {..})
+ -- 1xx responses don't have a body.
+ -- 2xx responses don't need a body to be completed.
+ -- 3xx:
+ | resStatus ≈ MovedPermanently
+ = txt ("The resource at " ⧺ path ⧺ " has been moved to ")
+ <+>
+ eelem "a" += sattr "href" loc
+ += txt loc
+ <+>
+ txt " permanently."
+
+ | resStatus ≈ Found
+ = txt ("The resource at " ⧺ path ⧺ " is currently located at ")
+ <+>
+ eelem "a" += sattr "href" loc
+ += txt loc
+ <+>
+ txt ". This is not a permanent relocation."
+
+ | resStatus ≈ SeeOther
+ = txt ("The resource at " ⧺ path ⧺ " can be found at ")
+ <+>
+ eelem "a" += sattr "href" loc
+ += txt loc
+ <+>
+ txt "."
+
+ | resStatus ≈ TemporaryRedirect
+ = txt ("The resource at " ⧺ path ⧺ " is temporarily located at ")
+ <+>
+ eelem "a" += sattr "href" loc
+ += txt loc
+ <+>
+ txt "."
+
+ -- 4xx:
+ | resStatus ≈ BadRequest
+ = txt "The server could not understand the request you sent."
+ | resStatus ≈ Unauthorized
+ = txt ("You need a valid authentication to access " ⧺ path)
+ | resStatus ≈ Forbidden
+ = txt ("You don't have permission to access " ⧺ path)
+ | resStatus ≈ NotFound
+ = txt ("The requested URL " ⧺ path ⧺ " was not found on this server.")
+ | resStatus ≈ Gone
+ = txt ("The resource at " ⧺ path ⧺ " was here in past times, but has gone permanently.")
+ | resStatus ≈ RequestEntityTooLarge
+ = txt ("The request entity you sent for " ⧺ path ⧺ " was too large to accept.")
+ | resStatus ≈ RequestURITooLarge
+ = txt "The request URI you sent was too large to accept."
+
+ -- 5xx:
+ | resStatus ≈ InternalServerError
+ = txt ("An internal server error has occured during the process of your request to " ⧺ path)
+ | resStatus ≈ ServiceUnavailable
+ = txt "The service is temporarily unavailable. Try later."
+
+ | otherwise
+ = none
where
- path :: String
- path = let uri = reqURI $ fromJust req
- in
- uriPath uri
+ path ∷ String
+ path = uriPath $ reqURI $ fromJust req
- loc :: String
- loc = fromJust $ getHeader "Location" res
+ loc ∷ String
+ loc = A.toString $ fromJust $ getHeader "Location" res