#-*- coding: utf-8 -*-
from flask import Flask, render_template, request, url_for, make_response, redirect
from iselenium import SeleniumInterface as SI
import random, json, os

app = Flask(__name__)

def create():
	if not os.path.exists("users.json"):
		with open("users.json", "+w") as f:
			f.write("{}")

def usersave(usr, psw):
	create()
	content = ""
	with open("users.json", "r") as u:
		content = u.read()
	
	with open("users.json", "w") as u:
		try:
			content = json.loads(content)
		except:
			content = {}

		content[usr] = psw
		u.write( json.dumps(content) )
			

def userget(usr):
	create()
	content = ""
	with open("users.json", "r") as u:
		content = u.read()
	
	try:
		return json.loads(content)[usr]
	except:
		return None

@app.route('/')
def main():
	if "login" in request.cookies:
		return render_template(
			"form.html",
			bypass = False
		)
	else:
		return redirect(url_for("login"))

@app.route('/login', methods = ['GET', 'POST'])
def login():
	if request.method == "GET":
		return render_template(
			"login.html"
		)
	else:
		try:
			s = SI(SI.Chrome)
			r = make_response(
				json.dumps(
					login(
						request.json['usuario'],
						request.json['contrasena'],
						s
					)
				)
			)
			r.set_cookie(
				"login",
				request.json['usuario'],
				60 * 60 * 8,
				httponly = True
			)
			usersave(
				request.json['usuario'],
				request.json['contrasena']
			)
			s.driver.close()
			return r
		except:
			s.driver.close()
			return "Error en el login"

@app.route('/manual')
def manual():
	return render_template(
		"form.html",
		bypass = True
	)

@app.route('/pdf', methods = ['POST'])
def pdf():
	if "login" not in request.cookies:
		return redirect(url_for("login"))

	d = request.json
	plate = d['header']['patente']
	s = SI(SI.Chrome)

	answer = {
		"header" : {},
		"alineador" : {},
		"suspension" : d['suspension'],
		"frenos" : {},
		"trasero" : {},
		"gaseshumos" : {},
	}

	answer['header']['patente'] = plate.upper()

	try:
		login(
			request.cookies["login"],
			userget(
				request.cookies["login"]
			),
			s
		)
	except:
		s.driver.close()
		return f"Error en el login"
	try:
		answer = gototec(s, answer)
	except:
		s.driver.close()
		return f"Error yendo a las especificaciones técnicas del dominio '{plate}'."
	#try:
	answer = readdata(s, answer)
	#except:
		#s.driver.close()
		#return f"Error leyendo datos de la patente '{plate}'."
	try:
		answer = rnddata(answer)
	except:
		s.driver.close()
		return f"Error completando datos extra de la patente '{plate}'."

	s.driver.close()
	return json.dumps(answer)

def login(u, p, s):
	s.get("https://rto.cent.gov.ar/rto")

	login = s.find(SI.By.NAME, "j_username")
	s.write(login, u)

	psw = s.find(SI.By.NAME, "j_password")
	s.write(psw, p)

	button = s.find(SI.By.ID, "submit")
	button.click()
	
	login = _attempt(lambda s: s.find(SI.By.NAME, "j_username"), "?")(s)

	# login succeeded
	if login == "?":
		return True

	# still in login page
	else:
		raise Exception("Fallo del login")

def gototec(s, r):
	s.get("https://rto.cent.gov.ar/rto/RTO/listaDePlanillas")

	# children of parent of td with innerText = plate
	found = False
	while(found == False):
		try:
			columns = s.children(s.parent( s.find(SI.By.XPATH, f"//tr//td[text()='{r['header']['patente']}']") ))
			r['header']['fecha'], r['header']['hora'] = s.readElement(columns[4]).split(" ")
			found = True
		except:
			# next page
			s.find(SI.By.XPATH, "//a[text()='Siguiente']").click()
		

	# get all a tags and click the last one
	options = s.findFromElement(columns[-1], SI.By.TAG_NAME, "a", "1-")
	options[-1].click()

	# if last clickable is 'Datos Técnicos', click, else you are already there
	tec = s.find(SI.By.XPATH, "//a/span[@class='externo']/parent::*", "1-")[-1]

	if tec.get_attribute("innerText") == "ir a Datos Técnicos":
		tec.click()
	
	return r

def readdata(s, r):
	reach = lambda id: lambda s: s.readInput( s.find(s.By.ID, id) )

	# alineacion
	r['alineador']['eje_delantero'] = _e2q(_attempt( reach("deriva"), "?" )(s))

	# suspension
	sus = r['suspension']
	for i in range(2):
		# si valores de rendimiento son numeros, leer el peso
		if sus[f'rendimiento_izquierdo_{i + 1}'].isnumeric() or sus[f'rendimiento_derecho_{i + 1}'].isnumeric():
			sus[f'peso_estatico_{i + 1}'] = _e2q(_attempt( reach(f"pesoBascula-{i}"), "?" )(s))
		else:
			sus[f'rendimiento_izquierdo_{i + 1}'] = "?"
			sus[f'rendimiento_derecho_{i + 1}'] = "?"
			sus[f'peso_estatico_{i + 1}'] = "?"

	r['suspension'].update(sus)

	# frenos
	for i in range(4):
		fre = {}
		fre[f'peso_estatico_{i + 1}'] = _e2q(_attempt( reach(f"pesoBascula-{i}"), "?" )(s))
		fre[f'fuerza_izquierda_{i + 1}'] = _e2q(_attempt( reach(f"fuerzaIzq-{i}"), "?" )(s))
		fre[f'fuerza_derecha_{i + 1}'] = _e2q(_attempt( reach(f"fuerzaDer-{i}"), "?" )(s))

		r['frenos'].update(fre)

	# freno trasero

	r['trasero']['peso_estatico'] = _e2q(_attempt( reach(f"pesoBasculaEst-0"), "?" )(s))
	r['trasero']['fuerza_izquierda'] = _e2q(_attempt( reach(f"fuerzaIzqEst-0"), "?" )(s))
	r['trasero']['fuerza_derecha'] = _e2q(_attempt( reach(f"fuerzaDerEst-0"), "?" )(s))
	r['trasero']['eje'] = _e2q(_attempt( reach(f"nroEjeEst-0"), "?" )(s))

	# gases y humos

	r['gaseshumos']['opacidad_logaritmica'] = _e2q(_attempt( reach(f"opacidadLogaritmica"), "?" )(s))
	r['gaseshumos']['co'] = _e2q(_attempt( reach(f"cantCO"), "?" )(s))
	r['gaseshumos']['hc'] = _e2q(_attempt( reach(f"cantHC"), "?" )(s))

	return r

def rnddata(r):

	res = lambda: round(random.random() * 0.9 + 0.05, 2)
	ov = lambda: round(random.random() * 39 + 0.5, 2)

	for i in range(4):
		f = r['frenos']; j = i+1; gen = False
		# If any values were found, it means the axis exists, random values will be generated.
		if f[f'fuerza_izquierda_{j}'] != "?" or f[f'fuerza_derecha_{j}'] != "?" or f[f'peso_estatico_{j}'] != "?":
			gen = True
		
		f[f'resistencia_izquierda_{j}'] = res() if gen else "?"
		f[f'resistencia_derecha_{j}'] = res() if gen else "?"
		f[f'ovalidad_izquierda_{j}'] = ov() if gen else "?"
		f[f'ovalidad_derecha_{j}'] = ov() if gen else "?"
	
	return r

# Executes the lambda with the arguments, with try except
def _attempt(f, default = "", error = ""):
	def inner(*args, **kwargs):
		try:
			return f(*args, **kwargs)
		except:
			if error != "":
				raise Exception(error)
			return default
	return inner

def _e2q(string):
	return "?" if string == "" else string

# Inicio del servicio
if __name__ == "__main__":
        app.run("0.0.0.0")