@@ -66,15 +66,14 @@ module Servant.API.ContentTypes
6666 , AllMime(..)
6767 , AllMimeRender(..)
6868 , AllMimeUnrender(..)
69− , FromFormUrlEncoded(..)
70− , ToFormUrlEncoded(..)
7169 , eitherDecodeLenient
7270 , canHandleAcceptH
7371 ) where
7472
7573import Control.Arrow (left)
7674import Control.Monad.Compat
77−import Data.Aeson (FromJSON(..), ToJSON(..), encode)
75+import Data.Aeson (FromJSON (..), ToJSON (..),
76+ encode)
7877import Data.Aeson.Parser (value)
7978import Data.Aeson.Types (parseEither)
8079import Data.Attoparsec.ByteString.Char8 (endOfInput, parseOnly,
@@ -82,10 +81,8 @@ import Data.Attoparsec.ByteString.Char8 (endOfInput, parseOnly,
8281import qualified Data.ByteString as BS
8382import Data.ByteString.Lazy (ByteString, fromStrict,
8483 toStrict)
85−import qualified Data.ByteString.Lazy as B
8684import qualified Data.ByteString.Lazy.Char8 as BC
8785import Data.Maybe (isJust)
88−import Data.Monoid.Compat
8986import Data.String.Conversions (cs)
9087import qualified Data.Text as TextS
9188import qualified Data.Text.Encoding as TextS
@@ -94,8 +91,9 @@ import qualified Data.Text.Lazy.Encoding as TextL
9491import Data.Typeable
9592import GHC.Generics (Generic)
9693import qualified Network.HTTP.Media as M
97−import Network.URI (escapeURIString,
98− isUnreserved, unEscapeString)
94+import Web.FormUrlEncoded (FromForm, ToForm,
95+ urlEncodeAsForm,
96+ urlDecodeAsForm)
9997import Prelude ()
10098import Prelude.Compat
10199
@@ -290,12 +288,12 @@ instance OVERLAPPABLE_
290288 ToJSON a => MimeRender JSON a where
291289 mimeRender _ = encode
292290
293−-- | @encodeFormUrlEncoded . toFormUrlEncoded@
291+-- | @urlEncodeAsForm"
294292-- Note that the @mimeUnrender p (mimeRender p x) == Right x@ law only
295293-- holds if every element of x is non-null (i.e., not @("", "")@)
296294instance OVERLAPPABLE_
297− ToFormUrlEncoded a => MimeRender FormUrlEncoded a where
298− mimeRender _ = encodeFormUrlEncoded . toFormUrlEncoded
295+ ToForm a => MimeRender FormUrlEncoded a where
296+ mimeRender _ = urlEncodeAsForm
299297
300298-- | `TextL.encodeUtf8`
301299instance MimeRender PlainText TextL.Text where
@@ -348,11 +346,11 @@ eitherDecodeLenient input =
348346instance FromJSON a => MimeUnrender JSON a where
349347 mimeUnrender _ = eitherDecodeLenient
350348
351−-- | @decodeFormUrlEncoded >=> fromFormUrlEncoded@
349+-- | @urlDecodeAsForm@
352350-- Note that the @mimeUnrender p (mimeRender p x) == Right x@ law only
353351-- holds if every element of x is non-null (i.e., not @("", "")@)
354−instance FromFormUrlEncoded a => MimeUnrender FormUrlEncoded a where
355− mimeUnrender _ = decodeFormUrlEncoded >=> fromFormUrlEncoded
352+instance FromForm a => MimeUnrender FormUrlEncoded a where
353+ mimeUnrender _ = left TextS.unpack . urlDecodeAsForm
356354
357355-- | @left show . TextL.decodeUtf8'@
358356instance MimeUnrender PlainText TextL.Text where
@@ -375,49 +373,6 @@ instance MimeUnrender OctetStream BS.ByteString where
375373 mimeUnrender _ = Right . toStrict
376374
377375
378−--------------------------------------------------------------------------
379−-- * FormUrlEncoded
380−
381−-- | A type that can be converted to @application/x-www-form-urlencoded@
382−class ToFormUrlEncoded a where
383− toFormUrlEncoded :: a -> [(TextS.Text, TextS.Text)]
384−
385−instance ToFormUrlEncoded [(TextS.Text, TextS.Text)] where
386− toFormUrlEncoded = id
387−
388−-- | A type that can be converted from @application/x-www-form-urlencoded@,
389−-- with the possibility of failure.
390−class FromFormUrlEncoded a where
391− fromFormUrlEncoded :: [(TextS.Text, TextS.Text)] -> Either String a
392−
393−instance FromFormUrlEncoded [(TextS.Text, TextS.Text)] where
394− fromFormUrlEncoded = return
395−
396−encodeFormUrlEncoded :: [(TextS.Text, TextS.Text)] -> ByteString
397−encodeFormUrlEncoded xs =
398− let escape :: TextS.Text -> ByteString
399− escape = cs . escapeURIString isUnreserved . cs
400− encodePair :: (TextS.Text, TextS.Text) -> ByteString
401− encodePair (k, "") = escape k
402− encodePair (k, v) = escape k <> "=" <> escape v
403− in B.intercalate "&" $ map encodePair xs
404−
405−decodeFormUrlEncoded :: ByteString -> Either String [(TextS.Text, TextS.Text)]
406−decodeFormUrlEncoded "" = return []
407−decodeFormUrlEncoded q = do
408− let xs :: [TextS.Text]
409− xs = TextS.splitOn "&" . cs $ q
410− parsePair :: TextS.Text -> Either String (TextS.Text, TextS.Text)
411− parsePair p =
412− case TextS.splitOn "=" p of
413− [k,v] -> return ( unescape k
414− , unescape v
415− )
416− [k] -> return ( unescape k, "" )
417− _ -> Left $ "not a valid pair: " <> cs p
418− unescape :: TextS.Text -> TextS.Text
419− unescape = cs . unEscapeString . cs . TextS.intercalate "%20" . TextS.splitOn "+"
420− mapM parsePair xs
421376
422377-- $setup
423378-- >>> import Servant.API