#!/usr/bin/env python
'Atom Script, by Sean B. Palmer'

import os, re, time

domain = os.environ.get('SERVER_NAME')
script = os.environ.get('SCRIPT_NAME')
r_title = re.compile('(?s)<title>(.*?), by (.*?)</title>')
r_entry = re.compile('(?s)<div title="(.*?)">(.*?)</div>')

if script: 
   index = open('index.html').read()
   title, author = r_title.search(index).groups()
   label = script.lstrip('/').rsplit('/', 1)[0]

   entries = []
   for (date, content) in r_entry.findall(index): 
      try: t = time.strptime(date, '%d %B %Y, %H:%M')
      except ValueError: continue
      updated = time.strftime('%Y-%m-%dT%H:%M:%SZ', t)
      content = content.replace(']]>', ']]>]]><![CDATA[')
      entries.append((date, updated, content))

   print 'Content-Type: application/atom+xml; charset=utf-8\n'
   print '<feed xmlns="http://www.w3.org/2005/Atom">'
   print ' <title>%s</title>' % title
   print ' <link href="http://%s/%s/"/>' % (domain, label)
   print ' <id>tag:%s,2006:%s</id>' % (domain, label)
   if entries: 
      print ' <updated>%s</updated>' % entries[0][1]
      for (date, updated, content) in entries: 
         print ' <entry xml:base="http://%s/%s/">' % (domain, label)
         print '  <id>tag:%s,2006:%s:%s</id>' % (domain, label, updated)
         print '  <title>%s</title>' % date
         print '  <updated>%s</updated>' % updated
         print '  <author><name>%s</name></author>' % author
         print '  <content type="html"><![CDATA[%s]]></content>' % content
         print ' </entry>'
   print '</feed>'