module Rakka.Utils
- ( parseYesOrNo
+ ( yesOrNo
+ , trueOrFalse
+ , parseYesOrNo
, maybeA
- , defaultTo
, deleteIfEmpty
+ , chomp
+ , guessMIMEType
+ , isSafeChar
+ , mkQueryString
)
where
+import qualified Codec.Binary.UTF8.String as UTF8
import Control.Arrow
import Control.Arrow.ArrowList
+import qualified Data.ByteString.Lazy as Lazy (ByteString)
+import qualified Data.ByteString.Lazy.Char8 as L8 hiding (ByteString)
+import Magic
+import Network.HTTP.Lucu
+import Network.URI
+import System.IO.Unsafe
+
+
+yesOrNo :: Bool -> String
+yesOrNo True = "yes"
+yesOrNo False = "no"
+
+
+trueOrFalse :: Bool -> String
+trueOrFalse True = "true"
+trueOrFalse False = "false"
parseYesOrNo :: ArrowChoice a => a String Bool
(x:_) -> returnA -< Just x
-defaultTo :: ArrowChoice a => b -> a (Maybe b) b
-defaultTo def
- = proc m -> case m of
- Nothing -> returnA -< def
- Just x -> returnA -< x
-
-
deleteIfEmpty :: (ArrowList a, ArrowChoice a) => a String String
deleteIfEmpty
= proc str -> do case str of
"" -> none -< ()
- _ -> returnA -< str
\ No newline at end of file
+ _ -> returnA -< str
+
+
+chomp :: String -> String
+chomp = reverse . snd . break (/= '\n') . reverse
+
+
+guessMIMEType :: Lazy.ByteString -> MIMEType
+guessMIMEType = read . unsafePerformIO . magicString magic . L8.unpack
+ where
+ magic :: Magic
+ magic = unsafePerformIO
+ $ do m <- magicOpen [MagicMime]
+ magicLoadDefault m
+ return m
+
+
+isSafeChar :: Char -> Bool
+isSafeChar c
+ | c == '/' = True
+ | isReserved c = False
+ | c > ' ' && c <= '~' = True
+ | otherwise = False
+
+
+mkQueryString :: [(String, String)] -> String
+mkQueryString [] = ""
+mkQueryString ((k, v) : xs) = encode k ++ "=" ++ encode v ++
+ if xs == [] then
+ ""
+ else
+ ';' : mkQueryString(xs)
+ where
+ encode :: String -> String
+ encode = escapeURIString isSafeChar . UTF8.encodeString
\ No newline at end of file