Skip to content

Commit 8951051

Browse files
committed
Resolving CR comments
1 parent 2a9db46 commit 8951051

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

src/ServiceManagement/Services/Commands.Utilities/Websites/WebsitesClient.cs

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
using System.Linq;
2121
using System.Net;
2222
using System.Net.Http;
23+
using System.Net.Http.Headers;
2324
using System.Web;
2425
using System.Xml.Linq;
2526
using Microsoft.Build.Evaluation;
@@ -1361,25 +1362,30 @@ private DeploymentBaseOptions CreateRemoteDeploymentBaseOptions(string websiteNa
13611362
TempAgent = false
13621363
};
13631364

1364-
var azureSessionUserAgent = GetUserAgentForAzureSession();
1365+
var azureSessionUserAgent = GetDeploymentBaseOptionsUserAgent();
13651366
if (!string.IsNullOrEmpty(azureSessionUserAgent))
13661367
{
13671368
remoteBaseOptions.UserAgent = azureSessionUserAgent;
13681369
}
13691370

13701371
return remoteBaseOptions;
13711372
}
1372-
1373+
13731374
/// <summary>
1374-
/// Gets user agent for azure session
1375+
/// Gets remote deployment base options useragent using AzureSession/WebSiteManagementClient.
13751376
/// </summary>
1376-
/// <returns>useragent string </returns>
1377-
private string GetUserAgentForAzureSession()
1377+
/// <returns>useragent string.</returns>
1378+
private string GetDeploymentBaseOptionsUserAgent()
13781379
{
13791380
var userAgent = string.Empty;
1380-
if (AzureSession.ClientFactory.UserAgents != null)
1381+
var managementClient = this.WebsiteManagementClient as WebSiteManagementClient;
1382+
var userAgents = AzureSession.ClientFactory.UserAgents.Any()
1383+
? AzureSession.ClientFactory.UserAgents as IEnumerable<ProductInfoHeaderValue>
1384+
: (managementClient != null) ? managementClient.UserAgent : null;
1385+
1386+
if (userAgents != null)
13811387
{
1382-
foreach (var agent in AzureSession.ClientFactory.UserAgents)
1388+
foreach (var agent in userAgents)
13831389
{
13841390
if (agent != null && agent.Product != null && !string.IsNullOrEmpty(agent.Product.Name))
13851391
{

0 commit comments

Comments
 (0)