module Network.HTTP.Utils
( trim
, trimL
, trimR
, crlf
, lf
, sp
, split
, splitBy
, readsOne
, dropWhileTail
, chopAtDelim
, toUTF8BS
, fromUTF8BS
) where
import Data.Bits
import Data.Char
import Data.List ( elemIndex )
import Data.Maybe ( fromMaybe )
import Data.Word ( Word8 )
import qualified Data.ByteString as BS
crlf :: String
crlf :: String
crlf = "\r\n"
lf :: String
lf :: String
lf = "\n"
sp :: String
sp :: String
sp = " "
split :: Eq a => a -> [a] -> Maybe ([a],[a])
split :: a -> [a] -> Maybe ([a], [a])
split delim :: a
delim list :: [a]
list = case a
delim a -> [a] -> Maybe Int
forall a. Eq a => a -> [a] -> Maybe Int
`elemIndex` [a]
list of
Nothing -> Maybe ([a], [a])
forall a. Maybe a
Nothing
Just x :: Int
x -> ([a], [a]) -> Maybe ([a], [a])
forall a. a -> Maybe a
Just (([a], [a]) -> Maybe ([a], [a])) -> ([a], [a]) -> Maybe ([a], [a])
forall a b. (a -> b) -> a -> b
$ Int -> [a] -> ([a], [a])
forall a. Int -> [a] -> ([a], [a])
splitAt Int
x [a]
list
trim :: String -> String
trim :: String -> String
trim xs :: String
xs = String -> String
trimR (String -> String
trimL String
xs)
trimL :: String -> String
trimL :: String -> String
trimL xs :: String
xs = (Char -> Bool) -> String -> String
forall a. (a -> Bool) -> [a] -> [a]
dropWhile Char -> Bool
isSpace String
xs
trimR :: String -> String
trimR :: String -> String
trimR str :: String
str = String -> Maybe String -> String
forall a. a -> Maybe a -> a
fromMaybe "" (Maybe String -> String) -> Maybe String -> String
forall a b. (a -> b) -> a -> b
$ (Char -> Maybe String -> Maybe String)
-> Maybe String -> String -> Maybe String
forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr Char -> Maybe String -> Maybe String
trimIt Maybe String
forall a. Maybe a
Nothing String
str
where
trimIt :: Char -> Maybe String -> Maybe String
trimIt x :: Char
x (Just xs :: String
xs) = String -> Maybe String
forall a. a -> Maybe a
Just (Char
xChar -> String -> String
forall a. a -> [a] -> [a]
:String
xs)
trimIt x :: Char
x Nothing
| Char -> Bool
isSpace Char
x = Maybe String
forall a. Maybe a
Nothing
| Bool
otherwise = String -> Maybe String
forall a. a -> Maybe a
Just [Char
x]
splitBy :: Eq a => a -> [a] -> [[a]]
splitBy :: a -> [a] -> [[a]]
splitBy _ [] = []
splitBy c :: a
c xs :: [a]
xs =
case (a -> Bool) -> [a] -> ([a], [a])
forall a. (a -> Bool) -> [a] -> ([a], [a])
break (a -> a -> Bool
forall a. Eq a => a -> a -> Bool
==a
c) [a]
xs of
(_,[]) -> [[a]
xs]
(as :: [a]
as,_:bs :: [a]
bs) -> [a]
as [a] -> [[a]] -> [[a]]
forall a. a -> [a] -> [a]
: a -> [a] -> [[a]]
forall a. Eq a => a -> [a] -> [[a]]
splitBy a
c [a]
bs
readsOne :: Read a => (a -> b) -> b -> String -> b
readsOne :: (a -> b) -> b -> String -> b
readsOne f :: a -> b
f n :: b
n str :: String
str =
case ReadS a
forall a. Read a => ReadS a
reads String
str of
((v :: a
v,_):_) -> a -> b
f a
v
_ -> b
n
dropWhileTail :: (a -> Bool) -> [a] -> [a]
dropWhileTail :: (a -> Bool) -> [a] -> [a]
dropWhileTail f :: a -> Bool
f ls :: [a]
ls =
case (a -> Maybe [a] -> Maybe [a]) -> Maybe [a] -> [a] -> Maybe [a]
forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr a -> Maybe [a] -> Maybe [a]
chop Maybe [a]
forall a. Maybe a
Nothing [a]
ls of { Just xs :: [a]
xs -> [a]
xs; Nothing -> [] }
where
chop :: a -> Maybe [a] -> Maybe [a]
chop x :: a
x (Just xs :: [a]
xs) = [a] -> Maybe [a]
forall a. a -> Maybe a
Just (a
xa -> [a] -> [a]
forall a. a -> [a] -> [a]
:[a]
xs)
chop x :: a
x _
| a -> Bool
f a
x = Maybe [a]
forall a. Maybe a
Nothing
| Bool
otherwise = [a] -> Maybe [a]
forall a. a -> Maybe a
Just [a
x]
chopAtDelim :: Eq a => a -> [a] -> ([a],[a])
chopAtDelim :: a -> [a] -> ([a], [a])
chopAtDelim elt :: a
elt xs :: [a]
xs =
case (a -> Bool) -> [a] -> ([a], [a])
forall a. (a -> Bool) -> [a] -> ([a], [a])
break (a -> a -> Bool
forall a. Eq a => a -> a -> Bool
==a
elt) [a]
xs of
(_,[]) -> ([a]
xs,[])
(as :: [a]
as,_:bs :: [a]
bs) -> ([a]
as,[a]
bs)
toUTF8BS :: String -> BS.ByteString
toUTF8BS :: String -> ByteString
toUTF8BS = [Word8] -> ByteString
BS.pack ([Word8] -> ByteString)
-> (String -> [Word8]) -> String -> ByteString
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> [Word8]
encodeStringUtf8
fromUTF8BS :: BS.ByteString -> String
fromUTF8BS :: ByteString -> String
fromUTF8BS = [Word8] -> String
decodeStringUtf8 ([Word8] -> String)
-> (ByteString -> [Word8]) -> ByteString -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ByteString -> [Word8]
BS.unpack
encodeStringUtf8 :: String -> [Word8]
encodeStringUtf8 :: String -> [Word8]
encodeStringUtf8 [] = []
encodeStringUtf8 (c :: Char
c:cs :: String
cs)
| Char
c Char -> Char -> Bool
forall a. Ord a => a -> a -> Bool
<= '\x07F' = Word8
w8
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: String -> [Word8]
encodeStringUtf8 String
cs
| Char
c Char -> Char -> Bool
forall a. Ord a => a -> a -> Bool
<= '\x7FF' = (0xC0 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. Int -> Word8
w8ShiftR 6 )
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Word8
w8 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: String -> [Word8]
encodeStringUtf8 String
cs
| Char
c Char -> Char -> Bool
forall a. Ord a => a -> a -> Bool
<= '\xD7FF'= (0xE0 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. Int -> Word8
w8ShiftR 12 )
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Int -> Word8
w8ShiftR 6 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Word8
w8 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: String -> [Word8]
encodeStringUtf8 String
cs
| Char
c Char -> Char -> Bool
forall a. Ord a => a -> a -> Bool
<= '\xDFFF'= 0xEF Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: 0xBF Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: 0xBD
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: String -> [Word8]
encodeStringUtf8 String
cs
| Char
c Char -> Char -> Bool
forall a. Ord a => a -> a -> Bool
<= '\xFFFF'= (0xE0 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. Int -> Word8
w8ShiftR 12 )
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Int -> Word8
w8ShiftR 6 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Word8
w8 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: String -> [Word8]
encodeStringUtf8 String
cs
| Bool
otherwise = (0xf0 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. Int -> Word8
w8ShiftR 18 )
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Int -> Word8
w8ShiftR 12 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Int -> Word8
w8ShiftR 6 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: (0x80 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.|. (Word8
w8 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F))
Word8 -> [Word8] -> [Word8]
forall a. a -> [a] -> [a]
: String -> [Word8]
encodeStringUtf8 String
cs
where
w8 :: Word8
w8 = Int -> Word8
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Char -> Int
ord Char
c) :: Word8
w8ShiftR :: Int -> Word8
w8ShiftR :: Int -> Word8
w8ShiftR = Int -> Word8
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Int -> Word8) -> (Int -> Int) -> Int -> Word8
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Int -> Int -> Int
forall a. Bits a => a -> Int -> a
shiftR (Char -> Int
ord Char
c)
decodeStringUtf8 :: [Word8] -> String
decodeStringUtf8 :: [Word8] -> String
decodeStringUtf8 = [Word8] -> String
go
where
go :: [Word8] -> String
go :: [Word8] -> String
go [] = []
go (c :: Word8
c : cs :: [Word8]
cs)
| Word8
c Word8 -> Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= 0x7F = Int -> Char
chr (Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral Word8
c) Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs
| Word8
c Word8 -> Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= 0xBF = Char
replacementChar Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs
| Word8
c Word8 -> Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= 0xDF = Word8 -> [Word8] -> String
twoBytes Word8
c [Word8]
cs
| Word8
c Word8 -> Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= 0xEF = Int -> Int -> [Word8] -> Int -> String
moreBytes 3 0x800 [Word8]
cs (Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8 -> Int) -> Word8 -> Int
forall a b. (a -> b) -> a -> b
$ Word8
c Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0xF)
| Word8
c Word8 -> Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= 0xF7 = Int -> Int -> [Word8] -> Int -> String
moreBytes 4 0x10000 [Word8]
cs (Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8 -> Int) -> Word8 -> Int
forall a b. (a -> b) -> a -> b
$ Word8
c Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x7)
| Word8
c Word8 -> Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= 0xFB = Int -> Int -> [Word8] -> Int -> String
moreBytes 5 0x200000 [Word8]
cs (Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8 -> Int) -> Word8 -> Int
forall a b. (a -> b) -> a -> b
$ Word8
c Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3)
| Word8
c Word8 -> Word8 -> Bool
forall a. Ord a => a -> a -> Bool
<= 0xFD = Int -> Int -> [Word8] -> Int -> String
moreBytes 6 0x4000000 [Word8]
cs (Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8 -> Int) -> Word8 -> Int
forall a b. (a -> b) -> a -> b
$ Word8
c Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x1)
| Bool
otherwise = Char
replacementChar Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs
twoBytes :: Word8 -> [Word8] -> String
twoBytes :: Word8 -> [Word8] -> String
twoBytes c0 :: Word8
c0 (c1 :: Word8
c1:cs' :: [Word8]
cs')
| Word8
c1 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0xC0 Word8 -> Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0x80
= let d :: Int
d = (Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8
c0 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x1F) Int -> Int -> Int
forall a. Bits a => a -> Int -> a
`shiftL` 6)
Int -> Int -> Int
forall a. Bits a => a -> a -> a
.|. Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8
c1 Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0x3F)
in if Int
d Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
>= 0x80
then Int -> Char
chr Int
d Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs'
else Char
replacementChar Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs'
twoBytes _ cs' :: [Word8]
cs' = Char
replacementChar Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs'
moreBytes :: Int -> Int -> [Word8] -> Int -> [Char]
moreBytes :: Int -> Int -> [Word8] -> Int -> String
moreBytes 1 overlong :: Int
overlong cs' :: [Word8]
cs' acc :: Int
acc
| Int
overlong Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
<= Int
acc Bool -> Bool -> Bool
&& Int
acc Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
<= 0x10FFFF Bool -> Bool -> Bool
&& (Int
acc Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
< 0xD800 Bool -> Bool -> Bool
|| 0xDFFF Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
< Int
acc)
= Int -> Char
chr Int
acc Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs'
| Bool
otherwise
= Char
replacementChar Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs'
moreBytes byteCount :: Int
byteCount overlong :: Int
overlong (cn :: Word8
cn:cs' :: [Word8]
cs') acc :: Int
acc
| Word8
cn Word8 -> Word8 -> Word8
forall a. Bits a => a -> a -> a
.&. 0xC0 Word8 -> Word8 -> Bool
forall a. Eq a => a -> a -> Bool
== 0x80
= Int -> Int -> [Word8] -> Int -> String
moreBytes (Int
byteCountInt -> Int -> Int
forall a. Num a => a -> a -> a
-1) Int
overlong [Word8]
cs'
((Int
acc Int -> Int -> Int
forall a. Bits a => a -> Int -> a
`shiftL` 6) Int -> Int -> Int
forall a. Bits a => a -> a -> a
.|. Word8 -> Int
forall a b. (Integral a, Num b) => a -> b
fromIntegral Word8
cn Int -> Int -> Int
forall a. Bits a => a -> a -> a
.&. 0x3F)
moreBytes _ _ cs' :: [Word8]
cs' _
= Char
replacementChar Char -> String -> String
forall a. a -> [a] -> [a]
: [Word8] -> String
go [Word8]
cs'
replacementChar :: Char
replacementChar = '\xfffd'