mirror of
https://github.com/simplex-chat/simplexmq.git
synced 2026-04-01 13:56:15 +00:00
f3408d9bb6
* explicit exports * more empty exports * add exports * reorder * use correct ControlProtocol type for xftp router --------- Co-authored-by: Evgeny @ SimpleX Chat <259188159+evgeny-simplex@users.noreply.github.com>
50 lines
1.6 KiB
Haskell
50 lines
1.6 KiB
Haskell
module Simplex.Messaging.Agent.QueryString
|
|
( QueryStringParams (..),
|
|
QSPEscaping (..),
|
|
queryParam,
|
|
queryParamParser,
|
|
queryParam_,
|
|
queryParamStr,
|
|
) where
|
|
|
|
import Data.Attoparsec.ByteString.Char8 (Parser)
|
|
import qualified Data.Attoparsec.ByteString.Char8 as A
|
|
import Data.ByteString.Char8 (ByteString)
|
|
import qualified Data.ByteString.Char8 as B
|
|
import Data.List (find)
|
|
import qualified Network.HTTP.Types as Q
|
|
import Simplex.Messaging.Encoding.String
|
|
import Simplex.Messaging.Parsers (parseAll)
|
|
|
|
data QueryStringParams = QSP QSPEscaping Q.SimpleQuery
|
|
deriving (Show)
|
|
|
|
data QSPEscaping = QEscape | QNoEscaping
|
|
deriving (Show)
|
|
|
|
instance StrEncoding QueryStringParams where
|
|
strEncode (QSP esc q) = case esc of
|
|
QEscape -> Q.renderSimpleQuery False q
|
|
QNoEscaping ->
|
|
Q.renderQueryPartialEscape False $
|
|
map (\(n, v) -> (n, [Q.QN v])) q
|
|
strP = QSP QEscape . Q.parseSimpleQuery <$> A.takeTill (\c -> c == ' ' || c == '\n')
|
|
|
|
queryParam :: StrEncoding a => ByteString -> QueryStringParams -> Parser a
|
|
queryParam = queryParamParser strP
|
|
|
|
queryParamParser :: Parser a -> ByteString -> QueryStringParams -> Parser a
|
|
queryParamParser p name q =
|
|
case queryParamStr name q of
|
|
Just s -> either fail pure $ parseAll p s
|
|
_ -> fail $ "no qs param " <> B.unpack name
|
|
|
|
queryParam_ :: StrEncoding a => ByteString -> QueryStringParams -> Parser (Maybe a)
|
|
queryParam_ name q =
|
|
case queryParamStr name q of
|
|
Just p -> either fail pure $ parseAll strP p
|
|
_ -> pure Nothing
|
|
|
|
queryParamStr :: ByteString -> QueryStringParams -> Maybe ByteString
|
|
queryParamStr name (QSP _ q) = snd <$> find ((== name) . fst) q
|