From 19598ffd9bb751de6852633bad298f168c5e1b93 Mon Sep 17 00:00:00 2001
From: Halil ibrahim Kalkan <hi_kalkan@yahoo.com>
Date: Thu, 21 Feb 2019 15:04:29 +0300
Subject: [PATCH] Show tenant switch only for anonymous users.

---
 .../Components/TenantSwitch/Default.cshtml    | 27 ++++++++++---------
 .../TenantSwitch/TenantSwitchViewComponent.cs | 16 ++++++++---
 2 files changed, 28 insertions(+), 15 deletions(-)

diff --git a/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/Default.cshtml b/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/Default.cshtml
index 33419549f0..515d4cc7dc 100644
--- a/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/Default.cshtml
+++ b/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/Default.cshtml
@@ -1,14 +1,17 @@
 @using Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy.Components.TenantSwitch
 @model TenantSwitchViewComponent.TenantSwitchViewModel
-<li class="nav-item">
-    <a abp-button="Link" id="TenantSwitchToolbarLink" href="#">
-        @if (Model.Tenant == null)
-        {
-            <text>@@host</text>
-        }
-        else
-        {
-            <text>@@@Model.Tenant.Name</text>
-        }
-    </a>
-</li>
\ No newline at end of file
+@if (!Model.CurrentUser.IsAuthenticated)
+{
+    <li class="nav-item">
+        <a abp-button="Link" id="TenantSwitchToolbarLink" href="#">
+            @if (Model.Tenant == null)
+            {
+                <text>@@host</text>
+            }
+            else
+            {
+                <text>@@@Model.Tenant.Name</text>
+            }
+        </a>
+    </li>
+}
\ No newline at end of file
diff --git a/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/TenantSwitchViewComponent.cs b/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/TenantSwitchViewComponent.cs
index 739dc71b5b..bfcaebed0e 100644
--- a/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/TenantSwitchViewComponent.cs
+++ b/framework/src/Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy/Volo/Abp/AspNetCore/Mvc/UI/MultiTenancy/Components/TenantSwitch/TenantSwitchViewComponent.cs
@@ -1,6 +1,7 @@
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Mvc;
 using Volo.Abp.MultiTenancy;
+using Volo.Abp.Users;
 
 namespace Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy.Components.TenantSwitch
 {
@@ -12,18 +13,25 @@ namespace Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy.Components.TenantSwitch
         public const int Order = -1_000_000;
 
         protected ITenantStore TenantStore { get; }
-
         protected ICurrentTenant CurrentTenant { get; }
+        protected ICurrentUser CurrentUser { get; }
 
-        public TenantSwitchViewComponent(ITenantStore tenantStore, ICurrentTenant currentTenant)
+        public TenantSwitchViewComponent(
+            ITenantStore tenantStore, 
+            ICurrentTenant currentTenant,
+            ICurrentUser currentUser)
         {
             TenantStore = tenantStore;
             CurrentTenant = currentTenant;
+            CurrentUser = currentUser;
         }
 
         public async Task<IViewComponentResult> InvokeAsync()
         {
-            var model = new TenantSwitchViewModel();
+            var model = new TenantSwitchViewModel
+            {
+                CurrentUser = CurrentUser
+            };
 
             if (CurrentTenant.Id.HasValue)
             {
@@ -36,6 +44,8 @@ namespace Volo.Abp.AspNetCore.Mvc.UI.MultiTenancy.Components.TenantSwitch
         public class TenantSwitchViewModel
         {
             public TenantInfo Tenant { get; set; }
+
+            public ICurrentUser CurrentUser { get; set; }
         }
     }
 }