mirror of
https://github.com/bunny-lab-io/Borealis.git
synced 2025-12-16 06:45:48 -07:00
Re-implemented enrollmentcode commandline arguments for bootstrapper script.
This commit is contained in:
45
Borealis.ps1
45
Borealis.ps1
@@ -10,7 +10,7 @@ param(
|
|||||||
[switch]$EngineTests,
|
[switch]$EngineTests,
|
||||||
[switch]$EngineProduction,
|
[switch]$EngineProduction,
|
||||||
[switch]$EngineDev,
|
[switch]$EngineDev,
|
||||||
[string]$InstallerCode = ''
|
[string]$EnrollmentCode = ''
|
||||||
)
|
)
|
||||||
|
|
||||||
# Preselect menu choices from CLI args (optional)
|
# Preselect menu choices from CLI args (optional)
|
||||||
@@ -1034,6 +1034,7 @@ function InstallOrUpdate-BorealisAgent {
|
|||||||
config_file_watcher_interval = 2
|
config_file_watcher_interval = 2
|
||||||
agent_id = ''
|
agent_id = ''
|
||||||
regions = @{}
|
regions = @{}
|
||||||
|
enrollment_code = ''
|
||||||
installer_code = ''
|
installer_code = ''
|
||||||
}
|
}
|
||||||
$config = [ordered]@{}
|
$config = [ordered]@{}
|
||||||
@@ -1058,40 +1059,44 @@ function InstallOrUpdate-BorealisAgent {
|
|||||||
$config['regions'] = @{}
|
$config['regions'] = @{}
|
||||||
}
|
}
|
||||||
|
|
||||||
$existingInstallerCode = ''
|
$existingEnrollmentCode = ''
|
||||||
if ('installer_code' -in $config.Keys -and $null -ne $config['installer_code']) {
|
if ('enrollment_code' -in $config.Keys -and $null -ne $config['enrollment_code']) {
|
||||||
$existingInstallerCode = [string]$config['installer_code']
|
$existingEnrollmentCode = [string]$config['enrollment_code']
|
||||||
|
} elseif ('installer_code' -in $config.Keys -and $null -ne $config['installer_code']) {
|
||||||
|
$existingEnrollmentCode = [string]$config['installer_code']
|
||||||
}
|
}
|
||||||
|
|
||||||
$providedInstallerCode = ''
|
$providedEnrollmentCode = ''
|
||||||
if ($InstallerCode -and $InstallerCode.Trim()) {
|
if ($EnrollmentCode -and $EnrollmentCode.Trim()) {
|
||||||
$providedInstallerCode = $InstallerCode.Trim()
|
$providedEnrollmentCode = $EnrollmentCode.Trim()
|
||||||
} elseif ($env:BOREALIS_INSTALLER_CODE -and $env:BOREALIS_INSTALLER_CODE.Trim()) {
|
} elseif ($env:BOREALIS_ENROLLMENT_CODE -and $env:BOREALIS_ENROLLMENT_CODE.Trim()) {
|
||||||
$providedInstallerCode = $env:BOREALIS_INSTALLER_CODE.Trim()
|
$providedEnrollmentCode = $env:BOREALIS_ENROLLMENT_CODE.Trim()
|
||||||
}
|
}
|
||||||
|
|
||||||
if (-not $providedInstallerCode) {
|
if (-not $providedEnrollmentCode) {
|
||||||
$defaultDisplay = if ($existingInstallerCode) { $existingInstallerCode } else { '' }
|
$defaultDisplay = if ($existingEnrollmentCode) { $existingEnrollmentCode } else { '' }
|
||||||
Write-Host ""; Write-Host "Set an installer code for agent enrollment." -ForegroundColor DarkYellow
|
Write-Host ""; Write-Host "Set an enrollment code for agent enrollment." -ForegroundColor DarkYellow
|
||||||
$inputCode = Read-Host ("Installer Code [{0}] (e.g. A4E1-••••-••••-••••-••••-••••-••••-350A)" -f $defaultDisplay)
|
$inputCode = Read-Host ("Enrollment Code [{0}] (e.g. A4E1-••••-••••-••••-••••-••••-••••-350A)" -f $defaultDisplay)
|
||||||
if ($inputCode -and $inputCode.Trim()) {
|
if ($inputCode -and $inputCode.Trim()) {
|
||||||
$providedInstallerCode = $inputCode.Trim()
|
$providedEnrollmentCode = $inputCode.Trim()
|
||||||
} elseif ($defaultDisplay) {
|
} elseif ($defaultDisplay) {
|
||||||
$providedInstallerCode = $defaultDisplay
|
$providedEnrollmentCode = $defaultDisplay
|
||||||
} else {
|
} else {
|
||||||
$providedInstallerCode = ''
|
$providedEnrollmentCode = ''
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
$config['installer_code'] = $providedInstallerCode
|
$config['enrollment_code'] = $providedEnrollmentCode
|
||||||
|
# Retain legacy key to avoid breaking existing agent readers
|
||||||
|
$config['installer_code'] = $providedEnrollmentCode
|
||||||
|
|
||||||
try {
|
try {
|
||||||
$configJson = $config | ConvertTo-Json -Depth 10
|
$configJson = $config | ConvertTo-Json -Depth 10
|
||||||
[System.IO.File]::WriteAllText($configPath, $configJson, $utf8NoBom)
|
[System.IO.File]::WriteAllText($configPath, $configJson, $utf8NoBom)
|
||||||
if ($providedInstallerCode) {
|
if ($providedEnrollmentCode) {
|
||||||
Write-Host "Installer code saved to agent_settings.json." -ForegroundColor Green
|
Write-Host "Enrollment code saved to agent_settings.json." -ForegroundColor Green
|
||||||
} else {
|
} else {
|
||||||
Write-Host "Installer code cleared in agent_settings.json." -ForegroundColor Yellow
|
Write-Host "Enrollment code cleared in agent_settings.json." -ForegroundColor Yellow
|
||||||
}
|
}
|
||||||
} catch {
|
} catch {
|
||||||
Write-AgentLog -FileName 'Install.log' -Message ("[CONFIG] Failed to persist agent_settings.json: {0}" -f $_.Exception.Message)
|
Write-AgentLog -FileName 'Install.log' -Message ("[CONFIG] Failed to persist agent_settings.json: {0}" -f $_.Exception.Message)
|
||||||
|
|||||||
Reference in New Issue
Block a user