update virtualhub cmdlets also

Author: nagula-ritvika

src/Network/Network/Cortex/VirtualHub/NewAzureRmVirtualHubCommand.cs

@@ -107,6 +107,7 @@ public class NewAzureRmVirtualHubCommand : VirtualHubBaseCmdlet
         public override void Execute()
         {
             base.Execute();
+            Dictionary<string, List<string>> auxAuthHeader = null;
 
             if (this.IsVirtualHubPresent(this.ResourceGroupName, this.Name))
             {
@@ -155,6 +156,19 @@ public override void Execute()
                     if (this.HubVnetConnection != null)
                     {
                         virtualHub.VirtualNetworkConnections.AddRange(this.HubVnetConnection);
+
+                        // get auth headers for cross-tenant hubvnet conn
+                        List<string> resourceIds = new List<string>();
+                        foreach (var connection in this.HubVnetConnection)
+                        {
+                            resourceIds.Add(connection.RemoteVirtualNetwork.Id);
+                        }
+
+                        var auxHeaderDictionary = GetAuxilaryAuthHeaderFromResourceIds(resourceIds);
+                        if (auxHeaderDictionary != null && auxHeaderDictionary.Count > 0)
+                        {
+                            auxAuthHeader = new Dictionary<string, List<string>>(auxHeaderDictionary);
+                        }
                     }
 
                     virtualHub.RouteTable = this.RouteTable;
@@ -169,7 +183,8 @@ public override void Execute()
                         this.ResourceGroupName,
                         this.Name,
                         virtualHub,
-                        this.Tag));
+                        this.Tag,
+                        auxAuthHeader));
                 });
         }
     }

src/Network/Network/Cortex/VirtualHub/UpdateAzureRmVirtualHubCommand.cs

@@ -107,6 +107,7 @@ public override void Execute()
         {
             base.Execute();
             PSVirtualHub virtualHubToUpdate = null;
+            Dictionary<string, List<string>> auxAuthHeader = null;
 
             if (ParameterSetName.Equals(CortexParameterSetNames.ByVirtualHubObject, StringComparison.OrdinalIgnoreCase))
             {
@@ -142,6 +143,19 @@ public override void Execute()
             {
                 virtualHubToUpdate.VirtualNetworkConnections = new List<PSHubVirtualNetworkConnection>();
                 virtualHubToUpdate.VirtualNetworkConnections.AddRange(this.HubVnetConnection);
+
+                // get auth headers for cross-tenant hubvnet conn
+                List<string> resourceIds = new List<string>();
+                foreach (var connection in this.HubVnetConnection)
+                {
+                    resourceIds.Add(connection.RemoteVirtualNetwork.Id);
+                }
+
+                var auxHeaderDictionary = GetAuxilaryAuthHeaderFromResourceIds(resourceIds);
+                if (auxHeaderDictionary != null && auxHeaderDictionary.Count > 0)
+                {
+                    auxAuthHeader = new Dictionary<string, List<string>>(auxHeaderDictionary);
+                }
             }
 
             //// VirtualHubRouteTable
@@ -166,7 +180,8 @@ public override void Execute()
                             this.ResourceGroupName,
                             this.Name,
                             virtualHubToUpdate,
-                            this.Tag));
+                            this.Tag,
+                            auxAuthHeader));
                     });
         }
     }