"""base xml and rss views

:organization: Logilab

:copyright: 2001-2009 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr

"""

__docformat__ = "restructuredtext en"

from time import timezone

from logilab.mtconverter import xml_escape

from cubicweb.selectors import non_final_entity, one_line_rset, appobject_selectable

from cubicweb.view import EntityView, AnyRsetView

from cubicweb.web.httpcache import MaxAgeHTTPCacheManager

from cubicweb.web.component import Component

from cubicweb.web.box import BoxTemplate

from cubicweb.common.uilib import simple_sgml_tag

_ = unicode

# base xml views ##############################################################

class XmlView(EntityView):

    """xml view for entities"""

    id = 'xml'

    title = _('xml')

    templatable = False

    content_type = 'text/xml'

    xml_root = 'rset'

    item_vid = 'xmlitem'

    def cell_call(self, row, col):

        self.wview(self.item_vid, self.rset, row=row, col=col)

    def call(self):

        """display a list of entities by calling their <item_vid> view"""

        self.w(u'<?xml version="1.0" encoding="%s"?>\n' % self.req.encoding)

        self.w(u'<%s size="%s">\n' % (self.xml_root, len(self.rset)))

        for i in xrange(self.rset.rowcount):

            self.cell_call(i, 0)

        self.w(u'</%s>\n' % self.xml_root)

class XmlItemView(EntityView):

    id = 'xmlitem'

    def cell_call(self, row, col):

        """ element as an item for an xml feed """

        entity = self.complete_entity(row, col)

        self.w(u'<%s>\n' % (entity.e_schema))

        for rschema, attrschema in entity.e_schema.attribute_definitions():

            attr = rschema.type

            try:

                value = entity[attr]

            except KeyError:

                # Bytes

                continue

            if value is not None:

                if attrschema == 'Bytes':

                    from base64 import b64encode

                    value = '<![CDATA[%s]]>' % b64encode(value.getvalue())

                elif isinstance(value, basestring):

                    value = xml_escape(value)

                self.w(u'  <%s>%s</%s>\n' % (attr, value, attr))

        self.w(u'</%s>\n' % (entity.e_schema))

class XmlRsetView(AnyRsetView):

    """dumps raw rset as xml"""

    id = 'rsetxml'

    title = _('xml export')

    templatable = False

    content_type = 'text/xml'

    xml_root = 'rset'

    def call(self):

        w = self.w

        rset, descr = self.rset, self.rset.description

        eschema = self.schema.eschema

        labels = self.columns_labels(False)

        w(u'<?xml version="1.0" encoding="%s"?>\n' % self.req.encoding)

        w(u'<%s query="%s">\n' % (self.xml_root, xml_escape(rset.printable_rql())))

        for rowindex, row in enumerate(self.rset):

            w(u' <row>\n')

            for colindex, val in enumerate(row):

                etype = descr[rowindex][colindex]

                tag = labels[colindex]

                attrs = {}

                if '(' in tag:

                    attrs['expr'] = tag

                    tag = 'funccall'

                if val is not None and not eschema(etype).is_final():

                    attrs['eid'] = val

                    # csvrow.append(val) # val is eid in that case

                    val = self.view('textincontext', rset,

                                    row=rowindex, col=colindex)

                else:

                    val = self.view('final', rset, displaytime=True,

                                    row=rowindex, col=colindex, format='text/plain')

                w(simple_sgml_tag(tag, val, **attrs))

            w(u' </row>\n')

        w(u'</%s>\n' % self.xml_root)

# RSS stuff ###################################################################

class RSSFeedURL(Component):

    id = 'rss_feed_url'

    __select__ = non_final_entity()

    def feed_url(self):

        return self.build_url(rql=self.limited_rql(), vid='rss')

class RSSEntityFeedURL(Component):

    id = 'rss_feed_url'

    __select__ = non_final_entity() & one_line_rset()

    def feed_url(self):

        return self.entity(0, 0).rss_feed_url()

class RSSIconBox(BoxTemplate):

    """just display the RSS icon on uniform result set"""

    id = 'rss'

    __select__ = (BoxTemplate.__select__

                  & appobject_selectable('components', 'rss_feed_url'))

    visible = False

    order = 999

    def call(self, **kwargs):

        try:

            rss = self.req.external_resource('RSS_LOGO')

        except KeyError:

            self.error('missing RSS_LOGO external resource')

            return

        urlgetter = self.vreg.select_component('rss_feed_url', self.req, self.rset)

        url = urlgetter.feed_url()

class RssView(XmlView):

    id = 'rss'

    title = _('rss')

    templatable = False

    content_type = 'text/xml'

    http_cache_manager = MaxAgeHTTPCacheManager

    cache_max_age = 60*60*2 # stay in http cache for 2 hours by default 

    def cell_call(self, row, col):

        self.wview('rssitem', self.rset, row=row, col=col)

    def call(self):

        """display a list of entities by calling their <item_vid> view"""

        req = self.req

        self.w(u'<?xml version="1.0" encoding="%s"?>\n' % req.encoding)

        self.w(u'''<rdf:RDF

 xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"

 xmlns:dc="http://purl.org/dc/elements/1.1/"

 xmlns="http://purl.org/rss/1.0/"

>''')

        self.w(u'  <channel rdf:about="%s">\n' % xml_escape(req.url()))

        self.w(u'    <title>%s RSS Feed</title>\n' % xml_escape(self.page_title()))

        self.w(u'    <description>%s</description>\n' % xml_escape(req.form.get('vtitle', '')))

        params = req.form.copy()

        params.pop('vid', None)

        self.w(u'    <link>%s</link>\n' % xml_escape(self.build_url(**params)))

        self.w(u'    <items>\n')

        self.w(u'      <rdf:Seq>\n')

        for entity in self.rset.entities():

            self.w(u'      <rdf:li resource="%s" />\n' % xml_escape(entity.absolute_url()))

        self.w(u'      </rdf:Seq>\n')

        self.w(u'    </items>\n')

        self.w(u'  </channel>\n')

        for i in xrange(self.rset.rowcount):

            self.cell_call(i, 0)

        self.w(u'</rdf:RDF>')

class RssItemView(EntityView):

    id = 'rssitem'

    date_format = '%%Y-%%m-%%dT%%H:%%M%+03i:00' % (timezone / 3600)

    def cell_call(self, row, col):

        entity = self.complete_entity(row, col)

        self.w(u'<item rdf:about="%s">\n' % xml_escape(entity.absolute_url()))

        self._marker('title', entity.dc_long_title())

        self._marker('link', entity.absolute_url())

        self._marker('description', entity.dc_description())

        self._marker('dc:date', entity.dc_date(self.date_format))

        if entity.creator:

            self.w(u'<author>')

            self._marker('name', entity.creator.name())

            email = entity.creator.get_email()

            if email:

                self._marker('email', email)

            self.w(u'</author>')

        self.w(u'</item>\n')

    def _marker(self, marker, value):

        if value:

            self.w(u'  <%s>%s</%s>\n' % (marker, xml_escape(value), marker))