public/Update-TNPluginFeed.ps1
|
function Update-TNPluginFeed { <# .SYNOPSIS Updates plugins or feeds .DESCRIPTION Updates plugins or feeds .PARAMETER SessionObject Optional parameter to force using specific SessionObjects. By default, each command will connect to all connected servers that have been connected to using Connect-TNServer .PARAMETER FilePath The path to the feed or plugin .PARAMETER Type The type of update, "Feed","ActivePlugin", "PassivePlugin", or "Lce" .PARAMETER Wait Wait for the upload to process before outputting results .PARAMETER EnableException By default, when something goes wrong we try to catch it, interpret it and give you a friendly warning message. This avoids overwhelming you with 'sea of red' exceptions, but is inconvenient because it basically disables advanced scripting. Using this switch turns this 'nice by default' feature off and enables you to catch exceptions with your own try/catch. .EXAMPLE PS C:\> Update-TNPluginFeed -Type ActivePlugin -FilePath "C:\sc\CM-253478-sc-plugins.tar.gz" -Wait Uploads an active plugin update C:\sc\CM-253478-sc-plugins.tar.gz and wait for it to process .EXAMPLE PS C:\> Update-TNPluginFeed -Type Feed -FilePath "C:\sc\CM-253480-SecurityCenterFeed48.tar.gz" Uploads an a feed file update C:\sc\CM-253478-sc-plugins.tar.gz and does not wait for the file to process #> [CmdletBinding()] param ( [Parameter(ValueFromPipelineByPropertyName)] [object[]]$SessionObject = (Get-TNSession), [Parameter(Mandatory, ValueFromPipelineByPropertyName)] [ValidateScript( { Test-Path -Path $_ })] [string]$FilePath, [Parameter(Mandatory, ValueFromPipelineByPropertyName)] [ValidateSet("Feed","ActivePlugin", "PassivePlugin", "Lce")] [string]$Type, [Parameter(ValueFromPipelineByPropertyName)] [switch]$Wait, [switch]$EnableException ) begin { switch ($Type) { "Feed" { $webpath = "sc" } "ActivePlugin" { $webpath = "active" } "PassivePlugin" { $webpath = "passive" } "Lce" { $webpath = "lce" } } } process { foreach ($session in $SessionObject) { $PSDefaultParameterValues["*:SessionObject"] = $session if (-not $session.sc) { Stop-PSFFunction -EnableException:$EnableException -Message "Only tenable.sc supported" -Continue } $files = Get-ChildItem -Path $FilePath foreach ($file in $files.FullName) { Write-ProgressHelper -TotalSteps 1 -StepNumber 1 -Activity "Uploading $file" -Message "Uploading..." -ExcludePercent $body = $file | Publish-File -Session $session -EnableException:$EnableException $params = @{ SessionObject = $session Method = "POST" Path = "/feed/$webpath/process" Parameter = $body ContentType = "application/json" } $null = Invoke-TnRequest @params | ConvertFrom-TNRestResponse if (-not $Wait) { Get-TNFeedStatus -Session $session } else { $stepnumber = 0 Start-Sleep -Seconds 2 $result = Get-TNFeedStatus | Select-Object *Running while ($result.SecurityCenterUpdateRunning -ne $true -and $result.ActivePluginsUpdateRunning -ne $true -and $result.PassivePluginsUpdateRunning -ne $true -and $result.IndustrialUpdateRunning -ne $true -and $result.LceRunning -ne $true -and $stepnumber -lt 100) { # Wait for it to start Write-ProgressHelper -TotalSteps 100 -StepNumber ($stepnumber++) -Activity "Running update for $Type" -Message "Status: File uploaded, waiting for update to start" Start-Sleep -Seconds 1 $result = Get-TNFeedStatus | Select-Object *Running } while ($result.SecurityCenterUpdateRunning -eq $true -or $result.ActivePluginsUpdateRunning -eq $true -or $result.PassivePluginsUpdateRunning -eq $true -or $result.IndustrialUpdateRunning -eq $true -or $result.LceRunning -eq $true) { Start-Sleep -Seconds 1 if ($stepnumber -eq 100) { $stepnumber = 0 } else { Write-ProgressHelper -TotalSteps 100 -StepNumber ($stepnumber++) -Activity "Running update for $Type" -Message "Status: Updating. This may take a while." } $result = Get-TNFeedStatus -Session $session | Select-Object *Running } Get-TNFeedStatus -Session $session } } } } } |