AzSKPreview.ADO

1.24.1

Framework/Listeners/UserReports/WritePsConsole.ps1

Set-StrictMode -Version Latest 
class WritePsConsole: FileOutputBase
{
    hidden static [WritePsConsole] $Instance = $null;
    hidden [string] $SummaryMarkerText = "------";
    static [WritePsConsole] GetInstance()
    {
        if ($null -eq  [WritePsConsole]::Instance)
        {
            [WritePsConsole]::Instance = [WritePsConsole]::new();
        }

        return [WritePsConsole]::Instance
    }

    [void] RegisterEvents()
    {        
        $this.UnregisterEvents();

        # Mandatory: Generate Run Identifier Event
        $this.RegisterEvent([AzSKRootEvent]::GenerateRunIdentifier, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $currentInstance.SetRunIdentifier([AzSKRootEventArgument] ($Event.SourceArgs | Select-Object -First 1));                         
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });
            
        
        $this.RegisterEvent([AzSKGenericEvent]::CustomMessage, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                if($Event.SourceArgs)
                {
                    $messages = @();
                    $messages += $Event.SourceArgs;
                    $messages | ForEach-Object {
                        $currentInstance.WriteMessageData($_);
                    }
                }
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

        $this.RegisterEvent([AzSKGenericEvent]::Exception, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $exceptionObj = $Event.SourceArgs | Select-Object -First 1
                #if(($null -ne $exceptionObj) -and ($null -ne $exceptionObj.Exception) -and (-not [String]::IsNullOrEmpty($exceptionObj.Exception.Message)))
                #{
                # $currentInstance.WriteMessage($exceptionObj.Exception.Message, [MessageType]::Error);
                # Write-Debug $exceptionObj
                #}
                #else
                #{
                    $currentInstance.WriteMessage($exceptionObj, [MessageType]::Error);                       
                #}
            }
            catch 
            {
                #Consuming the exception intentionally to prevent infinite loop of errors
                #$currentInstance.PublishException($_);
            }
        });
        

        $this.RegisterEvent([AzSKRootEvent]::CustomMessage, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                if($Event.SourceArgs -and $Event.SourceArgs.Messages)
                {
                    $Event.SourceArgs.Messages | ForEach-Object {
                        $currentInstance.WriteMessageData($_);
                    }
                }
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });
        
        $this.RegisterEvent([AzSKRootEvent]::CommandStarted, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $currentInstance.CommandStartedAction($Event);
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

        $this.RegisterEvent([AzSKRootEvent]::CommandError, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $currentInstance.WriteMessage($Event.SourceArgs.ExceptionMessage, [MessageType]::Error);  
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });
                
        $this.RegisterEvent([AzSKRootEvent]::CommandCompleted, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $messages = $Event.SourceArgs.Messages;
                if(($messages | Measure-Object).Count -gt 0 -and $Event.SourceArgs.Messages[0].Message -eq "RecommendationData")
                {
                    $reportObject = [RecommendedSecurityReport] $Event.SourceArgs.Messages[0].DataObject;
                    $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)
                    $currentInstance.WriteMessage("Current Combination", [MessageType]::Info)
                    $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)
                    if([string]::IsNullOrWhiteSpace($reportObject.ResourceGroupName))
                    {
                        $currentInstance.WriteMessage("ResourceGroup Name: Not Specified", [MessageType]::Default);    
                    }
                    else {
                        $currentInstance.WriteMessage("ResourceGroup Name: [$($reportObject.ResourceGroupName)]", [MessageType]::Default);
                    }

                    if(($reportObject.Input.Features | Measure-Object).Count -le 0)
                    {
                        $currentInstance.WriteMessage("Features: Not Specified", [MessageType]::Default);
                    }
                    else {
                        $featuresString = [String]::Join(",", $reportObject.Input.Features);
                        $currentInstance.WriteMessage("Features: [$featuresString]", [MessageType]::Default);
                    }

                    if(($reportObject.Input.Categories | Measure-Object).Count -le 0)
                    {
                        $currentInstance.WriteMessage("Categories: Not Specified", [MessageType]::Default);
                    }
                    else {
                        $categoriesString = [String]::Join(",", $reportObject.Input.Categories);
                        $currentInstance.WriteMessage("Categories: [$categoriesString]", [MessageType]::Default);
                    }
                    $currentInstance.WriteMessage([Constants]::UnderScoreLineLine, [MessageType]::Info)                    
                    $currentInstance.WriteMessage("Analysis & Recommendations:", [MessageType]::Info);
                    $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info);
                    $currentInstance.WriteMessage("Analysis of current feature group:", [MessageType]::Info);
                    if($null -ne $reportObject.Recommendations.CurrentFeatureGroup)
                    {
                        $currentInstance.WriteMessage("Current Group Ranking: $($reportObject.Recommendations.CurrentFeatureGroup.Ranking)", [MessageType]::Default);
                        $currentInstance.WriteMessage("No. of instances with same combination: $($reportObject.Recommendations.CurrentFeatureGroup.TotalOccurances)", [MessageType]::Default);
                        $featuresString = [String]::Join(",", $reportObject.Recommendations.CurrentFeatureGroup.Features);
                        $currentInstance.WriteMessage("Current Combination Features: $featuresString", [MessageType]::Default);
                        $categoriesString = [String]::Join(",", $reportObject.Recommendations.CurrentFeatureGroup.Categories);
                        $currentInstance.WriteMessage("Current Combination Categories: $categoriesString", [MessageType]::Default);
                        $currentInstance.WriteMessage("Measures: [Total Pass#: $($reportObject.Recommendations.CurrentFeatureGroup.TotalSuccessCount)] [Total Fail#: $($reportObject.Recommendations.CurrentFeatureGroup.TotalFailCount)] ", [MessageType]::Default);                                                                        
                    }
                    else {
                        $currentInstance.WriteMessage("Cannot find exact matching combination for the current user input.", [MessageType]::Default);
                    }
                    $currentInstance.WriteMessage([Constants]::SingleDashLine, [MessageType]::Info);
                    $currentInstance.WriteMessage("Recommendations based on categories:", [MessageType]::Info);
                    if(($reportObject.Recommendations.RecommendedFeatureGroups | Measure-Object).Count -gt 0)
                    {
                        $orderedRecommendations = $reportObject.Recommendations.RecommendedFeatureGroups | Sort-Object -Property Ranking
                        $orderedRecommendations | ForEach-Object {
                            $recommendation = $_;
                            $currentInstance.WriteMessage("Category Group Ranking: $($recommendation.Ranking)", [MessageType]::Default);
                            $currentInstance.WriteMessage("No. of instances with same combination: $($recommendation.TotalOccurances)", [MessageType]::Default);
                            $featuresString = [String]::Join(",", $recommendation.Features);
                            $currentInstance.WriteMessage("Feature combination: $featuresString", [MessageType]::Default);
                            $categoriesString = [String]::Join(",", $recommendation.Categories);
                            $currentInstance.WriteMessage("Category Combination: $categoriesString", [MessageType]::Default);
                            $currentInstance.WriteMessage("Measures: [Total Pass#: $($recommendation.TotalSuccessCount)] [Total Fail#: $($recommendation.TotalFailCount)] ", [MessageType]::Default);                                                                        
                            $currentInstance.WriteMessage([Constants]::SingleDashLine, [MessageType]::Info);
                        }
                    }

                    $currentInstance.WriteMessage(($dataObject | ConvertTo-Json -Depth 10), [MessageType]::Info)
                }
                else {
                    $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)
                    $currentInstance.WriteMessage("Logs have been exported to: '$([WriteFolderPath]::GetInstance().FolderPath)'", [MessageType]::Info)
                    $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)    
                }                                
                $currentInstance.FilePath = "";
                ##Print Error##
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

        # SVT events
        $this.RegisterEvent([SVTEvent]::CommandStarted, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $currentInstance.CommandStartedAction($Event);
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });
        
        $this.RegisterEvent([SVTEvent]::CommandError, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $currentInstance.WriteMessage($Event.SourceArgs.ExceptionMessage, [MessageType]::Error);  
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

        $this.RegisterEvent([SVTEvent]::CommandCompleted, {
            $currentInstance = [WritePsConsole]::GetInstance();
            $currentInstance.PushAIEventsfromHandler("WritePsConsole CommandCompleted"); 
            try 
            {
                if(($Event.SourceArgs | Measure-Object).Count -gt 0 -or $null -ne [PartialScanManager]::CollatedSummaryCount)
                {
                    
                    # Print summary
                    $currentInstance.PrintSummaryData($Event);
                    if($currentInstance.InvocationContext.MyCommand.Name -eq "Set-AzSKADOBaselineConfigurations"){
                        $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)
                        $currentInstance.PrintBaselineConfigurationData($Event);
                    }                
                    
                    $AttestControlParamFound = $currentInstance.InvocationContext.BoundParameters["AttestControls"];
                    if($null -eq $AttestControlParamFound)
                    {
                        $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)
                        $currentInstance.WriteMessage([Constants]::RemediationMsg, [MessageType]::Info)
                        #$currentInstance.WriteMessage([Constants]::AttestationReadMsg + [ConfigurationManager]::GetAzSKConfigData().AzSKRGName, [MessageType]::Info)
                        
                    }

                    #if auto bug logging is enabled and the path is valid or autoClosedBugs is enabled, print a summary of all bugs encountered
                    if(($currentInstance.InvocationContext.BoundParameters["AutoBugLog"] -and [BugLogPathManager]::GetIsPathValid()) -or $currentInstance.InvocationContext.BoundParameters["AutoCloseBugs"]){
                        $currentInstance.WriteMessage([Constants]::SingleDashLine, [MessageType]::Info)
                        $currentInstance.PrintBugSummaryData($Event);
                    }
                    $currentInstance.WriteMessage([Constants]::SingleDashLine, [MessageType]::Info)
                }

                $currentInstance.WriteMessage("Status and detailed logs have been exported to path - $([WriteFolderPath]::GetInstance().FolderPath)", [MessageType]::Info)
                $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)

                #change batch scan state to COMP
                if($currentInstance.InvocationContext.BoundParameters.ContainsKey('BatchScan')){
                    $CurrentResourceCount = $currentInstance.UpdateCurrentBatch();
                    $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Update)
                    if($currentInstance.InvocationContext.BoundParameters.ContainsKey('BatchScanMultipleProjects')){
                        $currentProjectDetails = $currentInstance.GetProjectCurrentBatch()
                        $CurrentProjectCount = $currentProjectDetails[0];
                        $TotalProjectCount = $currentProjectDetails[1];
                        $CurrentProject = $currentProjectDetails[2];
                        if($null -ne $CurrentProject){
                            $currentInstance.WriteMessage("Execution completed for current batch. Scanned $($CurrentResourceCount) resources. $($CurrentProjectCount) out of $($TotalProjectCount) projects have been completely scanned. Scan for $($CurrentProject) is in progress and will be scanned in next batch. Next scan will take place in a fresh PS Console. You may close this window now.", [MessageType]::Update)
                        }
                        else {
                            $currentInstance.WriteMessage("Execution completed for current batch. Scanned $($CurrentResourceCount) resources. $($CurrentProjectCount) out of $($TotalProjectCount) projects have been completely scanned. Next scan will take place in a fresh PS Console. You may close this window now.", [MessageType]::Update)
                        }                        
                    }
                    else {
                        $currentInstance.WriteMessage("Execution completed for current batch. Scanned $($CurrentResourceCount) resources. Next scan will take place in a fresh PS Console. You may close this window now.", [MessageType]::Update)
                    }            
                    
                    $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Update)
                }
                $currentInstance.FilePath = "";
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

        $this.RegisterEvent([SVTEvent]::EvaluationStarted, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                if($Event.SourceArgs.IsResource())
                {
                    $startHeading = ([Constants]::ModuleStartHeading -f $Event.SourceArgs.FeatureName, $Event.SourceArgs.ResourceContext.ResourceGroupName, $Event.SourceArgs.ResourceContext.ResourceName);
                }
                else
                {
                    $startHeading = ([Constants]::ModuleStartHeadingSub -f $Event.SourceArgs.FeatureName, $Event.SourceArgs.OrganizationContext.OrganizationName, $Event.SourceArgs.OrganizationContext.OrganizationId);                    
                }
                $currentInstance.WriteMessage($startHeading, [MessageType]::Info);
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

        $this.RegisterEvent([SVTEvent]::EvaluationCompleted, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                if($Event.SourceArgs -and $Event.SourceArgs.Count -ne 0)
                {
                    $props = $Event.SourceArgs[0];
                    if($props.IsResource())
                    {
                        $currentInstance.WriteMessage(([Constants]::CompletedAnalysis  -f $props.FeatureName, $props.ResourceContext.ResourceGroupName, $props.ResourceContext.ResourceName), [MessageType]::Update);
                    }
                    else
                    {
                        $currentInstance.WriteMessage(([Constants]::CompletedAnalysisSub  -f $props.FeatureName, $props.OrganizationContext.OrganizationName, $props.OrganizationContext.OrganizationId), [MessageType]::Update);
                    }
                }
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

        $this.RegisterEvent([SVTEvent]::EvaluationError, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $currentInstance.WriteMessage($Event.SourceArgs.ExceptionMessage, [MessageType]::Error);  
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });

          $this.RegisterEvent([SVTEvent]::ControlStarted, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                if($Event.SourceArgs.IsResource())
                {
                    $AnalysingControlHeadingMsg =([Constants]::AnalysingControlHeading  -f $Event.SourceArgs.FeatureName, $Event.SourceArgs.ControlItem.Description,$Event.SourceArgs.ResourceContext.ResourceName)
                }
                else
                {
                    $AnalysingControlHeadingMsg =([Constants]::AnalysingControlHeadingSub  -f $Event.SourceArgs.FeatureName, $Event.SourceArgs.ControlItem.Description,$Event.SourceArgs.OrganizationContext.OrganizationName)
                }
                $currentInstance.WriteMessage($AnalysingControlHeadingMsg, [MessageType]::Info)                             
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });
        
        $this.RegisterEvent([SVTEvent]::ControlDisabled, {
            $currentInstance = [WritePsConsole]::GetInstance();
            try 
            {
                $currentInstance.WriteMessage(("**Disabled**: [{0}]-[{1}]" -f 
                        $Event.SourceArgs.FeatureName, 
                        $Event.SourceArgs.ControlItem.Description), [MessageType]::Warning);    
            }
            catch 
            {
                $currentInstance.PublishException($_);
            }
        });
    }

        #Write message on powershell console with appropriate color
    [void] WriteMessage([PSObject] $message, [MessageType] $messageType)
    {
        if(-not $message)
        {
            return;
        }
        
        $colorCode = [System.ConsoleColor]::White

        switch($messageType)
        {
            ([MessageType]::Critical) {  
                $colorCode = [System.ConsoleColor]::Red              
            }
            ([MessageType]::Error) {
                $colorCode = [System.ConsoleColor]::Red             
            }
            ([MessageType]::Warning) {
                $colorCode = [System.ConsoleColor]::Yellow              
            }
            ([MessageType]::Info) {
                $colorCode = [System.ConsoleColor]::Cyan
            }  
            ([MessageType]::Update) {
                $colorCode = [System.ConsoleColor]::Green
            }
            ([MessageType]::Deprecated) {
                $colorCode = [System.ConsoleColor]::DarkYellow
            }
            ([MessageType]::Default) {
                $colorCode = [System.ConsoleColor]::White
            }           
        }   

        # FilePath check ensures to print detailed error objects on PS host
        $formattedMessage = [Helpers]::ConvertObjectToString($message, (-not [string]::IsNullOrEmpty($this.FilePath)));        
        Write-Host $formattedMessage -ForegroundColor $colorCode
        #if($message.GetType().FullName -eq "System.Management.Automation.ErrorRecord")
        #{
        $this.AddOutputLog([Helpers]::ConvertObjectToString($message, $false));
        #}
        #else
        #{
        # $this.AddOutputLog($message);
        #}
    }
    
    hidden [void] WriteMessage([PSObject] $message)
    {
        $this.WriteMessage($message, [MessageType]::Info);
    }

    hidden [void] WriteMessageData([MessageData] $messageData)
    {
        if($messageData)
        {
            $this.WriteMessage(("`r`n" + $messageData.Message), $messageData.MessageType);       
            if($messageData.DataObject)
            {
                #if (-not [string]::IsNullOrEmpty($messageData.Message))
                #{
                # $this.WriteMessage("`r`n");
                #}

                $this.WriteMessage($messageData.DataObject, $messageData.MessageType);       
            }
        }
    }

    hidden [void] AddOutputLog([string] $message, [bool] $includeTimeStamp)   
    {
        if([string]::IsNullOrEmpty($message) -or [string]::IsNullOrEmpty($this.FilePath))
        {
            return;
        }
             
        if($includeTimeStamp)
        {
            $message = (Get-Date -format "MM\/dd\/yyyy HH:mm:ss") + "-" + $message
        }

        Add-Content -Value $message -Path $this.FilePath        
    } 
        
    hidden [void] AddOutputLog([string] $message)   
    {
       $this.AddOutputLog($message, $false);  
    } 

    hidden [void] PrintSummaryData($event)
    {
        if (($event.SourceArgs | Measure-Object).Count -ne 0)
        {
            $summary = @($event.SourceArgs | select-object @{Name="VerificationResult"; Expression = {$_.ControlResults.VerificationResult}},@{Name="ControlSeverity"; Expression = {$_.ControlItem.ControlSeverity}})

            if(($summary | Measure-Object).Count -ne 0)
            {
                $summaryResult = @();

                $severities = @();
                $severities += $summary | Select-Object -Property ControlSeverity | Select-Object -ExpandProperty ControlSeverity -Unique;

                $verificationResults = @();
                $verificationResults += $summary | Select-Object -Property VerificationResult | Select-Object -ExpandProperty VerificationResult -Unique;

                if($severities.Count -ne 0)
                {
                    # Create summary matrix
                    $totalText = "Total";
                    $MarkerText = "MarkerText";
                    $rows = @();
                    $rows += $severities;
                    $rows += $MarkerText;
                    $rows += $totalText;
                    $rows += $MarkerText;
                    $rows | ForEach-Object {
                        $result = [PSObject]::new();
                        Add-Member -InputObject $result -Name "Summary" -MemberType NoteProperty -Value $_.ToString()
                        Add-Member -InputObject $result -Name $totalText -MemberType NoteProperty -Value 0

                        [Enum]::GetNames([VerificationResult]) | Where-Object { $verificationResults -contains $_ } |
                        ForEach-Object {
                            Add-Member -InputObject $result -Name $_.ToString() -MemberType NoteProperty -Value 0
                        };
                        $summaryResult += $result;
                    };

                    $totalRow = $summaryResult | Where-Object { $_.Summary -eq $totalText } | Select-Object -First 1;

                    $summary | Group-Object -Property ControlSeverity | ForEach-Object {
                        $item = $_;
                        $summaryItem = $summaryResult | Where-Object { $_.Summary -eq $item.Name } | Select-Object -First 1;
                        if($summaryItem)
                        {
                            $summaryItem.Total = $_.Count;
                            if($totalRow)
                            {
                                $totalRow.Total += $_.Count
                            }
                            $item.Group | Group-Object -Property VerificationResult | ForEach-Object {
                                $propName = $_.Name;
                                $summaryItem.$propName += $_.Count;
                                if($totalRow)
                                {
                                    $totalRow.$propName += $_.Count
                                }
                            };
                        }
                    };
                    $markerRows = $summaryResult | Where-Object { $_.Summary -eq $MarkerText } 
                    $markerRows | ForEach-Object { 
                        $markerRow = $_
                        Get-Member -InputObject $markerRow -MemberType NoteProperty | ForEach-Object {
                                $propName = $_.Name;
                                $markerRow.$propName = $this.SummaryMarkerText;                
                            }
                        };
                    if($summaryResult.Count -ne 0)
                    {        
                        $this.WriteMessage(($summaryResult | Format-Table | Out-String), [MessageType]::Info)
                    }
                }
            }
        }
        else
        {
            if([PartialScanManager]::CollatedSummaryCount.Count -ne 0)
            {    
                $nonNullProps = @();

                #get all verificationResults that are not 0 so that summary does not include null values
                [PartialScanManager]::CollatedSummaryCount | foreach-object {
                    $nonNullProps += $_.PSObject.Properties | Where-Object {$_.Value -ne 0 -and $_.Value -ne $this.SummaryMarkerText} | Select-Object -ExpandProperty Name
                }     
                $nonNullProps = $nonNullProps | Select -Unique
                $this.WriteMessage(([PartialScanManager]::CollatedSummaryCount | Format-Table -Property $nonNullProps | Out-String), [MessageType]::Info)
                [PartialScanManager]::CollatedSummaryCount = @()
            }
        }
    }

    hidden [void] PrintBaselineConfigurationData($event){
        $erroredControls = 0;
        $passedControls = @{Passed = @()}
        $fixedControls = @{Fixed = @()}
        if (($event.SourceArgs | Measure-Object).Count -ne 0){
            $event.SourceArgs | ForEach-Object {
                $item = $_
                if ($item -and $item.ControlResults){
                    $control = [PSCustomObject]@{
                        'Control' = $item.ControlItem.ControlID
                        'ResourceName' = $item.ResourceContext.ResourceName
                    }
                    if($item.ControlResults[0].VerificationResult -eq "Fixed"){
                        $fixedControls.Fixed+=$control
                    }
                    elseif($item.ControlResults[0].VerificationResult -eq "Passed"){
                        $passedControls.Passed+=$control
                    }
                    else{
                        $erroredControls++;
                    }
                }            

            }
            $totalControls = $fixedControls.Fixed.Count + $passedControls.Passed.Count+$erroredControls;
            $oldBaselineCompliance = ($passedControls.Passed.Count/$totalControls) *100;
            $newBaselineCompliance = (($passedControls.Passed.Count+$fixedControls.Fixed.Count)/$totalControls) *100;
            $currentInstance = [WritePsConsole]::GetInstance();
            $currentInstance.WriteMessage("Baseline compliance before configurations: "+$oldBaselineCompliance+"%`n",[MessageType]::Update);
            $currentInstance.WriteMessage("Baseline compliance after configurations: "+$newBaselineCompliance+"%`n",[MessageType]::Update);
            $currentInstance.WriteMessage([Constants]::SingleDashLine, [MessageType]::Update)
            if($passedControls.Passed.Count -gt 0){
                $currentInstance.WriteMessage("Following controls have been found to be already passing: ",[MessageType]::Update);
                $currentInstance.WriteMessage($passedControls.Passed,[MessageType]::Update);
            }
            if($fixedControls.Fixed.Count -gt 0){
                $currentInstance.WriteMessage([Constants]::SingleDashLine, [MessageType]::Update)
                $currentInstance.WriteMessage("Following controls have been fixed to increase baseline compliance: ",[MessageType]::Update);
                $currentInstance.WriteMessage($fixedControls.Fixed,[MessageType]::Update);
            }
            $currentInstance.WriteMessage([Constants]::SingleDashLine, [MessageType]::Update)
            

        }
    }

    #function to print metrics summary for all kinds of bugs encountered

    hidden [void] PrintBugSummaryData($event){

        [PSCustomObject[]] $summary = @();
        $currentInstance = [WritePsConsole]::GetInstance();
        # For -upc mode summary information is already available in static variable
        if($currentInstance.InvocationContext.BoundParameters["UsePartialCommits"]){
            $summary=[PartialScanManager]::CollatedBugSummaryCount
            $duplicateClosedBugCount=[PartialScanManager]::duplicateClosedBugCount
        }
        # In regular scan populate summary
        else {
            if (($event.SourceArgs | Measure-Object).Count -ne 0)
            {
                #gather all control results that have failed/verify as their control result
                #obtain their control severities
                $event.SourceArgs | ForEach-Object {
                    $item = $_
                    if ($item -and $item.ControlResults -and ($item.ControlResults[0].VerificationResult -eq "Failed" -or $item.ControlResults[0].VerificationResult -eq "Verify"))
                    {
                        $item
                        $item.ControlResults[0].Messages | ForEach-Object{
                            if($_.Message -eq "New Bug" -or $_.Message -eq "Active Bug" -or $_.Message -eq "Resolved Bug"){
                            $summary += [PSCustomObject]@{
                                BugStatus=$_.Message
                                ControlSeverity = $item.ControlItem.ControlSeverity;
                                
                            };
                        }
                        };
                    }
                };
            }
            #The following 2 integer variables help identify duplicate work items.
            $TotalWorkItemCount=0;
            $TotalControlsClosedCount=0;
            $bugsClosed=[AutoCloseBugManager]::ClosedBugs
            if($bugsClosed){
                $bugsClosed | ForEach-Object{
                    $TotalControlsClosedCount+=1
                    $item=$_
                    $item.ControlResults[0].Messages | ForEach-Object{
                    if($_.Message -eq "Closed Bug"){
                            $summary += [PSCustomObject]@{
                                    BugStatus=$_.Message
                                    ControlSeverity = $item.ControlItem.ControlSeverity;
                                    };
                            $TotalWorkItemCount+=1;

                        }
                    }
                }

            }
            $duplicateClosedBugCount=$TotalWorkItemCount- $TotalControlsClosedCount        
        }

        

        #if such bugs were found, print a summary table

        if($summary.Count -ne 0)
        {
            $summaryResult = @();

            $severities = @();
            $severities += $summary | Select-Object -Property ControlSeverity | Select-Object -ExpandProperty ControlSeverity -Unique;

            $bugStatusResult = @();
            $bugStatusResult += $summary | Select-Object -Property BugStatus | Select-Object -ExpandProperty BugStatus -Unique;
            $totalText = "Total";
            $MarkerText = "MarkerText";
            $rows = @();
            $rows += $severities;
            $rows += $MarkerText;
            $rows += $totalText;
            $rows += $MarkerText;
            $rows | ForEach-Object {
                $result = [PSObject]::new();
                Add-Member -InputObject $result -Name "Summary" -MemberType NoteProperty -Value $_.ToString()
                Add-Member -InputObject $result -Name $totalText -MemberType NoteProperty -Value 0
                
                $bugStatusResult |
                ForEach-Object {
                    Add-Member -InputObject $result -Name $_.ToString() -MemberType NoteProperty -Value 0
                };
                $summaryResult += $result;
            };
            $totalRow = $summaryResult | Where-Object { $_.Summary -eq $totalText } | Select-Object -First 1;

            $summary | Group-Object -Property ControlSeverity | ForEach-Object {
                $item = $_;
                $summaryItem = $summaryResult | Where-Object { $_.Summary -eq $item.Name } | Select-Object -First 1;
                if($summaryItem)
                {
                    $summaryItem.Total = $_.Count;
                    if($totalRow)
                    {
                        $totalRow.Total += $_.Count
                    }
                    $item.Group | Group-Object -Property BugStatus | ForEach-Object {
                        $propName = $_.Name;
                        $summaryItem.$propName += $_.Count;
                        if($totalRow)
                        {
                            $totalRow.$propName += $_.Count
                        }
                    };
                }
            };
            $markerRows = $summaryResult | Where-Object { $_.Summary -eq $MarkerText } 
                $markerRows | ForEach-Object { 
                    $markerRow = $_
                    Get-Member -InputObject $markerRow -MemberType NoteProperty | ForEach-Object {
                            $propName = $_.Name;
                            $markerRow.$propName = $this.SummaryMarkerText;                
                        }
                    };
                if($summaryResult.Count -ne 0)
                {        
                    $this.WriteMessage(($summaryResult | Format-Table | Out-String), [MessageType]::Info)
                }
                $currentInstance = [WritePsConsole]::GetInstance();
                $currentInstance.WriteMessage([Constants]::DoubleDashLine, [MessageType]::Info)
                $currentInstance.WriteMessage([Constants]::BugLogMsg, [MessageType]::Info)
                $currentInstance.WriteMessage("A summary of the bugs logged has been written to the following file: $([WriteFolderPath]::GetInstance().FolderPath)\BugSummary.Json", [MessageType]::Info)

            
        }
        #Print information about duplicate work items in Console summary
        if($duplicateClosedBugCount -gt 0){
            $currentInstance.WriteMessage("Count of duplicate closed work items : $duplicateClosedBugCount ", [MessageType]::Info)
        }
        #Clearing the static variables
        [PartialScanManager]::ControlResultsWithBugSummary = @();
        [PartialScanManager]::ControlResultsWithClosedBugSummary = @();
        [PartialScanManager]::CollatedBugSummaryCount = @();
        [PartialScanManager]::duplicateClosedBugCount = 0;


    }

    hidden [int] UpdateCurrentBatch(){
        if($PSCmdlet.MyInvocation.BoundParameters.ContainsKey("BatchScanMultipleProjects")){
            [BatchScanManagerForMultipleProjects] $batchScanMngr = [BatchScanManagerForMultipleProjects]:: GetInstance();
        }
        else {
            [BatchScanManager] $batchScanMngr = [BatchScanManager]:: GetInstance();
        }
        $batchStatus= $batchScanMngr.GetBatchStatus();
        $batchStatus.BatchScanState=[BatchScanState]::COMP;
        if($batchStatus.UpcError -eq 'False'){
            $batchStatus.UpcError = 'True';
        }
        $batchScanMngr.BatchScanTrackerObj = $batchStatus;
        $batchScanMngr.WriteToBatchTrackerFile();
        return $batchStatus.ResourceCount;
    }

    hidden [System.Object[]] GetProjectCurrentBatch(){
        
        [BatchScanManagerForMultipleProjects] $batchScanMngr = [BatchScanManagerForMultipleProjects]:: GetInstance();
        
        $batchStatus= $batchScanMngr.GetBatchStatus();
        $isCurrentProjectComplete = $batchScanMngr.IsCurrentProjectComplete()
        $currentProject = $batchStatus.ProjectsCount - $batchStatus.Projects.Count
        if($isCurrentProjectComplete){
            return $($currentProject+1), $batchStatus.ProjectsCount,$null;
        }
        return $currentProject, $batchStatus.ProjectsCount,$batchStatus.Projects[0];
    }


    hidden [void] CommandStartedAction($event)
    {
        $arg = $event.SourceArgs | Select-Object -First 1;
    
        $this.SetFilePath($arg.OrganizationContext, [FileOutputBase]::ETCFolderPath, "PowerShellOutput.LOG");      
        
        $currentVersion = $this.GetCurrentModuleVersion();
        $moduleName = $this.GetModuleName();
        $methodName = $this.InvocationContext.InvocationName;
        $verbndnoun = $methodName.Split('-')
        $aliasName = [CommandHelper]::Mapping | Where {$_.Verb -eq $verbndnoun[0] -and $_.Noun -eq $verbndnoun[1] }

        $this.WriteMessage([Constants]::DoubleDashLine + "`r`n$moduleName Version: $currentVersion `r`n" + [Constants]::DoubleDashLine , [MessageType]::Info);      
        # Version check message
        if($arg.Messages)
        {
            $arg.Messages | ForEach-Object {
                $this.WriteMessageData($_);
            }
        }
        
        if($aliasName)
        {
            $aliasName = $aliasName.ShortName 
            
            #Get List of parameters used with short alias
            $paramlist = @()
            $paramlist = $this.GetParamList()
            
            #Get command with short alias
            $cmID = $this.GetShortCommand($aliasName,$paramlist);

            $this.WriteMessage("Method Name: $methodName ($aliasName)`r`nInput Parameters: $(($paramlist | Out-String).TrimEnd()) `r`n`nYou can also use: $cmID `r`n" + [Constants]::DoubleDashLine , [MessageType]::Info);
        }
        else
        {
            $this.WriteMessage("Method Name: $methodName `r`nInput Parameters: $(($this.InvocationContext.BoundParameters | Out-String).TrimEnd()) `r`n" + [Constants]::DoubleDashLine , [MessageType]::Info);                           
        }
        
        $user = [ContextHelper]::GetCurrentSessionUser();
        $this.WriteMessage([ConfigurationManager]::GetAzSKConfigData().PolicyMessage + "`r`nUsing identity: " + $user,[MessageType]::Warning)
        
    }

    hidden [string] GetShortCommand($aliasName,$paramlist)
    {
        $aliasshort = $aliasName.ToLower()
            $cmID = "$aliasshort "
            #Looping on parameters and adding them to the short alias with key and value and if no alias found adding it as it is
            foreach($item in $paramlist)
            {
                $ky = $item.Alias
                $vl = $item.Value

                if($vl -eq $true)
                {
                    $vl = ""
                }
                if($ky)
                {
                    $cmID += "-$ky $vl "
                }
                else
                {
                    $ky = $item.Name
                    $cmID += "-$ky $vl "
                }
            }
        return $cmID;
    }

    hidden [psobject] GetParamList()
    {
        $paramlist = @()
            #Looping on parameters and creating list of smallest alias and creating parameter detail object
            $this.InvocationContext.BoundParameters.Keys | % {
                $key = $this.InvocationContext.MyCommand.Parameters.$_.Aliases #| Where {$_.Length -lt 5}
                $key = $key | Sort-Object length -Descending | select -Last 1
                $val = $this.InvocationContext.BoundParameters[$_]

                $myObject = New-Object System.Object

                $myObject | Add-Member -type NoteProperty -name Name -Value $_
                $myObject | Add-Member -type NoteProperty -name Alias -Value $key
                $myObject | Add-Member -type NoteProperty -name Value -Value $val

                $paramlist += $myObject
            }
        return $paramlist;
    }

}

class SVTSummary
{
    [VerificationResult] $VerificationResult = [VerificationResult]::Manual;
    [string] $ControlSeverity = [ControlSeverity]::High;
}