"""
Heartbeat - an event that fires every few minutes, called by a cron script.

This is used to trigger cleanup events, sending e-mails and many other
*periodic tasks*. To use the `heartbeat` you need to create a `heartbeat`
subscriber registered using :term:`ZCA`

Example use::

    from zope import component
    from naaya.core.heartbeat import IHeartbeat
    from Products.Naaya.interfaces import INySite

    @component.adapter(INySite, IHeartbeat)
    def cleanupUnsubmittedObjects(site, hb):
        site.cleanupUnsubmittedObjects(site.get_site_uid())

    component.provideHandler(cleanupUnsubmittedObjects)

or you can simply register a subscriber via zcml.

In `configure.zml`::

      <subscriber for="Products.Naaya.interfaces.INySite
                       Products.Naaya.interfaces.IHeartbeat"
                  handler=".my_subscriber" />

In `subscribers.py`::

      def my_subscriber(site, hearthbeat):
        #do stuff

This module also contains a few default subscribers for generic periodic tasks
in *Naaya*.

"""

from datetime import datetime, timedelta

from zope.interface import implements
from zope.component import adapter

from Products.Naaya.interfaces import INySite
from naaya.core.utils import cooldown
from interfaces import IHeartbeat

class Heartbeat(object):
    implements(IHeartbeat)
    def __init__(self):
        self.when = datetime.now()


@adapter(INySite, IHeartbeat)
def cleanup_unsubmitted_objects(site, hb):
    """
    Clean up unsubmitted objects (documents, stories) older than 1 day.
    """

    name = 'site cleanup unsubmitted %r' % physical_path(site)
    if cooldown(name, timedelta(hours=6)):
        return

    site.cleanupUnsubmittedObjects(site.get_site_uid())

def physical_path(ob):
    return '/'.join(ob.getPhysicalPath())

@adapter(INySite, IHeartbeat)
def rdfcalendar_cron(site, hb):
    """
    Update any `RDFCalendar` objects in the site's root
    """
    import transaction

    name = 'site rdfcalendar %r' % physical_path(site)
    if cooldown(name, timedelta(hours=6)):
        return

    transaction.commit() # commit earlier stuff; fresh transaction

    try:
        for rdfcal in site.objectValues(['RDF Calendar']):
            rdfcal.manage_updateChannels()
    except:
        site.log_current_error()
        transaction.abort()
    else:
        transaction.get().note("RDFCalendar cron %r" % physical_path(site))
        transaction.commit()

@adapter(INySite, IHeartbeat)
def linkchecker_cron(site, hb):
    """
    Run the link checker
    """
    import transaction

    name = 'site link checker %r' % physical_path(site)
    if cooldown(name, timedelta(days=7)):
        return

    transaction.commit() # commit earlier stuff; fresh transaction

    try:
        for link_checker in site.objectValues(['Naaya LinkChecker']):
            if hasattr(link_checker, 'cronCheck'):
                link_checker.cronCheck()
    except:
        site.log_current_error()
        transaction.abort()
    else:
        transaction.get().note("LinkChecker cron %r" % physical_path(site))
        transaction.commit()
