mirror of
https://github.com/simple-login/app.git
synced 2024-11-14 08:01:13 +01:00
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
|
from flask import render_template, request, redirect, url_for, flash
|
||
|
|
from flask_login import login_required, current_user
|
||
|
|
|
||
|
|
from app.dashboard.base import dashboard_bp
|
||
|
|
from app.log import LOG
|
||
|
|
from app.models import CustomDomain, Mailbox, SLDomain
|
||
|
|
|
||
|
|
|
||
|
|
@dashboard_bp.route("/subdomain", methods=["GET", "POST"])
|
||
|
|
@login_required
|
||
|
|
def subdomain_route():
|
||
|
|
if not current_user.can_use_subdomain:
|
||
|
|
flash("Unknown error, redirect to the home page", "error")
|
||
|
|
return redirect(url_for("dashboard.index"))
|
||
|
|
|
||
|
|
sl_domains = SLDomain.filter_by(can_use_subdomain=True).all()
|
||
|
|
subdomains = CustomDomain.filter_by(
|
||
|
|
user_id=current_user.id, is_sl_subdomain=True
|
||
|
|
).all()
|
||
|
|
|
||
|
|
errors = {}
|
||
|
|
|
||
|
|
if request.method == "POST":
|
||
|
|
if request.form.get("form-name") == "create":
|
||
|
|
if not current_user.is_premium():
|
||
|
|
flash("Only premium plan can add subdomain", "warning")
|
||
|
|
return redirect(request.url)
|
||
|
|
|
||
|
|
subdomain = request.form.get("subdomain").lower().strip()
|
||
|
|
domain = request.form.get("domain").lower().strip()
|
||
|
|
|
||
|
|
if domain not in [sl_domain.domain for sl_domain in sl_domains]:
|
||
|
|
LOG.e("Domain %s is tampered by %s", domain, current_user)
|
||
|
|
flash("Unknown error, refresh the page", "error")
|
||
|
|
return redirect(request.url)
|
||
|
|
|
||
|
|
full_domain = f"{subdomain}.{domain}"
|
||
|
|
|
||
|
|
if CustomDomain.get_by(domain=full_domain):
|
||
|
|
flash(f"{full_domain} already used", "error")
|
||
|
|
elif Mailbox.filter(
|
||
|
|
Mailbox.verified.is_(True),
|
||
|
|
Mailbox.email.endswith(f"@{full_domain}"),
|
||
|
|
).first():
|
||
|
|
flash(f"{full_domain} already used in a SimpleLogin mailbox", "error")
|
||
|
|
else:
|
||
|
|
new_custom_domain = CustomDomain.create(
|
||
|
|
is_sl_subdomain=True,
|
||
|
|
catch_all=True, # by default catch-all is enabled
|
||
|
|
domain=full_domain,
|
||
|
|
user_id=current_user.id,
|
||
|
|
verified=True,
|
||
|
|
dkim_verified=True,
|
||
|
|
spf_verified=True,
|
||
|
|
dmarc_verified=True,
|
||
|
|
ownership_verified=True,
|
||
|
|
commit=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
flash(
|
||
|
|
f"New subdomain {new_custom_domain.domain} is created",
|
||
|
|
"success",
|
||
|
|
)
|
||
|
|
|
||
|
|
return redirect(
|
||
|
|
url_for(
|
||
|
|
"dashboard.domain_detail",
|
||
|
|
custom_domain_id=new_custom_domain.id,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
return render_template(
|
||
|
|
"dashboard/subdomain.html",
|
||
|
|
sl_domains=sl_domains,
|
||
|
|
errors=errors,
|
||
|
|
subdomains=subdomains,
|
||
|
|
)
|