simple-login/app
1
1
Fork 0
mirror of https://github.com/simple-login/app.git synced 2025-02-24 07:43:54 +08:00
Code Issues Projects Releases Packages Wiki Activity

fix custom domain not correctly set on /v2/alias/custom/new and /v3/alias/custom/new

Browse source
This commit is contained in:
Son 2021-11-22 16:20:50 +01:00
parent 3add9e6db8
commit b5c2d9ee2a
1 changed files with 20 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
app/api/views/new_custom_alias.py
View file

@ -1,3 +1,6 @@
from typing import Optional
from email_validator import validate_email
from flask import g from flask import g
from flask import jsonify, request from flask import jsonify, request
from itsdangerous import SignatureExpired from itsdangerous import SignatureExpired
@ -22,10 +25,23 @@ from app.models import (
DomainDeletedAlias, DomainDeletedAlias,
Mailbox, Mailbox,
AliasMailbox, AliasMailbox,
SLDomain,
) )
from app.utils import convert_to_id from app.utils import convert_to_id
def get_custom_domain(alias_address) -> Optional[CustomDomain]:
alias_domain = validate_email(
alias_address, check_deliverability=False, allow_smtputf8=False
).domain
# handle the case a SLDomain is also a CustomDomain
if SLDomain.get_by(domain=alias_domain) is None:
custom_domain = CustomDomain.get_by(domain=alias_domain)
if custom_domain:
return custom_domain
@api_bp.route("/v2/alias/custom/new", methods=["POST"]) @api_bp.route("/v2/alias/custom/new", methods=["POST"])
@limiter.limit(ALIAS_LIMIT) @limiter.limit(ALIAS_LIMIT)
@require_api_auth @require_api_auth
@ -88,25 +104,14 @@ def new_custom_alias_v2():
LOG.d("full alias already used %s", full_alias) LOG.d("full alias already used %s", full_alias)
return jsonify(error=f"alias {full_alias} already exists"), 409 return jsonify(error=f"alias {full_alias} already exists"), 409
custom_domain_id = None custom_domain = get_custom_domain(full_alias)
if alias_suffix.startswith("@"):
alias_domain = alias_suffix[1:]
domain = CustomDomain.get_by(domain=alias_domain)
# check if the alias is currently in the domain trash
if domain and DomainDeletedAlias.get_by(domain_id=domain.id, email=full_alias):
LOG.d(f"Alias {full_alias} is currently in the {domain.domain} trash. ")
return jsonify(error=f"alias {full_alias} in domain trash"), 409
if domain:
custom_domain_id = domain.id
alias = Alias.create( alias = Alias.create(
user_id=user.id, user_id=user.id,
email=full_alias, email=full_alias,
mailbox_id=user.default_mailbox_id, mailbox_id=user.default_mailbox_id,
note=note, note=note,
custom_domain_id=custom_domain_id, custom_domain_id=custom_domain.id if custom_domain else None,
) )
Session.commit() Session.commit()
@ -208,12 +213,7 @@ def new_custom_alias_v3():
LOG.d("full alias already used %s", full_alias) LOG.d("full alias already used %s", full_alias)
return jsonify(error=f"alias {full_alias} already exists"), 409 return jsonify(error=f"alias {full_alias} already exists"), 409
custom_domain_id = None custom_domain = get_custom_domain(full_alias)
if alias_suffix.startswith("@"):
alias_domain = alias_suffix[1:]
domain = CustomDomain.get_by(domain=alias_domain)
if domain:
custom_domain_id = domain.id
alias = Alias.create( alias = Alias.create(
user_id=user.id, user_id=user.id,
@ -221,7 +221,7 @@ def new_custom_alias_v3():
note=note, note=note,
name=name or None, name=name or None,
mailbox_id=mailboxes[0].id, mailbox_id=mailboxes[0].id,
custom_domain_id=custom_domain_id, custom_domain_id=custom_domain.id if custom_domain else None,
) )
Session.flush() Session.flush()