Add reader environment

app/EpubParser.hs

@@ -5,12 +5,30 @@ import qualified Data.ByteString.Lazy as BL
 import qualified Data.ByteString.Lazy.Char8 as C8
 import Text.HTML.TagSoup (parseTags, fromAttrib, (~==), Tag, innerText, sections)
 import System.FilePath (takeDirectory)
+import Control.Monad.Reader
+import qualified Data.Text as T
+import Data.Maybe (fromMaybe)
 
-data BookInfo = BookInfo 
-  { title  :: String
-  , author :: String
+type EpubAction a = ReaderT EpubEnv IO a
+
+data EpubEnv = EpubEnv
+  { title      :: T.Text
+  , author     :: T.Text
+  , rootPrefix :: FilePath
   } deriving (Show)
 
+myEnv :: Archive -> [Tag String] -> EpubEnv
+myEnv archive tags =
+  let
+    opfPath = fromMaybe "" (getOpfPath archive)
+    prefix  = getRootPrefix opfPath
+
+  in
+    EpubEnv 
+      { title      = getTagText "dc:title" tags
+      , author     = getTagText "dc:creator" tags
+      , rootPrefix = prefix 
+      }
 
 openEpub :: FilePath -> IO (Either String Archive)
 openEpub path = toArchive <$> BL.readFile path >>= \archive -> maybe (pure $ Left "Error reading file") (\entry -> if fromEntry entry == C8.pack "application/epub+zip" then pure (Right archive) else pure (Left "Wrong mimetype")) (findEntryByPath "mimetype" archive)
@@ -39,8 +57,9 @@ getRootPrefix path =
 tagify :: String -> String
 tagify item = "<" ++ item ++ ">"
 
-getTagText :: String -> [Tag String] -> String
+getTagText :: String -> [Tag String] -> T.Text
 getTagText tagName tags =
   case sections (~== tagify tagName) tags of
-    (x:_) -> innerText x
-    []    -> ""
+    (x:_) -> T.pack (innerText x)
+    []    -> T.empty
+

svitak.cabal

@@ -63,7 +63,9 @@ executable svitak
                       bytestring >= 0.12.2.0,
                       epub-metadata >=5.4,
                       filepath >= 1.5.4.0,
+                      mtl >= 2.3.2,
                       tagsoup,
+                      text >= 2.1.2,
                       zip-archive >= 0.4.3.2
 
     hs-source-dirs:   app