Adding cmdlets for Azure Template Specs (Preview) with deployment support

src/Resources/ResourceManager/Implementation/CmdletBase/ResourceWithParameterCmdletBase.cs

@@ -18,8 +18,12 @@
 using System.Linq;
 using System.Management.Automation;
 using Microsoft.Azure.Commands.Common.Authentication.Abstractions;
+using Microsoft.Azure.Commands.ResourceManager.Cmdlets.Components;
+using Microsoft.Azure.Commands.ResourceManager.Cmdlets.SdkClient;
+using Microsoft.Azure.Commands.ResourceManager.Cmdlets.SdkModels;
 using Microsoft.Azure.Commands.ResourceManager.Cmdlets.Utilities;
 using Microsoft.WindowsAzure.Commands.Utilities.Common;
+using Newtonsoft.Json.Linq;
 
 namespace Microsoft.Azure.Commands.ResourceManager.Cmdlets.Implementation
 {
@@ -41,6 +45,10 @@ public abstract class ResourceWithParameterCmdletBase : ResourceManagerCmdletBas
         protected const string ParameterlessTemplateFileParameterSetName = "ByTemplateFileWithNoParameters";
         protected const string ParameterlessTemplateUriParameterSetName = "ByTemplateUriWithNoParameters";
 
+        protected const string TemplateSpecResourceIdParameterSetName = "ByTemplateSpecResourceId";
+        protected const string TemplateSpecResourceIdParameterFileParameterSetName = "ByTemplateSpecResourceIdAndParams";
+        protected const string TemplateSpecResourceIdParameterUriParameterSetName = "ByTemplateSpecResourceIdAndParamsUri";
+
         protected RuntimeDefinedParameterDictionary dynamicParameters;
 
         private Hashtable templateObject;
@@ -49,6 +57,8 @@ public abstract class ResourceWithParameterCmdletBase : ResourceManagerCmdletBas
 
         private string templateUri;
 
+        private string templateSpecId;
+
         protected ResourceWithParameterCmdletBase()
         {
             dynamicParameters = new RuntimeDefinedParameterDictionary();
@@ -68,6 +78,8 @@ protected ResourceWithParameterCmdletBase()
             Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "A file that has the template parameters.")]
         [Parameter(ParameterSetName = TemplateUriParameterFileParameterSetName,
             Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "A file that has the template parameters.")]
+        [Parameter(ParameterSetName = TemplateSpecResourceIdParameterFileParameterSetName,
+            Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "A file that has the template parameters.")]
         [ValidateNotNullOrEmpty]
         public string TemplateParameterFile { get; set; }
 
@@ -77,6 +89,8 @@ protected ResourceWithParameterCmdletBase()
             Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Uri to the template parameter file.")]
         [Parameter(ParameterSetName = TemplateUriParameterUriParameterSetName,
             Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Uri to the template parameter file.")]
+        [Parameter(ParameterSetName = TemplateSpecResourceIdParameterUriParameterSetName,
+            Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Uri to the template parameter file.")]
         [ValidateNotNullOrEmpty]
         public string TemplateParameterUri { get; set; }
 
@@ -113,12 +127,41 @@ protected ResourceWithParameterCmdletBase()
         [ValidateNotNullOrEmpty]
         public string TemplateUri { get; set; }
 
+        [Parameter(ParameterSetName = TemplateSpecResourceIdParameterSetName,
+            Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Resource ID of the templateSpec to be deployed.")]
+        [Parameter(ParameterSetName = TemplateSpecResourceIdParameterUriParameterSetName,
+            Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Resource ID of the templateSpec to be deployed.")]
+        [Parameter(ParameterSetName = TemplateSpecResourceIdParameterFileParameterSetName,
+            Mandatory = true, ValueFromPipelineByPropertyName = true, HelpMessage = "Resource ID of the templateSpec to be deployed.")]
+        [ValidateNotNullOrEmpty]
+        public string TemplateSpecId { get; set; }
+
         [Parameter(Mandatory = false, HelpMessage = "Skips the PowerShell dynamic parameter processing that checks if the provided template parameter contains all necessary parameters used by the template. " +
                                                     "This check would prompt the user to provide a value for the missing parameters, but providing the -SkipTemplateParameterPrompt will ignore this prompt and " +
                                                     "error out immediately if a parameter was found not to be bound in the template. For non-interactive scripts, -SkipTemplateParameterPrompt can be provided " +
                                                     "to provide a better error message in the case where not all required parameters are satisfied.")]
         public SwitchParameter SkipTemplateParameterPrompt { get; set; }
 
+        private TemplateSpecsSdkClient templateSpecsSdkClient;
+
+        /// <summary>
+        /// Gets or sets the Template Specs Azure sdk client wrapper
+        /// </summary>
+        public TemplateSpecsSdkClient TemplateSpecsSdkClient
+        {
+            get
+            {
+                if (this.templateSpecsSdkClient == null)
+                {
+                    this.templateSpecsSdkClient = new TemplateSpecsSdkClient(DefaultContext);
+                }
+
+                return this.templateSpecsSdkClient;
+            }
+
+            set { this.templateSpecsSdkClient = value; }
+        }
+
         public object GetDynamicParameters()
         {
             if (!this.IsParameterBound(c => c.SkipTemplateParameterPrompt))
@@ -185,6 +228,40 @@ public object GetDynamicParameters()
                             MyInvocation.MyCommand.Parameters.Keys.ToArray());
                     }
                 }
+                else if (!string.IsNullOrEmpty(TemplateSpecId) &&
+                    !TemplateSpecId.Equals(templateSpecId, StringComparison.OrdinalIgnoreCase))
+                {
+                    templateSpecId = TemplateSpecId;
+                    ResourceIdentifier resourceIdentifier = new ResourceIdentifier(templateSpecId);
+                    if(!resourceIdentifier.ResourceType.Equals("Microsoft.Resources/templateSpecs/versions", StringComparison.OrdinalIgnoreCase))
+                    {
+                        throw new PSArgumentException("No version found in Resource ID");
+                    }
+
+                    var templateSpecVersion = TemplateSpecsSdkClient.GetTemplateSpec(
+                        ResourceIdUtility.GetResourceName(templateSpecId).Split('/')[0],
+                        ResourceIdUtility.GetResourceGroupName(templateSpecId),
+                        resourceIdentifier.ResourceName).Versions.Single();
+
+                    var templateObj = JObject.Parse(templateSpecVersion.Template);
+
+                    if (string.IsNullOrEmpty(TemplateParameterUri))
+                    {
+                        dynamicParameters = TemplateUtility.GetTemplateParametersFromFile(
+                            templateObj,
+                            TemplateParameterObject,
+                            this.ResolvePath(TemplateParameterFile),
+                            MyInvocation.MyCommand.Parameters.Keys.ToArray());
+                    }
+                    else
+                    {
+                        dynamicParameters = TemplateUtility.GetTemplateParametersFromFile(
+                            templateObj,
+                            TemplateParameterObject,
+                            TemplateParameterUri,
+                            MyInvocation.MyCommand.Parameters.Keys.ToArray());
+                    }
+                }
             }
 
             RegisterDynamicParameters(dynamicParameters);

src/Resources/ResourceManager/Implementation/CmdletBase/TemplateSpecCmdletBase.cs

@@ -0,0 +1,64 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+using Microsoft.Azure.Commands.ResourceManager.Cmdlets.SdkClient;
+using Microsoft.Azure.Commands.ResourceManager.Common;
+using System;
+
+namespace Microsoft.Azure.Commands.ResourceManager.Cmdlets.Implementation
+{
+    public class TemplateSpecCmdletBase : AzureRMCmdlet
+    {
+        private TemplateSpecsSdkClient templateSpecsSdkClient;
+
+        /// <summary>
+        /// Gets or sets the Template Specs Azure sdk client wrapper
+        /// </summary>
+        public TemplateSpecsSdkClient TemplateSpecsSdkClient
+        {
+            get
+            {
+                if (this.templateSpecsSdkClient == null)
+                {
+                    this.templateSpecsSdkClient = new TemplateSpecsSdkClient(DefaultContext);
+                }
+
+                return this.templateSpecsSdkClient;
+            }
+
+            set { this.templateSpecsSdkClient = value; }
+        }
+
+        /// <summary>
+        /// Override method to extract inner errors.
+        /// </summary>
+        /// <param name="ex">exception</param>
+        protected override void WriteExceptionError(Exception ex)
+        {
+            var aggEx = ex as AggregateException;
+
+            if (aggEx != null && aggEx.InnerExceptions != null)
+            {
+                foreach (var e in aggEx.Flatten().InnerExceptions)
+                {
+                    WriteExceptionError(e);
+                }
+
+                return;
+            }
+
+            base.WriteExceptionError(ex);
+        }
+    }
+}

src/Resources/ResourceManager/Implementation/Deployments/GetAzureSubscriptionDeploymentWhatIfResultCmdlet.cs

@@ -57,6 +57,7 @@ public class GetAzureSubscriptionDeploymentWhatIfResultCmdlet : DeploymentWhatIf
             mode: DeploymentMode.Incremental,
             templateUri: this.TemplateUri ?? this.TryResolvePath(this.TemplateFile),
             templateObject: this.TemplateObject,
+            templateSpecId: TemplateSpecId,
             templateParametersUri: this.TemplateParameterUri,
             templateParametersObject: GetTemplateParameterObject(this.TemplateParameterObject),
             resultFormat: this.ResultFormat,

src/Resources/ResourceManager/Implementation/Deployments/NewAzureSubscriptionDeploymentCmdlet.cs

@@ -77,6 +77,7 @@ public class NewAzureSubscriptionDeploymentCmdlet : DeploymentCreateCmdlet
             DeploymentMode = DeploymentMode.Incremental,
             TemplateFile = this.TemplateUri ?? this.TryResolvePath(this.TemplateFile),
             TemplateObject = this.TemplateObject,
+            TemplateSpecId = TemplateSpecId,
             TemplateParameterObject = this.GetTemplateParameterObject(this.TemplateParameterObject),
             ParameterUri = this.TemplateParameterUri,
             DeploymentDebugLogLevel = this.GetDeploymentDebugLogLevel(this.DeploymentDebugLogLevel),
@@ -90,6 +91,7 @@ public class NewAzureSubscriptionDeploymentCmdlet : DeploymentCreateCmdlet
             mode: DeploymentMode.Incremental,
             templateUri: TemplateUri ?? this.TryResolvePath(TemplateFile),
             templateObject: this.TemplateObject,
+            templateSpecId: TemplateSpecId,
             templateParametersUri: this.TemplateParameterUri,
             templateParametersObject: GetTemplateParameterObject(this.TemplateParameterObject),
             resultFormat: this.WhatIfResultFormat,

src/Resources/ResourceManager/Implementation/ResourceGroupDeployments/GetAzureResourceGroupDeploymentWhatIfResultCmdlet.cs

@@ -57,6 +57,7 @@ public class GetAzureResourceGroupDeploymentWhatIfResultCmdlet : DeploymentWhatI
             resourceGroupName: this.ResourceGroupName,
             templateUri: this.TemplateUri ?? this.TryResolvePath(this.TemplateFile),
             templateObject: this.TemplateObject,
+            templateSpecId: TemplateSpecId,
             templateParametersUri: this.TemplateParameterUri,
             templateParametersObject: GetTemplateParameterObject(this.TemplateParameterObject),
             resultFormat: this.ResultFormat,

src/Resources/ResourceManager/Implementation/ResourceGroupDeployments/NewAzureResourceGroupDeploymentCmdlet.cs

@@ -35,7 +35,7 @@ public class NewAzureResourceGroupDeploymentCmdlet : DeploymentCreateCmdlet
     {
         [Alias("DeploymentName")]
         [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true,
-            HelpMessage = "The name of the deployment it's going to create. If not specified, defaults to the template file name when a template file is provided; defaults to the current time when a template object is provided, e.g. \"20131223140835\".")]
+             HelpMessage = "The name of the deployment it's going to create. If not specified, defaults to the template file name when a template file is provided; defaults to the current time when a template object is provided, e.g. \"20131223140835\".")]
         [ValidateNotNullOrEmpty]
         public string Name { get; set; }
 
@@ -87,6 +87,7 @@ public class NewAzureResourceGroupDeploymentCmdlet : DeploymentCreateCmdlet
             DeploymentMode = Mode,
             TemplateFile = TemplateUri ?? this.TryResolvePath(TemplateFile),
             TemplateObject = TemplateObject,
+            TemplateSpecId = TemplateSpecId,
             TemplateParameterObject = GetTemplateParameterObject(TemplateParameterObject),
             ParameterUri = TemplateParameterUri,
             DeploymentDebugLogLevel = GetDeploymentDebugLogLevel(DeploymentDebugLogLevel),
@@ -107,6 +108,7 @@ public class NewAzureResourceGroupDeploymentCmdlet : DeploymentCreateCmdlet
             resourceGroupName: this.ResourceGroupName,
             templateUri: this.TemplateUri ?? this.TryResolvePath(this.TemplateFile),
             templateObject: this.TemplateObject,
+            templateSpecId: TemplateSpecId,
             templateParametersUri: this.TemplateParameterUri,
             templateParametersObject: this.GetTemplateParameterObject(this.TemplateParameterObject),
             resultFormat: this.WhatIfResultFormat,