#!/usr/bin/python
# -*- coding: utf-8 -*-

import cgi
import os
import sys
import patchtracker.Conf as Conf
from patchtracker.Templates import ErrorTemplate, PatchTemplate, PackageVersTemplate, LetterTocTemplate, FrontPageTemplate
from patchtracker.DiffGzHandler import DiffGzHandler
import patchtracker.DB as DB
from patchtracker.DB import PatchTrackerDB
import pygments
from pygments.lexers import DiffLexer
from pygments.formatters import HtmlFormatter

class CmdHandler:
  def __init__(self, uri):
    if len(uri)<len(Conf.root_url) or uri[0:len(Conf.root_url)]!=Conf.root_url:
      self.error("Invalid URL!")

    self.db = PatchTrackerDB()
    args = uri[len(Conf.root_url)+1:].split("/")
    print "args:",args
    self.cmd = args[0]
    if self.cmd == "patch":
      patchtype,mode,pkgname,version = args[1:5]
      self.parsemode(mode)
      dh = self.make_diffhandler(pkgname,version)
      if patchtype == "series":
        self.patchname = args[5]
        self.content = dh.series().fetch(self.patchname)
      elif patchtype == "debianonly":
        self.patchname = "debian-dir only changes"
        self.content = dh.debiandir()
      elif patchtype == "nondebian":
        self.patchname = "direct (non packaging) changes"
        self.content = dh.nondebiandir()
      else:
        self.error("unhandled patch type '%s'"%(patchtype))
      self.pkgname = pkgname
      self.version = version
    elif self.cmd == "package":
      self.db.setFactory(DB.srcpkg_factory)
      self.name = args[1]
      self.version = args[2]
      self.srcpkg = self.db.findSourcePackage(self)
    elif self.cmd == "index":
      if len(args) < 2 or not len(args[1]):
        self.error("please provide a letter on which to index")
      else:
        self.letter = args[1]
        self.toc = self.db.findLetterToc(self.letter)
    else:
      # XXX better way to set up "frontpage"
      self.index = self.db.findMasterIndex()
      self.cmd = "frontpage"

  # XXX this assumes to much hard coded, but until there's a faster
  # XXX (i.e. database) way to query sourcepackage/diffs it will
  # XXX have to do...
  def make_diffhandler(self, pkgname, vers):
    file = None
    dfile = pkgname+"_"+vers+".diff.gz"
    for comp in ['main', 'contrib', 'non-free']:
      loc = os.sep.join([Conf.archive_root, 'pool', comp, pkgname[0], pkgname])
      try:
        test = os.sep.join([loc, dfile])
        os.stat(test)
        file = test
        break
      except:
        pass
    if file:
      return DiffGzHandler(file)
    else:
      self.error("can not find diff file for %s / %s"%(pkgname,vers))

  def parsemode(self, mode):
    if mode == "view" or mode == "dl":
      self.mode = mode
    else:
      self.error("unhandled display mode '%s'"%(mode))

  def error(self, msg):
    print "Content-Type: text/html\n\n"
    print ErrorTemplate(msg)
    sys.exit(1)

  def output(self):
    if self.cmd == "patch":
      if self.mode == "dl":
        print "Content-Type: text/x-diff\n\n"
        print self.content
      else:
        print "Content-Type: text/html\n\n"
        print PatchTemplate(pkg=self.pkgname,vers=self.version,patch=self.content,name=self.patchname)
    elif self.cmd == "package":
      print "Content-Type: text/html\n\n"
      print PackageVersTemplate(self.srcpkg,"XXX FIXME")
    elif self.cmd == "index":
      print "Content-Type: text/html\n\n"
      print LetterTocTemplate(self.letter, self.toc)
    elif self.cmd == "frontpage":
      print "Content-Type: text/html\n\n"
      print FrontPageTemplate(self.index)

if __name__ == "__main__":
  uri = os.getenv("REQUEST_URI")
  if not uri:
    uri = sys.argv[1]
  CmdHandler(uri).output()