#!/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

import my_db

from os import environ

from htmltmpl import TemplateManager

from webutils import get_template_processor, set_selection, sort_keywords

def output(prefs, folder, parents, children, bookmarks, keywords):

    tmpl = TemplateManager().prepare("folders.tmpl")

    tproc = get_template_processor(prefs)

    tproc.set('Parents', parents)

    tproc.set('folder', folder['id'])

    tproc.set('name', folder['name'])

    tproc.set('total', len(bookmarks))

    bookmarks.sort(lambda l,r: cmp(l['name'],r['name']))

    tproc.set('Bookmarks', bookmarks)

    tproc.set('Subfolders', children)

    keywords = sort_keywords(keywords[1:], prefs)

    tproc.set('Keywords', keywords)

    print tproc.process(tmpl)

if (__name__ == "__main__"):

    form = cgi.FieldStorage()

    db = my_db.connect(environ["REMOTE_USER"])

    if form.has_key('folder'):

    	folder = int(form['folder'].value)

    else:

    	folder = 0

    if form.has_key('action'):

    	if form['action'].value == 'add':

    		if form.has_key('sel'):

			keyword = int(form['sel'].value)

			db.add_folder(keyword, folder)

			print "<em>Folder successfully add.</em>"

    parents = db.get_folder_parents_and_self(folder)

    children = db.get_subfolders(folder)

    parents.reverse()

    selection = map(lambda e: e['keyword'], parents)

    exclusion = map(lambda e: e['keyword'], children)

    bookmarks = db.select_bookmarks(selection, exclusion)

    selected_keywords = set_selection(db,

		map(lambda e: e[0], bookmarks),

		selection, exclusion)

    keywords = []

    for keyword in selected_keywords:

    	if not keyword['id'] in exclusion:

		keywords.append(keyword)

    for child in children:

    	child['count'] = len(db.select_bookmarks(selection + [child['keyword']]))

    bookmarks = db.get_bookmarks(map(lambda bk: bk[0], bookmarks))

    bookmarks = map(lambda bk: bk.dict(), bookmarks)

    output(db.get_preferences(), parents.pop(), parents, children, bookmarks, keywords)