from os import environ
import requests
import time
from flask import *
from urllib.parse import quote

from drama.helpers.security import *
from drama.helpers.wrappers import *
from drama.classes import *
from drama.__main__ import app


def send_mail(to_address, subject, html, plaintext=None, files={},
			  from_address="Drama <noreply@mail.rdrama.net>"):

	url = "https://api.mailgun.net/v3/rdrama.net/messages"

	data = {"from": from_address,
			"to": [to_address],
			"subject": subject,
			"text": plaintext,
			"html": html,
			}

	return requests.post(url,
						 auth=(
							 "api", environ.get("MAILGUN_KEY").strip()),
						 data=data,
						 files=[("attachment", (k, files[k])) for k in files]
						 )


def send_verification_email(user, email=None):

	if not email:
		email = user.email

	url = f"https://{app.config['SERVER_NAME']}/activate"
	now = int(time.time())

	token = generate_hash(f"{email}+{user.id}+{now}")
	params = f"?email={quote(email)}&id={user.id}&time={now}&token={token}"

	link = url + params

	send_mail(to_address=email,
			  html=render_template("email/email_verify.html",
								   action_url=link,
								   v=user),
			  subject="Validate your Drama account email."
			  )


@app.route("/api/verify_email", methods=["POST"])
@is_not_banned
def api_verify_email(v):

	send_verification_email(v)

	return "", 204


@app.route("/activate", methods=["GET"])
@auth_desired
def activate(v):

	email = request.args.get("email", "")
	id = request.args.get("id", "")
	timestamp = int(request.args.get("time", "0"))
	token = request.args.get("token", "")

	if int(time.time()) - timestamp > 3600:
		return render_template("message.html", v=v, title="Verification link expired.",
							   message="That link has expired. Visit your settings to send yourself another verification email."), 410

	if not validate_hash(f"{email}+{id}+{timestamp}", token):
		abort(403)

	user = g.db.query(User).filter_by(id=id).first()
	if not user:
		abort(404)

	if user.is_activated and user.email == email:
		return render_template("message_success.html", v=v,
							   title="Email already verified.", message="Email already verified."), 404

	user.email = email
	user.is_activated = True

	if not any([b.badge_id == 2 for b in user.badges]):
		mail_badge = Badge(user_id=user.id,
						   badge_id=2,
						   created_utc=time.time())
		g.db.add(mail_badge)

	g.db.add(user)

	return render_template("message_success.html", v=v, title="Email verified.", message=f"Your email {email} has been verified. Thank you.")