preview backend now fully works!

[Rakka.git] / Rakka / Utils.hs

diff --git a/Rakka/Utils.hs b/Rakka/Utils.hs
index 9eb667c115971c10d816f0176e8d9bda2b4a1632..0fddc6d64127baf25a1e5adbe19b58b5e758d039 100644 (file)
--- a/Rakka/Utils.hs
+++ b/Rakka/Utils.hs
@@ -4,11 +4,17 @@ module Rakka.Utils
     , maybeA
     , deleteIfEmpty
     , chomp
+    , guessMIMEType
     )
     where
 
 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           System.IO.Unsafe
 
 
 yesOrNo :: Bool -> String
@@ -41,3 +47,13 @@ deleteIfEmpty
 
 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