2020-02-10 16:17:05 +00:00
|
|
|
from flask import render_template, request, redirect, url_for, flash
|
|
|
|
from flask_login import login_required, current_user
|
|
|
|
from flask_wtf import FlaskForm
|
2023-01-12 11:34:14 +00:00
|
|
|
from itsdangerous import TimestampSigner
|
2023-01-17 10:55:34 +00:00
|
|
|
from wtforms import validators, IntegerField
|
2020-02-10 16:17:05 +00:00
|
|
|
from wtforms.fields.html5 import EmailField
|
|
|
|
|
2023-01-11 21:08:52 +00:00
|
|
|
from app import parallel_limiter
|
2023-05-09 14:35:58 +00:00
|
|
|
from app.config import MAILBOX_SECRET
|
2020-02-10 16:17:05 +00:00
|
|
|
from app.dashboard.base import dashboard_bp
|
2021-10-12 12:36:47 +00:00
|
|
|
from app.db import Session
|
2020-02-10 16:17:05 +00:00
|
|
|
from app.log import LOG
|
2023-05-09 14:35:58 +00:00
|
|
|
from app.mailbox_utils import (
|
|
|
|
create_mailbox_and_send_verification,
|
|
|
|
set_mailbox_verified,
|
|
|
|
MailboxError,
|
|
|
|
delete_mailbox,
|
|
|
|
)
|
|
|
|
from app.models import Mailbox
|
2022-10-27 08:04:47 +00:00
|
|
|
from app.utils import CSRFValidationForm
|
2020-02-10 16:17:05 +00:00
|
|
|
|
|
|
|
|
|
|
|
class NewMailboxForm(FlaskForm):
|
|
|
|
email = EmailField(
|
|
|
|
"email", validators=[validators.DataRequired(), validators.Email()]
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2023-01-17 10:55:34 +00:00
|
|
|
class DeleteMailboxForm(FlaskForm):
|
|
|
|
mailbox_id = IntegerField(
|
|
|
|
validators=[validators.DataRequired()],
|
|
|
|
)
|
|
|
|
transfer_mailbox_id = IntegerField()
|
|
|
|
|
|
|
|
|
2020-02-10 16:17:05 +00:00
|
|
|
@dashboard_bp.route("/mailbox", methods=["GET", "POST"])
|
|
|
|
@login_required
|
2023-01-11 21:08:52 +00:00
|
|
|
@parallel_limiter.lock(only_when=lambda: request.method == "POST")
|
2020-02-10 16:17:05 +00:00
|
|
|
def mailbox_route():
|
2020-04-28 18:22:37 +00:00
|
|
|
mailboxes = (
|
2021-10-12 12:36:47 +00:00
|
|
|
Mailbox.filter_by(user_id=current_user.id)
|
2020-04-28 18:22:37 +00:00
|
|
|
.order_by(Mailbox.created_at.desc())
|
|
|
|
.all()
|
|
|
|
)
|
2020-02-10 16:17:05 +00:00
|
|
|
|
|
|
|
new_mailbox_form = NewMailboxForm()
|
2022-10-27 08:04:47 +00:00
|
|
|
csrf_form = CSRFValidationForm()
|
2023-01-17 10:55:34 +00:00
|
|
|
delete_mailbox_form = DeleteMailboxForm()
|
2020-02-10 16:17:05 +00:00
|
|
|
|
|
|
|
if request.method == "POST":
|
|
|
|
if request.form.get("form-name") == "delete":
|
2023-01-17 10:55:34 +00:00
|
|
|
if not delete_mailbox_form.validate():
|
|
|
|
flash("Invalid request", "warning")
|
|
|
|
return redirect(request.url)
|
2023-05-09 14:35:58 +00:00
|
|
|
try:
|
|
|
|
mailbox = delete_mailbox(
|
|
|
|
current_user,
|
|
|
|
delete_mailbox_form.mailbox_id.data,
|
|
|
|
delete_mailbox_form.transfer_mailbox_id.data,
|
|
|
|
)
|
|
|
|
except MailboxError as e:
|
|
|
|
flash(str(e), "error")
|
2020-02-23 08:41:53 +00:00
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
|
|
|
|
2021-08-15 15:50:47 +00:00
|
|
|
flash(
|
|
|
|
f"Mailbox {mailbox.email} scheduled for deletion."
|
|
|
|
f"You will receive a confirmation email when the deletion is finished",
|
|
|
|
"success",
|
|
|
|
)
|
2020-02-10 16:17:05 +00:00
|
|
|
|
2020-02-23 08:51:26 +00:00
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
|
|
|
if request.form.get("form-name") == "set-default":
|
2023-01-17 10:55:34 +00:00
|
|
|
if not csrf_form.validate():
|
|
|
|
flash("Invalid request", "warning")
|
|
|
|
return redirect(request.url)
|
|
|
|
mailbox_id = request.form.get("mailbox_id")
|
2020-02-23 08:51:26 +00:00
|
|
|
mailbox = Mailbox.get(mailbox_id)
|
|
|
|
|
|
|
|
if not mailbox or mailbox.user_id != current_user.id:
|
|
|
|
flash("Unknown error. Refresh the page", "warning")
|
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
|
|
|
|
|
|
|
if mailbox.id == current_user.default_mailbox_id:
|
|
|
|
flash("This mailbox is already default one", "error")
|
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
|
|
|
|
|
|
|
if not mailbox.verified:
|
|
|
|
flash("Cannot set unverified mailbox as default", "error")
|
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
|
|
|
|
|
|
|
current_user.default_mailbox_id = mailbox.id
|
2021-10-12 12:36:47 +00:00
|
|
|
Session.commit()
|
2020-02-23 08:51:26 +00:00
|
|
|
flash(f"Mailbox {mailbox.email} is set as Default Mailbox", "success")
|
|
|
|
|
2020-02-10 16:17:05 +00:00
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
|
|
|
|
|
|
|
elif request.form.get("form-name") == "create":
|
2020-02-29 11:12:55 +00:00
|
|
|
if not current_user.is_premium():
|
|
|
|
flash("Only premium plan can add additional mailbox", "warning")
|
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
2023-05-09 14:35:58 +00:00
|
|
|
mailbox_email = new_mailbox_form.email.data.lower().strip().replace(" ", "")
|
|
|
|
try:
|
|
|
|
new_mailbox = create_mailbox_and_send_verification(
|
|
|
|
current_user, mailbox_email
|
2020-09-17 15:02:50 +00:00
|
|
|
)
|
2023-05-09 14:35:58 +00:00
|
|
|
flash(
|
|
|
|
f"You are going to receive an email to confirm {mailbox_email}.",
|
|
|
|
"success",
|
|
|
|
)
|
|
|
|
return redirect(
|
|
|
|
url_for(
|
|
|
|
"dashboard.mailbox_detail_route",
|
|
|
|
mailbox_id=new_mailbox.id,
|
2020-03-08 21:59:19 +00:00
|
|
|
)
|
2023-05-09 14:35:58 +00:00
|
|
|
)
|
|
|
|
except MailboxError as e:
|
|
|
|
flash(str(e), "error")
|
2020-02-10 16:17:05 +00:00
|
|
|
|
|
|
|
return render_template(
|
|
|
|
"dashboard/mailbox.html",
|
|
|
|
mailboxes=mailboxes,
|
|
|
|
new_mailbox_form=new_mailbox_form,
|
2023-01-17 10:55:34 +00:00
|
|
|
delete_mailbox_form=delete_mailbox_form,
|
2022-10-27 08:04:47 +00:00
|
|
|
csrf_form=csrf_form,
|
2020-02-10 16:17:05 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
@dashboard_bp.route("/mailbox_verify")
|
|
|
|
def mailbox_verify():
|
2023-01-12 11:34:14 +00:00
|
|
|
s = TimestampSigner(MAILBOX_SECRET)
|
2020-02-10 16:17:05 +00:00
|
|
|
mailbox_id = request.args.get("mailbox_id")
|
|
|
|
|
|
|
|
try:
|
2023-01-12 11:34:14 +00:00
|
|
|
r_id = int(s.unsign(mailbox_id, max_age=900))
|
2020-03-09 08:17:40 +00:00
|
|
|
except Exception:
|
2020-02-10 16:17:05 +00:00
|
|
|
flash("Invalid link. Please delete and re-add your mailbox", "error")
|
2020-03-09 08:17:40 +00:00
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
2020-02-10 16:17:05 +00:00
|
|
|
else:
|
|
|
|
mailbox = Mailbox.get(r_id)
|
2020-05-06 08:06:05 +00:00
|
|
|
if not mailbox:
|
|
|
|
flash("Invalid link", "error")
|
|
|
|
return redirect(url_for("dashboard.mailbox_route"))
|
|
|
|
|
2023-05-09 14:35:58 +00:00
|
|
|
set_mailbox_verified(mailbox)
|
2020-02-10 16:17:05 +00:00
|
|
|
|
|
|
|
LOG.d("Mailbox %s is verified", mailbox)
|
2020-03-14 13:45:37 +00:00
|
|
|
|
|
|
|
return render_template("dashboard/mailbox_validation.html", mailbox=mailbox)
|