From d684740b9222480958863a2217329d93f497b463 Mon Sep 17 00:00:00 2001
From: KF7EEL <kf7eel@qsl.net>
Date: Wed, 8 Dec 2021 21:42:35 -0800
Subject: [PATCH] add message count to menu

---
 web/app.py                           | 9 ++++++++-
 web/templates/flask_user_layout.html | 6 +++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/web/app.py b/web/app.py
index 25adc08..4436ee1 100644
--- a/web/app.py
+++ b/web/app.py
@@ -767,7 +767,14 @@ def hbnet_web_service():
     # Use this to pass variables into Jinja2 templates
     @app.context_processor
     def global_template_config():
-        return dict(global_config={'mode': mode})
+        messages_waiting = 0
+        if current_user.is_authenticated == True:
+            mail_all = MailBox.query.filter_by(rcv_callsign=str(current_user.username).upper()).all()
+            messages_waiting = 0
+            for i in mail_all:
+                messages_waiting = messages_waiting + 1
+            
+        return dict(global_config={'mode': mode, 'messages': messages_waiting})
 
 
     # The Home page is accessible to anyone
diff --git a/web/templates/flask_user_layout.html b/web/templates/flask_user_layout.html
index 0a8728f..7637b23 100644
--- a/web/templates/flask_user_layout.html
+++ b/web/templates/flask_user_layout.html
@@ -121,7 +121,7 @@
         {% endif %}
 <!--
             <li class="nav-item">
-        <a class="nav-link" href="{{url}}/mail/{{ current_user.username or current_user.email }}"><i class="bi bi-mailbox"></i> Mailbox </a>
+        <a class="nav-link" href="{{url}}/mail/{{ current_user.username or current_user.email }}"><i class="bi bi-mailbox"></i> <strong>Mailbox</strong> </a>
       </li>
 -->
         
@@ -130,7 +130,11 @@
            <i class="bi bi-arrow-down-square"></i> Features 
           </a>
           <ul class="dropdown-menu" aria-labelledby="features_menu">
+    {% if global_config['messages'] > 0 %}
+            <li><a class="dropdown-item" href="{{url}}/mail/{{ current_user.username or current_user.email }}"><i class="bi bi-mailbox"></i> <strong>Mailbox - {{global_config['messages']}}</strong></a></li>
+    {% else %}
             <li><a class="dropdown-item" href="{{url}}/mail/{{ current_user.username or current_user.email }}"><i class="bi bi-mailbox"></i> Mailbox </a></li>
+    {% endif %}
             <li><a class="dropdown-item" href="{{url}}/discussion"><i class="bi bi-chat-right-quote"></i> Discussion </a></li>
     {% if global_config['mode'] == 'FULL' %}
             <li><a class="dropdown-item" href="{{url}}/generate_passphrase"><i class="bi bi-info-square"></i> Server(s) </a></li>