mirror of
https://github.com/simple-login/app.git
synced 2024-11-13 07:31:12 +01:00
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
import tldextract
|
|
from flask import g
|
|
from flask import jsonify, request
|
|
|
|
from app.api.base import api_bp, require_api_auth
|
|
from app.api.serializer import (
|
|
get_alias_info_v2,
|
|
serialize_alias_info_v2,
|
|
)
|
|
from app.config import MAX_NB_EMAIL_FREE_PLAN, ALIAS_LIMIT
|
|
from app.db import Session
|
|
from app.extensions import limiter
|
|
from app.log import LOG
|
|
from app.models import Alias, AliasUsedOn, AliasGeneratorEnum
|
|
from app.utils import convert_to_id
|
|
|
|
|
|
@api_bp.route("/alias/random/new", methods=["POST"])
|
|
@limiter.limit(ALIAS_LIMIT)
|
|
@require_api_auth
|
|
def new_random_alias():
|
|
"""
|
|
Create a new random alias
|
|
Input:
|
|
(Optional) note
|
|
Output:
|
|
201 if success
|
|
|
|
"""
|
|
user = g.user
|
|
if not user.can_create_new_alias():
|
|
LOG.d("user %s cannot create new random alias", user)
|
|
return (
|
|
jsonify(
|
|
error=f"You have reached the limitation of a free account with the maximum of "
|
|
f"{MAX_NB_EMAIL_FREE_PLAN} aliases, please upgrade your plan to create more aliases"
|
|
),
|
|
400,
|
|
)
|
|
|
|
note = None
|
|
data = request.get_json(silent=True)
|
|
if data:
|
|
note = data.get("note")
|
|
|
|
# custom alias suggestion and suffix
|
|
hostname = request.args.get("hostname")
|
|
if hostname and user.include_website_in_one_click_alias:
|
|
LOG.d("Use %s to create new alias", hostname)
|
|
# keep only the domain name of hostname, ignore TLD and subdomain
|
|
# for ex www.groupon.com -> groupon
|
|
ext = tldextract.extract(hostname)
|
|
prefix_suggestion = ext.domain
|
|
prefix_suggestion = convert_to_id(prefix_suggestion)
|
|
|
|
alias = Alias.create_new(user, prefix_suggestion, note=note)
|
|
Session.commit()
|
|
|
|
else:
|
|
scheme = user.alias_generator
|
|
mode = request.args.get("mode")
|
|
if mode:
|
|
if mode == "word":
|
|
scheme = AliasGeneratorEnum.word.value
|
|
elif mode == "uuid":
|
|
scheme = AliasGeneratorEnum.uuid.value
|
|
else:
|
|
return jsonify(error=f"{mode} must be either word or uuid"), 400
|
|
|
|
alias = Alias.create_new_random(user=user, scheme=scheme, note=note)
|
|
Session.commit()
|
|
|
|
if hostname:
|
|
AliasUsedOn.create(alias_id=alias.id, hostname=hostname, user_id=alias.user_id)
|
|
Session.commit()
|
|
|
|
return (
|
|
jsonify(alias=alias.email, **serialize_alias_info_v2(get_alias_info_v2(alias))),
|
|
201,
|
|
)
|