EAMonitor

0.0.4

Commands/Initialize-EAMonitor.ps1

Function Initialize-EAMonitor{
    [CmdletBinding()]
    Param(
        [Parameter(Mandatory = $true, ParameterSetName = "Sqlite")]
        [string]$SqliteFilePath,
        [Parameter(Mandatory = $true, ParameterSetName = "Sql")]
        [string]$SqlServer,
        [Parameter(Mandatory = $true, ParameterSetName = "Sql")]
        [string]$Database,
        [Parameter(Mandatory = $false, ParameterSetName = "Sql")]
        [bool]$IntegratedSecurity = $false,
        [Parameter(Mandatory = $true, ParameterSetName = "ConnectionString")]
        [string]$ConnectionString,
        [Parameter(Mandatory = $true, ParameterSetName = "ConnectionString")]
        [ValidateSet('MSSQL', 'SQLite')]
        [string]$DatabaseType,
        [Parameter(Mandatory = $false, ParameterSetName = "ConnectionString")]
        [Parameter(Mandatory = $false, ParameterSetName = "Sql")]
        [Parameter(Mandatory = $false, ParameterSetName = "Sqlite")]
        [switch]$DetectMonitorModules,
        [Parameter(Mandatory = $false, ParameterSetName = "ConnectionString")]
        [Parameter(Mandatory = $false, ParameterSetName = "Sql")]
        [Parameter(Mandatory = $false, ParameterSetName = "Sqlite")]
        [switch]$CreateDb,
        [Parameter(Mandatory = $false, ParameterSetName = "ConnectionString")]
        [Parameter(Mandatory = $false, ParameterSetName = "Sql")]
        [Parameter(Mandatory = $false, ParameterSetName = "Sqlite")]
        [string]$Environment
    )
    begin{
        $Script:efPoshDbContextParams = @{
            'EnsureCreated' = $true -eq $CreateDb
            'RunMigrations' = $true -eq $CreateDb
        }
        $ParentDirectory = (Get-Item $PSScriptRoot).Parent
        [EAMonitor.Classes.EAMonitorModuleCache]::Clear()
    }
    Process{
        if($PSVersionTable.PSVersion.Major -gt 5){
            $Script:efPoshDbContextParams['AssemblyFile'] = [System.IO.Path]::Combine($ParentDirectory.FullName, "Dependencies", "net6.0", "EAMonitor.dll")
        }
        else{
            $Script:efPoshDbContextParams['AssemblyFile'] = [System.IO.Path]::Combine($ParentDirectory.FullName, "Dependencies", "net472", "EAMonitor.dll")
        }

        if($Environment){
            $Script:EAMonitorEnvironment = $Environment
        }
        
        if($PSCmdlet.ParameterSetName -eq 'Sqlite'){
            $Script:efPoshDbContextParams['SQLiteFile'] = $SqliteFilePath
            $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSqlite'
            if($PSVersionTable.PSVersion.Major -le 5){
                $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSqliteNet47'
            }
        }
        elseif($PSCmdlet.ParameterSetName -eq 'ConnectionString'){
            $Script:efPoshDbContextParams['DBType'] = $DatabaseType
            $Script:efPoshDbContextParams['ConnectionString'] = $ConnectionString
            if($DatabaseType -eq 'SQLite'){
                $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSqlite'
                if($PSVersionTable.PSVersion.Major -le 5){
                    $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSqliteNet47'
                }
            }
            elseif($DatabaseType -eq 'MSSQL'){
                $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSQL'
                if($PSVersionTable.PSVersion.Major -le 5){
                    $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSQLNet47'
                }
            }
        }
        elseif($PSCmdlet.ParameterSetName -eq 'Sql'){
            $Script:efPoshDbContextParams['MSSQLServer'] = $SqlServer
            $Script:efPoshDbContextParams['MSSQLDatabase'] = $Database
            $Script:efPoshDbContextParams['MSSQLIntegratedSecurity'] = $IntegratedSecurity
            $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSQL'
            if($PSVersionTable.PSVersion.Major -le 5){
                $Script:efPoshDbContextParams['ClassName'] = 'EAMonitorContextSQLNet47'
            }
        }
        else{
            throw "Congrats - this should be impossible! Open an issue and provide the parameters used to call this function"
            return
        }
        New-EAMonitorDbContext -Force
        
    }
    end{
        if($DetectMonitorModules){
            $ModulesToImport = Get-Module -Name 'EAMonitor-*' -ListAvailable
            foreach($module in $ModulesToImport){
                Import-Module $Module.Name -Force
            }
        }
        Import-EAMonitorScriptBlock -Type ProcessTestData -Name 'Default' -ScriptBlock {
            Param([Pester.Test]$Result)
            return [PSCustomObject]@{
                Test = $result.ExpandedPath
                Passed = $result.Passed
            }
        }
        Import-EAMonitorScriptBlock -Type 'SendNotification' -Name 'Default' -ScriptBlock {
            Param([EAMonitor.Classes.EAMonitorResult[]]$FailedResultArray)
            $MonitorName = $FailedResultArray[0].Monitor.Name
            $monSettings = Get-EAMonitorSetting -MonitorName $MonitorName -AsHashtable

            if([string]::IsNullOrEmpty($monSettings.SendMailTo)){
                return
            }
            
            $header = '<style>h2{font-family:Arial,Helvetica,sans-serif;color:#009;font-size:16px}table{font-size:12px;border:0;font-family:Arial,Helvetica,sans-serif}td{padding:4px;margin:0;border:0}th{background:#395870;background:linear-gradient(#49708f,#293f50);color:#fff;font-size:11px;text-transform:uppercase;padding:10px 15px;vertical-align:middle}tbody tr:nth-child(even){background:#f0f0f2}</style>'
            $env = Get-EAMonitorEnvironment
            $EnvironmentString = ''
            if(-not [string]::IsNullOrEmpty($env)){
                $EnvironmentString = " for environment $($env)"
            }
            $ReportBody = "Please review the failed monitor $($MonitorName) $($EnvironmentString). The test failures are:`n"
            $CompiledResults = @{}
            foreach($result in $FailedResultArray){
                if($CompiledResults[$result.TestResult.Name]){
                    $CompiledResults[$result.TestResult.Name] += $result
                }
                else{
                    $CompiledResults[$result.TestResult.Name] = @($result)
                }
            }
            foreach($key in $CompiledResults.Keys){
                $compiledResultsArray = $CompiledResults[$key]
                $ReportBody += $compiledResultsArray.Data | ConvertTo-Html -Fragment -PreContent "<h2>Test: $($key)</h2>"
            }

            $Report = ConvertTo-HTML -Body $ReportBody -Title "EAMonitor - Failed Monitors" -Head $header
            
            $emailMessage = New-Object System.Net.Mail.MailMessage
            $emailMessage.From = $monSettings.SendMailFrom
            foreach($email in $monSettings.SendMailTo.Split(';').Split(',')){
                $emailMessage.To.Add($email)
            }
            $emailMessage.Subject = "EAMonitor - Failed Monitors" 
            $emailMessage.IsBodyHtml = $true
            $emailMessage.Body = "$Report"
            $SMTPClient = New-Object System.Net.Mail.SmtpClient( $monSettings.SendMailSmtp , $monSettings.SendMailSmtpPort)
            if($monSettings.SendMailEnableSSl){
                $SMTPClient.EnableSsl = $monSettings.SendMailEnableSSl
            }
            if($MonSettings.SendMailCredentials){
                $Secret = Get-Secret -Name $MonSettings.SendMailCredentials
                $SMTPClient.Credentials = New-Object System.Net.NetworkCredential( $Secret.GetNetworkCredential().UserName, $Secret.GetNetworkCredential().Password );
            }
            
            $SMTPClient.Send( $emailMessage )
        }
    }
}