lib/core/icingaagent/misc/Start-IcingaAgentDirectorWizard.psm1
|
function Start-IcingaAgentDirectorWizard() { param( [string]$DirectorUrl, [string]$SelfServiceAPIKey = $null, $OverrideDirectorVars = $null, [bool]$RunInstaller = $FALSE, [switch]$ForceTemplateKey = $FALSE ); [hashtable]$DirectorOverrideArgs = @{ } if ([string]::IsNullOrEmpty($DirectorUrl)) { $DirectorUrl = (Get-IcingaAgentInstallerAnswerInput -Prompt 'Please specify the Url pointing to your Icinga Director (Example: "https://example.com/icingaweb2/director")' -Default 'v').answer; } [bool]$HostKnown = $FALSE; [string]$TemplateKey = $SelfServiceAPIKey; if ($null -eq $OverrideDirectorVars -And $RunInstaller -eq $FALSE) { if ((Get-IcingaAgentInstallerAnswerInput -Prompt 'Do you want to manually override arguments provided by the Director API?' -Default 'n').result -eq 0) { $OverrideDirectorVars = $TRUE; } else { $OverrideDirectorVars = $FALSE; } } $LocalAPIKey = Get-IcingaPowerShellConfig -Path 'IcingaDirector.SelfService.ApiKey'; if ($ForceTemplateKey) { if ($SelfServiceAPIKey -eq $LocalAPIKey) { $ForceTemplateKey = $FALSE; } } if ($ForceTemplateKey -eq $FALSE) { if ([string]::IsNullOrEmpty($LocalAPIKey)) { $LegacyTokenPath = Join-Path -Path Get-IcingaAgentConfigDirectory -ChildPath 'icingadirector.token'; if (Test-Path $LegacyTokenPath) { $SelfServiceAPIKey = Get-Content -Path $LegacyTokenPath; Set-IcingaPowerShellConfig -Path 'IcingaDirector.SelfService.ApiKey' -Value $SelfServiceAPIKey; } else { $ForceTemplateKey = $TRUE; } } else { $SelfServiceAPIKey = $LocalAPIKey; } } if ([string]::IsNullOrEmpty($SelfServiceAPIKey)) { $SelfServiceAPIKey = (Get-IcingaAgentInstallerAnswerInput -Prompt 'Please enter your Self-Service API key' -Default 'v').answer; } else { if ($ForceTemplateKey -eq $FALSE) { $HostKnown = $TRUE; } } if ([string]::IsNullOrEmpty($LocalAPIKey) -eq $FALSE -And $LocalAPIKey -ne $TemplateKey -And $ForceTemplateKey -eq $FALSE) { try { $Arguments = Get-IcingaDirectorSelfServiceConfig -DirectorUrl $DirectorUrl -ApiKey $LocalAPIKey; } catch { Write-IcingaConsoleError 'Your local stored host key is no longer valid. Using provided template key'; return Start-IcingaAgentDirectorWizard ` -DirectorUrl $DirectorUrl ` -SelfServiceAPIKey $TemplateKey ` -OverrideDirectorVars $OverrideDirectorVars ` -ForceTemplateKey; } } else { try { $Arguments = Get-IcingaDirectorSelfServiceConfig -DirectorUrl $DirectorUrl -ApiKey $SelfServiceAPIKey; } catch { Write-IcingaConsoleError ([string]::Format('Failed to connect to your Icinga Director at "{0}". Please try again', $DirectorUrl)); return Start-IcingaAgentDirectorWizard ` -SelfServiceAPIKey ((Get-IcingaAgentInstallerAnswerInput -Prompt 'Please re-enter your SelfService API Key for the Host-Template in case the key is no longer assigned to your host' -Default 'v' -DefaultInput $SelfServiceAPIKey).answer) ` -OverrideDirectorVars $OverrideDirectorVars; } } $Arguments = Convert-IcingaDirectorSelfServiceArguments -JsonInput $Arguments; if ($OverrideDirectorVars -eq $TRUE -And -Not $RunInstaller) { $DirectorOverrideArgs = Start-IcingaDirectorAPIArgumentOverride -Arguments $Arguments; foreach ($entry in $DirectorOverrideArgs.Keys) { if ($Arguments.ContainsKey($entry)) { $Arguments[$entry] = $DirectorOverrideArgs[$entry]; } } } if ($HostKnown -eq $FALSE) { while ($TRUE) { [bool]$RegisterFailed = $FALSE; try { $SelfServiceAPIKey = Register-IcingaDirectorSelfServiceHost -DirectorUrl $DirectorUrl -ApiKey $SelfServiceAPIKey -Hostname (Get-IcingaHostname @Arguments) -Endpoint $Arguments.IcingaMaster; if ([string]::IsNullOrEmpty($SelfServiceAPIKey) -eq $FALSE) { break; } else { $RegisterFailed = $TRUE; } } catch { $RegisterFailed = $TRUE; } if ($RegisterFailed) { $SelfServiceAPIKey = (Get-IcingaAgentInstallerAnswerInput -Prompt ([string]::Format('Failed to register host within Icinga Director. Full error: "{0}". Please re-enter your SelfService API Key. If this prompt continues ensure you are using an Agent template or drop your host key at "Hosts -> {1} -> Agent"', $_.Exception.Message, (Get-IcingaHostname @Arguments))) -Default 'v' -DefaultInput $SelfServiceAPIKey).answer; } } # Host is already registered if ($null -eq $SelfServiceAPIKey) { Write-IcingaConsoleError 'The wizard is unable to complete as this host is already registered but the local API key is not stored within the config' return; } $Arguments = Get-IcingaDirectorSelfServiceConfig -DirectorUrl $DirectorUrl -ApiKey $SelfServiceAPIKey; $Arguments = Convert-IcingaDirectorSelfServiceArguments -JsonInput $Arguments; if ($OverrideDirectorVars -eq $TRUE -And -Not $RunInstaller) { $DirectorOverrideArgs = Start-IcingaDirectorAPIArgumentOverride -Arguments $Arguments; foreach ($entry in $DirectorOverrideArgs.Keys) { if ($Arguments.ContainsKey($entry)) { $Arguments[$entry] = $DirectorOverrideArgs[$entry]; } } } } $IcingaTicket = Get-IcingaDirectorSelfServiceTicket -DirectorUrl $DirectorUrl -ApiKey $SelfServiceAPIKey; $DirectorOverrideArgs.Add( 'DirectorUrl', $DirectorUrl ); $DirectorOverrideArgs.Add( 'Ticket', $IcingaTicket ); $DirectorOverrideArgs.Add( 'OverrideDirectorVars', 0 ); if ([string]::IsNullOrEmpty($TemplateKey) -eq $FALSE) { $DirectorOverrideArgs.Add( 'SelfServiceAPIKey', $TemplateKey ); } return @{ 'Arguments' = $Arguments; 'Overrides' = $DirectorOverrideArgs; }; } function Start-IcingaDirectorAPIArgumentOverride() { param( $Arguments ); $NewArguments = @{}; Write-IcingaConsoleNotice 'Please follow the wizard and manually override all entries you intend to'; Write-IcingaConsoleNotice '===='; foreach ($entry in $Arguments.Keys) { $value = (Get-IcingaAgentInstallerAnswerInput -Prompt ([string]::Format('Please enter the new value for the argument "{0}"', $entry)) -Default 'v' -DefaultInput $Arguments[$entry]).answer; if ($Arguments[$entry] -is [array] -Or ($value -is [string] -And $value.Contains(','))) { if ([string]::IsNullOrEmpty($value) -eq $FALSE) { while ($value.Contains(', ')) { $value = $value.Replace(', ', ','); } [array]$tmpArray = $value.Split(','); if ($null -ne (Compare-Object -ReferenceObject $Arguments[$entry] -DifferenceObject $tmpArray)) { $NewArguments.Add( $entry, $tmpArray ); } } continue; } elseif ($Arguments[$entry] -is [bool]) { if ($value -eq 'true' -or $value -eq 'y' -or $value -eq '1' -or $value -eq 'yes' -or $value -eq 1) { $value = 1; } else { $value = 0; } } if ($Arguments[$entry] -ne $value) { $NewArguments.Add($entry, $value); } } return $NewArguments; } Export-ModuleMember -Function @( 'Start-IcingaAgentDirectorWizard' ); |