#-}
-- |Utility functions used internally in this package.
module Network.HTTP.Lucu.Utils
- ( Host
+ ( Scheme
+ , Host
, PathSegment
, Path
, splitBy
, quoteStr
, parseWWWFormURLEncoded
+ , uriCIScheme
, uriHost
, uriPathSegments
, trim
where
import Control.Applicative hiding (empty)
import Control.Monad hiding (mapM)
-import Data.Ascii (Ascii, AsciiBuilder)
+import Data.Ascii (Ascii, AsciiBuilder, CIAscii)
import qualified Data.Ascii as A
import Data.ByteString (ByteString)
import qualified Data.ByteString.Char8 as BS
import Data.Char
import Data.Collections
import Data.Collections.BaseInstances ()
+import Data.Convertible.Base
+import Data.Convertible.Instances.Ascii ()
+import Data.Convertible.Instances.Text ()
+import Data.Convertible.Instances.Time ()
import Data.Maybe
import Data.Monoid.Unicode
-import Data.Ratio
import Data.Text (Text)
-import qualified Data.Text as T
import Data.Time
-import Data.Time.Clock.POSIX
import Network.URI
import Prelude hiding (last, mapM, null, reverse)
import Prelude.Unicode
import System.Directory
-import System.Time (ClockTime(..))
+
+-- |'Scheme' represents an URI scheme.
+type Scheme = CIAscii
-- |'Host' represents an IP address or a host name in an URI
-- authority.
-- >>> quoteStr "ab\"c"
-- "\"ab\\\"c\""
quoteStr ∷ Ascii → AsciiBuilder
-quoteStr str = A.toAsciiBuilder "\"" ⊕
- go (A.toByteString str) (∅) ⊕
- A.toAsciiBuilder "\""
+quoteStr str = cs ("\"" ∷ Ascii) ⊕
+ go (cs str) (∅) ⊕
+ cs ("\"" ∷ Ascii)
where
go ∷ ByteString → AsciiBuilder → AsciiBuilder
go bs ab
→ ab ⊕ b2ab x
| otherwise
→ go (BS.tail y)
- (ab ⊕ b2ab x ⊕ A.toAsciiBuilder "\\\"")
+ (ab ⊕ b2ab x ⊕ cs ("\\\"" ∷ Ascii))
b2ab ∷ ByteString → AsciiBuilder
- b2ab = A.toAsciiBuilder ∘ A.unsafeFromByteString
+ b2ab = cs ∘ A.unsafeFromByteString
-- |>>> parseWWWFormURLEncoded "aaa=bbb&ccc=ddd"
-- [("aaa", "bbb"), ("ccc", "ddd")]
-- THINKME: We could gain some performance by using attoparsec
-- here.
| src ≡ "" = []
- | otherwise = do pairStr ← splitBy (\ c → c ≡ ';' ∨ c ≡ '&') (A.toString src)
+ | otherwise = do pairStr ← splitBy (\ c → c ≡ ';' ∨ c ≡ '&') (cs src)
let (key, value) = break (≡ '=') pairStr
return ( unescape key
, unescape $ case value of
plusToSpace '+' = ' '
plusToSpace c = c
+-- |>>> uriCIScheme "http://example.com/foo/bar"
+-- "http"
+uriCIScheme ∷ URI → CIAscii
+{-# INLINE uriCIScheme #-}
+uriCIScheme = convertUnsafe ∘ fst ∘ fromJust ∘ back ∘ uriScheme
+
-- |>>> uriHost "http://example.com/foo/bar"
-- "example.com"
uriHost ∷ URI → Host
{-# INLINE uriHost #-}
-uriHost = CI.mk ∘ T.pack ∘ uriRegName ∘ fromJust ∘ uriAuthority
+uriHost = CI.mk ∘ cs ∘ uriRegName ∘ fromJust ∘ uriAuthority
-- |>>> uriPathSegments "http://example.com/foo/bar"
-- ["foo", "bar"]
-- |Get the modification time of a given file.
getLastModified ∷ FilePath → IO UTCTime
-getLastModified = (clockTimeToUTC <$>) ∘ getModificationTime
- where
- clockTimeToUTC ∷ ClockTime → UTCTime
- clockTimeToUTC (TOD sec picoSec)
- = posixSecondsToUTCTime ∘ fromRational
- $ sec % 1 + picoSec % (1000 ⋅ 1000 ⋅ 1000 ⋅ 1000)
+getLastModified = (cs <$>) ∘ getModificationTime