Bootstrap.ps1

param (
	# Params allow for manually launching the script for debugging
	[Parameter(Mandatory = $false)]
	[string]$JobId = $null,
	[Parameter(Mandatory = $false)]
	[string]$SessionId = $null
)

function FinishBootstrap {
	param(
		[switch]$Failed
	)

	if ($config.logging.enabled) {
		Stop-Logging

		if ($newLogfile) {
			# Rename log file to include the job name.
			try {
				Rename-Item -Path $logFile -NewName "$(Split-Path $newLogfile -Leaf)" -ErrorAction Stop
			}
			catch {
				Write-Output "ERROR: Failed to rename log file: $_" | Out-File $logFile -Append
			}
		}
	}
	if ($Failed) {
		exit 1
	}
	else {
		exit 0
	}
}

# Import modules
Import-Module Veeam.Backup.PowerShell -DisableNameChecking
Import-Module "$PSScriptRoot\resources\Logger.psm1"
Import-Module "$PSScriptRoot\resources\JsonValidator.psm1"
Import-Module "$PSScriptRoot\resources\VBRSessionInfo.psm1"

$IsDebug = $false

if ($JobId -and -not $SessionId) {
	Write-Output 'INFO: JobId is provided but not SessionId. Attempting to retrieve the last session ID for the job.'
	$SessionId = (Get-VBRBackupSession | Where-Object {$_.JobId -eq $JobId} | Sort-Object EndTimeUTC -Descending | Select-Object -First 1).Id.ToString()
	if ($null -eq $SessionId) {
		Write-Output 'ERROR: Failed to retrieve the session ID for the provided job ID. Please check the job ID.'
		FinishBootstrap -Failed
	}
}
elseif ($SessionId -and -not $JobId) {
	Write-Output 'INFO: SessionId is provided but not JobId. Attempting to retrieve the last job ID for the session.'
	$JobId = (Get-VBRBackupSession -Id $SessionId | Sort-Object EndTimeUTC -Descending | Select-Object -First 1).JobId.ToString()
	if ($null -eq $JobId) {
		Write-Output 'ERROR: Failed to retrieve the job ID for the provided session ID. Please check the session ID.'
		FinishBootstrap -Failed
	}
}
if ($JobId -and $SessionId) {
	Write-Output 'INFO: JobId and SessionId found. Using them to start the script.'
	$IsDebug = $true
}

# Set vars
$configFile = "$PSScriptRoot\config\conf.json"
$date = (Get-Date -UFormat %Y-%m-%d_%T).Replace(':', '.')
$logFile = "$PSScriptRoot\log\$($date)_Bootstrap.log"
$idRegex = '[0-9a-f]{8}-([0-9a-f]{4}-){3}[0-9a-f]{12}'
$supportedTypes = 'Backup', 'EpAgentBackup', 'Replica', 'BackupToTape', 'FileToTape'

# Start logging to file
Start-Logging -Path $logFile

# Log version
Write-LogMessage -Tag 'INFO' -Message "Version: $(Get-Content "$PSScriptRoot\resources\version.txt" -Raw)"

# Retrieve configuration.
## Pull config to PSCustomObject
$config = Get-Content -Raw $configFile | ConvertFrom-Json # TODO: import config from param instead of later as file. Can then improve logging flow.

# Stop logging and remove log file if logging is disable in config.
if (-not $config.logging.enabled) {
	Stop-Logging
	Remove-Item $logFile -Force -ErrorAction SilentlyContinue
}

## Pull raw config and format for later.
## This is necessary since $config as a PSCustomObject was not passed through correctly with Start-Process and $powershellArguments.
$configRaw = (Get-Content -Raw $configFile).Replace('"', '\"').Replace("`n", '').Replace("`t", '').Replace('  ', ' ')

## Test config.
$validationResult = Test-JsonValid -JsonPath $configFile -SchemaPath "$PSScriptRoot\config\schema.json"
if ($validationResult.IsValid) {
	Write-LogMessage -Tag 'INFO' -Message 'Configuration validated successfully.'
}
else {
	Write-LogMessage -Tag 'ERROR' -Message "Failed to validate configuration: $($validationResult.Message)"
}

if ($IsDebug) {
	Write-LogMessage -Tag 'INFO' -Message "JobId: $JobId"
	Write-LogMessage -Tag 'INFO' -Message "SessionId: $SessionId"
}
else {
	Write-LogMessage -Tag 'INFO' -Message 'Attempting to retrieve job ID and session ID from the parent process.'

	# Get the command line used to start the Veeam session.
	$parentPid = (Get-CimInstance Win32_Process -Filter "processid='$PID'").parentprocessid.ToString()
	$parentCmd = (Get-CimInstance Win32_Process -Filter "processid='$parentPID'").CommandLine

	# Get the Veeam job and session IDs
	$JobId = ([regex]::Matches($parentCmd, $idRegex)).Value[0]
	$SessionId = ([regex]::Matches($parentCmd, $idRegex)).Value[1]
}

# Get the Veeam job details and hide warnings to mute the warning regarding deprecation of the use of some cmdlets to get certain job type details.
# At time of writing, there is no alternative way to discover the job time.
Write-LogMessage -Tag 'INFO' -Message 'Getting VBR job details'
$job = Get-VBRJob -WarningAction SilentlyContinue | Where-Object {$_.Id.ToString() -eq $JobId}
if ($job) {
	$JobType = $job.JobType
}
else {
	# Can't locate non-tape job so check if it's a tape job
	Write-LogMessage -Tag 'DEBUG' -Message "Job with ID $JobId not found in VBR jobs. Checking for tape jobs."
	$job = Get-VBRTapeJob -WarningAction SilentlyContinue | Where-Object {$_.Id.ToString() -eq $JobId}
	if ($job) {
		$JobType = $job.Type
	}
	else {
		Write-LogMessage -Tag 'ERROR' -Message "Job with ID $JobId not found in tape jobs. Exiting."
		FinishBootstrap -Failed
	}
}


# Get the session information and name.
Write-LogMessage -Tag 'INFO' -Message 'Getting VBR session information'
try {
	$sessionInfo = Get-VBRSessionInfo -SessionId $SessionId -JobId $JobId -JobType $JobType
}
catch {
	Write-LogMessage -Tag 'ERROR' -Message "Failed to retrieve session information: $_"
	FinishBootstrap -Failed
}
$jobName = $sessionInfo.JobName
$vbrSessionLogger = $sessionInfo.Session.Logger

$vbrLogEntry = $vbrSessionLogger.AddLog('[VeeamNotify] Parsing job & session information...')

# Quit if job type is not supported.
if ($JobType -notin $supportedTypes) {
	Write-LogMessage -Tag 'ERROR' -Message "Job type '$($JobType)' is not supported."
	FinishBootstrap -Failed
}

Write-LogMessage -Tag 'INFO' -Message "Bootstrap script for Veeam job '$jobName' (job $JobId session $SessionId) - Session & job detection complete."

# Set log file name based on job
## Replace spaces if any in the job name
if ($jobName -match ' ') {
	$logJobName = $jobName.Replace(' ', '_')
}
else {
	$logJobName = $jobName
}
$newLogfile = "$PSScriptRoot\log\$($date)-$($logJobName).log"

# Build argument string for the alert sender script.
$powershellArguments = "-NoProfile -File $PSScriptRoot\AlertSender.ps1", `
	"-JobId `"$JobId`"", `
	"-SessionId `"$SessionId`"", `
	"-JobType `"$JobType`"", `
	"-Config `"$configRaw`"", `
	"-Logfile `"$newLogfile`""

$vbrSessionLogger.UpdateSuccess($vbrLogEntry, '[VeeamNotify] Parsed job & session information.') | Out-Null

# Start a new new script in a new process with some of the information gathered here.
# This allows Veeam to finish the current session faster and allows us gather information from the completed job.
try {
	$powershellExePath = (Get-Command -Name 'powershell.exe' -ErrorAction Stop).Path
	Write-LogMessage -Tag 'INFO' -Message 'Launching AlertSender.ps1...'
	$vbrLogEntry = $vbrSessionLogger.AddLog('[VeeamNotify] Launching Alert Sender...')
	Start-Process -FilePath "$powershellExePath" -Verb runAs -ArgumentList $powershellArguments -WindowStyle hidden -ErrorAction Stop
	Write-LogMessage -Tag 'INFO' -Message 'AlertSender.ps1 launched successfully.'
	$vbrSessionLogger.UpdateSuccess($vbrLogEntry, '[VeeamNotify] Launched Alert Sender.') | Out-Null
}
catch {
	Write-LogMessage -Tag 'ERROR' -Message "Failed to launch AlertSender.ps1: $_"
	$vbrSessionLogger.UpdateErr($vbrLogEntry, '[VeeamNotify] Failed to launch Alert Sender.', "Please check the log: $newLogfile") | Out-Null
	FinishBootstrap -Failed
}

FinishBootstrap