X-Git-Url: http://git.cielonegro.org/gitweb.cgi?a=blobdiff_plain;f=Network%2FHTTP%2FLucu%2FUtils.hs;h=9212747d9cf4ea782491240f1c8e201c29715dc4;hb=f62b6f07bbf1eefcf552163d8f7daa6e0862ed5d;hp=df19a76d251fdd42c66161e381ea6c45bece2be1;hpb=e624f0db8c4610b36da9e4463a656e0cb8a104dd;p=Lucu.git diff --git a/Network/HTTP/Lucu/Utils.hs b/Network/HTTP/Lucu/Utils.hs index df19a76..9212747 100644 --- a/Network/HTTP/Lucu/Utils.hs +++ b/Network/HTTP/Lucu/Utils.hs @@ -1,42 +1,75 @@ +-- |Utility functions used internally in the Lucu httpd. These +-- functions may be useful too for something else. module Network.HTTP.Lucu.Utils - ( splitBy -- (a -> Bool) -> [a] -> [[a]] - , joinWith -- [a] -> [[a]] -> [a] - , trim -- (a -> Bool) -> [a] -> [a] - , noCaseEq -- String -> String -> Bool - , isWhiteSpace -- Char -> Bool + ( splitBy + , joinWith + , trim + , isWhiteSpace + , quoteStr + , parseWWWFormURLEncoded ) where -import Control.Monad.Trans -import Data.Char -import Data.List -import Foreign -import Foreign.C - +import Data.List hiding (last) +import Network.URI +import Prelude hiding (last) +-- |> splitBy (== ':') "ab:c:def" +-- > ==> ["ab", "c", "def"] splitBy :: (a -> Bool) -> [a] -> [[a]] -splitBy isSeparator src - = case break isSeparator src - of (last , [] ) -> last : [] - (first, sep:rest) -> first : splitBy isSeparator rest - +splitBy isSep src + = case break isSep src + of (last , [] ) -> [last] + (first, _sep:rest) -> first : splitBy isSep rest +-- |> joinWith ":" ["ab", "c", "def"] +-- > ==> "ab:c:def" joinWith :: [a] -> [[a]] -> [a] joinWith separator xs - = foldr (++) [] $ intersperse separator xs - + = separator `seq` xs `seq` + foldr (++) [] $! intersperse separator xs +-- |> trim (== '_') "__ab_c__def___" +-- > ==> "ab_c__def" trim :: (a -> Bool) -> [a] -> [a] -trim p = trimTail . trimHead +trim p = p `seq` trimTail . trimHead where trimHead = dropWhile p trimTail = reverse . trimHead . reverse +-- |@'isWhiteSpace' c@ is 'Prelude.True' iff c is one of SP, HT, CR +-- and LF. +isWhiteSpace :: Char -> Bool +isWhiteSpace ' ' = True +isWhiteSpace '\t' = True +isWhiteSpace '\r' = True +isWhiteSpace '\n' = True +isWhiteSpace _ = False +{-# INLINE isWhiteSpace #-} -noCaseEq :: String -> String -> Bool -noCaseEq a b - = (map toLower a) == (map toLower b) +-- |> quoteStr "abc" +-- > ==> "\"abc\"" +-- +-- > quoteStr "ab\"c" +-- > ==> "\"ab\\\"c\"" +quoteStr :: String -> String +quoteStr str = str `seq` + foldr (++) "" (["\""] ++ map quote str ++ ["\""]) + where + quote :: Char -> String + quote '"' = "\\\"" + quote c = [c] -isWhiteSpace :: Char -> Bool -isWhiteSpace = flip elem " \t\r\n" +-- |> parseWWWFormURLEncoded "aaa=bbb&ccc=ddd" +-- > ==> [("aaa", "bbb"), ("ccc", "ddd")] +parseWWWFormURLEncoded :: String -> [(String, String)] +parseWWWFormURLEncoded src + | src == "" = [] + | otherwise = do pairStr <- splitBy (\ c -> c == ';' || c == '&') src + let (key, value) = break (== '=') pairStr + return ( unEscapeString key + , unEscapeString $ case value of + ('=':val) -> val + val -> val + )