#!/usr/bin/python

import sys, os, cgi, re
import cgitb; cgitb.enable()

THIS_SCRIPT = "/cgi-bin/album.py"
ALBUMS_PATH  = "../albums"
SMALL_EXT = ".small.jpg"
THUMB_EXT = ".thumb.jpg"
EXIF_EXT = ".exif.txt"


# Required header that tells the browser how to render the text.
print "Content-Type: text/html\n\n"

#---------------------------------------------------------------------------
class Photo:
  def __init__(self, albumName, name):
    self.name = name
    self.albumName = albumName

  def __str__(self):
    return self.getImageBase() + ":" + self.getUrl()

  def getAlbumName(self):
    return self.albumName

  def getName(self):
    return self.name

  def getImageBase(self):
    return os.path.join(ALBUMS_PATH, self.albumName, self.name)
    
  def getThumbFile(self):
    return self.getImageBase() + THUMB_EXT

  def getSmallFile(self):
    return self.getImageBase() + SMALL_EXT

  def getExifFile(self):
    return self.getImageBase() + EXIF_EXT

  def getAlbumUrl(self):
    return "%s?album=%s" % (THIS_SCRIPT, self.albumName)

  def getUrl(self):
    return "%s?album=%s&photo=%s" % (THIS_SCRIPT, self.albumName, self.name)


class Album:
  def __init__(self, name):
    self.name = name
    self.date = "?"
    self.title = "?"
    self.count = -1
    self.parseName()
    self.photos = []

  def parseName(self):
    m = re.match("^(\d\d)-(\d\d)-(\d\d)-(.*)$", self.name)
    if not m:
      return 
    else:
      self.date = m.group(1)
      y = m.group(1)
      mo = m.group(2)
      d = m.group(3)
      self.date = mo + "/" + d + "/" + y
      self.title = m.group(4)
      self.formatTitle()

  def formatTitle(self):
    t = self.title
    if t == "":
      return
    self.title = ""
    t = t.replace("_", " ")
    toks = t.split()
    for tok in toks:
      tok = tok[0].upper() + tok[1:]
      self.title += tok + " "
    self.title = self.title.strip()

  def getDate(self):
    return self.date

  def getTitle(self):
    return self.title

  def getName(self):
    return self.name

  def getUrl(self):
    return "%s?album=%s" % (THIS_SCRIPT, self.name)

  def getCount(self):
    self.getPhotoList()
    return self.count

  def getPhotoList(self):
    # load on demand
    if not self.photos == []:
      return self.photos
    # currently if we find a thumb, we assume the small is also there
    ext = ".thumb.jpg"
    files = os.listdir(os.path.join(ALBUMS_PATH, self.name))
    files.sort()
    photos = []
    for f in files:
      extStart = f.find(ext)
      if extStart > -1:
        # strip off the thumb extension
        photo = Photo(self.name, f[:extStart])
        photos.append(photo)
    self.count = len(photos)
    return photos

#---------------------------------------------------------------------------
def show_albums():
  print "<br>"
  print "<br>"
  print "<center>"
  print "<h1>Photo Albums</h1>"
  print "<table cellspacing=5>"
  albums = os.listdir(ALBUMS_PATH)
  albums.sort()
  albums.reverse()
  for aN in albums:
    a = Album(aN)
    photos = a.getPhotoList()
    #print "<a href=\"%s?album=%s\">%s (%s)</a>" % (THIS_SCRIPT, a.getName(), a.getName(), a.getCount())
    print "<tr>"
    print "<td align=right>%s</td>" % a.getDate()
    print "<td><a href=\"%s?album=%s\">%s</a></td>" % (THIS_SCRIPT, a.getName(), a.getTitle())
    print "<td align=right>%s pics</td>" % a.getCount()
    #print "<a href=\"%s?album=%s\">%s %s (%s)</a>" % (THIS_SCRIPT, a.getName(), a.getDate(), a.getName(), a.getCount())
    print "</tr>"
  print "</table>"
  print "</center>"


#---------------------------------------------------------------------------
def show_one_album(album):
  #photos = getPhotoList(album)
  #for p in photos:
  #  print p
  #return

  a = Album(album)
  photos = a.getPhotoList()
  print "<br>"
  print "<center>"
  print "<h1 align=center>%s (%s)</h1>" % (a.getTitle(), a.getDate())
  print "<a href=\"%s\" style=\"position:relative; bottom:15\">" \
        "[all albums]</a>" % THIS_SCRIPT
  print "<table border=0 align=center cellspacing=5 cellpadding=0><tr>"
  i = 1
  TABLE_COLS = 5
  for p in photos:
    print "<td align=center width=110 height=110 style=\"border:solid 1px #AAA\">"
    print "<a href=\"%s\">" % (p.getUrl())
    print "<img border=0 src=\"%s\"></a>" % p.getThumbFile()
    print "<br>"
    # put the hyperlink twice to avoid having a <CR> in between which produces
    # the annoying underscore/dash artifact
    #print "<a href=\"%s\">" % (url)
    #print "<font size=-1>%s</font></a>" % (thumbToViewableName(p))
    print "</td>"
    if (i % TABLE_COLS) == 0:
      print "</tr><tr>"
    i = i + 1
  print "</tr></table>"
  print "</center>"

#---------------------------------------------------------------------------
def getPrevAndNextPhotos(photos, curPhoto):
  prevPhoto = None
  nextPhoto = None
  for i in range(len(photos)):
    p = photos[i]
    if curPhoto.getName() == p.getName():
      if i > 0:
        prevPhoto = photos[i-1]
      if i < len(photos)-1:
        nextPhoto = photos[i+1]
      break
  return (prevPhoto, nextPhoto)


def show_photo(albumName, photoName):
  photo = Photo(albumName, photoName)
  album = Album(albumName)
  photos = album.getPhotoList()
  [prevPhoto, nextPhoto] = getPrevAndNextPhotos(photos, photo)
  #print "<br>"
  #print "<center>"


  # big table
  print "<table border=0 cellpadding=5><tr valign=top>"
  print "<td>"

  # left table
  print "<table border=0 cellpadding=0><tr valign=top align=center><td>"

  # control panel ---
  print "<table border=0 cellpadding=10><tr>"
  # prev
  if prevPhoto:
    print "<td><a href=\"%s\">Prev</a></td>" % prevPhoto.getUrl()
  else:
    print "<td>Prev</td>"
  # current
  print "<td align=center>"
  print "<a href=\"%s\">[thumbnails]</a>" % photo.getAlbumUrl()
  print "</td>"
  # next
  if nextPhoto:
    print "<td><a href=\"%s\">Next</a></td>" % nextPhoto.getUrl()
  else:
    print "<td>Next</td>"
  # end: control panel
  print "</tr></table>"
  #print "<br>"
  # --- 

  # left table cont.
  print "</td></tr><tr><td>"

  # exif information if it exists
  if os.path.exists(photo.getExifFile()):
    print "<table bgcolor=BBBBFF><tr><td align=left>"
    print "<font size=-1>"
    file = open(photo.getExifFile(), 'r')
    for line in file:
      line = line.strip()
      if not line == "":
        print "%s<br>" % line
    print "</font>"
    print "</td></tr></table>"
  # end exif table

  print "</td></tr></table>"
  # end left table

  # big table cont.
  print "</td><td>"

  print "<img src=\"%s\" style=\"border:solid 1px #AAA\">" % photo.getSmallFile()

  print "</td>"
  print "</tr></table>"
  # end big table


  #print "</center>"


#def show_photo(album, photoName):
#  photo = Photo(album, photoName)
#  photos = getPhotoList(album)
#  [prevPhoto, nextPhoto] = getPrevAndNextPhotos(photos, photo)
#  #print "<br>"
#  print "<center>"
#
#  # control panel ---
#  print "<table border=0 cellpadding=10><tr>"
#  # prev
#  if prevPhoto:
#    print "<td><a href=\"%s\">Prev</a></td>" % prevPhoto.getUrl()
#  else:
#    print "<td>Prev</td>"
#  # current
#  print "<td align=center>"
#  print "<a href=\"%s\">Up</a>" % photo.getAlbumUrl()
#  print "</td>"
#  # next
#  if nextPhoto:
#    print "<td><a href=\"%s\">Next</a></td>" % nextPhoto.getUrl()
#  else:
#    print "<td>Next</td>"
#  # end: control panel
#  print "</tr></table>"
#  #print "<br>"
#  # --- 
#
#  # image
#  print "<table border=0 cellpadding=5><tr valign=top>"
#  print "<td>"
#  print "<img src=\"%s\">" % photo.getSmallFile()
#  print "</td>"
#  print "<td>"
#  # exif information if it exists
#  if os.path.exists(photo.getExifFile()):
#    print "<table bgcolor=BBBBFF><tr><td align=left>"
#    print "<font size=-1>"
#    file = open(photo.getExifFile(), 'r')
#    for line in file:
#      line = line.strip()
#      if not line == "":
#        print "%s<br>" % line
#    print "</font>"
#    print "</td></tr></table>"
#  # end
#  print "</td>"
#  print "</tr></table>"



  ## image name
  #print "<br>"
  #print album
  #print "<br>"
  #print photo.getName()

  print "</center>"

#---------------------------------------------------------------------------
def main():
  form = cgi.FieldStorage()
  if (form.has_key("photo")):
    show_photo(form["album"].value, form["photo"].value)
  elif (form.has_key("album")):
    show_one_album(form["album"].value)
  else:
    show_albums()
  #show_one_album("05-07-ultimate")

main()