pass custom headers for cross tenant hubvnet conn

Author: nagula-ritvika

src/Network/Network/Cortex/HubVnetConnection/NewAzureRmHubVirtualNetworkConnectionCommand.cs

@@ -137,7 +137,8 @@ public class NewHubVirtualNetworkConnectionCommand : HubVnetConnectionBaseCmdlet
         public override void Execute()
         {
             base.Execute();
-            
+            Dictionary<string, List<string>> auxAuthHeader = null;
+
             if (ParameterSetName.Contains(CortexParameterSetNames.ByVirtualHubObject))
             {
                 this.ResourceGroupName = this.ParentObject.ResourceGroupName;
@@ -182,6 +183,14 @@ public override void Execute()
                 parentVirtualHub.VirtualNetworkConnections = new List<PSHubVirtualNetworkConnection>();
             }
 
+            List<string> resourceIds = new List<string>();
+            resourceIds.Add(hubVnetConnection.RemoteVirtualNetwork.Id);
+            var auxHeaderDictionary = GetAuxilaryAuthHeaderFromResourceIds(resourceIds);
+            if (auxHeaderDictionary != null && auxHeaderDictionary.Count > 0)
+            {
+                auxAuthHeader = new Dictionary<string, List<string>>(auxHeaderDictionary);
+            }
+
             parentVirtualHub.VirtualNetworkConnections.Add(hubVnetConnection);
 
             ConfirmAction(
@@ -190,6 +199,7 @@ public override void Execute()
                 () =>
                 {
                     WriteVerbose(String.Format(Properties.Resources.CreatingLongRunningOperationMessage, this.ResourceGroupName, this.Name));
+                    this.VirtualHubClient.CreateOrUpdateWithHttpMessagesAsync(this.ResourceGroupName, this.ParentResourceName, )
                     this.CreateOrUpdateVirtualHub(this.ResourceGroupName, this.ParentResourceName, parentVirtualHub, parentVirtualHub.Tag);
                     var createdVirtualHub = this.GetVirtualHub(this.ResourceGroupName, this.ParentResourceName);
 

src/Network/Network/Cortex/VirtualHub/VirtualHubBaseCmdlet.cs

@@ -71,13 +71,26 @@ public bool IsVirtualHubPresent(string resourceGroupName, string name)
             return psVirtualHub == null ? false : true;
         }
 
-        public PSVirtualHub CreateOrUpdateVirtualHub(string resourceGroupName, string virtualHubName, PSVirtualHub virtualHub, Hashtable tags)
+        public PSVirtualHub CreateOrUpdateVirtualHub(string resourceGroupName, string virtualHubName, PSVirtualHub virtualHub, Hashtable tags, Dictionary<string, List<string>> customHeaders = null)
         {
             var virtualHubModel = NetworkResourceManagerProfile.Mapper.Map<MNM.VirtualHub>(virtualHub);
             virtualHubModel.Location = virtualHub.Location;
             virtualHubModel.Tags = TagsConversionHelper.CreateTagDictionary(tags, validate: true);
+            MNM.VirtualHub virtualHubCreatedOrUpdated;
+
+            if (customHeaders == null)
+            {
+                virtualHubCreatedOrUpdated = this.VirtualHubClient.CreateOrUpdate(resourceGroupName, virtualHubName, virtualHubModel);
+            }
+            else
+            {
+                // Execute the create call and pass the custom headers. 
+                using (var _result = this.VirtualHubClient.CreateOrUpdateWithHttpMessagesAsync(resourceGroupName, virtualHubName, virtualHubModel, customHeaders).GetAwaiter().GetResult())
+                {
+                    virtualHubCreatedOrUpdated = _result.Body;
+                }
+            }
 
-            var virtualHubCreatedOrUpdated = this.VirtualHubClient.CreateOrUpdate(resourceGroupName, virtualHubName, virtualHubModel);
             PSVirtualHub hubToReturn = this.ToPsVirtualHub(virtualHubCreatedOrUpdated);
             hubToReturn.ResourceGroupName = resourceGroupName;