From e98bd42438408bf4a89edaed586491a7017dff41 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?P=C4=93teris=20Caune?= <cuu508@gmail.com>
Date: Sat, 19 Jan 2019 18:51:29 +0200
Subject: [PATCH] Fix project names in the top navigation dropdown menu.

---
 hc/accounts/middleware.py |  5 ++++-
 templates/base.html       | 19 ++++++-------------
 2 files changed, 10 insertions(+), 14 deletions(-)

diff --git a/hc/accounts/middleware.py b/hc/accounts/middleware.py
index 7e84dc46..3a9521a6 100644
--- a/hc/accounts/middleware.py
+++ b/hc/accounts/middleware.py
@@ -1,3 +1,4 @@
+from django.db.models import Q
 from hc.accounts.models import Profile, Project
 
 
@@ -9,7 +10,9 @@ class TeamAccessMiddleware(object):
         if not request.user.is_authenticated:
             return self.get_response(request)
 
-        projects_q = Project.objects.filter(member__user_id=request.user.id)
+        is_owner = Q(owner=request.user)
+        is_member = Q(member__user_id=request.user.id)
+        projects_q = Project.objects.filter(is_owner | is_member)
         projects_q = projects_q.select_related("owner")
         request.get_projects = lambda: list(projects_q)
 
diff --git a/templates/base.html b/templates/base.html
index 349c2b31..7a3e20a7 100644
--- a/templates/base.html
+++ b/templates/base.html
@@ -116,7 +116,7 @@
                 <li class="dropdown">
                     <a id="nav-email" href="#" class="dropdown-toggle" data-toggle="dropdown" role="button">
                         {% if check %}
-                            {{ check.project.owner.profile }}
+                            {{ check.project }}
                         {% else %}
                             {{ request.project }}
                         {% endif %}
@@ -124,26 +124,19 @@
                     </a>
                     <ul class="dropdown-menu">
                     {% with projects=request.get_projects %}
-                        {% if projects %}
-                        <li class="dropdown-header">{{ request.user.profile }}</li>
-                        <li>
-                            <a href="{% url 'hc-switch-team' request.user.username %}" class="active">Checks</a>
-                        </li>
-                        {% endif %}
-
-                        <li><a href="{% url 'hc-profile' %}">Account Settings</a></li>
-                        <li role="separator" class="divider"></li>
-
                         {% for project in projects %}
                         <li class="dropdown-header">{{ project }}</li>
                         <li>
                             <a href="{% url 'hc-switch-team' project.owner.username %}">Checks</a>
                         </li>
+                        {% if project.owner == request.user %}
+                            <li><a href="{% url 'hc-profile' %}">Account Settings</a></li>
+                        {% endif %}
                         <li role="separator" class="divider"></li>
                         {% endfor %}
-
-                        <li><a href="{% url 'hc-logout' %}">Log Out</a></li>
                     {% endwith %}
+
+                    <li><a href="{% url 'hc-logout' %}">Log Out</a></li>
                     </ul>
                 </li>
             </ul>