from keywords import get_keywords

from webutils import load_index

from templates import Template

def print_kw(keywords, prefs):

    tmpl = Template("edit_kw.tmpl", prefs, "", "")

    tmpl.set('Keywords', keywords)

    print tmpl.process()

def print_confirm(step, sel_keywords, keywords, prefs):

    if step == 'delete':

    	tmpl = Template("kw_delete.tmpl", prefs)

    elif step == 'merge':

    	tmpl = Template("kw_merge.tmpl", prefs)

    elif step == 'rename':

    	tmpl = Template("kw_rename.tmpl", prefs)

    tmpl.set("Selected", sel_keywords)

    tmpl.set("Keywords", keywords)

    print tmpl.process()

def list_kw(db, prefs):

    kw = db.get_all_keywords()[1:]

    kw = map(lambda elem: {

   	 'id' : elem[0],

	 'keyword' : elem[1] }, kw)

    kw.sort(lambda l,r: cmp(l['keyword'],r['keyword']))

    print_kw(kw, prefs)

def confirm_kw(step, db, prefs, form):

    ids = get_keywords(form, 'kw')

    keywords = map(lambda e: { 'id': e[0], 'keyword': e[1]},

   	 db.get_all_keywords()[1:])

    keywords.sort(lambda x, y: cmp(x['keyword'],y['keyword']))

    sel_keywords = filter(lambda e: e['id'] in ids, keywords)

    keywords = filter(lambda e: e['id'] not in ids, keywords)

    print_confirm(step, sel_keywords, keywords, prefs)

def do_merge(db, prefs, form):

	fromids = get_keywords(form, 'id')

	toids = get_keywords(form, 'kw')

	remove_them = form.has_key('remove')

	if remove_them:

		lastid = toids.pop()

	status = "result"

	status_msg = []

	for id in fromids:

		fromkw = db.get_keyword(id)

		for dest in toids:

			tokw = db.get_keyword(dest)

			status_msg.append("Adding '%s' into '%s'." % (fromkw, tokw))

			db.add_keywords(id, dest) 

		if remove_them:

			tokw = db.get_keyword(lastid)

			status_msg.append("Merging '%s' into '%s'." % (fromkw, tokw))

			db.merge_keywords(id, lastid)

	load_index(db, prefs, form, status, "<br/>".join(status_msg))

def do_rename(db, prefs, form):

	ids = get_keywords(form, 'id')

	status = "result"

	status_msg = []

	for id in ids:

		oldname = db.get_keyword(id)

		newname = form['name%d' % id].value

		status_msg.append("Renaming '%s' into '%s'." % (oldname, newname))

		db.update_keyword(id, newname)

	load_index(db, prefs, form, status, "<br/>".join(status_msg))

def do_delete(db, prefs, form):

	ids = get_keywords(form, 'id')

	status = "result"

	status_msg = []

	for id in ids:

		kw = db.get_keyword(id)

		status_msg.append("Removing '%s'." % kw)

		db.remove_keyword(id)

	load_index(db, prefs, form, status, "<br/>".join(status_msg))

def do_it(action, db, prefs, form):

	step = 'list'

	if form.has_key('step'):

		step = form['step'].value

	if step == 'list':

		list_kw(db, prefs)

	elif step == 'cancel':

		load_index(db, prefs, form, "err", "Operation cancel")

	elif step == 'do_delete':

		do_delete(db, prefs, form)

	elif step == 'do_merge':

		do_merge(db, prefs, form)

	elif step == 'do_rename':

		do_rename(db, prefs, form)

	else:

		confirm_kw(step, db, prefs, form)