Last active
October 24, 2022 16:40
-
-
Save laymanstake/00e2d1b38514b89c4496bee4d8590369 to your computer and use it in GitHub Desktop.
PowerShell script, which can download almost all the latest packages (except nssm, which is not github hosted) and then proceed to configure them on the same machine. Functions to download and create services are created in a way that select set of package can be downloaded or installed as putting everything on the same server may not be good idea.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #Requires -RunAsAdministrator | |
| #Requires -Version 3.0 | |
| #Requires -Modules BitsTransfer | |
| # Author : Nitish Kumar | |
| # Download and Install Prometheus, Grafana and Windows Exporter | |
| # version 1.0 | 19/10/2022 Initial version | |
| # version 1.1 | 21/10/2022 Cleaned up the script to supress error messages, added override switch | |
| # version 1.2 | 21/10/2022 Added a crude way to update Prometheus configuration for adding nodes at mass scale | |
| # version 1.3 | 22/10/2022 Added remove services option. Sorted nssm output encoding issue to find install path | |
| # version 1.4 | 23/10/2022 Ready for mass deployment, sample code parts added | |
| # USAGES: | |
| # For getting the installers need to add the below line | |
| # Get-Executables # By default it downloads all four packages | |
| # Get-Executables -packages Prometheus, Grafana # In case you want to download select packages | |
| # For services setup, need to add the lines like below | |
| # New-Services # By default, it would assume that all 4 packages need to be installed in c:\program files\<package_name> & executables are at c:\%username%\downloads\Executables | |
| # New-Services -Override # Adding Override switch would remove existing installations & install fresh except nssm | |
| # New-Services - packages Grafana, Windows_exporter # this would install only two of the packages given | |
| # For prometheus config, need to add the lines like below, same way can be extended to many nodes using csv file or table in a loop | |
| # Set-PrometheusConfig -jobName "Windows_exporter" -scrape_interval "5s" -targetAddress <IPAddress or FQDN> -targetPort <port> -prometheusInstallPath <Installation folder> # Except Scrape interview, rest all parameters are mandatory | |
| # REFER TO COMMENTED PARTS AT END OF SCRIPT FOR MASS DEPLOYMENT CASES | |
| Import-Module BitsTransfer | |
| # Function to download excutables for Prometheus, Grafana, Windows_exporter and nssm | |
| function Get-Executables { | |
| [CmdletBinding()] | |
| Param( | |
| [Parameter(ValueFromPipeline = $true,mandatory=$false)] | |
| [ValidateSet('Prometheus','Grafana','Windows_exporter','nssm')] | |
| [String[]]$packages = @('Prometheus','Grafana','Windows_exporter','nssm'), | |
| [Parameter(ValueFromPipeline = $true,mandatory=$false)]$executableDownloadPath = "$env:userprofile\downloads\" | |
| ) | |
| if (Test-Path -Path $executableDownloadPath"Executables") { | |
| #Clean up the old files in the path before downloading new ones | |
| $OldFiles = (get-Childitem $executableDownloadPath"Executables" ).FullName | Select-String -Pattern $packages | |
| $OldFiles | ForEach-Object {Remove-Item $_.Line} | |
| } else { | |
| New-Item -Path $executableDownloadPath -Name "Executables" -ItemType "directory" | Out-Null | |
| } | |
| if ($packages -match 'Prometheus'){ | |
| $url = Invoke-RestMethod -uri https://api.github.com/repos/prometheus/prometheus/releases/latest | Select-Object -ExpandProperty assets | Where-Object{$_.Name.EndsWith("windows-amd64.zip")} | select -expand browser_download_url | |
| $file = Split-Path $url -Leaf | |
| Write-Host "Downloading $file to $($executableDownloadPath)Executables..." -ForegroundColor GREEN | |
| Start-BitsTransfer -Source $url -Destination $executableDownloadPath"Executables\"$file | |
| } | |
| if ($packages -match 'Grafana'){ | |
| $grafanaVersion = (Split-Path (Invoke-RestMethod -uri https://api.github.com/repos/grafana/grafana/releases/latest).html_url -Leaf ).Substring(1) | |
| $file = "grafana-" + $grafanaVersion + ".windows-amd64.zip" | |
| $url = "https://dl.grafana.com/oss/release/" + $file | |
| Write-Host "Downloading $file to $($executableDownloadPath)Executables..." -ForegroundColor GREEN | |
| Start-BitsTransfer -Source $url -Destination $executableDownloadPath"Executables\"$file | |
| } | |
| if ($packages -match 'Windows_exporter'){ | |
| $url = Invoke-RestMethod -uri https://api.github.com/repos/prometheus-community/windows_exporter/releases/latest | Select-Object -ExpandProperty assets | Where-Object{$_.Name.EndsWith("amd64.exe")} | select -expand browser_download_url | |
| $file = Split-Path $url -Leaf | |
| Write-Host "Downloading $file to $($executableDownloadPath)Executables..." -ForegroundColor GREEN | |
| Start-BitsTransfer -Source $url -Destination $executableDownloadPath"Executables\"$file | |
| } | |
| if ($packages -match 'nssm'){ | |
| $file = Split-Path("https://nssm.cc/release/nssm-2.24.zip") -Leaf | |
| Write-Host "Downloading $file to $($executableDownloadPath)Executables..." -ForegroundColor GREEN | |
| Start-BitsTransfer -Source https://nssm.cc/release/nssm-2.24.zip -Destination $executableDownloadPath"Executables\"$file | |
| } | |
| } | |
| # Function to install services for Prometheus, Grafana, Windows_exporter and nssm | |
| function New-Services { | |
| [CmdletBinding()] | |
| Param( | |
| [Parameter(ValueFromPipeline = $true,mandatory=$false)] | |
| [ValidateSet('Prometheus','Grafana','Windows_exporter','nssm')] | |
| [String[]]$packages = @('nssm','Prometheus','Grafana','Windows_exporter'), | |
| [Parameter(ValueFromPipeline = $true,mandatory=$false)]$servicePath = "$env:programfiles\", | |
| [Parameter(ValueFromPipeline = $true,mandatory=$false)]$Override, | |
| [Parameter(ValueFromPipeline = $true,mandatory=$false)]$executableDownloadPath = "$env:userprofile\downloads\Executables" | |
| ) | |
| $packages | |
| $servicePath | |
| $Override | |
| $executableDownloadPath | |
| if($Override.IsPresent){ | |
| Write-Host "Proceeding to wipe out earlier installations of $($packages -join ","). Close the script if you do not wish so" -ForegroundColor RED | |
| Pause | |
| } | |
| Write-Host "Checking if nssm is setup ..." -ForegroundColor Yellow | |
| $nssmExists = (where.exe nssm) | |
| if ($nssmExists) {Write-Host "nssm found to be installed at $nssmExists ..."} else {Write-Host "nssm not installed yet ..."} | |
| ForEach($package in $packages){ | |
| # Install nssm which would help creating services for Prometheus and Grafana | |
| if ($package -match 'nssm' -AND !($nssmExists)){ | |
| # Create folder for nssm in Installation path | |
| New-Item -Path $servicePath -Name $package -ItemType "directory" | Out-Null | |
| Write-Host "Setting up $package ..." -ForegroundColor GREEN | |
| # Extract nssm zip to Install path | |
| $filePath = (get-childitem $executableDownloadPath | Where-Object {$_.Name.StartsWith($package.ToLower())}).versioninfo.filename | |
| $shell = New-Object -ComObject Shell.Application | |
| $zipFile = $shell.NameSpace($filePath + "\" + ((get-childitem $executableDownloadPath | Where-Object {$_.Name.StartsWith($package)}).Name -replace ".zip")) | |
| $destinationFolder = $shell.NameSpace($servicePath + $package) | |
| $copyFlags = 0x00 | |
| $copyFlags += 0x04 # Hide progress dialogs | |
| $copyFlags += 0x10 # Overwrite existing files | |
| Write-Host "Copying files for $package ..." -ForegroundColor GREEN | |
| $destinationFolder.CopyHere($zipFile.Items(), $copyFlags) | |
| # Add nssm to system path permanently | |
| $oldpath = (Get-ItemProperty -Path 'Registry::HKEY_LOCAL_MACHINE\System\CurrentControlSet\Control\Session Manager\Environment' -Name PATH).path | |
| $newpath = $oldpath + ";" + $servicePath + $package + "\win64" | |
| Set-ItemProperty -Path 'Registry::HKEY_LOCAL_MACHINE\System\CurrentControlSet\Control\Session Manager\Environment' -Name PATH -Value $newPath | |
| # Refresh the path variable for current session | |
| $env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path","User") | |
| $nssmExists = (where.exe nssm) | |
| if ($nssmExists) {Write-Host "nssm is now installed at $nssmExists ..."} else {Write-Host "nssm not installed yet ..."} | |
| } | |
| # Install Prometheus as Service | |
| if ($package -match 'prometheus' -AND ($nssmExists)){ | |
| if($Override.IsPresent -AND (Get-Service prometheus-service)){ | |
| Write-Host "Proceeding to wipe out $package. Close the script if you do not wish so" -ForegroundColor RED | |
| Pause | |
| Remove-Services -packages $package | |
| } | |
| New-Item -Path $servicePath -Name $package -ItemType "directory" | Out-Null | |
| # Extract Prometheus installer zip to Install path | |
| $filePath = (get-childitem $executableDownloadPath | Where-Object {$_.Name.StartsWith($package.ToLower())}).versioninfo.filename | |
| $shell = New-Object -ComObject Shell.Application | |
| $zipFile = $shell.NameSpace($filePath + "\" + ((get-childitem $executableDownloadPath | Where-Object {$_.Name.StartsWith($package.ToLower())}).Name -replace ".zip")) | |
| $destinationFolder = $shell.NameSpace($servicePath + $package) | |
| $copyFlags = 0x00 | |
| $copyFlags += 0x04 # Hide progress dialogs | |
| $copyFlags += 0x10 # Overwrite existing files | |
| Write-Host "Copying files for $package Service from $filePath ..." -ForegroundColor GREEN | |
| $destinationFolder.CopyHere($zipFile.Items(), $copyFlags) | |
| Write-Host "Setting up $package ..." -ForegroundColor GREEN | |
| $prometheusServiceName = 'prometheus-service' | |
| $prometheusInstallPath = $servicePath + $package | |
| $prometheusServiceUsername = "NT SERVICE\$prometheusServiceName" | |
| Write-Host "Configuring the $prometheusServiceName service..." -ForegroundColor GREEN | |
| nssm install $prometheusServiceName "`"$prometheusInstallPath\prometheus.exe`"" | |
| nssm set $prometheusServiceName Start SERVICE_AUTO_START | |
| nssm set $prometheusServiceName AppRotateFiles 1 | |
| nssm set $prometheusServiceName AppRotateOnline 1 | |
| nssm set $prometheusServiceName AppRotateSeconds 86400 | |
| nssm set $prometheusServiceName AppRotateBytes 1048576 | |
| nssm set $prometheusServiceName AppStdout "`"$prometheusInstallPath\logs\service-stdout.log`"" | |
| nssm set $prometheusServiceName AppStderr "`"$prometheusInstallPath\logs\service-stderr.log`"" | |
| nssm set $prometheusServiceName AppDirectory "`"$prometheusInstallPath`"" | |
| # Parameters are passed via a Here-String, so do not touch formatting if to use a path with space | |
| nssm set $prometheusServiceName AppParameters ` | |
| @" | |
| --config.file=""$prometheusInstallPath\prometheus.yml"" --storage.tsdb.path=""$prometheusInstallPath\data"" --storage.tsdb.retention.time=168h --web.console.libraries=""$prometheusInstallPath\console_libraries"" --web.console.templates=""$prometheusInstallPath\consoles"" --web.listen-address=:9090 | |
| "@ | |
| $result = sc.exe sidtype $prometheusServiceName unrestricted | |
| if ($result -ne '[SC] ChangeServiceConfig2 SUCCESS') { | |
| throw "sc.exe sidtype failed with $result" | |
| } | |
| $result = sc.exe config $prometheusServiceName obj= $prometheusServiceUsername | |
| if ($result -ne '[SC] ChangeServiceConfig SUCCESS') { | |
| throw "sc.exe config failed with $result" | |
| } | |
| $result = sc.exe failure $prometheusServiceName reset= 0 actions= restart/1000 | |
| if ($result -ne '[SC] ChangeServiceConfig2 SUCCESS') { | |
| throw "sc.exe failure failed with $result" | |
| } | |
| $acl = Get-Acl $prometheusInstallPath | |
| $user = New-Object System.Security.Principal.NTAccount($prometheusServiceUsername) | |
| $acl.SetOwner([System.Security.Principal.NTAccount] $user) | |
| $acl.SetAccessRuleProtection($True, $False) | |
| $ruleSystem = New-Object System.Security.AccessControl.FileSystemAccessRule('SYSTEM','FullControl','ObjectInherit, ContainerInherit','None','Allow') | |
| $ruleAdmins = New-Object System.Security.AccessControl.FileSystemAccessRule('Administrators','FullControl','ObjectInherit, ContainerInherit','None','Allow') | |
| $ruleService = New-Object System.Security.AccessControl.FileSystemAccessRule($user,'ReadAndExecute','ObjectInherit, ContainerInherit','None','Allow') | |
| $acl.AddAccessRule($ruleSystem) | |
| $acl.AddAccessRule($ruleAdmins) | |
| $acl.AddAccessRule($ruleService) | |
| Set-Acl $prometheusInstallPath $acl | Out-Null | |
| 'data','logs' | ForEach-Object { | |
| mkdir $prometheusInstallPath\$_ | Out-Null | |
| $acl = Get-Acl $prometheusInstallPath\$_ | |
| $acl.SetOwner([System.Security.Principal.NTAccount] $prometheusServiceUsername) | |
| $acl.SetAccessRuleProtection($True, $False) | |
| $acl.AddAccessRule($ruleSystem) | |
| $acl.AddAccessRule($ruleAdmins) | |
| $ruleService = New-Object System.Security.AccessControl.FileSystemAccessRule($user,'FullControl','ObjectInherit, ContainerInherit','None','Allow') | |
| $acl.AddAccessRule($ruleService) | |
| Set-Acl $prometheusInstallPath\$_ $acl | Out-Null | |
| Write-Host "Checking the prometheus configuration..." -ForegroundColor GREEN | |
| &$prometheusInstallPath\promtool.exe check config $prometheusInstallPath\prometheus.yml | |
| Write-Host "Starting the $prometheusServiceName service..." -ForegroundColor GREEN | |
| Start-Service $prometheusServiceName | |
| # Create Firewall rule for Prometheus Server | |
| $suppressMsg = New-NetFirewallRule -DisplayName 'Prometheus Server' -Profile @('Public','Private','Domain') -Direction Inbound -Action Allow -Protocol TCP -LocalPort 9090 -ErrorAction SilentlyContinue -ErrorVariable MyErrorVar | |
| } | |
| } | |
| # Install Grafana as Service | |
| if ($package -match 'Grafana' -AND ($nssmExists)){ | |
| if($Override.IsPresent){ | |
| Write-Host "Proceeding to wipe out $package. Close the script if you do not wish so" -ForegroundColor RED | |
| Pause | |
| Remove-Services -packages $package | |
| New-Item -Path $servicePath -Name $package -ItemType "directory" | Out-Null | |
| } | |
| Write-Host "Setting up $package ..." -ForegroundColor GREEN | |
| $grafanaServiceName = 'grafana-server' | |
| $grafanaInstallPath = $servicePath + "Grafana" | |
| $grafanaServiceUsername = "NT SERVICE\$grafanaServiceName" | |
| # Extract Grafana installer zip to Install path | |
| $filePath = (get-childitem $executableDownloadPath | Where-Object {$_.Name.StartsWith($package.ToLower())}).versioninfo.filename | |
| $shell = New-Object -ComObject Shell.Application | |
| $rawPath = $filePath + "\" + ((get-childitem $executableDownloadPath | Where-Object {$_.Name.StartsWith($package.ToLower())}).Name -replace ".zip") | |
| $actualPath = $rawPath.Substring(0,$rawPath.Length-14) | |
| $zipFile = $shell.NameSpace($actualPath) | |
| $destinationFolder = $shell.NameSpace($servicePath + $package) | |
| $copyFlags = 0x00 | |
| $copyFlags += 0x04 # Hide progress dialogs | |
| $copyFlags += 0x10 # Overwrite existing files | |
| Write-Host "Copying data for the $($grafanaServiceName) service from $filePath, may take some time..." -ForegroundColor YELLOW | |
| $destinationFolder.CopyHere($zipFile.Items(), $copyFlags) | |
| $confPath = $servicePath + $package | |
| Copy-Item $confPath\Conf\defaults.ini $confPath\Conf\custom.ini | |
| Write-Host "Configuring the $($grafanaServiceName) service..." -ForegroundColor GREEN | |
| nssm install $grafanaServiceName "`"$grafanaInstallPath\bin\grafana-server.exe`"" | |
| nssm set $grafanaServiceName Start SERVICE_AUTO_START | |
| nssm set $grafanaServiceName AppRotateFiles 1 | |
| nssm set $grafanaServiceName AppRotateOnline 1 | |
| nssm set $grafanaServiceName AppRotateSeconds 86400 | |
| nssm set $grafanaServiceName AppRotateBytes 1048576 | |
| nssm set $grafanaServiceName AppStdout "`"$grafanaInstallPath\logs\service-stdout.log`"" | |
| nssm set $grafanaServiceName AppStderr "`"$grafanaInstallPath\logs\service-stderr.log`"" | |
| nssm set $grafanaServiceName AppDirectory "`"$grafanaInstallPath\bin`"" | |
| $result = sc.exe sidtype $grafanaServiceName unrestricted | |
| if ($result -ne '[SC] ChangeServiceConfig2 SUCCESS') { | |
| throw "sc.exe sidtype failed with $result" | |
| } | |
| $result = sc.exe config $grafanaServiceName obj= $grafanaServiceUsername | |
| if ($result -ne '[SC] ChangeServiceConfig SUCCESS') { | |
| throw "sc.exe config failed with $result" | |
| } | |
| $result = sc.exe failure $grafanaServiceName reset= 0 actions= restart/1000 | |
| if ($result -ne '[SC] ChangeServiceConfig2 SUCCESS') { | |
| throw "sc.exe failure failed with $result" | |
| } | |
| $acl = Get-Acl $grafanaInstallPath | |
| $user = New-Object System.Security.Principal.NTAccount($grafanaServiceUsername) | |
| $acl.SetOwner([System.Security.Principal.NTAccount] $user) | |
| $acl.SetAccessRuleProtection($True, $False) | |
| $ruleSystem = New-Object System.Security.AccessControl.FileSystemAccessRule('SYSTEM','FullControl','ObjectInherit, ContainerInherit','None','Allow') | |
| $ruleAdmins = New-Object System.Security.AccessControl.FileSystemAccessRule('Administrators','FullControl','ObjectInherit, ContainerInherit','None','Allow') | |
| $ruleService = New-Object System.Security.AccessControl.FileSystemAccessRule($user,'ReadAndExecute','ObjectInherit, ContainerInherit','None','Allow') | |
| $acl.AddAccessRule($ruleSystem) | |
| $acl.AddAccessRule($ruleAdmins) | |
| $acl.AddAccessRule($ruleService) | |
| Set-Acl $grafanaInstallPath $acl | Out-Null | |
| 'data','logs' | ForEach-Object { | |
| mkdir $grafanaInstallPath\$_ | Out-Null | |
| $acl = Get-Acl $grafanaInstallPath\$_ | |
| $acl.SetOwner([System.Security.Principal.NTAccount] $grafanaServiceUsername) | |
| $acl.SetAccessRuleProtection($True, $False) | |
| $acl.AddAccessRule($ruleSystem) | |
| $acl.AddAccessRule($ruleAdmins) | |
| $ruleService = New-Object System.Security.AccessControl.FileSystemAccessRule($user,'FullControl','ObjectInherit, ContainerInherit','None','Allow') | |
| $acl.AddAccessRule($ruleService) | |
| Set-Acl $grafanaInstallPath\$_ $acl | Out-Null | |
| Write-Host "Starting the $grafanaServiceName service..." -ForegroundColor GREEN | |
| Start-Service $grafanaServiceName | |
| # Create Firewall rule for Grafana Server | |
| $suppressMsg = New-NetFirewallRule -DisplayName 'Grafana Server' -Profile @('Public','Private','Domain') -Direction Inbound -Action Allow -Protocol TCP -LocalPort 3000 -ErrorAction SilentlyContinue -ErrorVariable MyErrorVar | |
| } | |
| } | |
| # Install Windows Exporter as Service | |
| if ($package -match 'windows_exporter'){ | |
| if($Override.IsPresent){ | |
| Write-Host "Proceeding to wipe out $package. Close the script if you do not wish so" -ForegroundColor RED | |
| Pause | |
| Remove-Services -packages $package | |
| New-Item -Path $servicePath -Name $package -ItemType "directory" | Out-Null | |
| } | |
| Write-Host "Setting up $package ..." -ForegroundColor GREEN | |
| # Extract Windows Exporter installer zip to Install path | |
| $filePath = (get-childitem $executableDownloadPath | Where-Object {$_.Name.StartsWith($package.ToLower())}).versioninfo.filename | |
| $windowsExporterInstallPath = $servicePath + $package | |
| Write-Host "Copying files for $package Service from $filePath ..." -ForegroundColor GREEN | |
| Copy-Item $filePath $windowsExporterInstallPath | |
| $exeName = split-path $filePath -leaf | |
| $ServicePath = """$windowsExporterInstallPath\$exeName"" --log.format logger:eventlog?name=windows_exporter --collectors.enabled ""cpu,cs,net,service,process,tcp,logical_disk,os,system,textfile,thermalzone"" --collector.textfile.directory C:\custom_metrics\" | |
| $params = @{ | |
| Name = "Windows_exporter" | |
| BinaryPathName = $ServicePath | |
| DisplayName = "Windows Exporter" | |
| StartupType = "Automatic" | |
| Description = "To run windows exporter service" | |
| } | |
| New-Service @params | |
| Write-Host "Starting the Windows Exporter service..." -ForegroundColor GREEN | |
| Start-Service Windows_exporter | |
| # Create Firewall rule for Windows Exporter | |
| $suppressMsg = New-NetFirewallRule -DisplayName 'Windows Exporter' -Profile @('Public','Private','Domain') -Direction Inbound -Action Allow -Protocol TCP -LocalPort 9182 -ErrorAction SilentlyContinue -ErrorVariable MyErrorVar | |
| } | |
| } | |
| } | |
| # Function to configure Prometheus | |
| function Set-PrometheusConfig{ | |
| [CmdletBinding()] | |
| Param( | |
| [Parameter(ValueFromPipeline = $true,mandatory=$true)]$jobName, | |
| [Parameter(ValueFromPipeline = $true,mandatory=$true)]$targetAddress, | |
| [Parameter(ValueFromPipeline = $true,mandatory=$true)]$targetPort, | |
| [Parameter(ValueFromPipeline = $true,mandatory=$false)]$scrape_interval="10s" | |
| ) | |
| # Encoding of nssm output causes issue, so forcing to unicode temporarily | |
| $originalEncoding = [Console]::OutputEncoding | |
| [Console]::OutputEncoding = [System.Text.Encoding]::Unicode | |
| $prometheusInstallPath =(nssm get prometheus-service AppDirectory) # This gives error msg due to NT Service in config, can be ignored | |
| [Console]::OutputEncoding = $originalEncoding | |
| if((Test-Path $prometheusInstallPath\prometheus.yml) -AND (Test-Path $prometheusInstallPath\promtool.exe)){ | |
| $backupYaml = Get-Content $prometheusInstallPath\prometheus.yml | |
| $backupYamlUpdated = $backupYaml | |
| $foundAt = ($backupYaml | Select-String "Job_name: ""$jobName""" ).LineNumber | |
| $i = $foundAt | |
| if($foundAt){ | |
| do { | |
| if(($backupYaml[$i] | Select-String "targets:" ).LineNumber){ | |
| $backupYamlUpdated[$i] = $backupYamlUpdated[$i] -replace ']',",`"$targetAddress`:$targetPort`"]" | |
| Write-Host "Targets changed to " $backupYaml[$i] | |
| } | |
| $i++ | |
| } while ( $null -eq ($backupYaml[$i-1] | Select-String "targets:" ).LineNumber ) | |
| $backupYamlUpdated | Set-Content $prometheusInstallPath\prometheus.yml | |
| $backupYaml | Set-Content $prometheusInstallPath\prometheus-copy.yml | |
| }else { | |
| add-content $prometheusInstallPath\prometheus.yml @" | |
| - job_name: "$jobName" | |
| scrape_interval: $scrape_interval | |
| static_configs: | |
| - targets: ["$targetAddress`:$targetPort"] | |
| "@ | |
| } | |
| if((& $prometheusInstallPath\promtool.exe check config $prometheusInstallPath\prometheus.yml) -match "SUCCESS"){ | |
| Write-Host "Configuration updated successfully." -ForegroundColor Green | |
| Write-Host "Restarting the prometheus-service service..." -ForegroundColor GREEN | |
| Get-Service prometheus-service | Restart-Service | |
| } else { | |
| $backupYaml | Set-Content $prometheusInstallPath\prometheus.yml | |
| Write-Host "Incorrect config value: Changes rolled back." -ForegroundColor Red | |
| } | |
| } else { | |
| Write-Host "The given path $prometheusInstallPath doens't include either yml or promtool so we can not continue." | |
| } | |
| } | |
| function Remove-Services { | |
| [CmdletBinding()] | |
| Param( | |
| [Parameter(ValueFromPipeline = $true,mandatory=$true)] | |
| [ValidateSet('Prometheus','Grafana','Windows_exporter','nssm')] | |
| [String[]]$packages | |
| ) | |
| ForEach($package in $packages){ | |
| # Remove nssm from system | |
| if($package -match "nssm"){ | |
| $nssmExists = (where.exe nssm) | |
| if ($nssmExists) {Write-Host "nssm found to be installed at $nssmExists ..."} else {Write-Host "nssm not installed yet ..."} | |
| $nssmDependents = (Get-CimInstance Win32_Service | Where-Object{$_.PathName -like "*nssm.exe*"} | Select-Object Name, DisplayName, StartMode, State) | |
| if($nssmDependents.count -ne 0){ | |
| Write-Host "There are $($nssmDependents.Count) depedent service(s): $($nssmDependents.name -join ","). Remove them first..." -ForegroundColor Red | |
| } else { | |
| if($nssmExists){ | |
| $servicePath = Split-Path $nssmExists | |
| try{ | |
| Get-Process nssm -ErrorAction SilentlyContinue | Stop-Process -Force -ErrorAction SilentlyContinue | |
| Remove-Item $servicePath -Recurse -Force | |
| if((Split-Path ($servicePath) -Parent).EndsWith("nssm")){ | |
| Remove-Item (Split-Path ($servicePath) -Parent) -Recurse -Force | |
| } | |
| If ((Read-Host 'Do you want to update system path as well (y/n) ') -eq 'Y') { | |
| $OldPath = (Get-ItemProperty -Path 'Registry::HKEY_LOCAL_MACHINE\System\CurrentControlSet\Control\Session Manager\Environment' -Name PATH).path | |
| Write-Host "Old Path is: `n$OldPath" | |
| $NewPath = ($OldPath -split ";" | where-object{$_ -ne "$servicePath" -AND $_ -ne ""}) -join ";" | |
| Set-ItemProperty -Path 'Registry::HKEY_LOCAL_MACHINE\System\CurrentControlSet\Control\Session Manager\Environment' -Name PATH -Value "$NewPath" | |
| Write-Host "Updated new Path is: `n$NewPath" | |
| } | |
| Write-host "$package has been removed from the system ..." -ForegroundColor GREEN | |
| } catch { | |
| Write-Host "Unable to stop nssm process.. can't remove nssm." | |
| } | |
| } | |
| } | |
| } | |
| # Remove Grafana from system | |
| if($package -match "Grafana"){ | |
| if(Get-Service grafana-server -ErrorAction SilentlyContinue){ | |
| # Encoding of nssm output causes issue, so forcing to unicode temporarily | |
| $originalEncoding = [Console]::OutputEncoding | |
| [Console]::OutputEncoding = [System.Text.Encoding]::Unicode | |
| $GrafanaInstallPath =(nssm get grafana-server AppDirectory) # This gives error msg due to NT Service in config, can be ignored | |
| [Console]::OutputEncoding = $originalEncoding | |
| Stop-Service grafana-server -ErrorAction SilentlyContinue | |
| Get-Process grafana-server -ErrorAction SilentlyContinue | Stop-Process -Force -ErrorAction SilentlyContinue | |
| Wait-Process -Name grafana-server -ErrorAction SilentlyContinue | |
| nssm remove grafana-server confirm | |
| $GrafanaInstallPath = Split-Path ($GrafanaInstallPath) -Parent | |
| If ((Read-Host "Do you want to delete $GrafanaInstallPath as well (y/n) ") -ieq 'Y') { | |
| Remove-Item $GrafanaInstallPath -Recurse -Force | |
| Write-Host "$GrafanaInstallPath deleted." -ForegroundColor Yellow | |
| } | |
| } else { | |
| Write-Host "Count not find $package service in system ... " -ForegroundColor Green | |
| } | |
| } | |
| # Remove Prometheus from the system | |
| if($package -match "Prometheus"){ | |
| if(Get-Service prometheus-service -ErrorAction SilentlyContinue){ | |
| # Encoding of nssm output causes issue, so forcing to unicode temporarily | |
| $originalEncoding = [Console]::OutputEncoding | |
| [Console]::OutputEncoding = [System.Text.Encoding]::Unicode | |
| $prometheusInstallPath =(nssm get prometheus-service AppDirectory) # This gives error msg due to NT Service in config, can be ignored | |
| [Console]::OutputEncoding = $originalEncoding | |
| Stop-Service prometheus-service -ErrorAction SilentlyContinue | |
| Get-Process prometheus-service -ErrorAction SilentlyContinue | Stop-Process -Force -ErrorAction SilentlyContinue | |
| Wait-Process -Name prometheus-service -ErrorAction SilentlyContinue | |
| nssm remove prometheus-service confirm | |
| If ((Read-Host "Do you want to delete $prometheusInstallPath as well (y/n) ") -ieq 'Y') { | |
| Remove-Item $prometheusInstallPath -Recurse -Force | |
| Write-Host "$prometheusInstallPath deleted." -ForegroundColor Yellow | |
| } | |
| } else { | |
| Write-Host "Count not find $package service in system ... " -ForegroundColor Green | |
| } | |
| } | |
| # Remove Windows Exporter from the system | |
| if($package -match "Windows_Exporter"){ | |
| if(Get-Service Windows_exporter -ErrorAction SilentlyContinue){ | |
| $WinExporterInstallPath = split-path ((Get-CimInstance Win32_Service | Where-Object {$_.Name -eq "Windows_exporter"}).PathName -split "--")[0] | |
| Stop-Service Windows_exporter -ErrorAction SilentlyContinue | |
| Get-Process windows_exporter* -ErrorAction SilentlyContinue | Stop-Process -Force -ErrorAction SilentlyContinue | |
| Wait-Process windows_exporter* -ErrorAction SilentlyContinue | |
| Remove-Service Windows_exporter -ErrorAction SilentlyContinue | |
| If ((Read-Host "Do you want to delete $WinExporterInstallPath as well (y/n) ") -ieq 'Y') { | |
| Remove-Item $WinExporterInstallPath -Recurse -Force | |
| Write-Host "$WinExporterInstallPath deleted." -ForegroundColor Yellow | |
| } | |
| Write-host "$package has been removed from the system ..." -ForegroundColor GREEN | |
| } else { | |
| Write-Host "Count not find $package service in system ... " -ForegroundColor Green | |
| } | |
| } | |
| } | |
| } | |
| #Write-Host "Download all the latest executables ...." -ForegroundColor GREEN | |
| #Get-Executables nssm | |
| # Sample code if you want to download the packages on remote machine | |
| #$credential = Get-Credential | |
| #Invoke-Command -ComputerName $env:COMPUTERNAME -ScriptBlock ${function:get-executables} -Credential $credential -ArgumentList @("nssm","prometheus"), "$env:userprofile\" | |
| # Sample code if you want to install the packages on remote machine | |
| #$param = @{ | |
| # packages = @("nssm", "prometheus") | |
| # servicePath = "c:\down\" | |
| # Override = $true | |
| # executableDownloadPath = "$env:userprofile\Executables" | |
| #} | |
| #$ScriptBlock = [scriptblock]::Create("${function:New-Services}") | |
| #Invoke-Command -ComputerName $env:COMPUTERNAME -ScriptBlock $ScriptBlock -Credential $credential -ArgumentList ($param.packages, $param.servicePath, $param.Override,$param.executableDownloadPath) | |
| # Sample code if you want to install the packages on local machine | |
| #Write-Host "Setting up services ...." -ForegroundColor GREEN | |
| #New-Services | |
| # Sample code if you want to download the packages on remote machine | |
| #$credential = Get-Credential | |
| #Invoke-Command -ComputerName $env:COMPUTERNAME -ScriptBlock ${function:get-executables} -Credential $credential -ArgumentList @("nssm","prometheus"), "$env:userprofile\" | |
| # Sample code if you want to remove the packages from remote machine | |
| #$param = @{ | |
| # packages = @("nssm", "prometheus") | |
| #} | |
| #$ScriptBlock = [scriptblock]::Create("${function:Remove-Services}") | |
| #Invoke-Command -ComputerName $env:COMPUTERNAME -ScriptBlock $ScriptBlock -Credential $credential -ArgumentList ($param.packages) | |
| # Sample code if you want to remove the service(s) from local machine | |
| #Remove-Services -packages nssm, Prometheus | |
| # Update Prometheus config to scrape current host metrics | Can be done for other hosts using similar syntax | |
| #$HostAddress = (Test-Connection -ComputerName $env:ComputerName -IPv4 -Count 1).Address.IPAddressToString | |
| #Set-PrometheusConfig -jobName "Windows_exporter" -scrape_interval "5s" -targetAddress $HostAddress -targetPort 9182 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment