Allow to rotate BEFORE a full backup

[dsmonrot.git] / dsmonrot.ps1

diff --git a/dsmonrot.ps1 b/dsmonrot.ps1
index 353e0dbaae7cd1321f12961481a4d3bda0d0253d..b07d5ef3722452bc3f0f87f366d3d710fdd26c2e 100644 (file)
--- a/dsmonrot.ps1
+++ b/dsmonrot.ps1
@@ -16,6 +16,11 @@
 # Keep backup for this amount of months (excluding the current month),\r
 # -1 for indefinite\r
 [Int32]$keepMonths = 2\r
+# Rotate BEFORE the beginning of a full backup (default is after a successful\r
+# full backup)\r
+# WARNING: If this option is set to $True and the full backup fails you have\r
+# NO backup\r
+$rotateBeforeBackup = $False\r
 # Path to Drive Snapshot\r
 [String]$dsPath = "C:\Users\Patrick\Desktop\DSMonRot\snapshot.exe"\r
 # Path to Drive Snapshot log file (specify only the file name if you set\r
@@ -46,7 +51,7 @@
 # To address of email notification\r
 [String]$emailToAddress = "patrick@test.local"\r
 # Subject of email notification\r
-[String]$emailSubject = "DSMonRot"\r
+[String]$emailSubject = "DSMonRot on $env:computername"\r
 # Mail server\r
 [String]$emailMailserver = "localhost"\r
 # SMTP Port\r
@@ -70,21 +75,47 @@ function Send-Email([String]$body) {
        Write-Host "Sending email: $emailToAddress, $body"\r
        \r
        try {\r
-               $smtp = New-Object System.Net.Mail.SmtpClient($emailMailServer, $emailPort);\r
+               $smtp = New-Object System.Net.Mail.SmtpClient($emailMailServer, $emailPort)\r
 \r
                $smtp.EnableSSL = $emailSSL\r
 \r
                if($emailAuth) {\r
-                       $smtp.Credentials = New-Object System.Net.NetworkCredential($emailUser, $emailPassword);\r
+                       $smtp.Credentials = New-Object System.Net.NetworkCredential($emailUser, $emailPassword)\r
                }\r
 \r
-               $smtp.Send($emailFromAddress, $emailToAddress, $emailSubject, $body);\r
+               $smtp.Send($emailFromAddress, $emailToAddress, $emailSubject, $body)\r
        }\r
        catch {\r
                Write-Host "Could not send email: $_.Exception.Message"\r
        }\r
 }\r
 \r
+function Rotate-Backup {\r
+       if($keepMonths -lt 0) {\r
+               return\r
+       }\r
+\r
+       Write-Host "Rotating"\r
+       \r
+       $keepMonthsCount = $keepMonths\r
+       \r
+       Get-ChildItem $backupDir -Directory | Where-Object {($_.Name -ne $currMonth) -and ($_.Name -match "^\d{4,}-\d{2}$")} | Sort-Object -Descending |\r
+       Foreach-Object {\r
+               Write-Host $_ "=>" $_.FullName\r
+               \r
+               if($keepMonthsCount -ge 0) {\r
+                       $keepMonthsCount--\r
+               }\r
+               \r
+               if($keepMonthsCount -eq -1) {\r
+                       Write-Host "Deleting $_"\r
+                       Remove-Item -Recurse -Force $_.FullName\r
+               }\r
+       }\r
+}\r
+\r
+Write-Host "Started at" (Get-Date -format "yyyy-MM-dd HH:mm:ss")\r
+\r
 $errorMessages = @()\r
 \r
 $smbConnected = $False\r
@@ -111,7 +142,7 @@ if($smbDrive) {
                $smbConnected = $True\r
        }\r
        Catch {\r
-               Write-Host "Could not connect to network drive: $_.Exception.Message"\r
+               Write-Host "Could not connect to network drive $smbDrive`: $_.Exception.Message"\r
                exit\r
        }\r
 }\r
@@ -161,18 +192,21 @@ if((Test-Path $backupTarget) -and (Test-Path $backupTargetFull) -and (Test-Path
                & $dsPath $dsArgs\r
                \r
                if($LastExitCode -ne 0) {\r
-                       Write-Host "Error code: $LastExitCode"\r
+                       Write-Host "Drive Snapshot failed to backup! Exit code: $LastExitCode"\r
+                       $errorMessages += "Drive Snapshot failed to backup! Exit code: $LastExitCode"\r
+               }\r
+               else {\r
+                       $success = $True\r
                }\r
        }\r
        else {\r
                Write-Host "Directory $backupTargetDiff already exists!"\r
-               \r
                $errorMessages += "Directory $backupTargetDiff already exists!"\r
        }\r
 }\r
 else {\r
        Write-Host "Full backup"\r
-       \r
+\r
        if(!(Test-Path $backupTarget)) {\r
                Write-Host "Creating directory $backupTarget"\r
                New-Item -ItemType directory -Path $backupTarget\r
@@ -183,6 +217,10 @@ else {
                New-Item -ItemType directory -Path $backupTargetFull\r
        }\r
        \r
+       if($rotateBeforeBackup) {\r
+               Rotate-Backup\r
+       }\r
+       \r
        $dsLogPath = if($dsLogFileToBackup) { "$backupTargetFull\$dsLogFile" } else { $dsLogFile }\r
 \r
        $dsArgs = @($disksToBackup, "--logfile:$dsLogPath", "$backupTargetFull\`$disk.sna") + $dsAdditionalArgs\r
@@ -191,39 +229,32 @@ else {
        & $dsPath $dsArgs\r
        \r
        if($LastExitCode -ne 0) {\r
-               Write-Host "Error code: $LastExitCode"\r
+               Write-Host "Drive Snapshot failed to backup! Exit code: $LastExitCode"\r
+               $errorMessages += "Drive Snapshot failed to backup! Exit code: $LastExitCode"\r
+       }\r
+       else {\r
+               $success = $True\r
        }\r
        \r
-       $success = $False\r
-}\r
-\r
-if($isDiff -eq $False -and $success -eq $True -and $keepMonths -ge 0) {\r
-       Write-Host "Rotating"\r
-       \r
-       $keepMonthsCount = $keepMonths\r
-       \r
-       Get-ChildItem $backupDir -Directory | Where-Object {($_.Name -ne $currMonth) -and ($_.Name -match "^\d{4,}-\d{2}$")} | Sort-Object -Descending |\r
-       Foreach-Object {\r
-               Write-Host $_ "=>" $_.FullName\r
-               \r
-               if($keepMonthsCount -ge 0) {\r
-                       $keepMonthsCount--\r
-               }\r
-               \r
-               Write-Host $keepMonthsCount\r
-               \r
-               if($keepMonthsCount -eq -1) {\r
-                       Write-Host "Deleting $_"\r
-                       Remove-Item -Recurse -Force $_.FullName\r
-               }\r
+       if($rotateBeforeBackup -eq $False -and $success -eq $True) {\r
+               Rotate-Backup\r
        }\r
 }\r
 \r
 if($smbConnected) {\r
        Write-Host "Disconnecting network drive"\r
-       Remove-PSDrive $smbDrive\r
+       \r
+       try {\r
+               Remove-PSDrive $smbDrive -ErrorAction Stop\r
+       }\r
+       catch {\r
+               Write-Host "Could not disconnect network drive $smbDrive`: $_.Exception.Message"\r
+               $errorMessages +=  "Could not disconnect network drive $smbDrive`: $_.Exception.Message"\r
+       }\r
 }\r
 \r
 if($emailOnError -and $errorMessages.Count -gt 0) {\r
        Send-Email ("Error:`n"+($errorMessages -join "`n"))\r
-}
\ No newline at end of file
+}\r
+\r
+Write-Host "Ended at" (Get-Date -format "yyyy-MM-dd HH:mm:ss")
\ No newline at end of file