From aafc45eb5eed3839a5210a7d48928d975df6a296 Mon Sep 17 00:00:00 2001 From: Joris Guyonvarch Date: Sat, 11 Apr 2015 20:38:08 +0200 Subject: Handle a configuration file to save the url --- src/Parser/Utils.hs | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'src/Parser/Utils.hs') diff --git a/src/Parser/Utils.hs b/src/Parser/Utils.hs index 8527777..16fe3d2 100644 --- a/src/Parser/Utils.hs +++ b/src/Parser/Utils.hs @@ -6,27 +6,28 @@ module Parser.Utils import Data.List (find, findIndex) import Data.Maybe (listToMaybe) +import Data.Text (Text) import qualified Data.Text as T import Text.HTML.TagSoup -getTagsBetween :: String -> String -> [Tag T.Text] -> [Tag T.Text] +getTagsBetween :: String -> String -> [Tag Text] -> [Tag Text] getTagsBetween beginSelector endSelector = takeWhile (~/= endSelector) . drop 1 . dropWhile (~/= beginSelector) -getTagAttribute :: String -> T.Text -> [Tag T.Text] -> Maybe T.Text +getTagAttribute :: String -> Text -> [Tag Text] -> Maybe Text getTagAttribute selector attribute tags = find (~== selector) tags >>= maybeTagAttribute attribute -getTagTextAfter :: String -> [Tag T.Text] -> Maybe T.Text +getTagTextAfter :: String -> [Tag Text] -> Maybe Text getTagTextAfter selector tags = case findIndex (~== selector) tags of Just index -> fmap T.strip $ safeGetAt (index + 1) tags >>= maybeTagText Nothing -> Nothing -maybeTagAttribute :: T.Text -> Tag T.Text -> Maybe T.Text +maybeTagAttribute :: Text -> Tag Text -> Maybe Text maybeTagAttribute name (TagOpen _ xs) = fmap snd . find (\(x, _) -> x == name) $ xs maybeTagAttribute attribute _ = Nothing -- cgit v1.2.3