src/Untitled-1.ps1
|
class Computer { [string] $Namespace [string] $ClusterId [string] $DatacenterId [string] $Id Computer([string]$Id) { $this.Id = $Id if ($Id -notmatch "(\w+)(\d+)([\w\d]+)") { throw "Invalid Computer ID '$Id'" } $_, $this.Namespace, $this.ClusterId, $this.DatacenterId = $Matches } } class AuditResult { [boolean] $Healthy [Computer] $Computer } function testNode($node) { } function Test-Node { [CmdletBinding()] Param( [Parameter(Mandatory, ParameterSetName = "Nodes", ValueFromPipeline)] [ValidateNotNullOrEmpty()] [string[]] $Node, [Parameter(Mandatory, ParameterSetName = "File")] [ValidateSet( { Test-Path $_ })] [ValidateNotNullOrEmpty()] [string] $Path ) begin { if ($PSCmdlet.ParameterSetName -eq "File") { $Node = Get-Content $Path } } process { testNode $_ } } function readPromptOption([string] $message, [string[]] $options) { $optionsPrompt = "( $($options -join " / ") )" do { $value = Read-Prompt "$message $optionsPrompt" if ($value -notin $options) { Write-Error "Please enter $optionsPrompt" } else { return $value } } while ($true) } function Start-Troubleshooter { [Diagnostics.CodeAnalysis.SuppressMessageAttribute("PSAvoidUsingWriteHost", "")] [CmdletBinding(SupportsShouldProcess)] Param( [Parameter(Mandatory)] [ValidateNotNullOrEmpty()] [string]$Service ) Write-Host "Welcome! Let's start troubleshooting issues with '$Service'" $comfortable = readPromptOption "Are you comfortable troubleshooting this issue?" @("y", "n") switch ($comfortable) { "y" { Write-Host "Good. Let's start finding bad nodes..." $nodes = Get-Content $env:NodeListPath | ? { $_ -like "*$Service*" } $auditResults = $nodes | % { testNode $_ } Write-Host "Audit complete.`nResults:" $auditResults ` | Sort-Object "Id" ` | % { Write-Host $_.Id -ForegroundColor (if ($_.Healthy) { "Green" } else { "Red" }) } $badNodes = $auditResults | ? { -not $_.Health } if ($badNodes) { Write-Host "Found some bad nodes!" -ForegroundColor Red $proceed = readPromptOption "Fix all?" @("y", "n") if ($proceed) { $badNodes | % { Invoke-RestMethod -Method Get -Uri "$env:AutohealWebhookUri&computer=$($_.Id)" } Write-Host "Successfully completed" -ForegroundColor Green } else { Write-Host "Ending without taking action." -ForegroundColor Yellow } } else { Write-Host "All clear! This is probably an application layer issue" -ForegroundColor Green } } "n" { if ($PSCmdlet.ShouldProcess("ShouldProcess?")) { Invoke-RestMethod -Method Get -Uri $env:EscalationWebHookUri } Write-Host "Don't worry. We've escalated the issue to avoid preventable accidents." } } } |