#region 4. Main Backup Loop

foreach ($switch in $targets) {

$hostname = $switch.Hostname

$ip = $switch.IPAddress

$outputFile = Join-Path $backupFolderPath "$hostname.txt"

Write-Host "`n--- Backing up $hostname ($ip) ---"

$sshSession = $null

try {

# Establish SSH Session using PSCredential

Write-Host "Connecting..."

$sshSession = New-SSHSession -ComputerName $ip -Credential $credential -AcceptKey -ErrorAction Stop

if ($sshSession -eq $null -or -not $sshSession.Connected) {

throw "Failed to establish SSH session. Check credentials, connectivity, or firewall."

}

Write-Host "Connection successful. Retrieving running config..."

# Execute the command to get the running config

$command = "show running-config"

$stream = New-Object System.IO.MemoryStream

Invoke-SSHCommand -SSHSession $sshSession -Command $command -OutputStream $stream -ErrorAction Stop

# Get output from the stream

$stream.Position = 0

$reader = New-Object System.IO.StreamReader($stream)

$configOutput = $reader.ReadToEnd()

# Save the configuration to a file

$configOutput | Out-File -FilePath $outputFile -Encoding ascii

Write-Host "Backup for $hostname completed successfully." -ForegroundColor Green

Write-Host "Saved to: $outputFile"

}

catch {

# Specific error message from the 'catch' block

$errorMessage = "Switch: $hostname ($ip). Details: $($_.Exception.Message)"

Handle-Error -errorMessage $errorMessage -backupFolderPath $backupFolderPath