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/Utils.hs | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'src/Parser/Utils.hs')

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