#!/usr/bin/python

from templates import Template
from keywords import set_selection, sort_keywords

def do_folder_action(db, form, folder):
   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>"
		return keyword
   elif form['action'].value == 'delete':
   	if form.has_key('delete'):
		keyword = int(form['delete'].value)
		db.remove_folders(keyword)
		print "<em>Folder successfully remove.</em>"
		return keyword
   return -1

def sort_kw_with_removed(l, r):
	result = l['removed'] - r['removed']
	if result == 0:
		result = l['count'] - r['count']
	return -int(result)

def output(prefs, folder, parents, children, bookmarks, keywords):
    tmpl = Template("folders.tmpl", prefs)
    tmpl.set('Parents', parents)
    tmpl.set('folder', folder['id'])
    tmpl.set('name', folder['name'])
    tmpl.set('total', folder['count'])
    tmpl.set('subtotal', len(bookmarks))
    bookmarks.sort(lambda l,r: cmp(l['name'],r['name']))
    tmpl.set('Bookmarks', bookmarks)
    tmpl.set('Subfolders', children)
    keywords.sort(sort_kw_with_removed)
    tmpl.set('Keywords', keywords)
    print tmpl.process()

def do_folders(db, prefs, form = None):
    folder = 0
    if form:
    	if form.has_key('folder'):
    		folder = int(form['folder'].value)
   	if form.has_key('action'):
		do_folder_action(db, form, folder)
    parents = db.get_folder_parents_and_self(folder)
    parents.reverse()
    children = db.get_subfolders(folder)
    selection = map(lambda e: e['keyword'], parents)
    exclusion = map(lambda e: e['keyword'], children)
    folders_keywords = []
    for parent in parents:
	folders_keywords.append(parent['keyword'])
	parent['count'] = db.get_bookmarks_count(folders_keywords)
    for child in children:
	folders_keywords.append(child['keyword'])
	# This is really selection, not folders_keywords
	child['count'] = db.get_bookmarks_count(selection + [child['keyword']])
    bookmarks = map(lambda bk: bk[0], db.select_bookmarks(selection))
    allkw = db.get_keywords(bookmarks)
    keywords = []
    for keyword in allkw:
	if keyword[0] not in folders_keywords:
		kw = { 'id': keyword[0], 'keyword': keyword[1] }
		kwsel = selection + [kw['id']]
		kw['count'] = db.get_bookmarks_count(kwsel)
		kw['removed'] = db.get_bookmarks_count(kwsel, exclusion)
		keywords.append(kw)
    del allkw, folders_keywords
    bookmarks = db.select_bookmarks(selection, exclusion)
    if len(bookmarks) <> 0:
   	bookmarks = db.get_bookmarks(map(lambda bk: bk[0], bookmarks))
    	bookmarks = map(lambda bk: bk.dict(), bookmarks)
    output(prefs, parents.pop(), parents, children, bookmarks, keywords)