From 69e69017b75d1cdaa1fd2aef2818de5111b29735 Mon Sep 17 00:00:00 2001
From: Joris
Date: Thu, 14 Jul 2016 11:57:12 +0000
Subject: Update code and fix parsers

---
 src/Parser/Utils.hs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

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

diff --git a/src/Parser/Utils.hs b/src/Parser/Utils.hs
index d72a1ce..98694bb 100644
--- a/src/Parser/Utils.hs
+++ b/src/Parser/Utils.hs
@@ -5,14 +5,16 @@ module Parser.Utils
   , getTagAttributes
   , getTagAttribute
   , getTagTextAfter
+  , hasClass
   ) where
 
 import Data.List (find, findIndex)
-import Data.Maybe (listToMaybe, catMaybes)
+import Data.Maybe (listToMaybe, catMaybes, isJust)
 import Data.Text (Text)
 import qualified Data.Text as T
 
 import Text.HTML.TagSoup
+import Text.HTML.TagSoup.Match (tagOpen)
 
 getTagsBefore :: String -> [Tag Text] -> [Tag Text]
 getTagsBefore selector = takeWhile (~/= selector)
@@ -43,7 +45,15 @@ getTagTextAfter selector tags =
 maybeTagAttribute :: Text -> Tag Text -> Maybe Text
 maybeTagAttribute name (TagOpen _ xs) =
   fmap snd . find (\(x, _) -> x == name) $ xs
-maybeTagAttribute attribute _ = Nothing
+maybeTagAttribute _ _ = Nothing
 
 safeGetAt :: Int -> [a] -> Maybe a
 safeGetAt index = listToMaybe . drop index
+
+hasClass :: Text -> Text -> Tag Text -> Bool
+hasClass selector className =
+  tagOpen ((==) selector) (isJust . find matchClass)
+  where matchClass (name, values) =
+          (  name == (T.pack "class")
+          && (isJust . find ((==) className) . T.words $ values)
+          )
-- 
cgit v1.2.3