from flask import render_template, request, jsonify
from www import app
import json, logging, os, glob
from lxml import etree as et
import config

def list_all(d, ext):

	if not os.path.isdir(d):
		logging.error(d + " is not a valid directory.")
		return None

	return [os.path.basename(f) for f in glob.glob(os.path.join(d, "*." + ext))]

def read_xml(d, fn):
	fp = os.path.join(d, fn)
	if not os.path.isfile(fp):
		return None

	root = et.parse(fp).getroot()
	xml_data = []
	for m in root.findall('mails/mail'):
		nbr_str = m.find('nbr').text
		date_str = m.find('date').text
		from_str = m.find('from').text
		subject_str = m.find('subject').text
		xml_data.append({'nbr': nbr_str, 'date': date_str, 'from': from_str, 'subject': subject_str})

	return xml_data


def update_nbr_xml(d, fn, nbr, new_nbr):
	fp = os.path.join(d, fn)
	if not os.path.isfile(fp):
		return False

	tr = et.parse(fp)
	m = tr.xpath(".//nbr[text()='" + nbr + "']")

	if m is None or len(m) > 1:
		return False


	# replace nbr
	m[0].text = new_nbr

	# order all mails according to their nbr
	mails = tr.find('mails')
	print(mails)

	mails[:] = sorted(mails, key=lambda x: float(x.find('nbr').text))

	tr.write(fp)
	return True





@app.route('/')
def index():
	return render_template("index.html")

@app.route('/xml', methods = ['GET', 'POST'])
def xml():
	if request.method == 'GET':
		li = list_all(config.xml['path'], 'xml')
		li = sorted(li, key=lambda x: int(x.split('.')[0]))
		return render_template("xml_all.html", files=li)

@app.route('/xml/<path:fn>', methods = ['GET', 'POST'])
def xmlfn(fn):
	if request.method == 'GET':
		data = read_xml(config.xml['path'], fn)
		if data is not None:
			return render_template("xml.html", fn=fn, data=data)
		else:
			return "File: " + fn + "does not exist."
	elif request.method == 'POST':
		data = request.form		
		print(data)
		a = data.get('action')
		if a == "update":
			logging.info("POST update for " + fn + " -- " + data.get('nbr') + " ++ " + data.get('new_nbr'))
			if update_nbr_xml(config.xml['path'], fn, data.get('nbr'), data.get('new_nbr')):
				return "ok"
		return "-"