X-Git-Url: http://git.cielonegro.org/gitweb.cgi?p=Lucu.git;a=blobdiff_plain;f=Network%2FHTTP%2FLucu%2FAbortion.hs;h=064a97ebd8671232b9053a6382d93c85927df318;hp=0099576e62e04c631ae00a528ea0c55cb02eb6c5;hb=51eda5b;hpb=1789cee5ee66d2f7f2b26280be2f13eac4df7980 diff --git a/Network/HTTP/Lucu/Abortion.hs b/Network/HTTP/Lucu/Abortion.hs index 0099576..064a97e 100644 --- a/Network/HTTP/Lucu/Abortion.hs +++ b/Network/HTTP/Lucu/Abortion.hs @@ -1,117 +1,50 @@ {-# LANGUAGE - Arrows - , DeriveDataTypeable - , TypeOperators - , UnicodeSyntax + UnicodeSyntax #-} - -- |Aborting the computation of 'Network.HTTP.Lucu.Resource.Resource' -- in any 'Prelude.IO' monads or arrows. module Network.HTTP.Lucu.Abortion - ( Abortion(..) + ( Abortion + , mkAbortion + , mkAbortion' , abort - , abortPurely - , abortSTM - , abortA - , abortPage ) where -import Control.Arrow.ArrowIO -import Control.Arrow.ListArrow -import Control.Arrow.Unicode -import Control.Concurrent.STM import Control.Exception import Control.Monad.Trans import Data.Ascii (Ascii, CIAscii) +import Data.Monoid.Unicode import Data.Text (Text) -import qualified Data.Text as T -import qualified Data.Text.Lazy as Lazy -import Data.Typeable -import Network.HTTP.Lucu.Config -import Network.HTTP.Lucu.DefaultPage +import Network.HTTP.Lucu.Abortion.Internal import Network.HTTP.Lucu.Headers -import Network.HTTP.Lucu.Request import Network.HTTP.Lucu.Response import Prelude.Unicode -import Text.XML.HXT.Arrow.WriteDocument -import Text.XML.HXT.Arrow.XmlArrow -import Text.XML.HXT.Arrow.XmlState - -data Abortion = Abortion { - aboStatus ∷ !StatusCode - , aboHeaders ∷ !Headers - , aboMessage ∷ !(Maybe Text) - } deriving (Eq, Show, Typeable) -instance Exception Abortion - --- |Computation of @'abort' status headers msg@ aborts the --- 'Network.HTTP.Lucu.Resource.Resource' monad with given status, --- additional response headers, and optional message string. --- --- What this really does is to throw a special --- 'Control.Exception.Exception'. The exception will be caught by the --- Lucu system. --- --- 1. If the 'Network.HTTP.Lucu.Resource.Resource' is in the /Deciding --- Header/ or any precedent states, it is possible to use the --- @status@ and such like as a HTTP response to be sent to the --- client. --- --- 2. Otherwise the HTTP response can't be modified anymore so the --- only possible thing the system can do is to dump it to the --- stderr. See --- 'Network.HTTP.Lucu.Config.cnfDumpTooLateAbortionToStderr'. --- --- Note that the status code doesn't have to be an error code so you --- can use this action for redirection as well as error reporting e.g. --- --- > abort MovedPermanently --- > [("Location", "http://example.net/")] --- > (Just "It has been moved to example.net") -abort ∷ MonadIO m ⇒ StatusCode → [ (CIAscii, Ascii) ] → Maybe Text → m a +-- |Construct an 'Abortion' with additional headers and an optional +-- message text. +mkAbortion ∷ StatusCode sc ⇒ sc → [(CIAscii, Ascii)] → Maybe Text → Abortion +{-# INLINE mkAbortion #-} +mkAbortion sc hdr msg + = Abortion { + aboStatus = fromStatusCode sc + , aboHeaders = toHeaders hdr + , aboMessage = msg + } + +-- |Construct an 'Abortion' without any additional headers but with a +-- message text. +mkAbortion' ∷ StatusCode sc ⇒ sc → Text → Abortion +{-# INLINE mkAbortion' #-} +mkAbortion' sc msg + = Abortion { + aboStatus = fromStatusCode sc + , aboHeaders = (∅) + , aboMessage = Just msg + } + +-- |Throw an 'Abortion' in a 'MonadIO', including the very +-- 'Network.HTTP.Lucu.Resource.Resource' monad. +abort ∷ MonadIO m ⇒ Abortion → m a {-# INLINE abort #-} -abort status headers - = liftIO ∘ throwIO ∘ Abortion status (toHeaders headers) - --- |This is similar to 'abort' but computes it with --- 'System.IO.Unsafe.unsafePerformIO'. -abortPurely ∷ StatusCode → [ (CIAscii, Ascii) ] → Maybe Text → a -{-# INLINE abortPurely #-} -abortPurely status headers - = throw ∘ Abortion status (toHeaders headers) - --- |Computation of @'abortSTM' status headers msg@ just computes --- 'abort' in a 'Control.Monad.STM.STM' monad. -abortSTM ∷ StatusCode → [ (CIAscii, Ascii) ] → Maybe Text → STM a -{-# INLINE abortSTM #-} -abortSTM status headers - = throwSTM ∘ Abortion status (toHeaders headers) - --- | Computation of @'abortA' -< (status, (headers, msg))@ just --- computes 'abort' in an 'Control.Arrow.ArrowIO.ArrowIO'. -abortA ∷ ArrowIO (⇝) ⇒ (StatusCode, ([ (CIAscii, Ascii) ], Maybe Text)) ⇝ c -{-# INLINE abortA #-} -abortA = proc (status, (headers, msg)) → - arrIO throwIO ⤙ Abortion status (toHeaders headers) msg - --- aboMessage が Just なら單に mkDefaultPage に渡すだけで良いので樂だが、 --- Nothing の場合は getDefaultPage を使ってデフォルトのメッセージを得な --- ければならない。 -abortPage ∷ Config → Maybe Request → Response → Abortion → Lazy.Text -abortPage conf reqM res abo - = case aboMessage abo of - Just msg - → let [html] = runLA ( mkDefaultPage conf (aboStatus abo) (txt $ T.unpack msg) - ⋙ - writeDocumentToString [ withIndent True ] - ) () - in - Lazy.pack html - Nothing - → let res' = res { resStatus = aboStatus abo } - res'' = foldl (∘) id [setHeader name value - | (name, value) ← fromHeaders $ aboHeaders abo] res' - in - getDefaultPage conf reqM res'' +abort = liftIO ∘ throwIO