mirror of
https://github.com/fankes/komari-agent.git
synced 2025-10-18 02:29:22 +08:00
262 lines
11 KiB
PowerShell
262 lines
11 KiB
PowerShell
# Windows PowerShell installation script for Komari Agent
|
|
|
|
# Logging functions with colors
|
|
function Log-Info { param([string]$Message) Write-Host "$Message" -ForegroundColor Cyan }
|
|
function Log-Success { param([string]$Message) Write-Host "$Message" -ForegroundColor Green }
|
|
function Log-Warning { param([string]$Message) Write-Host "[WARNING] $Message" -ForegroundColor Yellow }
|
|
function Log-Error { param([string]$Message) Write-Host "[ERROR] $Message" -ForegroundColor Red }
|
|
function Log-Step { param([string]$Message) Write-Host "$Message" -ForegroundColor Magenta }
|
|
function Log-Config { param([string]$Message) Write-Host "- $Message" -ForegroundColor White }
|
|
|
|
# Default parameters
|
|
$InstallDir = Join-Path $Env:ProgramFiles "Komari"
|
|
$ServiceName = "komari-agent"
|
|
$GitHubProxy = ""
|
|
$KomariArgs = @()
|
|
$InstallVersion = ""
|
|
|
|
# Parse script arguments
|
|
for ($i = 0; $i -lt $args.Count; $i++) {
|
|
switch ($args[$i]) {
|
|
"--install-dir" { $InstallDir = $args[$i + 1]; $i++; continue }
|
|
"--install-service-name" { $ServiceName = $args[$i + 1]; $i++; continue }
|
|
"--install-ghproxy" { $GitHubProxy = $args[$i + 1]; $i++; continue }
|
|
"--install-version" { $InstallVersion = $args[$i + 1]; $i++; continue }
|
|
Default { $KomariArgs += $args[$i] }
|
|
}
|
|
}
|
|
|
|
# Ensure running as Administrator
|
|
if (-not ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()
|
|
).IsInRole([Security.Principal.WindowsBuiltinRole]::Administrator)) {
|
|
Log-Error "Please run this script as Administrator."
|
|
exit 1
|
|
}
|
|
|
|
# Prepare GitHub proxy display
|
|
if ($GitHubProxy -ne '') { $ProxyDisplay = $GitHubProxy } else { $ProxyDisplay = '(direct)' }
|
|
|
|
# Detect architecture early for constructing binary name
|
|
switch ($env:PROCESSOR_ARCHITECTURE) {
|
|
'AMD64' { $arch = 'amd64' }
|
|
'ARM64' { $arch = 'arm64' }
|
|
'x86' { $arch = '386' }
|
|
Default { Log-Error "Unsupported architecture: $env:PROCESSOR_ARCHITECTURE"; exit 1 }
|
|
}
|
|
|
|
# Ensure installation directory exists for nssm and agent
|
|
Log-Step "Ensuring installation directory exists: $InstallDir"
|
|
New-Item -ItemType Directory -Path $InstallDir -Force -ErrorAction SilentlyContinue | Out-Null # Ensure $InstallDir exists
|
|
|
|
# Check for nssm and download if not present
|
|
$nssmExeToUse = Join-Path $InstallDir "nssm.exe"
|
|
|
|
# First, check if nssm is in PATH and is functional
|
|
$nssmCmd = Get-Command nssm -ErrorAction SilentlyContinue
|
|
if ($nssmCmd) {
|
|
Log-Info "nssm found in PATH at $($nssmCmd.Source)."
|
|
try {
|
|
$nssmVersionOutput = nssm version 2>&1
|
|
Log-Info "Detected nssm version: $nssmVersionOutput"
|
|
}
|
|
catch {
|
|
Log-Warning "nssm found in PATH failed to execute 'nssm version'. Will attempt to use/download local copy. Error: $_"
|
|
$nssmCmd = $null # Force re-evaluation for local copy or download
|
|
}
|
|
}
|
|
|
|
# If nssm not found in PATH or the one in PATH failed, check local $InstallDir
|
|
if (-not $nssmCmd) {
|
|
if (Test-Path $nssmExeToUse) {
|
|
Log-Info "nssm found at $nssmExeToUse. Attempting to use it by adding $InstallDir to PATH."
|
|
$env:Path = "$($InstallDir);$($env:Path)"
|
|
$nssmCmd = Get-Command nssm -ErrorAction SilentlyContinue
|
|
if ($nssmCmd) {
|
|
try {
|
|
$nssmVersionOutput = nssm version 2>&1
|
|
}
|
|
catch {
|
|
Log-Warning "nssm from $InstallDir failed to execute 'nssm version'. Error: $_"
|
|
$nssmCmd = $null # Mark as unusable
|
|
}
|
|
}
|
|
else {
|
|
Log-Warning "Failed to make nssm from $nssmExeToUse available via PATH. Will attempt download."
|
|
}
|
|
}
|
|
}
|
|
|
|
# If still no usable nssm command, proceed to download
|
|
if (-not $nssmCmd) {
|
|
Log-Info "nssm not found or not usable. Attempting to download to $InstallDir..."
|
|
$NssmVersion = "2.24"
|
|
$NssmZipUrl = "https://nssm.cc/release/nssm-$NssmVersion.zip"
|
|
$TempNssmZipPath = Join-Path $env:TEMP "nssm-$NssmVersion.zip"
|
|
$TempExtractDir = Join-Path $env:TEMP "nssm_extract_temp"
|
|
|
|
try {
|
|
Log-Info "Downloading nssm from $NssmZipUrl..."
|
|
Invoke-WebRequest -Uri $NssmZipUrl -OutFile $TempNssmZipPath -UseBasicParsing
|
|
|
|
if (Test-Path $TempExtractDir) { Remove-Item -Recurse -Force $TempExtractDir }
|
|
New-Item -ItemType Directory -Path $TempExtractDir -Force | Out-Null
|
|
Expand-Archive -Path $TempNssmZipPath -DestinationPath $TempExtractDir -Force
|
|
|
|
$NssmSourceDirInsideZip = "nssm-$NssmVersion" # Used for Get-ChildItem search path
|
|
# The path part within the extracted nssm folder, e.g., "nssm-2.24\win32"
|
|
# 'win32' nssm is used for both 'amd64' and 'arm64' PowerShell architectures.
|
|
$NssmArchSubDir = Join-Path "nssm-$NssmVersion" "win32"
|
|
$NssmSourceExePath = Join-Path (Join-Path $TempExtractDir $NssmArchSubDir) "nssm.exe"
|
|
|
|
if (-not (Test-Path $NssmSourceExePath)) {
|
|
Log-Error "Could not find nssm.exe at expected path: $NssmSourceExePath after extraction."
|
|
# Fallback search for nssm.exe within the extracted directory
|
|
$foundNssmFallback = Get-ChildItem -Path $TempExtractDir -Recurse -Filter "nssm.exe" |
|
|
Where-Object { $_.FullName -like "*$NssmArchSubDir\nssm.exe" } |
|
|
Select-Object -First 1
|
|
if ($foundNssmFallback) {
|
|
Log-Warning "Found nssm.exe at $($foundNssmFallback.FullName) using fallback search. Using this."
|
|
$NssmSourceExePath = $foundNssmFallback.FullName
|
|
}
|
|
else {
|
|
Log-Error "nssm.exe ($NssmArchSubDir) still not found in $TempExtractDir. Please install nssm manually (from https://nssm.cc) and ensure it's in your PATH."
|
|
exit 1
|
|
}
|
|
}
|
|
|
|
Copy-Item -Path $NssmSourceExePath -Destination $nssmExeToUse -Force
|
|
|
|
$env:Path = "$($InstallDir);$($env:Path)"
|
|
$nssmCmd = Get-Command nssm -ErrorAction SilentlyContinue # Re-check after adding to PATH
|
|
if ($nssmCmd) {
|
|
Log-Success "Downloaded nssm is now configured and available in PATH."
|
|
}
|
|
else {
|
|
Log-Error "Failed to configure downloaded nssm in PATH from $nssmExeToUse. Please ensure $InstallDir is in your system PATH or nssm is installed globally."
|
|
exit 1
|
|
}
|
|
}
|
|
catch {
|
|
Log-Error "Failed to download or configure nssm: $_"
|
|
Log-Error "Please install nssm manually from https://nssm.cc and ensure nssm.exe is in your PATH."
|
|
exit 1
|
|
}
|
|
finally {
|
|
if (Test-Path $TempNssmZipPath) { Remove-Item $TempNssmZipPath -Force -ErrorAction SilentlyContinue }
|
|
if (Test-Path $TempExtractDir) { Remove-Item $TempExtractDir -Recurse -Force -ErrorAction SilentlyContinue }
|
|
}
|
|
}
|
|
|
|
# Final check that nssm is operational
|
|
try {
|
|
$nssmVersionOutput = nssm version 2>&1
|
|
}
|
|
catch {
|
|
Log-Error "nssm command failed to execute even after setup attempts. Please check the nssm installation and PATH. Error: $_"
|
|
exit 1
|
|
}
|
|
|
|
Log-Step "Installation configuration:"
|
|
Log-Config "Service name: $ServiceName"
|
|
Log-Config "Install directory: $InstallDir"
|
|
Log-Config "GitHub proxy: $ProxyDisplay"
|
|
Log-Config "Agent arguments: $($KomariArgs -join ' ')"
|
|
if ($InstallVersion -ne "") {
|
|
Log-Config "Specified agent version: $InstallVersion"
|
|
} else {
|
|
Log-Config "Agent version: Latest"
|
|
}
|
|
|
|
# Paths
|
|
$BinaryName = "komari-agent-windows-$arch.exe"
|
|
$AgentPath = Join-Path $InstallDir "komari-agent.exe"
|
|
|
|
# Uninstall previous service and binary
|
|
function Uninstall-Previous {
|
|
Log-Step "Checking for existing service..."
|
|
# Check if service exists using nssm status, as Get-Service might not work for nssm services if not properly registered
|
|
$serviceStatus = nssm status $ServiceName 2>&1
|
|
if ($serviceStatus -notmatch "SERVICE_STOPPED" -and $serviceStatus -notmatch "does not exist") {
|
|
Log-Info "Stopping service $ServiceName..."
|
|
nssm stop $ServiceName 2>&1 | Out-Null
|
|
}
|
|
# Attempt to remove the service using nssm
|
|
# We check if it exists first by trying to get its status.
|
|
# nssm remove will succeed if the service exists, and fail otherwise.
|
|
# We add confirm to avoid interactive prompts.
|
|
$removeOutput = nssm remove $ServiceName confirm 2>&1
|
|
if ($LASTEXITCODE -eq 0) {
|
|
}
|
|
elseif ($removeOutput -match "Can't open service! (The specified service does not exist as an installed service.)" -or $removeOutput -match "No such service" -or $removeOutput -match "does not exist") {
|
|
Log-Info "Service $ServiceName does not exist or was already removed."
|
|
}
|
|
else {
|
|
# If nssm remove fails for other reasons, try sc.exe delete as a fallback for older installations
|
|
$svc = Get-Service -Name $ServiceName -ErrorAction SilentlyContinue
|
|
if ($svc) {
|
|
Stop-Service $ServiceName -Force -ErrorAction SilentlyContinue
|
|
sc.exe delete $ServiceName | Out-Null
|
|
}
|
|
}
|
|
|
|
if (Test-Path $AgentPath) {
|
|
Log-Warning "Removing old binary..."
|
|
Remove-Item $AgentPath -Force
|
|
}
|
|
}
|
|
Uninstall-Previous
|
|
|
|
$versionToInstall = ""
|
|
if ($InstallVersion -ne "") {
|
|
Log-Info "Attempting to install specified version: $InstallVersion"
|
|
$versionToInstall = $InstallVersion
|
|
}
|
|
else {
|
|
$ApiUrl = "https://api.github.com/repos/komari-monitor/komari-agent/releases/latest"
|
|
try {
|
|
Log-Step "Fetching latest release version from GitHub API..."
|
|
$release = Invoke-RestMethod -Uri $ApiUrl -UseBasicParsing
|
|
$versionToInstall = $release.tag_name
|
|
Log-Success "Latest version fetched: $versionToInstall"
|
|
}
|
|
catch {
|
|
Log-Error "Failed to fetch latest version: $_"
|
|
exit 1
|
|
}
|
|
}
|
|
Log-Success "Installing Komari Agent version: $versionToInstall"
|
|
|
|
# Construct download URL
|
|
$BinaryName = "komari-agent-windows-$arch.exe"
|
|
$DownloadUrl = if ($GitHubProxy) { "$GitHubProxy/https://github.com/komari-monitor/komari-agent/releases/download/$versionToInstall/$BinaryName" } else { "https://github.com/komari-monitor/komari-agent/releases/download/$versionToInstall/$BinaryName" }
|
|
|
|
# Download and install
|
|
New-Item -ItemType Directory -Path $InstallDir -Force | Out-Null
|
|
Log-Info "URL: $DownloadUrl"
|
|
try {
|
|
Invoke-WebRequest -Uri $DownloadUrl -OutFile $AgentPath -UseBasicParsing
|
|
}
|
|
catch {
|
|
Log-Error "Download failed: $_"
|
|
exit 1
|
|
}
|
|
Log-Success "Downloaded and saved to $AgentPath"
|
|
|
|
# Register and start service
|
|
Log-Step "Configuring Windows service with nssm..."
|
|
$argString = $KomariArgs -join ' '
|
|
# Ensure InstallDir and AgentPath are quoted if they contain spaces
|
|
$quotedAgentPath = "`"$AgentPath`""
|
|
nssm install $ServiceName $quotedAgentPath $argString
|
|
# Set display name and startup type using nssm
|
|
nssm set $ServiceName DisplayName "Komari Agent Service"
|
|
nssm set $ServiceName Start SERVICE_AUTO_START
|
|
nssm set $ServiceName AppExit Default Restart
|
|
nssm set $ServiceName AppRestartDelay 5000
|
|
# Start the service using nssm
|
|
nssm start $ServiceName
|
|
Log-Success "Service $ServiceName installed and started using nssm."
|
|
|
|
Log-Success "Komari Agent installation completed!"
|
|
Log-Config "Service name: $ServiceName"
|
|
Log-Config "Arguments: $argString" |