From 6c4a173de544cdc17988a752a3838bf6a7a2269b Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Thu, 28 Nov 2019 22:05:32 +0000
Subject: [PATCH] create /api/alias/new using api-key as authentication

---
 app/api/__init__.py       |  1 +
 app/api/base.py           |  3 ++
 app/api/views/__init__.py |  0
 app/api/views/index.py    | 90 +++++++++++++++++++++++++++++++++++++++
 server.py                 |  3 ++
 5 files changed, 97 insertions(+)
 create mode 100644 app/api/__init__.py
 create mode 100644 app/api/base.py
 create mode 100644 app/api/views/__init__.py
 create mode 100644 app/api/views/index.py

diff --git a/app/api/__init__.py b/app/api/__init__.py
new file mode 100644
index 00000000..86bd305b
--- /dev/null
+++ b/app/api/__init__.py
@@ -0,0 +1 @@
+from .views import index
diff --git a/app/api/base.py b/app/api/base.py
new file mode 100644
index 00000000..4b383883
--- /dev/null
+++ b/app/api/base.py
@@ -0,0 +1,3 @@
+from flask import Blueprint
+
+api_bp = Blueprint(name="api", import_name=__name__, url_prefix="/api")
diff --git a/app/api/views/__init__.py b/app/api/views/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/app/api/views/index.py b/app/api/views/index.py
new file mode 100644
index 00000000..40ae1ece
--- /dev/null
+++ b/app/api/views/index.py
@@ -0,0 +1,90 @@
+import random
+
+import arrow
+from flask import jsonify, request
+
+from app.api.base import api_bp
+from app.config import EMAIL_DOMAIN
+from app.extensions import db
+from app.log import LOG
+from app.models import ApiKey, AliasUsedOn, GenEmail, User, DeletedAlias
+from app.utils import random_string
+
+
+@api_bp.route("/alias/new", methods=["GET", "POST"])
+def index():
+    """
+    the incoming request must provide a valid api-key in "Authentication" header and
+    the payload must contain "hostname"
+    """
+    api_code = request.headers.get("Authentication")
+    api_key = ApiKey.get_by(code=api_code)
+
+    if not api_key:
+        return jsonify(error="Wrong api key"), 401
+
+    data = request.get_json()
+    if not data:
+        return jsonify(error="hostname must be provided")
+
+    hostname = data.get("hostname")
+
+    # Update api key stats
+    api_key.last_used = arrow.now()
+    api_key.times += 1
+    db.session.commit()
+
+    user = api_key.user
+
+    q = db.session.query(AliasUsedOn, GenEmail, User).filter(
+        AliasUsedOn.gen_email_id == GenEmail.id,
+        GenEmail.user_id == user.id,
+        AliasUsedOn.hostname == hostname,
+    )
+
+    r = q.first()
+    if r:
+        _, alias, _ = r
+        LOG.d("found alias %s %s %s", alias, hostname, user)
+        return jsonify(alias=alias.email)
+
+    # use a custom alias for this user
+    if user.is_premium():
+        LOG.d("create new custom alias %s %s", hostname, user)
+
+        # generate a custom email
+        found = False
+        while not found:
+            email_suffix = random_string(6)
+            email_prefix = hostname.replace(".", "_")
+            full_email = f"{email_prefix}.{email_suffix}@{EMAIL_DOMAIN}"
+
+            # check if email already exists. Very rare that an email is already used
+            if GenEmail.get_by(email=full_email) or DeletedAlias.get_by(
+                    email=full_email
+            ):
+                LOG.warning("full_email already used %s. Retry", full_email)
+            else:
+                found = True
+
+        gen_email = GenEmail.create(email=full_email, user_id=user.id, custom=True)
+        db.session.flush()
+
+        AliasUsedOn.create(gen_email_id=gen_email.id, hostname=hostname)
+        db.session.commit()
+
+        return jsonify(alias=full_email)
+    else:
+        # choose randomly from user non-custom alias
+        aliases = db.session.query(GenEmail).filter(GenEmail.user_id == user.id).all()
+        if aliases:
+            alias = random.choice(aliases)
+        else:
+            LOG.d("user %s has no alias, create one", user)
+            alias = GenEmail.create_new_gen_email(user.id)
+            db.session.commit()
+
+        AliasUsedOn.create(gen_email_id=alias.id, hostname=hostname)
+        db.session.commit()
+
+        return jsonify(alias=alias.email)
diff --git a/server.py b/server.py
index 580e24be..3f698c29 100644
--- a/server.py
+++ b/server.py
@@ -12,6 +12,7 @@ from flask_login import current_user
 from sentry_sdk.integrations.flask import FlaskIntegration
 
 from app.admin_model import SLModelView, SLAdminIndexView
+from app.api.base import api_bp
 from app.auth.base import auth_bp
 from app.config import (
     DB_URI,
@@ -38,6 +39,7 @@ from app.models import (
     RedirectUri,
     Subscription,
     PlanEnum,
+    ApiKey,
 )
 from app.monitor.base import monitor_bp
 from app.oauth.base import oauth_bp
@@ -153,6 +155,7 @@ def register_blueprints(app: Flask):
     app.register_blueprint(oauth_bp, url_prefix="/oauth2")
 
     app.register_blueprint(discover_bp)
+    app.register_blueprint(api_bp)
 
 
 def set_index_page(app):