From cf6f05769add4c53b5237ab47bc70b823da5f366 Mon Sep 17 00:00:00 2001
From: Alex Aragon <aragcar@gmail.com>
Date: Mon, 29 May 2017 17:35:28 -0500
Subject: [PATCH] fix hidden count_message in menu

---
 main/inc/ajax/message.ajax.php        |  2 +-
 main/template/default/layout/menu.tpl | 14 ++++++++++----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/main/inc/ajax/message.ajax.php b/main/inc/ajax/message.ajax.php
index fca6c49616..db05411ceb 100755
--- a/main/inc/ajax/message.ajax.php
+++ b/main/inc/ajax/message.ajax.php
@@ -42,7 +42,7 @@ switch ($action) {
             }
             $total_invitations = intval($number_of_new_messages_of_friend) + $group_pending_invitations + intval($count_unread_message);
         }
-        $total_invitations = !empty($total_invitations) ? Display::badge($total_invitations) : '';
+        //$total_invitations = !empty($total_invitations) ? Display::badge($total_invitations) : '';
 
         echo $total_invitations;
         break;
diff --git a/main/template/default/layout/menu.tpl b/main/template/default/layout/menu.tpl
index 1651a49c1b..2f3387286d 100755
--- a/main/template/default/layout/menu.tpl
+++ b/main/template/default/layout/menu.tpl
@@ -24,16 +24,22 @@
             {% if _u.logged == 1 %}
                 <script>
                     $(document).ready(function () {
-                        $("#count_message")
-                            .load('{{ _p.web_main }}inc/ajax/message.ajax.php?a=get_count_message', function () {
+                        $.get('{{ _p.web_main }}inc/ajax/message.ajax.php?a=get_count_message', function(data){
+                            var message = data;
+                            if( message === "0"){
+                                $("#count_message_li").addClass('hidden');
+                            } else {
                                 $("#count_message_li").removeClass('hidden');
-                            });
+                                $("#count_message").append(message);
+                            }
+                        });
                     });
                 </script>
+                
                 <ul class="nav navbar-nav navbar-right">
                     <li id="count_message_li" class="hidden">
                         <a href="{{ message_url }}">
-                            <span id="count_message"></span>
+                            <span id="count_message" class="badge badge-warning"></span>
                         </a>
                     </li>
                     {% if _u.status != 6 %}