add upload feature to DSA manifest creation
This commit is contained in:
+40
-13
@@ -2,15 +2,18 @@
|
||||
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Downloads and runs the DSA manifest collector.
|
||||
Downloads and runs the DSA manifest collector, with optional upload and cleanup.
|
||||
.DESCRIPTION
|
||||
Tools.ps1 must be dot-sourced by the caller first. Set $DSAManifestSource
|
||||
in the caller's scope before using Run-Script -LivePSScript Get-DSAManifest.
|
||||
The output path is generated in $Global:OutputDirectory with the filename
|
||||
dsa-manifest-HOSTNAME-yyyyMMddHHmmss.csv, using the computer name and current local date and time.
|
||||
Set $Zip to 'true', '$true', 'yes', 'y', or '1' to include --zip.
|
||||
Use 'false', '$false', 'no', 'n', or '0' to disable ZIP; an empty value also disables it.
|
||||
Values are case-insensitive and surrounding whitespace is ignored.
|
||||
$Zip enables ZIP compression; $Upload sends the result to the FileDrop;
|
||||
$DeleteAfterUpload removes the local result only after a confirmed upload.
|
||||
All three accept 'true', '$true', 'yes', 'y', or '1' to enable, and
|
||||
'false', '$false', 'no', 'n', or '0' to disable. Empty values disable the option.
|
||||
Values are case-insensitive and surrounding whitespace is ignored. When
|
||||
$Upload is false, the local CSV or ZIP is retained regardless of $DeleteAfterUpload.
|
||||
#>
|
||||
|
||||
If ( [string]::IsNullOrWhiteSpace([string]$DSAManifestSource) ) {
|
||||
@@ -18,16 +21,20 @@ If ( [string]::IsNullOrWhiteSpace([string]$DSAManifestSource) ) {
|
||||
}
|
||||
|
||||
# Normalize RMM text values without treating every nonempty string as true.
|
||||
$DsaZipValue = ([string]$Zip).Trim().ToLowerInvariant()
|
||||
If ( $DsaZipValue -in @('true', '$true', 'yes', 'y', '1') ) {
|
||||
$DsaZipEnabled = $true
|
||||
}
|
||||
ElseIf ( $DsaZipValue -in @('', 'false', '$false', 'no', 'n', '0') ) {
|
||||
$DsaZipEnabled = $false
|
||||
}
|
||||
Else {
|
||||
Throw 'Invalid $Zip value. Use true, $true, yes, y, 1, false, $false, no, n, or 0.'
|
||||
$DsaOptions = @{ Zip = $Zip; Upload = $Upload; DeleteAfterUpload = $DeleteAfterUpload }
|
||||
ForEach ( $DsaOptionName in @('Zip', 'Upload', 'DeleteAfterUpload') ) {
|
||||
$DsaOptionValue = ([string]$DsaOptions[$DsaOptionName]).Trim().ToLowerInvariant()
|
||||
If ( $DsaOptionValue -in @('true', '$true', 'yes', 'y', '1') ) {
|
||||
$DsaOptions[$DsaOptionName] = $true
|
||||
}
|
||||
ElseIf ( $DsaOptionValue -in @('', 'false', '$false', 'no', 'n', '0') ) {
|
||||
$DsaOptions[$DsaOptionName] = $false
|
||||
}
|
||||
Else {
|
||||
Throw ('Invalid ${0} value. Use true, $true, yes, y, 1, false, $false, no, n, or 0.' -f $DsaOptionName)
|
||||
}
|
||||
}
|
||||
$DsaZipEnabled = $DsaOptions.Zip
|
||||
|
||||
$DSAManifestOutput = Join-Path -Path $Global:OutputDirectory -ChildPath (
|
||||
'dsa-manifest-{0}-{1}.csv' -f $Env:COMPUTERNAME, (Get-Date -Format 'yyyyMMddHHmmss')
|
||||
@@ -65,6 +72,7 @@ $DsaCommandLine = ($DsaArguments | ForEach-Object {
|
||||
'"{0}"' -f ($_ -replace '(\\*)"', '$1$1\"' -replace '(\\+)$', '$1$1')
|
||||
}) -join ' '
|
||||
|
||||
$DsaCollectionTime = Get-Date
|
||||
Write-Output "Running dsa-collect.exe (ZIP compression: ${DsaZipEnabled})"
|
||||
$DsaProcess = Start-Process -FilePath $DsaExecutablePath -ArgumentList $DsaCommandLine `
|
||||
-Wait -PassThru -WindowStyle Hidden -ErrorAction Stop
|
||||
@@ -80,3 +88,22 @@ If ( !(Test-Path -LiteralPath $DsaResultPath -PathType Leaf) ) {
|
||||
Throw "dsa-collect.exe did not create the expected output: ${DsaResultPath}"
|
||||
}
|
||||
Write-Output "DSA manifest created: ${DsaResultPath}"
|
||||
|
||||
If ( $DsaOptions.Upload ) {
|
||||
Write-Output "Uploading DSA manifest: ${DsaResultPath}"
|
||||
$DsaUploadReceipt = Upload-File -Path $DsaResultPath `
|
||||
-Subject "DSA manifest file uploaded from ${Env:COMPUTERNAME}" `
|
||||
-Message "DSA collection performed on ${DSAManifestSource} at $($DsaCollectionTime.ToString('yyyy-MM-dd HH:mm:ss zzz'))" `
|
||||
-ErrorAction Stop
|
||||
If ( [string]::IsNullOrWhiteSpace([string]$DsaUploadReceipt.AttachmentId) -or
|
||||
[string]::IsNullOrWhiteSpace([string]$DsaUploadReceipt.Status) ) {
|
||||
Throw 'The upload did not return a confirmation receipt. The local manifest has been retained.'
|
||||
}
|
||||
Write-Output "DSA manifest uploaded: ${DsaResultPath}"
|
||||
|
||||
If ( $DsaOptions.DeleteAfterUpload ) {
|
||||
Remove-Item -LiteralPath $DsaResultPath -Force -ErrorAction Stop
|
||||
Write-Output "Deleted uploaded DSA manifest: ${DsaResultPath}"
|
||||
}
|
||||
$DsaUploadReceipt
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user