From a14b3c3d4447cbeb77e4146fba7e0d31393e625e Mon Sep 17 00:00:00 2001
From: Joris Guyonvarch
Date: Sat, 18 Apr 2015 11:03:56 +0200
Subject: Parsing images in detail and show URLs in notifications

---
 src/Parser/Detail.hs | 22 +++++++++++++---------
 src/Parser/Utils.hs  | 14 +++++++++++---
 2 files changed, 24 insertions(+), 12 deletions(-)

(limited to 'src/Parser')

diff --git a/src/Parser/Detail.hs b/src/Parser/Detail.hs
index 4144964..3a91ac2 100644
--- a/src/Parser/Detail.hs
+++ b/src/Parser/Detail.hs
@@ -14,12 +14,16 @@ import Parser.Utils
 parseDetail :: Text -> Detail
 parseDetail page =
   let tags = parseTags page
-      descriptionTags = getTagsBetween "<div itemprop=description>" "</div>" tags
-      description =
-        if null descriptionTags
-          then
-            Nothing
-          else
-            let replaceBr = map (\tag -> if tag ~== "<br>" then TagText (T.pack "\n") else tag)
-            in  Just . T.strip . renderTags . replaceBr $ descriptionTags
-  in  Detail { description = description }
+      description = parseDescription tags
+      images = getTagAttributes "<meta itemprop=image>" (T.pack "content") tags
+  in  Detail { description = description, images = images }
+
+parseDescription :: [Tag Text] -> Maybe Text
+parseDescription tags =
+  let descriptionTags = getTagsBetween "<div itemprop=description>" "</div>" tags
+  in  if null descriptionTags
+        then
+          Nothing
+        else
+          let replaceBr = map (\tag -> if tag ~== "<br>" then TagText (T.pack "\n") else tag)
+          in  Just . T.strip . renderTags . replaceBr $ descriptionTags
diff --git a/src/Parser/Utils.hs b/src/Parser/Utils.hs
index 16fe3d2..c03ab03 100644
--- a/src/Parser/Utils.hs
+++ b/src/Parser/Utils.hs
@@ -1,11 +1,12 @@
 module Parser.Utils
   ( getTagsBetween
+  , getTagAttributes
   , getTagAttribute
   , getTagTextAfter
   ) where
 
 import Data.List (find, findIndex)
-import Data.Maybe (listToMaybe)
+import Data.Maybe (listToMaybe, catMaybes)
 import Data.Text (Text)
 import qualified Data.Text as T
 
@@ -17,9 +18,16 @@ getTagsBetween beginSelector endSelector =
   . drop 1
   . dropWhile (~/= beginSelector)
 
+getTagAttributes :: String -> Text -> [Tag Text] -> [Text]
+getTagAttributes selector attribute =
+  catMaybes
+  . fmap (maybeTagAttribute attribute)
+  . filter (~== selector)
+
 getTagAttribute :: String -> Text -> [Tag Text] -> Maybe Text
-getTagAttribute selector attribute tags =
-  find (~== selector) tags >>= maybeTagAttribute attribute
+getTagAttribute selector attribute =
+  listToMaybe
+  . getTagAttributes selector attribute
 
 getTagTextAfter :: String -> [Tag Text] -> Maybe Text
 getTagTextAfter selector tags =
-- 
cgit v1.2.3