From 62c3b4cd02564b27747410438f3ef763b31349f1 Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Fri, 30 Aug 2019 22:42:06 +0200
Subject: [PATCH] highlight newly created gen-email

---
 app/config.py                                |  3 +++
 app/dashboard/templates/dashboard/index.html |  4 ++--
 app/dashboard/views/custom_alias.py          |  9 ++++++---
 app/dashboard/views/index.py                 | 16 ++++++++++++++--
 static/style.css                             |  6 ++++++
 5 files changed, 31 insertions(+), 7 deletions(-)

diff --git a/app/config.py b/app/config.py
index d3e941a8..e60ffdd6 100644
--- a/app/config.py
+++ b/app/config.py
@@ -86,3 +86,6 @@ FACEBOOK_CLIENT_SECRET = os.environ["FACEBOOK_CLIENT_SECRET"]
 
 # in seconds
 AVATAR_URL_EXPIRATION = 3600 * 24 * 7  # 1h*24h/d*7d=1week
+
+# session key
+HIGHLIGHT_GEN_EMAIL_ID = "highlight_gen_email_id"
\ No newline at end of file
diff --git a/app/dashboard/templates/dashboard/index.html b/app/dashboard/templates/dashboard/index.html
index 0c6aec09..14621d1e 100644
--- a/app/dashboard/templates/dashboard/index.html
+++ b/app/dashboard/templates/dashboard/index.html
@@ -50,7 +50,7 @@
             </thead>
             <tbody>
             {% for gen_email in gen_emails %}
-              <tr>
+              <tr {% if gen_email.id == highlight_gen_email_id %} class="highlight-row" {% endif %}>
                 <td>
                   <div>
                     <a href="mailto: {{ gen_email.email }}">{{ gen_email.email }}</a>
@@ -77,7 +77,7 @@
                       {% if gen_email.enabled %}
                         <button class="btn btn-secondary btn-sm"
                             {% if loop.index ==1 %}
-                                data-intro="By triggering the test email, 
+                                data-intro="By triggering the test email,
                             SimpleLogin server will send an email to this alias
                             and this email should arrive to your personal email inbox 🚀"
                             {% endif %}
diff --git a/app/dashboard/views/custom_alias.py b/app/dashboard/views/custom_alias.py
index 1129cd3c..4c36fd77 100644
--- a/app/dashboard/views/custom_alias.py
+++ b/app/dashboard/views/custom_alias.py
@@ -1,9 +1,9 @@
-from flask import render_template, redirect, url_for, flash, request
+from flask import render_template, redirect, url_for, flash, request, session
 from flask_login import login_required, current_user
 from flask_wtf import FlaskForm
 from wtforms import StringField, validators
 
-from app.config import EMAIL_DOMAIN
+from app.config import EMAIL_DOMAIN, HIGHLIGHT_GEN_EMAIL_ID
 from app.dashboard.base import dashboard_bp
 from app.extensions import db
 from app.log import LOG
@@ -43,10 +43,13 @@ def custom_alias():
             else:
                 # create the new alias
                 LOG.d("create custom alias %s for user %s", full_email, current_user)
-                GenEmail.create(email=full_email, user_id=current_user.id, custom=True)
+                gen_email = GenEmail.create(
+                    email=full_email, user_id=current_user.id, custom=True
+                )
                 db.session.commit()
 
                 flash(f"Email alias {full_email} has been created", "success")
+                session[HIGHLIGHT_GEN_EMAIL_ID] = gen_email.id
 
                 return redirect(url_for("dashboard.index"))
 
diff --git a/app/dashboard/views/index.py b/app/dashboard/views/index.py
index e945d66b..b7519538 100644
--- a/app/dashboard/views/index.py
+++ b/app/dashboard/views/index.py
@@ -1,8 +1,9 @@
-from flask import render_template, request, redirect, url_for, flash
+from flask import render_template, request, redirect, url_for, flash, session
 from flask_login import login_required, current_user
 from sqlalchemy.orm import joinedload
 
 from app import email_utils
+from app.config import HIGHLIGHT_GEN_EMAIL_ID
 from app.dashboard.base import dashboard_bp
 from app.extensions import db
 from app.log import LOG
@@ -12,6 +13,13 @@ from app.models import GenEmail, ClientUser
 @dashboard_bp.route("/", methods=["GET", "POST"])
 @login_required
 def index():
+    # after creating a gen email, it's helpful to highlight it
+    highlight_gen_email_id = session.get(HIGHLIGHT_GEN_EMAIL_ID)
+
+    # reset as it should not persist
+    if highlight_gen_email_id:
+        del session[HIGHLIGHT_GEN_EMAIL_ID]
+
     # User generates a new email
     if request.method == "POST":
         if request.form.get("form-name") == "trigger-email":
@@ -44,6 +52,7 @@ SimpleLogin team.
 
                 LOG.d("generate new email %s for user %s", gen_email, current_user)
                 flash(f"Email {gen_email.email} has been created", "success")
+                session[HIGHLIGHT_GEN_EMAIL_ID] = gen_email.id
             else:
                 flash(f"You need to upgrade your plan to create new email.", "warning")
 
@@ -95,5 +104,8 @@ SimpleLogin team.
     )
 
     return render_template(
-        "dashboard/index.html", client_users=client_users, gen_emails=gen_emails
+        "dashboard/index.html",
+        client_users=client_users,
+        gen_emails=gen_emails,
+        highlight_gen_email_id=highlight_gen_email_id,
     )
diff --git a/static/style.css b/static/style.css
index 893f477f..46b8a899 100644
--- a/static/style.css
+++ b/static/style.css
@@ -40,4 +40,10 @@
     border-radius: 50%;
 
     margin-top: 10px;
+}
+
+/* highlighted table row */
+.highlight-row {
+    background-color: #eee;
+    font-weight: bolder;
 }
\ No newline at end of file