{-# OPTIONS -fno-warn-incomplete-patterns #-}

--------------------------------------------------------------------
-- |
-- Module    : Text.Feed.Constructor
-- Copyright : (c) Galois, Inc. 2008,
--             (c) Sigbjorn Finne 2009-
-- License   : BSD3
--
-- Maintainer: Sigbjorn Finne <sof@forkIO.com>
-- Stability : provisional
-- Description: Module for an abstraction layer between different kinds of feeds.
--
--------------------------------------------------------------------
module Text.Feed.Constructor
  ( FeedKind(..)
  , newFeed -- :: FeedKind  -> Feed
  , feedFromRSS -- :: RSS       -> Feed
  , feedFromAtom -- :: Atom.Feed -> Feed
  , feedFromRDF -- :: RSS1.Feed -> Feed
  , feedFromXML -- :: Element   -> Feed
  , getFeedKind -- :: Feed      -> FeedKind
  , FeedSetter -- type _ a = a -> Feed -> Feed
  , addItem -- :: FeedSetter Item
  , withFeedTitle -- :: FeedSetter Text
  , withFeedHome -- :: FeedSetter URLString
  , withFeedHTML -- :: FeedSetter URLString
  , withFeedDescription -- :: FeedSetter Text
  , withFeedPubDate -- :: FeedSetter DateString
  , withFeedLastUpdate -- :: FeedSetter DateString
  , withFeedDate -- :: FeedSetter DateString
  , withFeedLogoLink -- :: FeedSetter URLString
  , withFeedLanguage -- :: FeedSetter Text
  , withFeedCategories -- :: FeedSetter [(Text, Maybe Text)]
  , withFeedGenerator -- :: FeedSetter Text
  , withFeedItems -- :: FeedSetter [Item]
  , newItem -- :: FeedKind   -> Item
  , getItemKind -- :: Item       -> FeedKind
  , atomEntryToItem -- :: Atom.Entry -> Item
  , rssItemToItem -- :: RSS.Item   -> Item
  , rdfItemToItem -- :: RSS1.Item  -> Item
  , ItemSetter -- type _ a = a -> Item -> Item
  , withItemTitle -- :: ItemSetter Text
  , withItemLink -- :: ItemSetter URLString
  , withItemPubDate -- :: ItemSetter DateString
  , withItemDate -- :: ItemSetter DateString
  , withItemAuthor -- :: ItemSetter Text
  , withItemCommentLink -- :: ItemSetter Text
  , withItemEnclosure -- :: Text -> Maybe Text -> ItemSetter Integer
  , withItemFeedLink -- :: Text -> ItemSetter Text
  , withItemId -- :: Bool   -> ItemSetter Text
  , withItemCategories -- :: ItemSetter [(Text, Maybe Text)]
  , withItemDescription -- :: ItemSetter Text
  , withItemRights -- :: ItemSetter Text
  ) where

import Prelude.Compat

import Text.Feed.Types as Feed.Types

import Text.Atom.Feed as Atom
import Text.DublinCore.Types
import Text.RSS.Syntax as RSS
import Text.RSS1.Syntax as RSS1

import Data.XML.Compat
import Data.XML.Types as XML

import Data.Char (toLower)
import Data.Maybe (fromMaybe, mapMaybe)
import Data.Text (Text, pack)

-- ToDo:
--
--  - complete set of constructors over feeds
--  - provide a unified treatment of date string reps.
--    (i.e., I know they differ across formats, but ignorant what
--    the constraints are at the moment.)
-- | Construct an empty feed document, intending to output it in
-- the 'fk' feed format.
newFeed :: FeedKind -> Feed.Types.Feed
newFeed :: FeedKind -> Feed
newFeed FeedKind
fk =
  case FeedKind
fk of
    FeedKind
AtomKind ->
      Feed -> Feed
AtomFeed
        (Text -> TextContent -> Text -> Feed
Atom.nullFeed
           Text
"feed-id-not-filled-in"
           (Text -> TextContent
TextString Text
"dummy-title")
           Text
"dummy-and-bogus-update-date")
    RSSKind Maybe Text
mbV ->
      let def :: RSS
def = Text -> Text -> RSS
RSS.nullRSS Text
"dummy-title" Text
"default-channel-url"
       in RSS -> Feed
RSSFeed forall a b. (a -> b) -> a -> b
$ forall b a. b -> (a -> b) -> Maybe a -> b
maybe RSS
def (\Text
v -> RSS
def {rssVersion :: Text
RSS.rssVersion = Text
v}) Maybe Text
mbV
    RDFKind Maybe Text
mbV ->
      let def :: Feed
def = Text -> Text -> Feed
RSS1.nullFeed Text
"default-channel-url" Text
"dummy-title"
       in Feed -> Feed
RSS1Feed forall a b. (a -> b) -> a -> b
$ forall b a. b -> (a -> b) -> Maybe a -> b
maybe Feed
def (\Text
v -> Feed
def {feedVersion :: Text
RSS1.feedVersion = Text
v}) Maybe Text
mbV

feedFromRSS :: RSS.RSS -> Feed.Types.Feed
feedFromRSS :: RSS -> Feed
feedFromRSS = RSS -> Feed
RSSFeed

feedFromAtom :: Atom.Feed -> Feed.Types.Feed
feedFromAtom :: Feed -> Feed
feedFromAtom = Feed -> Feed
AtomFeed

feedFromRDF :: RSS1.Feed -> Feed.Types.Feed
feedFromRDF :: Feed -> Feed
feedFromRDF = Feed -> Feed
RSS1Feed

feedFromXML :: XML.Element -> Feed.Types.Feed
feedFromXML :: Element -> Feed
feedFromXML = Element -> Feed
XMLFeed

getFeedKind :: Feed.Types.Feed -> FeedKind
getFeedKind :: Feed -> FeedKind
getFeedKind Feed
f =
  case Feed
f of
    Feed.Types.AtomFeed {} -> FeedKind
AtomKind
    Feed.Types.RSSFeed RSS
r ->
      Maybe Text -> FeedKind
RSSKind
        (case RSS -> Text
RSS.rssVersion RSS
r of
           Text
"2.0" -> forall a. Maybe a
Nothing
           Text
v -> forall a. a -> Maybe a
Just Text
v)
    Feed.Types.RSS1Feed Feed
r ->
      Maybe Text -> FeedKind
RDFKind
        (case Feed -> Text
RSS1.feedVersion Feed
r of
           Text
"1.0" -> forall a. Maybe a
Nothing
           Text
v -> forall a. a -> Maybe a
Just Text
v)
    Feed.Types.XMLFeed {} -> Maybe Text -> FeedKind
RSSKind (forall a. a -> Maybe a
Just Text
"2.0") -- for now, just a hunch..

addItem :: Feed.Types.Item -> Feed.Types.Feed -> Feed.Types.Feed
addItem :: Item -> Feed -> Feed
addItem Item
it Feed
f =
  case (Item
it, Feed
f) of
    (Feed.Types.AtomItem Entry
e, Feed.Types.AtomFeed Feed
fe) ->
      Feed -> Feed
Feed.Types.AtomFeed Feed
fe {feedEntries :: [Entry]
Atom.feedEntries = Entry
e forall a. a -> [a] -> [a]
: Feed -> [Entry]
Atom.feedEntries Feed
fe}
    (Feed.Types.RSSItem RSSItem
e, Feed.Types.RSSFeed RSS
r) ->
      RSS -> Feed
Feed.Types.RSSFeed
        RSS
r {rssChannel :: RSSChannel
RSS.rssChannel = (RSS -> RSSChannel
RSS.rssChannel RSS
r) {rssItems :: [RSSItem]
RSS.rssItems = RSSItem
e forall a. a -> [a] -> [a]
: RSSChannel -> [RSSItem]
RSS.rssItems (RSS -> RSSChannel
RSS.rssChannel RSS
r)}}
    (Feed.Types.RSS1Item Item
e, Feed.Types.RSS1Feed Feed
r)
         -- note: do not update the channel item URIs at this point;
         -- will delay doing so until serialization.
     -> Feed -> Feed
Feed.Types.RSS1Feed Feed
r {feedItems :: [Item]
RSS1.feedItems = Item
e forall a. a -> [a] -> [a]
: Feed -> [Item]
RSS1.feedItems Feed
r}
    (Item, Feed)
_ ->
      forall a. HasCallStack => [Char] -> a
error [Char]
"addItem: currently unable to automatically convert items from one feed type to another"

withFeedItems :: FeedSetter [Feed.Types.Item]
withFeedItems :: FeedSetter [Item]
withFeedItems [Item]
is Feed
fe =
  forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr
    Item -> Feed -> Feed
addItem
    (case Feed
fe of
       Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedEntries :: [Entry]
Atom.feedEntries = []}
       Feed.Types.RSSFeed RSS
f -> RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssItems :: [RSSItem]
rssItems = []}}
       Feed.Types.RSS1Feed Feed
f -> Feed -> Feed
Feed.Types.RSS1Feed Feed
f {feedItems :: [Item]
feedItems = []})
    [Item]
is

newItem :: FeedKind -> Feed.Types.Item
newItem :: FeedKind -> Item
newItem FeedKind
fk =
  case FeedKind
fk of
    FeedKind
AtomKind ->
      Entry -> Item
Feed.Types.AtomItem forall a b. (a -> b) -> a -> b
$
      Text -> TextContent -> Text -> Entry
Atom.nullEntry
        Text
"entry-id-not-filled-in"
        (Text -> TextContent
TextString Text
"dummy-entry-title")
        Text
"dummy-and-bogus-entry-update-date"
    RSSKind {} -> RSSItem -> Item
Feed.Types.RSSItem forall a b. (a -> b) -> a -> b
$ Text -> RSSItem
RSS.nullItem Text
"dummy-rss-item-title"
    RDFKind {} ->
      Item -> Item
Feed.Types.RSS1Item forall a b. (a -> b) -> a -> b
$ Text -> Text -> Text -> Item
RSS1.nullItem Text
"dummy-item-uri" Text
"dummy-item-title" Text
"dummy-item-link"

getItemKind :: Feed.Types.Item -> FeedKind
getItemKind :: Item -> FeedKind
getItemKind Item
f =
  case Item
f of
    Feed.Types.AtomItem {} -> FeedKind
AtomKind
    Feed.Types.RSSItem {} -> Maybe Text -> FeedKind
RSSKind (forall a. a -> Maybe a
Just Text
"2.0") -- good guess..
    Feed.Types.RSS1Item {} -> Maybe Text -> FeedKind
RDFKind (forall a. a -> Maybe a
Just Text
"1.0")
    Feed.Types.XMLItem {} -> Maybe Text -> FeedKind
RSSKind (forall a. a -> Maybe a
Just Text
"2.0")

type FeedSetter a = a -> Feed.Types.Feed -> Feed.Types.Feed

withFeedTitle :: FeedSetter Text
withFeedTitle :: FeedSetter Text
withFeedTitle Text
tit Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedTitle :: TextContent
feedTitle = Text -> TextContent
TextString Text
tit}
    Feed.Types.RSSFeed RSS
f -> RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssTitle :: Text
rssTitle = Text
tit}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed Feed
f {feedChannel :: Channel
feedChannel = (Feed -> Channel
feedChannel Feed
f) {channelTitle :: Text
channelTitle = Text
tit}}
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"title"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"title" Text
tit)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f

withFeedHome :: FeedSetter URLString
withFeedHome :: FeedSetter Text
withFeedHome Text
url Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedLinks :: [Link]
feedLinks = Link
newSelf forall a. a -> [a] -> [a]
: Feed -> [Link]
Atom.feedLinks Feed
f}
      -- ToDo: fix, the <link> element is for the HTML home of the channel, not the
      -- location of the feed itself. Struggling to find if there is a common way
      -- to represent this outside of RSS 2.0 standard elements..
    Feed.Types.RSSFeed RSS
f -> RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssLink :: Text
rssLink = Text
url}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed Feed
f {feedChannel :: Channel
feedChannel = (Feed -> Channel
feedChannel Feed
f) {channelURI :: Text
channelURI = Text
url}}
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"link"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"link" Text
url)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f
  where
    newSelf :: Link
newSelf = (Text -> Link
nullLink Text
url) {linkRel :: Maybe (Either Text Text)
linkRel = forall a. a -> Maybe a
Just (forall a b. a -> Either a b
Left Text
"self"), linkType :: Maybe Text
linkType = forall a. a -> Maybe a
Just Text
"application/atom+xml"}

-- | 'withFeedHTML' sets the URL where an HTML version of the
-- feed is published.
withFeedHTML :: FeedSetter URLString
withFeedHTML :: FeedSetter Text
withFeedHTML Text
url Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedLinks :: [Link]
feedLinks = Link
newAlt forall a. a -> [a] -> [a]
: Feed -> [Link]
Atom.feedLinks Feed
f}
    Feed.Types.RSSFeed RSS
f -> RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssLink :: Text
rssLink = Text
url}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed Feed
f {feedChannel :: Channel
feedChannel = (Feed -> Channel
feedChannel Feed
f) {channelLink :: Text
channelLink = Text
url}}
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"link"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"link" Text
url)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f
  where
    newAlt :: Link
newAlt = (Text -> Link
nullLink Text
url) {linkRel :: Maybe (Either Text Text)
linkRel = forall a. a -> Maybe a
Just (forall a b. a -> Either a b
Left Text
"alternate"), linkType :: Maybe Text
linkType = forall a. a -> Maybe a
Just Text
"text/html"}

-- | 'withFeedHTML' sets the URL where an HTML version of the
-- feed is published.
withFeedDescription :: FeedSetter Text
withFeedDescription :: FeedSetter Text
withFeedDescription Text
desc Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedSubtitle :: Maybe TextContent
feedSubtitle = forall a. a -> Maybe a
Just (Text -> TextContent
TextString Text
desc)}
    Feed.Types.RSSFeed RSS
f ->
      RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssDescription :: Text
rssDescription = Text
desc}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed Feed
f {feedChannel :: Channel
feedChannel = (Feed -> Channel
feedChannel Feed
f) {channelDesc :: Text
channelDesc = Text
desc}}
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"description"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"description" Text
desc)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f

withFeedPubDate :: FeedSetter Text
withFeedPubDate :: FeedSetter Text
withFeedPubDate Text
dateStr Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedUpdated :: Text
feedUpdated = Text
dateStr}
    Feed.Types.RSSFeed RSS
f ->
      RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssPubDate :: Maybe Text
rssPubDate = forall a. a -> Maybe a
Just Text
dateStr}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed forall a b. (a -> b) -> a -> b
$
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isDate forall a b. (a -> b) -> a -> b
$ Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f) of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f) {channelDC :: [DCItem]
RSS1.channelDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
dateStr} forall a. a -> [a] -> [a]
: [DCItem]
bs}
            }
        ([DCItem]
_, []) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f)
                  { channelDC :: [DCItem]
RSS1.channelDC =
                      DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Date, dcText :: Text
dcText = Text
dateStr} forall a. a -> [a] -> [a]
:
                      Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f)
                  }
            }
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"pubDate"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"pubDate" Text
dateStr)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f
  where
    isDate :: DCItem -> Bool
isDate DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Date

withFeedLastUpdate :: FeedSetter DateString
withFeedLastUpdate :: FeedSetter Text
withFeedLastUpdate Text
dateStr Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedUpdated :: Text
feedUpdated = Text
dateStr}
    Feed.Types.RSSFeed RSS
f ->
      RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssLastUpdate :: Maybe Text
rssLastUpdate = forall a. a -> Maybe a
Just Text
dateStr}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed forall a b. (a -> b) -> a -> b
$
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isDate forall a b. (a -> b) -> a -> b
$ Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f) of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f) {channelDC :: [DCItem]
RSS1.channelDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
dateStr} forall a. a -> [a] -> [a]
: [DCItem]
bs}
            }
        ([DCItem]
_, []) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f)
                  { channelDC :: [DCItem]
RSS1.channelDC =
                      DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Date, dcText :: Text
dcText = Text
dateStr} forall a. a -> [a] -> [a]
:
                      Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f)
                  }
            }
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"lastUpdate"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"lastUpdate" Text
dateStr)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f
  where
    isDate :: DCItem -> Bool
isDate DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Date

-- | 'withFeedDate dt' is the composition of 'withFeedPubDate'
-- and 'withFeedLastUpdate', setting both publication date and
-- last update date to 'dt'. Notice that RSS2.0 is the only format
-- supporting both pub and last-update.
withFeedDate :: FeedSetter DateString
withFeedDate :: FeedSetter Text
withFeedDate Text
dt Feed
f = FeedSetter Text
withFeedPubDate Text
dt (FeedSetter Text
withFeedLastUpdate Text
dt Feed
f)

withFeedLogoLink :: URLString -> FeedSetter URLString
withFeedLogoLink :: Text -> FeedSetter Text
withFeedLogoLink Text
imgURL Text
lnk Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f ->
      Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedLogo :: Maybe Text
feedLogo = forall a. a -> Maybe a
Just Text
imgURL, feedLinks :: [Link]
feedLinks = Link
newSelf forall a. a -> [a] -> [a]
: Feed -> [Link]
Atom.feedLinks Feed
f}
    Feed.Types.RSSFeed RSS
f ->
      RSS -> Feed
Feed.Types.RSSFeed
        RSS
f
          { rssChannel :: RSSChannel
rssChannel =
              (RSS -> RSSChannel
rssChannel RSS
f) {rssImage :: Maybe RSSImage
rssImage = forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ Text -> Text -> Text -> RSSImage
RSS.nullImage Text
imgURL (RSSChannel -> Text
rssTitle (RSS -> RSSChannel
rssChannel RSS
f)) Text
lnk}
          }
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed forall a b. (a -> b) -> a -> b
$
      Feed
f
        { feedImage :: Maybe Image
feedImage = forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ Text -> Text -> Text -> Image
RSS1.nullImage Text
imgURL (Channel -> Text
RSS1.channelTitle (Feed -> Channel
RSS1.feedChannel Feed
f)) Text
lnk
        , feedChannel :: Channel
feedChannel = (Feed -> Channel
feedChannel Feed
f) {channelImageURI :: Maybe Text
channelImageURI = forall a. a -> Maybe a
Just Text
imgURL}
        }
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"image"
                            then forall a. a -> Maybe a
Just
                                   (forall t. ToNode t => Name -> t -> Element
unode
                                      Name
"image"
                                      [forall t. ToNode t => Name -> t -> Element
unode Name
"url" Text
imgURL, forall t. ToNode t => Name -> t -> Element
unode Name
"title" Text
title, forall t. ToNode t => Name -> t -> Element
unode Name
"link" Text
lnk])
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f
      where title :: Text
title =
              case forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (Name -> Element -> Maybe Element
findChild Name
"title") (Name -> Element -> Maybe Element
findChild Name
"channel" Element
f) of
                Just (Just Element
e1) -> Element -> Text
strContent Element
e1
                Maybe (Maybe Element)
_ -> Text
"feed_title" -- shouldn't happen..
  where
    newSelf :: Link
newSelf = (Text -> Link
nullLink Text
lnk) {linkRel :: Maybe (Either Text Text)
linkRel = forall a. a -> Maybe a
Just (forall a b. a -> Either a b
Left Text
"self"), linkType :: Maybe Text
linkType = forall a. a -> Maybe a
Just Text
"application/atom+xml"}

withFeedLanguage :: FeedSetter Text
withFeedLanguage :: FeedSetter Text
withFeedLanguage Text
lang Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f -> Feed -> Feed
Feed.Types.AtomFeed Feed
f {feedAttrs :: [Attr]
Atom.feedAttrs = Attr
langAttr forall a. a -> [a] -> [a]
: Feed -> [Attr]
Atom.feedAttrs Feed
f}
      where langAttr :: Attr
langAttr = (Name
name, [Text -> Content
ContentText Text
lang])
            name :: Name
name = Name {nameLocalName :: Text
nameLocalName = Text
"lang", nameNamespace :: Maybe Text
nameNamespace = forall a. Maybe a
Nothing, namePrefix :: Maybe Text
namePrefix = forall a. a -> Maybe a
Just Text
"xml"}
    Feed.Types.RSSFeed RSS
f ->
      RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssLanguage :: Maybe Text
rssLanguage = forall a. a -> Maybe a
Just Text
lang}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed forall a b. (a -> b) -> a -> b
$
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isLang forall a b. (a -> b) -> a -> b
$ Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f) of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f) {channelDC :: [DCItem]
RSS1.channelDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
lang} forall a. a -> [a] -> [a]
: [DCItem]
bs}
            }
        ([DCItem]
_, []) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f)
                  { channelDC :: [DCItem]
RSS1.channelDC =
                      DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Language, dcText :: Text
dcText = Text
lang} forall a. a -> [a] -> [a]
:
                      Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f)
                  }
            }
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"language"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"language" Text
lang)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f
  where
    isLang :: DCItem -> Bool
isLang DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Language

withFeedCategories :: FeedSetter [(Text, Maybe Text)]
withFeedCategories :: FeedSetter [(Text, Maybe Text)]
withFeedCategories [(Text, Maybe Text)]
cats Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f ->
      Feed -> Feed
Feed.Types.AtomFeed
        Feed
f
          { feedCategories :: [Category]
Atom.feedCategories =
              forall a b. (a -> b) -> [a] -> [b]
map (\(Text
t, Maybe Text
mb) -> (Text -> Category
Atom.newCategory Text
t) {catScheme :: Maybe Text
Atom.catScheme = Maybe Text
mb}) [(Text, Maybe Text)]
cats forall a. [a] -> [a] -> [a]
++ Feed -> [Category]
feedCategories Feed
f
          }
    Feed.Types.RSSFeed RSS
f ->
      RSS -> Feed
Feed.Types.RSSFeed
        RSS
f
          { rssChannel :: RSSChannel
rssChannel =
              (RSS -> RSSChannel
rssChannel RSS
f)
                { rssCategories :: [RSSCategory]
RSS.rssCategories =
                    forall a b. (a -> b) -> [a] -> [b]
map (\(Text
t, Maybe Text
mb) -> (Text -> RSSCategory
RSS.newCategory Text
t) {rssCategoryDomain :: Maybe Text
RSS.rssCategoryDomain = Maybe Text
mb}) [(Text, Maybe Text)]
cats forall a. [a] -> [a] -> [a]
++
                    RSSChannel -> [RSSCategory]
RSS.rssCategories (RSS -> RSSChannel
rssChannel RSS
f)
                }
          }
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed
        Feed
f
          { feedChannel :: Channel
feedChannel =
              (Feed -> Channel
feedChannel Feed
f)
                { channelDC :: [DCItem]
RSS1.channelDC =
                    forall a b. (a -> b) -> [a] -> [b]
map (\(Text
t, Maybe Text
_) -> DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Subject, dcText :: Text
dcText = Text
t}) [(Text, Maybe Text)]
cats forall a. [a] -> [a] -> [a]
++
                    Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
feedChannel Feed
f)
                }
          }
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    (forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr
                       (\(Text
t, Maybe Text
mb) Element
acc ->
                          Element -> Element -> Element
addChild
                            (forall t. ToNode t => Name -> t -> Element
unode
                               Name
"category"
                               (forall b a. b -> (a -> b) -> Maybe a -> b
maybe (forall a. a -> [a] -> [a]
: []) (\Text
v Attr
x -> [Text -> Text -> Attr
mkAttr Text
"domain" Text
v, Attr
x]) Maybe Text
mb (Text -> Text -> Attr
mkAttr Text
"term" Text
t)))
                            Element
acc)
                       Element
e
                       [(Text, Maybe Text)]
cats)
             else forall a. Maybe a
Nothing)
        Element
f

withFeedGenerator :: FeedSetter (Text, Maybe URLString)
withFeedGenerator :: FeedSetter (Text, Maybe Text)
withFeedGenerator (Text
gen, Maybe Text
mbURI) Feed
fe =
  case Feed
fe of
    Feed.Types.AtomFeed Feed
f ->
      Feed -> Feed
Feed.Types.AtomFeed forall a b. (a -> b) -> a -> b
$
      Feed
f {feedGenerator :: Maybe Generator
Atom.feedGenerator = forall a. a -> Maybe a
Just ((Text -> Generator
Atom.nullGenerator Text
gen) {genURI :: Maybe Text
Atom.genURI = Maybe Text
mbURI})}
    Feed.Types.RSSFeed RSS
f ->
      RSS -> Feed
Feed.Types.RSSFeed RSS
f {rssChannel :: RSSChannel
rssChannel = (RSS -> RSSChannel
rssChannel RSS
f) {rssGenerator :: Maybe Text
rssGenerator = forall a. a -> Maybe a
Just Text
gen}}
    Feed.Types.RSS1Feed Feed
f ->
      Feed -> Feed
Feed.Types.RSS1Feed forall a b. (a -> b) -> a -> b
$
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isSource forall a b. (a -> b) -> a -> b
$ Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f) of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f) {channelDC :: [DCItem]
RSS1.channelDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
gen} forall a. a -> [a] -> [a]
: [DCItem]
bs}
            }
        ([DCItem]
_, []) ->
          Feed
f
            { feedChannel :: Channel
RSS1.feedChannel =
                (Feed -> Channel
RSS1.feedChannel Feed
f)
                  { channelDC :: [DCItem]
RSS1.channelDC =
                      DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Source, dcText :: Text
dcText = Text
gen} forall a. a -> [a] -> [a]
: Channel -> [DCItem]
RSS1.channelDC (Feed -> Channel
RSS1.feedChannel Feed
f)
                  }
            }
    Feed.Types.XMLFeed Element
f ->
      Element -> Feed
Feed.Types.XMLFeed forall a b. (a -> b) -> a -> b
$
      (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
        (\Element
e ->
           if Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
== Name
"channel"
             then forall a. a -> Maybe a
Just
                    ((Element -> Maybe Element) -> Element -> Element
mapMaybeChildren
                       (\Element
e2 ->
                          if Element -> Name
elementName Element
e2 forall a. Eq a => a -> a -> Bool
== Name
"generator"
                            then forall a. a -> Maybe a
Just (forall t. ToNode t => Name -> t -> Element
unode Name
"generator" Text
gen)
                            else forall a. Maybe a
Nothing)
                       Element
e)
             else forall a. Maybe a
Nothing)
        Element
f
  where
    isSource :: DCItem -> Bool
isSource DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Source

-- Item constructors (all the way to the end):
atomEntryToItem :: Atom.Entry -> Feed.Types.Item
atomEntryToItem :: Entry -> Item
atomEntryToItem = Entry -> Item
Feed.Types.AtomItem

rssItemToItem :: RSS.RSSItem -> Feed.Types.Item
rssItemToItem :: RSSItem -> Item
rssItemToItem = RSSItem -> Item
Feed.Types.RSSItem

rdfItemToItem :: RSS1.Item -> Feed.Types.Item
rdfItemToItem :: Item -> Item
rdfItemToItem = Item -> Item
Feed.Types.RSS1Item

type ItemSetter a = a -> Feed.Types.Item -> Feed.Types.Item

-- | 'withItemPubDate dt' associates the creation\/ publication date 'dt'
-- with a feed item.
withItemPubDate :: ItemSetter DateString
withItemPubDate :: ItemSetter Text
withItemPubDate Text
dt Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e -> Entry -> Item
Feed.Types.AtomItem Entry
e {entryUpdated :: Text
Atom.entryUpdated = Text
dt}
    Feed.Types.RSSItem RSSItem
i -> RSSItem -> Item
Feed.Types.RSSItem RSSItem
i {rssItemPubDate :: Maybe Text
RSS.rssItemPubDate = forall a. a -> Maybe a
Just Text
dt}
    Feed.Types.RSS1Item Item
i ->
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isDate forall a b. (a -> b) -> a -> b
$ Item -> [DCItem]
RSS1.itemDC Item
i of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) -> Item -> Item
Feed.Types.RSS1Item Item
i {itemDC :: [DCItem]
RSS1.itemDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
dt} forall a. a -> [a] -> [a]
: [DCItem]
bs}
        ([DCItem]
_, []) ->
          Item -> Item
Feed.Types.RSS1Item
            Item
i {itemDC :: [DCItem]
RSS1.itemDC = DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Date, dcText :: Text
dcText = Text
dt} forall a. a -> [a] -> [a]
: Item -> [DCItem]
RSS1.itemDC Item
i}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"pubDate" Text
dt) forall a b. (a -> b) -> a -> b
$ (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"pubDate") Element
i
  where
    isDate :: DCItem -> Bool
isDate DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Date

-- | 'withItemDate' is a synonym for 'withItemPubDate'.
withItemDate :: ItemSetter DateString
withItemDate :: ItemSetter Text
withItemDate = ItemSetter Text
withItemPubDate

-- | 'withItemTitle myTitle' associates a new title, 'myTitle',
-- with a feed item.
withItemTitle :: ItemSetter Text
withItemTitle :: ItemSetter Text
withItemTitle Text
tit Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e -> Entry -> Item
Feed.Types.AtomItem Entry
e {entryTitle :: TextContent
Atom.entryTitle = Text -> TextContent
TextString Text
tit}
    Feed.Types.RSSItem RSSItem
i -> RSSItem -> Item
Feed.Types.RSSItem RSSItem
i {rssItemTitle :: Maybe Text
RSS.rssItemTitle = forall a. a -> Maybe a
Just Text
tit}
    Feed.Types.RSS1Item Item
i -> Item -> Item
Feed.Types.RSS1Item Item
i {itemTitle :: Text
RSS1.itemTitle = Text
tit}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"title" Text
tit) forall a b. (a -> b) -> a -> b
$ (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"title") Element
i

-- | 'withItemAuthor auStr' associates new author info
-- with a feed item.
withItemAuthor :: ItemSetter Text
withItemAuthor :: ItemSetter Text
withItemAuthor Text
au Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e ->
      Entry -> Item
Feed.Types.AtomItem
        Entry
e {entryAuthors :: [Person]
Atom.entryAuthors = [Person
nullPerson {personName :: Text
personName = Text
au, personURI :: Maybe Text
personURI = forall a. a -> Maybe a
Just Text
au}]}
    Feed.Types.RSSItem RSSItem
i -> RSSItem -> Item
Feed.Types.RSSItem RSSItem
i {rssItemAuthor :: Maybe Text
RSS.rssItemAuthor = forall a. a -> Maybe a
Just Text
au}
    Feed.Types.RSS1Item Item
i ->
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isAuthor forall a b. (a -> b) -> a -> b
$ Item -> [DCItem]
RSS1.itemDC Item
i of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) -> Item -> Item
Feed.Types.RSS1Item Item
i {itemDC :: [DCItem]
RSS1.itemDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
au} forall a. a -> [a] -> [a]
: [DCItem]
bs}
        ([DCItem]
_, []) ->
          Item -> Item
Feed.Types.RSS1Item
            Item
i {itemDC :: [DCItem]
RSS1.itemDC = DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Creator, dcText :: Text
dcText = Text
au} forall a. a -> [a] -> [a]
: Item -> [DCItem]
RSS1.itemDC Item
i}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"author" Text
au) forall a b. (a -> b) -> a -> b
$ (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"author") Element
i
  where
    isAuthor :: DCItem -> Bool
isAuthor DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Creator

-- | 'withItemFeedLink name myFeed' associates the parent feed URL 'myFeed'
-- with a feed item. It is labelled as 'name'.
withItemFeedLink :: Text -> ItemSetter Text
withItemFeedLink :: Text -> ItemSetter Text
withItemFeedLink Text
tit Text
url Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e ->
      Entry -> Item
Feed.Types.AtomItem
        Entry
e
          { entrySource :: Maybe Source
Atom.entrySource =
              forall a. a -> Maybe a
Just Source
Atom.nullSource {sourceId :: Maybe Text
sourceId = forall a. a -> Maybe a
Just Text
url, sourceTitle :: Maybe TextContent
sourceTitle = forall a. a -> Maybe a
Just (Text -> TextContent
TextString Text
tit)}
          }
    Feed.Types.RSSItem RSSItem
i -> RSSItem -> Item
Feed.Types.RSSItem RSSItem
i {rssItemSource :: Maybe RSSSource
RSS.rssItemSource = forall a. a -> Maybe a
Just (Text -> Text -> RSSSource
RSS.nullSource Text
url Text
tit)}
    Feed.Types.RSS1Item Item
i -> Item -> Item
Feed.Types.RSS1Item Item
i {itemTitle :: Text
RSS1.itemTitle = Text
tit}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"source" ([Text -> Text -> Attr
mkAttr Text
"url" Text
url], Text
tit)) forall a b. (a -> b) -> a -> b
$
      (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"source") Element
i

-- | 'withItemCommentLink url' sets the URL reference to the comment page to 'url'.
withItemCommentLink :: ItemSetter Text
withItemCommentLink :: ItemSetter Text
withItemCommentLink Text
url Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e ->
      Entry -> Item
Feed.Types.AtomItem
        Entry
e {entryLinks :: [Link]
Atom.entryLinks = ((Text -> Link
nullLink Text
url) {linkRel :: Maybe (Either Text Text)
linkRel = forall a. a -> Maybe a
Just (forall a b. a -> Either a b
Left Text
"replies")}) forall a. a -> [a] -> [a]
: Entry -> [Link]
Atom.entryLinks Entry
e}
    Feed.Types.RSSItem RSSItem
i -> RSSItem -> Item
Feed.Types.RSSItem RSSItem
i {rssItemComments :: Maybe Text
RSS.rssItemComments = forall a. a -> Maybe a
Just Text
url}
    Feed.Types.RSS1Item Item
i ->
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isRel forall a b. (a -> b) -> a -> b
$ Item -> [DCItem]
RSS1.itemDC Item
i of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) -> Item -> Item
Feed.Types.RSS1Item Item
i {itemDC :: [DCItem]
RSS1.itemDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
url} forall a. a -> [a] -> [a]
: [DCItem]
bs}
        ([DCItem]
_, []) ->
          Item -> Item
Feed.Types.RSS1Item
            Item
i {itemDC :: [DCItem]
RSS1.itemDC = DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Relation, dcText :: Text
dcText = Text
url} forall a. a -> [a] -> [a]
: Item -> [DCItem]
RSS1.itemDC Item
i}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"comments" Text
url) forall a b. (a -> b) -> a -> b
$ (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"comments") Element
i
  where
    isRel :: DCItem -> Bool
isRel DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Relation

-- | 'withItemEnclosure url mbTy len' sets the URL reference to the comment page to 'url'.
withItemEnclosure :: Text -> Maybe Text -> ItemSetter (Maybe Integer)
withItemEnclosure :: Text -> Maybe Text -> ItemSetter (Maybe Integer)
withItemEnclosure Text
url Maybe Text
ty Maybe Integer
mb_len Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e ->
      Entry -> Item
Feed.Types.AtomItem
        Entry
e
          { entryLinks :: [Link]
Atom.entryLinks =
              ((Text -> Link
nullLink Text
url)
                 { linkRel :: Maybe (Either Text Text)
linkRel = forall a. a -> Maybe a
Just (forall a b. a -> Either a b
Left Text
"enclosure")
                 , linkType :: Maybe Text
linkType = Maybe Text
ty
                 , linkLength :: Maybe Text
linkLength = forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ([Char] -> Text
pack forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Show a => a -> [Char]
show) Maybe Integer
mb_len
                 }) forall a. a -> [a] -> [a]
:
              Entry -> [Link]
Atom.entryLinks Entry
e
          }
    Feed.Types.RSSItem RSSItem
i ->
      RSSItem -> Item
Feed.Types.RSSItem
        RSSItem
i {rssItemEnclosure :: Maybe RSSEnclosure
RSS.rssItemEnclosure = forall a. a -> Maybe a
Just (Text -> Maybe Integer -> Text -> RSSEnclosure
nullEnclosure Text
url Maybe Integer
mb_len (forall a. a -> Maybe a -> a
fromMaybe Text
"text/html" Maybe Text
ty))}
    Feed.Types.RSS1Item Item
i ->
      Item -> Item
Feed.Types.RSS1Item
        Item
i
          { itemContent :: [ContentInfo]
RSS1.itemContent =
              ContentInfo
nullContentInfo {contentURI :: Maybe Text
contentURI = forall a. a -> Maybe a
Just Text
url, contentFormat :: Maybe Text
contentFormat = Maybe Text
ty} forall a. a -> [a] -> [a]
: Item -> [ContentInfo]
RSS1.itemContent Item
i
          }
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild
        ((forall t. ToNode t => Name -> t -> Element
unode Name
"enclosure" Text
url)
           {elementAttributes :: [Attr]
elementAttributes = [Text -> Text -> Attr
mkAttr Text
"length" Text
"0", Text -> Text -> Attr
mkAttr Text
"type" (forall a. a -> Maybe a -> a
fromMaybe Text
"text/html" Maybe Text
ty)]}) forall a b. (a -> b) -> a -> b
$
      (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"enclosure") Element
i

-- | 'withItemId isURL id' associates new unique identifier with a feed item.
-- If 'isURL' is 'True', then the id is assumed to point to a valid web resource.
withItemId :: Bool -> ItemSetter Text
withItemId :: Bool -> ItemSetter Text
withItemId Bool
isURL Text
idS Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e -> Entry -> Item
Feed.Types.AtomItem Entry
e {entryId :: Text
Atom.entryId = Text
idS}
    Feed.Types.RSSItem RSSItem
i ->
      RSSItem -> Item
Feed.Types.RSSItem
        RSSItem
i {rssItemGuid :: Maybe RSSGuid
RSS.rssItemGuid = forall a. a -> Maybe a
Just (Text -> RSSGuid
nullGuid Text
idS) {rssGuidPermanentURL :: Maybe Bool
rssGuidPermanentURL = forall a. a -> Maybe a
Just Bool
isURL}}
    Feed.Types.RSS1Item Item
i ->
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break DCItem -> Bool
isId forall a b. (a -> b) -> a -> b
$ Item -> [DCItem]
RSS1.itemDC Item
i of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) -> Item -> Item
Feed.Types.RSS1Item Item
i {itemDC :: [DCItem]
RSS1.itemDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
idS} forall a. a -> [a] -> [a]
: [DCItem]
bs}
        ([DCItem]
_, []) ->
          Item -> Item
Feed.Types.RSS1Item
            Item
i {itemDC :: [DCItem]
RSS1.itemDC = DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Identifier, dcText :: Text
dcText = Text
idS} forall a. a -> [a] -> [a]
: Item -> [DCItem]
RSS1.itemDC Item
i}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"guid" ([Text -> Text -> Attr
mkAttr Text
"isPermaLink" (forall {a}. Show a => a -> Text
showBool Bool
isURL)], Text
idS)) forall a b. (a -> b) -> a -> b
$
      (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"guid") Element
i
  where
    showBool :: a -> Text
showBool a
x = [Char] -> Text
pack forall a b. (a -> b) -> a -> b
$ forall a b. (a -> b) -> [a] -> [b]
map Char -> Char
toLower (forall a. Show a => a -> [Char]
show a
x)
    isId :: DCItem -> Bool
isId DCItem
dc = DCItem -> DCInfo
dcElt DCItem
dc forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Identifier

-- | 'withItemDescription desc' associates a new descriptive string (aka summary)
-- with a feed item.
withItemDescription :: ItemSetter Text
withItemDescription :: ItemSetter Text
withItemDescription Text
desc Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e -> Entry -> Item
Feed.Types.AtomItem Entry
e {entrySummary :: Maybe TextContent
Atom.entrySummary = forall a. a -> Maybe a
Just (Text -> TextContent
TextString Text
desc)}
    Feed.Types.RSSItem RSSItem
i -> RSSItem -> Item
Feed.Types.RSSItem RSSItem
i {rssItemDescription :: Maybe Text
RSS.rssItemDescription = forall a. a -> Maybe a
Just Text
desc}
    Feed.Types.RSS1Item Item
i -> Item -> Item
Feed.Types.RSS1Item Item
i {itemDesc :: Maybe Text
RSS1.itemDesc = forall a. a -> Maybe a
Just Text
desc}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"description" Text
desc) forall a b. (a -> b) -> a -> b
$ (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"description") Element
i

-- | 'withItemRights rightStr' associates the rights information 'rightStr'
-- with a feed item.
withItemRights :: ItemSetter Text
withItemRights :: ItemSetter Text
withItemRights Text
desc Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e -> Entry -> Item
Feed.Types.AtomItem Entry
e {entryRights :: Maybe TextContent
Atom.entryRights = forall a. a -> Maybe a
Just (Text -> TextContent
TextString Text
desc)}
     -- Note: per-item copyright information isn't supported by RSS2.0 (and earlier editions),
     -- you can only attach this at the feed/channel level. So, there's not much we can do
     -- except dropping the information on the floor here. (Rolling our own attribute or
     -- extension element is an option, but would prefer if someone else had started that
     -- effort already.
    Feed.Types.RSSItem {} -> Item
fi
    Feed.Types.RSS1Item Item
i ->
      case forall a. (a -> Bool) -> [a] -> ([a], [a])
break ((forall a. Eq a => a -> a -> Bool
== DCInfo
DC_Rights) forall b c a. (b -> c) -> (a -> b) -> a -> c
. DCItem -> DCInfo
dcElt) forall a b. (a -> b) -> a -> b
$ Item -> [DCItem]
RSS1.itemDC Item
i of
        ([DCItem]
as, DCItem
dci:[DCItem]
bs) -> Item -> Item
Feed.Types.RSS1Item Item
i {itemDC :: [DCItem]
RSS1.itemDC = [DCItem]
as forall a. [a] -> [a] -> [a]
++ DCItem
dci {dcText :: Text
dcText = Text
desc} forall a. a -> [a] -> [a]
: [DCItem]
bs}
        ([DCItem]
_, []) ->
          Item -> Item
Feed.Types.RSS1Item
            Item
i {itemDC :: [DCItem]
RSS1.itemDC = DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Rights, dcText :: Text
dcText = Text
desc} forall a. a -> [a] -> [a]
: Item -> [DCItem]
RSS1.itemDC Item
i}
     -- Since we're so far assuming that a shallow XML rep. of an item
     -- is of RSS2.0 ilk, pinning on the rights info is hard (see above.)
    Feed.Types.XMLItem {} -> Item
fi

-- | 'withItemTitle myLink' associates a new URL, 'myLink',
-- with a feed item.
withItemLink :: ItemSetter URLString
withItemLink :: ItemSetter Text
withItemLink Text
url Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e ->
      Entry -> Item
Feed.Types.AtomItem Entry
e {entryLinks :: [Link]
Atom.entryLinks = Text -> [Link] -> [Link]
replaceAlternate Text
url (Entry -> [Link]
Atom.entryLinks Entry
e)}
    Feed.Types.RSSItem RSSItem
i -> RSSItem -> Item
Feed.Types.RSSItem RSSItem
i {rssItemLink :: Maybe Text
RSS.rssItemLink = forall a. a -> Maybe a
Just Text
url}
    Feed.Types.RSS1Item Item
i -> Item -> Item
Feed.Types.RSS1Item Item
i {itemLink :: Text
RSS1.itemLink = Text
url}
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      Element -> Element -> Element
addChild (forall t. ToNode t => Name -> t -> Element
unode Name
"link" Text
url) forall a b. (a -> b) -> a -> b
$ (Element -> Bool) -> Element -> Element
filterChildren (\Element
e -> Element -> Name
elementName Element
e forall a. Eq a => a -> a -> Bool
/= Name
"link") Element
i
  where
    replaceAlternate :: Text -> [Link] -> [Link]
replaceAlternate Text
_ [] = []
    replaceAlternate Text
x (Link
lr:[Link]
xs)
      | forall {a}. IsString a => Maybe (Either a a) -> a
toStr (Link -> Maybe (Either Text Text)
Atom.linkRel Link
lr) forall a. Eq a => a -> a -> Bool
== Text
"alternate" = Link
lr {linkHref :: Text
Atom.linkHref = Text
x} forall a. a -> [a] -> [a]
: [Link]
xs
      | Bool
otherwise = Link
lr forall a. a -> [a] -> [a]
: Text -> [Link] -> [Link]
replaceAlternate Text
x [Link]
xs
    toStr :: Maybe (Either a a) -> a
toStr Maybe (Either a a)
Nothing = a
""
    toStr (Just (Left a
x)) = a
x
    toStr (Just (Right a
x)) = a
x

withItemCategories :: ItemSetter [(Text, Maybe Text)]
withItemCategories :: ItemSetter [(Text, Maybe Text)]
withItemCategories [(Text, Maybe Text)]
cats Item
fi =
  case Item
fi of
    Feed.Types.AtomItem Entry
e ->
      Entry -> Item
Feed.Types.AtomItem
        Entry
e
          { entryCategories :: [Category]
Atom.entryCategories =
              forall a b. (a -> b) -> [a] -> [b]
map (\(Text
t, Maybe Text
mb) -> (Text -> Category
Atom.newCategory Text
t) {catScheme :: Maybe Text
Atom.catScheme = Maybe Text
mb}) [(Text, Maybe Text)]
cats forall a. [a] -> [a] -> [a]
++ Entry -> [Category]
entryCategories Entry
e
          }
    Feed.Types.RSSItem RSSItem
i ->
      RSSItem -> Item
Feed.Types.RSSItem
        RSSItem
i
          { rssItemCategories :: [RSSCategory]
RSS.rssItemCategories =
              forall a b. (a -> b) -> [a] -> [b]
map (\(Text
t, Maybe Text
mb) -> (Text -> RSSCategory
RSS.newCategory Text
t) {rssCategoryDomain :: Maybe Text
RSS.rssCategoryDomain = Maybe Text
mb}) [(Text, Maybe Text)]
cats forall a. [a] -> [a] -> [a]
++
              RSSItem -> [RSSCategory]
rssItemCategories RSSItem
i
          }
    Feed.Types.RSS1Item Item
i ->
      Item -> Item
Feed.Types.RSS1Item
        Item
i
          { itemDC :: [DCItem]
RSS1.itemDC =
              forall a b. (a -> b) -> [a] -> [b]
map (\(Text
t, Maybe Text
_) -> DCItem {dcElt :: DCInfo
dcElt = DCInfo
DC_Subject, dcText :: Text
dcText = Text
t}) [(Text, Maybe Text)]
cats forall a. [a] -> [a] -> [a]
++ Item -> [DCItem]
RSS1.itemDC Item
i
          }
    Feed.Types.XMLItem Element
i ->
      Element -> Item
Feed.Types.XMLItem forall a b. (a -> b) -> a -> b
$
      forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr
        (\(Text
t, Maybe Text
mb) Element
acc ->
           Element -> Element -> Element
addChild
             (forall t. ToNode t => Name -> t -> Element
unode Name
"category" (forall b a. b -> (a -> b) -> Maybe a -> b
maybe (forall a. a -> [a] -> [a]
: []) (\Text
v Attr
x -> [Text -> Text -> Attr
mkAttr Text
"domain" Text
v, Attr
x]) Maybe Text
mb (Text -> Text -> Attr
mkAttr Text
"term" Text
t)))
             Element
acc)
        Element
i
        [(Text, Maybe Text)]
cats

-- helpers..
filterChildren :: (XML.Element -> Bool) -> XML.Element -> XML.Element
filterChildren :: (Element -> Bool) -> Element -> Element
filterChildren Element -> Bool
pre Element
e =
  case Element -> [Node]
elementNodes Element
e of
    [] -> Element
e
    [Node]
cs -> Element
e {elementNodes :: [Node]
elementNodes = forall a b. (a -> Maybe b) -> [a] -> [b]
mapMaybe Node -> Maybe Node
filterElt [Node]
cs}
  where
    filterElt :: Node -> Maybe Node
filterElt xe :: Node
xe@(XML.NodeElement Element
el)
      | Element -> Bool
pre Element
el = forall a. a -> Maybe a
Just Node
xe
      | Bool
otherwise = forall a. Maybe a
Nothing
    filterElt Node
xe = forall a. a -> Maybe a
Just Node
xe

addChild :: XML.Element -> XML.Element -> XML.Element
addChild :: Element -> Element -> Element
addChild Element
a Element
b = Element
b {elementNodes :: [Node]
elementNodes = Element -> Node
XML.NodeElement Element
a forall a. a -> [a] -> [a]
: Element -> [Node]
elementNodes Element
b}

mapMaybeChildren :: (XML.Element -> Maybe XML.Element) -> XML.Element -> XML.Element
mapMaybeChildren :: (Element -> Maybe Element) -> Element -> Element
mapMaybeChildren Element -> Maybe Element
f Element
e =
  case Element -> [Node]
elementNodes Element
e of
    [] -> Element
e
    [Node]
cs -> Element
e {elementNodes :: [Node]
elementNodes = forall a b. (a -> b) -> [a] -> [b]
map Node -> Node
procElt [Node]
cs}
  where
    procElt :: Node -> Node
procElt xe :: Node
xe@(XML.NodeElement Element
el) =
      case Element -> Maybe Element
f Element
el of
        Maybe Element
Nothing -> Node
xe
        Just Element
el1 -> Element -> Node
XML.NodeElement Element
el1
    procElt Node
xe = Node
xe