From 7febe6e15b94624190cf75b2c35e3c064aae7462 Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Sun, 23 Feb 2020 15:41:53 +0700
Subject: [PATCH] cannot delete default mailbox

---
 app/dashboard/templates/dashboard/mailbox.html | 11 +++++++++--
 app/dashboard/views/mailbox.py                 |  4 ++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/app/dashboard/templates/dashboard/mailbox.html b/app/dashboard/templates/dashboard/mailbox.html
index 12727d32..b86affe8 100644
--- a/app/dashboard/templates/dashboard/mailbox.html
+++ b/app/dashboard/templates/dashboard/mailbox.html
@@ -43,11 +43,17 @@
                   🚫
                 </span>
               {% 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
+                </div>
+              {% endif %}
             </h5>
+
             <h6 class="card-subtitle mb-2 text-muted">
               Created {{ mailbox.created_at | dt }} <br>
               <span class="font-weight-bold">{{ mailbox.nb_alias() }}</span> aliases. <br>
+
             </h6>
 
             <a href="{{ url_for('dashboard.mailbox_detail_route', mailbox_id=mailbox.id) }}">Edit ➡</a>
@@ -60,7 +66,8 @@
                   <input type="hidden" name="form-name" value="delete">
                   <input type="hidden" class="mailbox" value="{{ mailbox.email }}">
                   <input type="hidden" name="mailbox-id" value="{{ mailbox.id }}">
-                  <span class="card-link btn btn-link text-danger float-right delete-mailbox">
+                  <span class="card-link btn btn-link text-danger float-right delete-mailbox
+                    {% if mailbox.id == current_user.default_mailbox_id %} disabled {% endif %}">
                     Delete
                   </span>
                 </form>
diff --git a/app/dashboard/views/mailbox.py b/app/dashboard/views/mailbox.py
index 5ce2ed5e..796c0001 100644
--- a/app/dashboard/views/mailbox.py
+++ b/app/dashboard/views/mailbox.py
@@ -44,6 +44,10 @@ def mailbox_route():
                 flash("Unknown error. Refresh the page", "warning")
                 return redirect(url_for("dashboard.mailbox_route"))
 
+            if mailbox.id == current_user.default_mailbox_id:
+                flash("You cannot delete default mailbox", "error")
+                return redirect(url_for("dashboard.mailbox_route"))
+
             email = mailbox.email
             Mailbox.delete(mailbox_id)
             db.session.commit()