#!/usr/bin/python

import sys
import traceback
sys.path.insert(0, "/home/fabien/lib/python")
sys.path.insert(0, "./lib")
sys.stderr = sys.stdout

print "Content-type: text/html; charset=iso-8859-1;"
print

# import cgitb; cgitb.enable()
import cgi
from htmltmpl import TemplateManager, TemplateProcessor
import time
from utils import unique
import my_db
from add_confirm import get_bk_from_form, get_new_kw_from_form
from index import load_index
import os

def add_new_keywords(db, newkw):
    kw = [0] # Default keyword, so that every bookmarks appears in
             # the bookmarks_keywords db.
    (ids, kws) = map(list,apply(zip, db.get_all_keywords()))
    for elem in unique(newkw):
    	if elem in kws:
		kw.append(ids[kws.index(elem)])
	else:
		id = db.add_keyword(elem)
		kw.append(id)
    return kw

def main(bk, kw):
    tmpl = TemplateManager().prepare("add_result.tmpl")
    tproc = TemplateProcessor()
    tproc.set("pagetitle", os.environ["REMOTE_USER"]+"'s XBELWeb Result")
    tproc.set("url", bk.url)
    tproc.set("name", bk.name)
    tproc.set("added", bk.added)
    tproc.set("desc", bk.desc)
    tproc.set("Keywords", kw)
    print tproc.process(tmpl)

if (__name__ == "__main__"):
    form = cgi.FieldStorage()
    db = my_db.connect(os.environ["REMOTE_USER"])
    action = form['action'].value
    if action == 'cancel':
      load_index(db)
    elif action == 'delete':
      db.remove_bookmark(int(form['id'].value))
      load_index(db)
    else:
      bk = get_bk_from_form(form)
      newkw = get_new_kw_from_form(form)
      kw = add_new_keywords(db, newkw)
      if bk.id == -1:
	id = db.add_bookmark(bk)
      else:
	db.update_bookmark(bk)
	id = bk.id
      db.update_keywords(id, kw)
      kw = map(lambda e: { 'keyword': e[1] }, db.get_keywords(id)[1:])
      main(bk, kw)