From ecb7e3ab2a2b79c016949de093b9d325b7236e1f Mon Sep 17 00:00:00 2001
From: Mikael Svenson <misvenso@microsoft.com>
Date: Sun, 18 Apr 2021 17:39:27 +0200
Subject: [PATCH] Added input field for optional tenant id

---
 .../SearchQueryTool/MainWindow.xaml           | 31 ++++++++++------
 .../SearchQueryTool/MainWindow.xaml.cs        | 36 +++++++++++--------
 2 files changed, 42 insertions(+), 25 deletions(-)

diff --git a/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml b/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml
index 4a3e5419..ed55381f 100644
--- a/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml
+++ b/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml
@@ -1123,6 +1123,7 @@
                                     <RowDefinition/>
                                     <RowDefinition/>
                                     <RowDefinition/>
+                                    <RowDefinition/>
                                 </Grid.RowDefinitions>
 
                                 <DockPanel HorizontalAlignment="Stretch">
@@ -1135,26 +1136,40 @@
 									LostFocus="SharePointSiteUrlTextBox_LostFocus"
 									TextAlignment="Left" />
                                 </DockPanel>
+
                                 <StackPanel Grid.Row="1" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal">
+                                    <Label Content="Tenant Id (optional):"/>
+                                    <TextBox x:Name="TenantIdTextBox" Height="26"
+									Margin="5,0,5,1" 
+									Text="" 
+									Padding="4"
+									TextAlignment="Left" Width="230">
+                                        <TextBox.ToolTip>
+                                            <TextBlock><Run Text="Needed for non sharepoint.com domains (contoso.com or guid)."/></TextBlock>
+                                        </TextBox.ToolTip>
+                                    </TextBox>
+                                </StackPanel>
+
+                                <StackPanel Grid.Row="3" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal">
                                     <Label Content="Timeout (sec):"/>
                                     <TextBox x:Name="WebRequestTimeoutTextBox" Height="26" 
 									Margin="5,0,5,1" 
 									Text="30" 
 									Padding="4"
-									TextAlignment="Left" Width="50" RenderTransformOrigin="1.847,0.53"/>
+									TextAlignment="Left" Width="50"/>
                                 </StackPanel>
-                                <StackPanel Grid.Row="2" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal">
+                                <StackPanel Grid.Row="4" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal">
                                     <Label Content="Accept:"/>
                                     <RadioButton x:Name="AcceptJsonRadioButton" Content="json" IsChecked="True" VerticalAlignment="Center" GroupName="AcceptFormat" />
                                     <Rectangle Width="10"/>
                                     <RadioButton x:Name="AcceptXmlRadioButton" Content="xml" VerticalAlignment="Center" GroupName="AcceptFormat" />
                                 </StackPanel>
-                                <DockPanel Grid.Row="3" Margin="0,10,0,0">
+                                <DockPanel Grid.Row="5" Margin="0,10,0,0">
                                     <Label Content="Method:" />
                                     <RadioButton VerticalContentAlignment="Center" IsChecked="True" Checked="HttpMethodModeRadioButton_Checked" x:Name="HttpGetMethodRadioButton" GroupName="HttpMethodMode" Content="HTTP GET"/>
                                     <RadioButton Margin="5,0,0,0" VerticalContentAlignment="Center" Checked="HttpMethodModeRadioButton_Checked" x:Name="HttpPostMethodRadioButton" GroupName="HttpMethodMode" Content="HTTP POST"/>
                                 </DockPanel>
-                                <StackPanel Grid.Row="4" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal">
+                                <StackPanel Grid.Row="6" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal">
                                     <Grid>
                                         <Grid.ColumnDefinitions>
                                             <ColumnDefinition/>
@@ -1220,15 +1235,9 @@
                                                 <Label x:Name="LoggedinLabel" Padding="5" Visibility="Hidden" Content="Successfully signed-in" FontWeight="Bold" Foreground="#FF275706"/>
                                             </StackPanel>
                                         </Grid>
-
-
-
                                     </Grid>
-
-
-
                                 </StackPanel>
-                                <StackPanel Grid.Row="5" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal"/>
+                                <StackPanel Grid.Row="7" Margin="0,10,0,0" HorizontalAlignment="Stretch" Orientation="Horizontal"/>
                             </Grid>
                         </Expander.Content>
                     </Expander>
diff --git a/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml.cs b/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml.cs
index 900f5d34..538f8070 100644
--- a/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml.cs
+++ b/Solutions/SharePoint.Search.QueryTool/SearchQueryTool/MainWindow.xaml.cs
@@ -61,6 +61,7 @@ private enum PropertyType
         private bool _enableExperimentalFeatures;
         private bool _firstInit = true;
         private static string _adalToken = null;
+        private static InteractiveAuthenticationProvider _interactiveProvider;
 
         public SearchPresetList SearchPresets { get; set; }
         private string PresetFolderPath { get; set; }
@@ -506,29 +507,36 @@ async Task AdalLogin(bool forcePrompt)
             const string clientId = "9bc3ab49-b65d-410a-85ad-de819febfddc";
             const string redirectUri = "https://oauth.spops.microsoft.com/";
 
-            if (forcePrompt || IsExpired(_adalToken, spUri.Host))
-            {
-                string tenant = spUri.Host.Replace("sharepoint", "onmicrosoft")
+            string tenant = spUri.Host.Replace("sharepoint", "onmicrosoft")
                     .Replace("-df", "")
                     .Replace("-admin", "");
 
-                InteractiveAuthenticationProvider interactiveProvider = new InteractiveAuthenticationProvider(clientId, tenant, new Uri(redirectUri));
-                _adalToken = await interactiveProvider.GetAccessTokenAsync(resourceUri);
+            if(!string.IsNullOrWhiteSpace(TenantIdTextBox.Text))
+            {
+                tenant = TenantIdTextBox.Text.Trim();
             }
-            _searchQueryRequest.Token = _searchSuggestionsRequest.Token = "Bearer " + _adalToken;
-        }
 
-        static bool IsExpired(string token, string host)
-        {
-            if (string.IsNullOrWhiteSpace(token))
+            if (_interactiveProvider == null || _interactiveProvider.TenantId != tenant)
             {
-                return true;
+                _interactiveProvider = new InteractiveAuthenticationProvider(clientId, tenant, new Uri(redirectUri));
             }
-            var jwtToken = new JwtSecurityToken(token);
-            if (jwtToken.Audiences.First().Contains(host) == false) return true;
-            return jwtToken.ValidTo < DateTime.UtcNow;
+
+            _adalToken = await _interactiveProvider.GetAccessTokenAsync(resourceUri);
+
+            _searchQueryRequest.Token = _searchSuggestionsRequest.Token = "Bearer " + _adalToken;
         }
 
+        //static bool IsExpired(string token, string host)
+        //{
+        //    if (string.IsNullOrWhiteSpace(token))
+        //    {
+        //        return true;
+        //    }
+        //    var jwtToken = new JwtSecurityToken(token);
+        //    if (jwtToken.Audiences.First().Contains(host) == false) return true;
+        //    return jwtToken.ValidTo < DateTime.UtcNow;
+        //}
+
         #endregion
 
         #region Event Handlers for Controls on the Search Query Tab