forked from filipg/twilight-library
65 lines
2.5 KiB
Haskell
65 lines
2.5 KiB
Haskell
|
|
{-# LANGUAGE Arrows, NoMonomorphismRestriction #-}
|
|
import ShadowLibrary.Core
|
|
|
|
import Text.XML.HXT.Core
|
|
import Text.XML.HXT.XPath
|
|
|
|
import Data.List
|
|
import Data.List.Utils (replace)
|
|
|
|
import Text.Regex.Posix
|
|
import Text.Printf
|
|
|
|
|
|
getLinkAndText xpathCondition = proc doc -> do
|
|
xpathTrees <- getXPathTrees xpathCondition -< doc
|
|
name <- getElemName -< xpathTrees
|
|
txt <- (listA (deep isText >>> getText) >>> arr (intercalate " ")) -< xpathTrees
|
|
href <- (getXPathTrees "//a" >>> getAttrValue "href") -< xpathTrees
|
|
returnA -< href, txt
|
|
|
|
|
|
extractNestedLinksWithText xpathCondition = proc url -> do
|
|
doc <- downloadDocument -< url
|
|
(link, text) <- getLinkAndText xpathCondition -< doc
|
|
uriFixed <- expandURIFixed -< (link, url)
|
|
returnA -< (uriFixed, text)
|
|
|
|
|
|
extractRecords = proc x -> do
|
|
(a, b) <- extractLinksWithText "//aside[@class='widget widget_maxmegamenu']//a[@class='mega-menu-link']" -< x -- pary adres-tytuł podstrony
|
|
(a', b') <- (extractLinksWithText "//aside[@class='widget widget_maxmegamenu']//a[@class='mega-menu-link']") -< a -- pobieramy podstronę i kolejne podstrony z menu
|
|
a'' <- (extractNestedLinksWithText "//big[a[contains(@href,'.pdf')][img]]") -< a' -- pobieramy stronę z adresu URL i wyciągamy linki z tej strony pasujące do wyrażenia XPathowego
|
|
returnA -< ((a'', b'), b)
|
|
-- ostatecznie wyjdą krotki (((adres URL, tytuł nr-u), tytuł podstrony 2), tytuł podstrony 1)
|
|
|
|
-- ... a tutaj te krotki przerabiamy do docelowej struktury ShadowItem
|
|
toShadowItem :: (((String, String), String), String) -> ShadowItem
|
|
toShadowItem (((url, releaseTitle), collectionTitle), categoryTitle) =
|
|
(defaultShadowItem url title) {
|
|
originalDate = Just date,
|
|
itype = "periodical",
|
|
format = Just "pdf",
|
|
finalUrl = url
|
|
}
|
|
where title = categoryTitle ++ (" " ++ collectionTitle)
|
|
date = getDate $ releaseTitle
|
|
|
|
|
|
getDate yearlyTitle =
|
|
case yearlyTitle =~~ "/.+(19[0-9][0-9]|20[0-9][0-9])/" :: Maybe [[String]] of
|
|
Just [[_, year]] -> year
|
|
-- otherwise -> error $ "unexpected yearlyTitle: " ++ yearlyTitle
|
|
otherwise -> yearlyTitle
|
|
|
|
|
|
main = do
|
|
let start = "http://zborbielawa.pl/archiwum/"
|
|
let shadowLibrary = ShadowLibrary {logoUrl=Nothing,
|
|
lname="Zbór Bielawa",
|
|
abbrev="ZboBiel",
|
|
lLevel=0,
|
|
webpage=start}
|
|
extractItemsStartingFromUrl shadowLibrary start (extractRecords >>> arr toShadowItem)
|