X-Git-Url: https://git.p6c8.net/dsmonrot.git/blobdiff_plain/a82b78082fdbb2f7614844cbdf9ec371699194b1..dc31e36391307b87cd67ab26cb1c5a69d83d5e41:/dsmonrot.ps1

diff --git a/dsmonrot.ps1 b/dsmonrot.ps1
index 82b5014..5732bd5 100644
--- a/dsmonrot.ps1
+++ b/dsmonrot.ps1
@@ -16,6 +16,11 @@
 # Keep backup for this amount of months (excluding the current month),
 # -1 for indefinite
 [Int32]$keepMonths = 2
+# Rotate BEFORE the beginning of a full backup (default is after a successful
+# full backup)
+# WARNING: If this option is set to $True and the full backup fails you could
+# have NO backup
+$rotateBeforeBackup = $False
 # Path to Drive Snapshot
 [String]$dsPath = "C:\Users\Patrick\Desktop\DSMonRot\snapshot.exe"
 # Path to Drive Snapshot log file (specify only the file name if you set
@@ -85,15 +90,43 @@ function Send-Email([String]$body) {
 	}
 }
 
-Write-Host "Started at" (Get-Date -format "yyyy-MM-dd HH:mm:ss")
+function Rotate-Backup {
+	if($keepMonths -lt 0) {
+		return
+	}
+
+	Write-Host "Rotating"
+	
+	$keepMonthsCount = $keepMonths
+	
+	Get-ChildItem $backupDir -Directory | Where-Object {($_.Name -ne $currMonth) -and ($_.Name -match "^\d{4,}-\d{2}$")} | Sort-Object -Descending |
+	Foreach-Object {
+		Write-Host $_ "=>" $_.FullName
+		
+		if($keepMonthsCount -ge 0) {
+			$keepMonthsCount--
+		}
+		
+		if($keepMonthsCount -eq -1) {
+			Write-Host "Deleting $_"
+			Remove-Item -Recurse -Force $_.FullName
+		}
+	}
+}
+
+$startTime = Get-Date -format "yyyy-MM-dd HH:mm:ss"
+Write-Host "Started at $startTime"
 
 $errorMessages = @()
 
 $smbConnected = $False
+$doBackup = $False
 $success = $False
+$isDiff = $False
+$dsCommand = ""
 
 if($smbDrive) {
-	Try {
+	try {
 		Write-Host "Connecting network drive"
 		
 		if($smbUser -and $smbPassword) {
@@ -112,114 +145,127 @@ if($smbDrive) {
 		
 		$smbConnected = $True
 	}
-	Catch {
+	catch {
 		Write-Host "Could not connect to network drive $smbDrive`: $_.Exception.Message"
-		exit
+		$errorMessages += "Could not connect to network drive $smbDrive`: $_.Exception.Message"
 	}
 }
 
 if(!(Test-Path $backupDir)) {
 	Write-Host "Directory $backupDir does not exist!"
-	exit
+	$errorMessages += "Directory $backupDir does not exist!"
 }
 
-$currMonth = Get-Date -format "yyyy-MM"
-$currDay = Get-Date -format "yyyy-MM-dd"
+if($errorMessages.Count -eq 0) {
+	$currMonth = Get-Date -format "yyyy-MM"
+	$currDay = Get-Date -format "yyyy-MM-dd"
 
-Write-Host $currMonth
+	Write-Host $currMonth
 
-$backupTarget = $backupDir + "\" + $currMonth
-$backupTargetFull = $backupTarget + "\" + "Full"
+	$backupTarget = $backupDir + "\" + $currMonth
+	$backupTargetFull = $backupTarget + "\" + "Full"
 
-$backupTargetDiff = $backupTarget + "\" + "Diff-" + $currDay
+	$backupTargetDiff = $backupTarget + "\" + "Diff-" + $currDay
 
-Write-Host $backupTarget
+	Write-Host $backupTarget
 
-$isDiff = $False
-
-if((Test-Path $backupTarget) -and (Test-Path $backupTargetFull) -and (Test-Path "$backupTargetFull\*.hsh")) {
-	Write-Host "Differential backup"
-	
-	$isDiff = $True
-	
-	if(!(Test-Path $backupTargetDiff)) {
-		Write-Host "Creating directory $backupTargetDiff"
+	if((Test-Path $backupTarget) -and (Test-Path $backupTargetFull) -and (Test-Path "$backupTargetFull\*.hsh")) {
+		Write-Host "Differential backup"
 		
-		try {
-			New-Item -ItemType directory -Path $backupTargetDiff -ErrorAction Stop
+		$isDiff = $True
 		
-			$success = $True
-		}
-		catch {
-			Write-Host "Could not create directory $backupTargetDiff`: $_.Exception.Message"
-			exit
-		}
-		
-		$dsLogPath = if($dsLogFileToBackup) { "$backupTargetDiff\$dsLogFile" } else { $dsLogFile }
-		
-		$dsArgs = @($disksToBackup, "--logfile:$dsLogPath", "$backupTargetDiff\`$disk.sna", "-h$backupTargetFull\`$disk.hsh") + $dsAdditionalArgs
-		Write-Host $dsPath ($dsArgs -join " ")
-		
-		& $dsPath $dsArgs
-		
-		if($LastExitCode -ne 0) {
-			Write-Host "Drive Snapshot failed to backup! Exit code: $LastExitCode"
-			$errorMessages += "Drive Snapshot failed to backup! Exit code: $LastExitCode"
+		if(!(Test-Path $backupTargetDiff)) {
+			Write-Host "Creating directory $backupTargetDiff"
+			
+			try {
+				New-Item -ItemType directory -Path $backupTargetDiff -ErrorAction Stop
+				$doBackup = $True
+			}
+			catch {
+				Write-Host "Could not create directory $backupTargetDiff`: $_.Exception.Message"
+				$errorMessages += "Could not create directory $backupTargetDiff`: $_.Exception.Message"
+			}
+			
+			if($doBackup) {
+				$dsLogPath = if($dsLogFileToBackup) { "$backupTargetDiff\$dsLogFile" } else { $dsLogFile }
+				
+				$dsArgs = @($disksToBackup, "--logfile:$dsLogPath", "$backupTargetDiff\`$disk.sna", "-h$backupTargetFull\`$disk.hsh") + $dsAdditionalArgs
+				Write-Host $dsPath ($dsArgs -join " ")
+				
+				$dsCommand = "$dsPath $dsArgs"
+				
+				& $dsPath $dsArgs
+				
+				if($LastExitCode -ne 0) {
+					Write-Host "Drive Snapshot failed to backup! Exit code: $LastExitCode"
+					$errorMessages += "Drive Snapshot failed to backup! Exit code: $LastExitCode"
+				}
+				else {
+					$success = $True
+				}
+			}
 		}
 		else {
-			$success = $True
+			Write-Host "Directory $backupTargetDiff already exists!"
+			$errorMessages += "Directory $backupTargetDiff already exists!"
 		}
 	}
 	else {
-		Write-Host "Directory $backupTargetDiff already exists!"
-		$errorMessages += "Directory $backupTargetDiff already exists!"
-	}
-}
-else {
-	Write-Host "Full backup"
-	
-	if(!(Test-Path $backupTarget)) {
-		Write-Host "Creating directory $backupTarget"
-		New-Item -ItemType directory -Path $backupTarget
-	}
-	
-	if(!(Test-Path $backupTargetFull)) {
-		Write-Host "Creating directory $backupTargetFull"
-		New-Item -ItemType directory -Path $backupTargetFull
-	}
-	
-	$dsLogPath = if($dsLogFileToBackup) { "$backupTargetFull\$dsLogFile" } else { $dsLogFile }
-
-	$dsArgs = @($disksToBackup, "--logfile:$dsLogPath", "$backupTargetFull\`$disk.sna") + $dsAdditionalArgs
-	Write-Host $dsPath ($dsArgs -join " ")
-	
-	& $dsPath $dsArgs
-	
-	if($LastExitCode -ne 0) {
-		Write-Host "Drive Snapshot failed to backup! Exit code: $LastExitCode"
-		$errorMessages += "Drive Snapshot failed to backup! Exit code: $LastExitCode"
-	}
-	else {
-		$success = $True
-	}
-}
+		Write-Host "Full backup"
 
-if(!$isDiff -and $success -eq $True -and $keepMonths -ge 0) {
-	Write-Host "Rotating"
-	
-	$keepMonthsCount = $keepMonths
-	
-	Get-ChildItem $backupDir -Directory | Where-Object {($_.Name -ne $currMonth) -and ($_.Name -match "^\d{4,}-\d{2}$")} | Sort-Object -Descending |
-	Foreach-Object {
-		Write-Host $_ "=>" $_.FullName
+		if(!(Test-Path $backupTarget)) {
+			Write-Host "Creating directory $backupTarget"
+			
+			try {
+				New-Item -ItemType directory -Path $backupTarget -ErrorAction Stop
+			}
+			catch {
+				Write-Host "Could not create directory $backupTarget`: $_.Exception.Message"
+				$errorMessages += "Could not create directory $backupTarget`: $_.Exception.Message"
+			}
+		}
 		
-		if($keepMonthsCount -ge 0) {
-			$keepMonthsCount--
+		if(!(Test-Path $backupTargetFull)) {
+			Write-Host "Creating directory $backupTargetFull"
+			
+			try {
+				New-Item -ItemType directory -Path $backupTargetFull -ErrorAction Stop
+				$doBackup = $True
+			}
+			catch {
+				Write-Host "Could not create directory $backupTargetFull`: $_.Exception.Message"
+				$errorMessages += "Could not create directory $backupTargetFull`: $_.Exception.Message"
+			}
+		}
+		else {
+			$doBackup = $True
 		}
 		
-		if($keepMonthsCount -eq -1) {
-			Write-Host "Deleting $_"
-			Remove-Item -Recurse -Force $_.FullName
+		if($doBackup) {
+			if($rotateBeforeBackup) {
+				Rotate-Backup
+			}
+			
+			$dsLogPath = if($dsLogFileToBackup) { "$backupTargetFull\$dsLogFile" } else { $dsLogFile }
+
+			$dsArgs = @($disksToBackup, "--logfile:$dsLogPath", "$backupTargetFull\`$disk.sna") + $dsAdditionalArgs
+			Write-Host $dsPath ($dsArgs -join " ")
+			
+			$dsCommand = "$dsPath $dsArgs"
+			
+			& $dsPath $dsArgs
+			
+			if($LastExitCode -ne 0) {
+				Write-Host "Drive Snapshot failed to backup! Exit code: $LastExitCode"
+				$errorMessages += "Drive Snapshot failed to backup! Exit code: $LastExitCode"
+			}
+			else {
+				$success = $True
+			}
+			
+			if($rotateBeforeBackup -eq $False -and $success -eq $True) {
+				Rotate-Backup
+			}
 		}
 	}
 }
@@ -237,7 +283,15 @@ if($smbConnected) {
 }
 
 if($emailOnError -and $errorMessages.Count -gt 0) {
-	Send-Email ("Error:`n"+($errorMessages -join "`n"))
+	$emailBody  = "This is DSMonRot on $env:computername, started at $startTime.`n"
+	$emailBody += "An error occured while performing a backup. Below are the error messages and some status information.`n`n"
+	$emailBody += "Differential backup:    $isDiff`n"
+	$emailBody += "Backup successful:      $success`n"
+	$emailBody += "Drive Snapshot command: $dsCommand`n`n"
+	$emailBody += ($errorMessages -join "`n")
+
+	Send-Email ($emailBody)
 }
 
-Write-Host "Ended at" (Get-Date -format "yyyy-MM-dd HH:mm:ss")
\ No newline at end of file
+$endTime = Get-Date -format "yyyy-MM-dd HH:mm:ss"
+Write-Host "Ended at $endTime"
\ No newline at end of file