From 1fcbe329902fcd2e439f622cbe681cb213da154b Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Sun, 23 Feb 2020 15:51:26 +0700
Subject: [PATCH] Can set a mailbox as default

---
 .../templates/dashboard/mailbox.html          | 17 ++++++++++++++-
 app/dashboard/views/mailbox.py                | 21 +++++++++++++++++++
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/app/dashboard/templates/dashboard/mailbox.html b/app/dashboard/templates/dashboard/mailbox.html
index b86affe8..1d4d84f0 100644
--- a/app/dashboard/templates/dashboard/mailbox.html
+++ b/app/dashboard/templates/dashboard/mailbox.html
@@ -45,7 +45,7 @@
               {% endif %}
               {% if mailbox.id == current_user.default_mailbox_id %}
                 <div class="badge badge-primary float-right" data-toggle="tooltip"
-                     title="A random alias belongs to the default mailbox">Default Mailbox
+                     title="When a new random alias is created, it belongs to the default mailbox">Default Mailbox
                 </div>
               {% endif %}
             </h5>
@@ -61,6 +61,21 @@
 
           <div class="card-footer p-0">
             <div class="row">
+              {% if mailbox.verified %}
+                <div class="col">
+                  <form method="post">
+                    <input type="hidden" name="form-name" value="set-default">
+                    <input type="hidden" class="mailbox" value="{{ mailbox.email }}">
+                    <input type="hidden" name="mailbox-id" value="{{ mailbox.id }}">
+                    <button class="card-link btn btn-link
+                    {% if mailbox.id == current_user.default_mailbox_id %} disabled {% endif %}"
+                    >
+                    Set As Default Mailbox
+                  </button>
+                  </form>
+                </div>
+              {% endif %}
+
               <div class="col">
                 <form method="post">
                   <input type="hidden" name="form-name" value="delete">
diff --git a/app/dashboard/views/mailbox.py b/app/dashboard/views/mailbox.py
index 796c0001..f81b9a3f 100644
--- a/app/dashboard/views/mailbox.py
+++ b/app/dashboard/views/mailbox.py
@@ -53,6 +53,27 @@ def mailbox_route():
             db.session.commit()
             flash(f"Mailbox {email} has been deleted", "success")
 
+            return redirect(url_for("dashboard.mailbox_route"))
+        if request.form.get("form-name") == "set-default":
+            mailbox_id = request.form.get("mailbox-id")
+            mailbox = Mailbox.get(mailbox_id)
+
+            if not mailbox or mailbox.user_id != current_user.id:
+                flash("Unknown error. Refresh the page", "warning")
+                return redirect(url_for("dashboard.mailbox_route"))
+
+            if mailbox.id == current_user.default_mailbox_id:
+                flash("This mailbox is already default one", "error")
+                return redirect(url_for("dashboard.mailbox_route"))
+
+            if not mailbox.verified:
+                flash("Cannot set unverified mailbox as default", "error")
+                return redirect(url_for("dashboard.mailbox_route"))
+
+            current_user.default_mailbox_id = mailbox.id
+            db.session.commit()
+            flash(f"Mailbox {mailbox.email} is set as Default Mailbox", "success")
+
             return redirect(url_for("dashboard.mailbox_route"))
 
         elif request.form.get("form-name") == "create":