Display ActionRule based on Type

Author: Mahak Jain

src/AlertsManagement/AlertsManagement/ActionRuleCommands/GetAzureActionRule.cs

@@ -294,20 +294,37 @@ protected override void ProcessRecordInternal()
 
                     } while (!string.IsNullOrEmpty(nextPageLink) && currentCount < lastCount);
 
-                    WriteObject(resultList.Select((r) => new PSActionRule(r)), enumerateCollection: true);
+                    WriteObject(resultList.Select((r) => TransformOutput(r)), enumerateCollection: true);
                     break;
 
                 case ActionRuleByNameParameterSet:
-                    PSActionRule rulebyName = new PSActionRule(this.AlertsManagementClient.ActionRules.GetByNameWithHttpMessagesAsync(ResourceGroupName, Name).Result.Body);
-                    WriteObject(sendToPipeline: rulebyName);
+                    var rulebyName = this.AlertsManagementClient.ActionRules.GetByNameWithHttpMessagesAsync(ResourceGroupName, Name).Result.Body;
+                    WriteObject(sendToPipeline: TransformOutput(rulebyName));
                     break;
 
                 case ResourceIdParameterSet:
                     ExtractedInfo info = CommonUtils.ExtractFromActionRuleResourceId(ResourceId);
-                    PSActionRule ruleById = new PSActionRule(this.AlertsManagementClient.ActionRules.GetByNameWithHttpMessagesAsync(info.ResourceGroupName, info.Resource).Result.Body);
-                    WriteObject(sendToPipeline: ruleById);
+                    var ruleById = this.AlertsManagementClient.ActionRules.GetByNameWithHttpMessagesAsync(info.ResourceGroupName, info.Resource).Result.Body;
+                    WriteObject(sendToPipeline: TransformOutput(ruleById));
                     break;
             }
         }
+
+        private PSActionRule TransformOutput(ActionRule input)
+        {
+            string actionRuleType = input.Properties.GetType().Name;
+
+            switch (actionRuleType)
+            {
+                case "Suppression":
+                    return new PSSuppressionActionRule(input);
+                case "ActionGroup":
+                    return new PSActionGroupActionRule(input);
+                case "Diagnostics":
+                    return new PSDiagnosticsActionRule(input);
+                default:
+                    return new PSActionRule(input);
+            }
+        }
     }
 }

src/AlertsManagement/AlertsManagement/ActionRuleCommands/SetAzureActionRule.cs

@@ -288,7 +288,7 @@ public class SetAzureActionRule : AlertsManagementBaseCmdlet
 
         protected override void ProcessRecordInternal()
         {
-            PSActionRule actionRule = new PSActionRule();
+            ActionRule result = new ActionRule();
             if (ShouldProcess(
                 target: string.Format(Resources.TargetWithRG, this.Name, this.ResourceGroupName),
                 action: Resources.CreateOrUpdateActionRule_Action))
@@ -314,8 +314,8 @@ protected override void ProcessRecordInternal()
                             )
                         );
 
-                        actionRule = new PSActionRule(this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
-                            resourceGroupName: ResourceGroupName, actionRuleName: Name, actionRule: actionGroupAR).Result.Body);
+                        result = this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
+                            resourceGroupName: ResourceGroupName, actionRuleName: Name, actionRule: actionGroupAR).Result.Body;
                         break;
 
                     case BySimplifiedFormatSuppressionActionRuleParameterSet:
@@ -354,8 +354,8 @@ protected override void ProcessRecordInternal()
                             )
                         );
 
-                        actionRule = new PSActionRule(this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
-                            resourceGroupName: ResourceGroupName, actionRuleName: Name, actionRule: suppressionAR).Result.Body);
+                        result = this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
+                            resourceGroupName: ResourceGroupName, actionRuleName: Name, actionRule: suppressionAR).Result.Body;
                         break;
 
                     case BySimplifiedFormatDiagnosticsActionRuleParameterSet:
@@ -376,8 +376,8 @@ protected override void ProcessRecordInternal()
                             )
                         );
 
-                        actionRule = new PSActionRule(this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
-                            resourceGroupName: ResourceGroupName, actionRuleName: Name, actionRule: diagnosticsAR).Result.Body);
+                        result = this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
+                            resourceGroupName: ResourceGroupName, actionRuleName: Name, actionRule: diagnosticsAR).Result.Body;
 
                         break;
                     case ByInputObjectParameterSet:
@@ -386,36 +386,38 @@ protected override void ProcessRecordInternal()
                         {
                             case "ActionGroup":
                                 // Create Action Rule
+                                PSActionGroupActionRule actionGroupInputObject = (PSActionGroupActionRule)InputObject;
                                 ActionRule actionGroupARFromInputObject = new ActionRule(
                                     location: "Global",
                                     tags: new Dictionary<string, string>(),
                                     properties: new ActionGroup(
-                                        scope: JsonConvert.DeserializeObject<Scope>(InputObject.Scope),
-                                        conditions: JsonConvert.DeserializeObject<Conditions>(InputObject.Conditions),
-                                        actionGroupId: InputObject.ActionGroupId,
-                                        description: InputObject.Description,
-                                        status: InputObject.Status
+                                        scope: JsonConvert.DeserializeObject<Scope>(actionGroupInputObject.Scope),
+                                        conditions: JsonConvert.DeserializeObject<Conditions>(actionGroupInputObject.Conditions),
+                                        actionGroupId: actionGroupInputObject.ActionGroupId,
+                                        description: actionGroupInputObject.Description,
+                                        status: actionGroupInputObject.Status
                                     )
                                 );
 
-                                actionRule = new PSActionRule(this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
-                                    resourceGroupName: info.ResourceGroupName, actionRuleName: info.Resource, actionRule: actionGroupARFromInputObject).Result.Body);
+                                result = this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
+                                    resourceGroupName: info.ResourceGroupName, actionRuleName: info.Resource, actionRule: actionGroupARFromInputObject).Result.Body;
                                 break;
 
                             case "Suppression":
-                                SuppressionConfig configFromInputObject = new SuppressionConfig(recurrenceType: InputObject.SuppressionConfig.RecurrenceType);
-                                if (InputObject.SuppressionConfig.RecurrenceType != "Always")
+                                PSSuppressionActionRule suppressionInputObject = (PSSuppressionActionRule)InputObject;
+                                SuppressionConfig configFromInputObject = new SuppressionConfig(recurrenceType: suppressionInputObject.SuppressionConfig.RecurrenceType);
+                                if (suppressionInputObject.SuppressionConfig.RecurrenceType != "Always")
                                 {
                                     configFromInputObject.Schedule = new SuppressionSchedule(
-                                        startDate: InputObject.SuppressionConfig.StartDate,
-                                        endDate: InputObject.SuppressionConfig.EndDate,
-                                        startTime: InputObject.SuppressionConfig.StartTime,
-                                        endTime: InputObject.SuppressionConfig.EndTime
+                                        startDate: suppressionInputObject.SuppressionConfig.StartDate,
+                                        endDate: suppressionInputObject.SuppressionConfig.EndDate,
+                                        startTime: suppressionInputObject.SuppressionConfig.StartTime,
+                                        endTime: suppressionInputObject.SuppressionConfig.EndTime
                                         );
 
                                     if (ReccurentValue.Length > 0)
                                     {
-                                        configFromInputObject.Schedule.RecurrenceValues = InputObject.SuppressionConfig.RecurrenceValues;
+                                        configFromInputObject.Schedule.RecurrenceValues = suppressionInputObject.SuppressionConfig.RecurrenceValues;
                                     }
                                 }
 
@@ -424,39 +426,40 @@ protected override void ProcessRecordInternal()
                                     location: "Global",
                                     tags: new Dictionary<string, string>(),
                                     properties: new Suppression(
-                                        scope: JsonConvert.DeserializeObject<Scope>(InputObject.Scope),
-                                        conditions: JsonConvert.DeserializeObject<Conditions>(InputObject.Conditions),
-                                        description: InputObject.Description,
-                                        status: InputObject.Status,
+                                        scope: JsonConvert.DeserializeObject<Scope>(suppressionInputObject.Scope),
+                                        conditions: JsonConvert.DeserializeObject<Conditions>(suppressionInputObject.Conditions),
+                                        description: suppressionInputObject.Description,
+                                        status: suppressionInputObject.Status,
                                         suppressionConfig: configFromInputObject
                                     )
                                 );
 
-                                actionRule = new PSActionRule(this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
-                                    resourceGroupName: info.ResourceGroupName, actionRuleName: info.Resource, actionRule: suppressionARFromInputObject).Result.Body);
+                                result = this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
+                                    resourceGroupName: info.ResourceGroupName, actionRuleName: info.Resource, actionRule: suppressionARFromInputObject).Result.Body;
                                 break;
 
                             case "Diagnostics":
                                 // Create Action Rule
+                                PSDiagnosticsActionRule diagnosticsInputObject = (PSDiagnosticsActionRule)InputObject;
                                 ActionRule diagnosticsARFromInputObject = new ActionRule(
                                     location: "Global",
                                     tags: new Dictionary<string, string>(),
                                     properties: new Diagnostics(
-                                        scope: JsonConvert.DeserializeObject<Scope>(InputObject.Scope),
-                                        conditions: JsonConvert.DeserializeObject<Conditions>(InputObject.Conditions),
-                                        description: InputObject.Description,
-                                        status: InputObject.Status
+                                        scope: JsonConvert.DeserializeObject<Scope>(diagnosticsInputObject.Scope),
+                                        conditions: JsonConvert.DeserializeObject<Conditions>(diagnosticsInputObject.Conditions),
+                                        description: diagnosticsInputObject.Description,
+                                        status: diagnosticsInputObject.Status
                                     )
                                 );
 
-                                actionRule = new PSActionRule(this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
-                                    resourceGroupName: info.ResourceGroupName, actionRuleName: info.Resource, actionRule: diagnosticsARFromInputObject).Result.Body);
+                                result = this.AlertsManagementClient.ActionRules.CreateUpdateWithHttpMessagesAsync(
+                                    resourceGroupName: info.ResourceGroupName, actionRuleName: info.Resource, actionRule: diagnosticsARFromInputObject).Result.Body;
                                 break;
                         }
                         break;
                 }
 
-                WriteObject(sendToPipeline: actionRule);
+                WriteObject(sendToPipeline: TransformOutput(result));
             }
         }
 
@@ -565,5 +568,22 @@ private string DetermineScopeType(string value)
                 throw new PSInvalidOperationException(string.Format(Resources.InvalidScopeType_Exception));
             }
         }
+
+        private PSActionRule TransformOutput(ActionRule input)
+        {
+            string actionRuleType = input.Properties.GetType().Name;
+
+            switch (actionRuleType)
+            {
+                case "Suppression":
+                    return new PSSuppressionActionRule(input);
+                case "ActionGroup":
+                    return new PSActionGroupActionRule(input);
+                case "Diagnostics":
+                    return new PSDiagnosticsActionRule(input);
+                default:
+                    return new PSActionRule(input);
+            }
+        }
     }
 }

src/AlertsManagement/AlertsManagement/Az.AlertsManagement.psd1

@@ -0,0 +1,149 @@
+#
+# Module manifest for module 'Az.Cdn'
+#
+# Generated by: Microsoft Corporation
+#
+# Generated on: 5/29/2019
+#
+
+@{
+
+# Script module or binary module file associated with this manifest.
+# RootModule = ''
+
+# Version number of this module.
+ModuleVersion = '1.3.0'
+
+# Supported PSEditions
+CompatiblePSEditions = 'Core', 'Desktop'
+
+# ID used to uniquely identify this module
+GUID = '91832aaa-dc11-4583-8239-bce5fd531604'
+
+# Author of this module
+Author = 'Microsoft Corporation'
+
+# Company or vendor of this module
+CompanyName = 'Microsoft Corporation'
+
+# Copyright statement for this module
+Copyright = 'Microsoft Corporation. All rights reserved.'
+
+# Description of the functionality provided by this module
+Description = 'Microsoft Azure PowerShell - CDN service cmdlets for Azure Resource Manager in Windows PowerShell and PowerShell Core.
+
+For more information on Cdn, please visit the following: https://docs.microsoft.com/azure/cdn/'
+
+# Minimum version of the PowerShell engine required by this module
+PowerShellVersion = '5.1'
+
+# Name of the PowerShell host required by this module
+# PowerShellHostName = ''
+
+# Minimum version of the PowerShell host required by this module
+# PowerShellHostVersion = ''
+
+# Minimum version of Microsoft .NET Framework required by this module. This prerequisite is valid for the PowerShell Desktop edition only.
+DotNetFrameworkVersion = '4.7.2'
+
+# Minimum version of the common language runtime (CLR) required by this module. This prerequisite is valid for the PowerShell Desktop edition only.
+# CLRVersion = ''
+
+# Processor architecture (None, X86, Amd64) required by this module
+# ProcessorArchitecture = ''
+
+# Modules that must be imported into the global environment prior to importing this module
+RequiredModules = @(@{ModuleName = 'Az.Accounts'; ModuleVersion = '1.6.0'; })
+
+# Assemblies that must be loaded prior to importing this module
+RequiredAssemblies = '.\Microsoft.Azure.Management.Cdn.dll'
+
+# Script files (.ps1) that are run in the caller's environment prior to importing this module.
+# ScriptsToProcess = @()
+
+# Type files (.ps1xml) to be loaded when importing this module
+# TypesToProcess = @()
+
+# Format files (.ps1xml) to be loaded when importing this module
+# FormatsToProcess = @()
+
+# Modules to import as nested modules of the module specified in RootModule/ModuleToProcess
+NestedModules = @('.\Microsoft.Azure.PowerShell.Cmdlets.Cdn.dll')
+
+# Functions to export from this module, for best performance, do not use wildcards and do not delete the entry, use an empty array if there are no functions to export.
+FunctionsToExport = @()
+
+# Cmdlets to export from this module, for best performance, do not use wildcards and do not delete the entry, use an empty array if there are no cmdlets to export.
+CmdletsToExport = 'Get-AzCdnProfile', 'Get-AzCdnProfileSsoUrl', 'New-AzCdnProfile', 
+               'Remove-AzCdnProfile', 'Set-AzCdnProfile', 'Get-AzCdnOrigin', 
+               'Set-AzCdnOrigin', 'Get-AzCdnEndpointNameAvailability', 
+               'Get-AzCdnEndpoint', 'Publish-AzCdnEndpointContent', 
+               'New-AzCdnEndpoint', 'Unpublish-AzCdnEndpointContent', 
+               'Remove-AzCdnEndpoint', 'Set-AzCdnEndpoint', 'Start-AzCdnEndpoint', 
+               'Stop-AzCdnEndpoint', 'Test-AzCdnCustomDomain', 
+               'Get-AzCdnCustomDomain', 'New-AzCdnCustomDomain', 
+               'Remove-AzCdnCustomDomain', 'Enable-AzCdnCustomDomain', 
+               'Disable-AzCdnCustomDomain', 'Enable-AzCdnCustomDomainHttps', 
+               'Disable-AzCdnCustomDomainHttps', 'Get-AzCdnProfileResourceUsage', 
+               'Confirm-AzCdnEndpointProbeUrl', 'Get-AzCdnEndpointResourceUsage', 
+               'Get-AzCdnProfileSupportedOptimizationType', 
+               'Get-AzCdnSubscriptionResourceUsage', 'Get-AzCdnEdgeNode', 
+               'New-AzCdnDeliveryRule', 'New-AzCdnDeliveryRuleCondition', 
+               'New-AzCdnDeliveryRuleAction', 'New-AzCdnDeliveryPolicy'
+
+# Variables to export from this module
+# VariablesToExport = @()
+
+# Aliases to export from this module, for best performance, do not use wildcards and do not delete the entry, use an empty array if there are no aliases to export.
+AliasesToExport = 'Validate-AzCdnCustomDomain', 'Get-AzCdnEdgeNodes'
+
+# DSC resources to export from this module
+# DscResourcesToExport = @()
+
+# List of all modules packaged with this module
+# ModuleList = @()
+
+# List of all files packaged with this module
+# FileList = @()
+
+# Private data to pass to the module specified in RootModule/ModuleToProcess. This may also contain a PSData hashtable with additional module metadata used by PowerShell.
+PrivateData = @{
+
+    PSData = @{
+
+        # Tags applied to this module. These help with module discovery in online galleries.
+        Tags = 'Azure','ResourceManager','ARM','CDN'
+
+        # A URL to the license for this module.
+        LicenseUri = 'https://aka.ms/azps-license'
+
+        # A URL to the main website for this project.
+        ProjectUri = 'https://github.com/Azure/azure-powershell'
+
+        # A URL to an icon representing this module.
+        # IconUri = ''
+
+        # ReleaseNotes of this module
+        ReleaseNotes = '* Updated cmdlets to support rulesEngine feature based on API version 2019-04-15.'
+
+        # Prerelease string of this module
+        # Prerelease = ''
+
+        # Flag to indicate whether the module requires explicit user acceptance for install/update/save
+        # RequireLicenseAcceptance = $false
+
+        # External dependent modules of this module
+        # ExternalModuleDependencies = @()
+
+    } # End of PSData hashtable
+
+ } # End of PrivateData hashtable
+
+# HelpInfo URI of this module
+# HelpInfoURI = ''
+
+# Default prefix for commands exported from this module. Override the default prefix using Import-Module -Prefix.
+# DefaultCommandPrefix = ''
+
+}
+