#!/usr/bin/python

import sys
import traceback
if (__name__ == '__main__'):
	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 my_db
from os import environ
from urlparse import urljoin

def get_selection(form):
	kw = form.getvalue("kw")
	if not isinstance(kw, type([])):
		if kw:
			kw = [kw]
		else:
			kw = [0]
	return map(int, kw)

def set_selection(kw, sel):
	res = []
	for key in kw:
		chk = key[0] in sel
		res.append({'id':key[0],
			'keyword': key[1],
			'count' : key[2],
			'checked' : chk})
	res.sort(lambda l,r: cmp(l["keyword"],r["keyword"]))
	return res

def get_curl():
	return urljoin( 'http://' + environ["HTTP_HOST"] + environ["REQUEST_URI"], 'add.py')

def load_index(db):
    	kw = db.get_all_keywords()
	kw = set_selection(kw, [])
    	total = kw[0]['count']
    	kw = kw[1:]
    	exc = map(lambda e: int(e['id']), kw)
    	bookmarks = db.select_bookmarks([0], exc)
	if len(bookmarks)>0:
    		bookmarks = db.get_bookmarks(map(lambda x: x[0], bookmarks))
    	bookmarks = map(lambda bk: bk.dict(), bookmarks)
    	curl = get_curl();
    	main(bookmarks, kw, curl, total)

def main(bk, kw, curl, total):
    tmpl = TemplateManager().prepare("index.tmpl")
    tproc = TemplateProcessor()
    tproc.set("Bookmarks", bk)
    tproc.set("Keywords", kw)
    tproc.set("curl", curl)
    tproc.set("total", total)
    print tproc.process(tmpl)

if (__name__ == "__main__"):
    form = cgi.FieldStorage()
    db = my_db.connect()
    keywords = db.get_all_keywords()
    total = keywords[0][2]
    keywords = keywords[1:]
    selection = get_selection(form)
    if selection[0] == 0:
    	exc = map(lambda e: int(e[0]), keywords)
   	bookmarks = db.select_bookmarks([0], exc)
    else:
    	bookmarks = db.select_bookmarks(selection)
    keywords = set_selection(keywords, selection)
    if len(bookmarks) > 0:
    	bookmarks = db.get_bookmarks(map(lambda x: x[0], bookmarks))
    bookmarks = map(lambda bk: bk.dict(), bookmarks)
    curl = get_curl();
    main(bookmarks, keywords, curl, total)