Offline: Recursively resolve @imports in css files.

1 files changed, 35 insertions, 12 deletions

diff --git a/src/Text/Pandoc/Offline.hs b/src/Text/Pandoc/Offline.hs
index 3515d1994..c0d6edef9 100644
--- a/src/Text/Pandoc/Offline.hs
+++ b/src/Text/Pandoc/Offline.hs
@@ -33,14 +33,12 @@ the HTML using data URIs.
 module Text.Pandoc.Offline ( offline ) where
 import Text.HTML.TagSoup
 import Network.URI (isAbsoluteURI, parseURI, escapeURIString)
-import Network.Browser
 import Network.HTTP
 import Data.ByteString.Base64
-import System.IO
 import qualified Data.ByteString.Char8 as B
 import Data.ByteString (ByteString)
-import Data.ByteString.UTF8 (toString, fromString)
-import System.FilePath (takeExtension, dropExtension)
+import Data.ByteString.UTF8 (toString)
+import System.FilePath (takeExtension, dropExtension, takeDirectory, (</>))
 import Data.Char (toLower, isAscii, isAlphaNum)
 import Codec.Compression.GZip as Gzip
 import qualified Data.ByteString.Lazy as L
@@ -53,8 +51,8 @@ getItem f =
 
 openURL :: String -> IO ByteString
 openURL u = getResponseBody =<< simpleHTTP (getReq u)
-  where getReq u = case parseURI u of
-                     Nothing  -> error $ "Could not parse URI: " ++ u
+  where getReq v = case parseURI v of
+                     Nothing  -> error $ "Could not parse URI: " ++ v
                      Just u'  -> mkRequest GET u'
 
 mimeTypeFor :: String -> String
@@ -189,6 +187,7 @@ mimeTypeFor s = case lookup s mimetypes of
                 (".xul", "text/xul")
                 ]
 
+isOk :: Char -> Bool
 isOk c = isAscii c && isAlphaNum c
 
 convertTag :: Tag String -> IO (Tag String)
@@ -215,22 +214,46 @@ convertTag t@(TagOpen "link" as) =
            return $ TagOpen "link" (("href",enc) : [(x,y) | (x,y) <- as, x /= "href"]) 
 convertTag t = return t
 
+cssImports :: FilePath -> ByteString -> IO ByteString
+cssImports d orig =
+  case B.breakSubstring "@import" orig of
+       (x,y) | B.null y  -> return orig
+             | otherwise -> do
+                  rest <- handleImport d (B.drop 7 y) >>= cssImports d
+                  return $ x `B.append` rest
+
+-- @import url("blah");
+-- @import url(blah);
+-- @import "blah";
+handleImport :: FilePath -> ByteString -> IO ByteString
+handleImport d x = fmap (`B.append` rest) (getItem $ d </> url)
+  where lparenOrQuote c = c == '(' || c == '"'
+        rparenOrQuote c = c == ')' || c == '"'
+        url = toString
+              $ B.takeWhile (not . rparenOrQuote)
+              $ B.dropWhile lparenOrQuote
+              $ B.dropWhile (not . lparenOrQuote) x
+        rest = B.drop 1 $ B.dropWhile (/= ';') x
+
 getRaw :: Tag String -> String -> IO (ByteString, String)
 getRaw t src = do
   let ext = map toLower $ takeExtension src
-  let (ext',decompress) = if ext == ".gz"
-                             then (takeExtension $ dropExtension src, B.concat . L.toChunks . Gzip.decompress . L.fromChunks . (:[]))
-                             else (ext, id)
+  let (ext',decomp) = if ext == ".gz"
+                         then (takeExtension $ dropExtension src, B.concat . L.toChunks . Gzip.decompress . L.fromChunks . (:[]))
+                         else (ext, id)
   let mime = case fromAttrib "type" t of
                   []  -> mimeTypeFor ext'
                   x   -> x
   raw <- getItem src
-  return (decompress raw, mime)
+  result <- if mime == "text/css"
+               then cssImports (takeDirectory src) $ decomp raw
+               else return $ decomp raw
+  return (result, mime)
 
 offline :: String -> IO String
 offline inp = do
   let tags = parseTags inp
-  out <- mapM convertTag tags
+  out' <- mapM convertTag tags
   return $ renderTagsOptions renderOptions{ optMinimize = (\t -> t == "br"
-                 || t == "img" || t == "meta" || t == "link" ) } out
+                 || t == "img" || t == "meta" || t == "link" ) } out'