Added TangoUpdate to Full Restore on PPC.

1 files changed, 30 insertions, 0 deletions

diff --git a/Software/Visual_Studio/PPC/Tango.PPC.Common/BackupRestore/DefaultBackupManager.cs b/Software/Visual_Studio/PPC/Tango.PPC.Common/BackupRestore/DefaultBackupManager.cs
index d32df734d..1462e1490 100644
--- a/Software/Visual_Studio/PPC/Tango.PPC.Common/BackupRestore/DefaultBackupManager.cs
+++ b/Software/Visual_Studio/PPC/Tango.PPC.Common/BackupRestore/DefaultBackupManager.cs
@@ -20,6 +20,7 @@ using Tango.BL.Entities;
 using Tango.PPC.Common.Authentication;
 using Tango.Integration.Upgrade;
 using Tango.Core.IO;
+using Tango.BL.Enumerations;
 
 namespace Tango.PPC.Common.BackupRestore
 {
@@ -524,6 +525,7 @@ namespace Tango.PPC.Common.BackupRestore
                                 OnProgress(BackupRestoreStage.RestoringFirmware, 100, 100, false);
                                 stream.Dispose();
                                 LogManager.Log("Full backup restored successfully.");
+                                UpdateTangoVersionAfterRestore(backupFile);
                                 OnProgress(BackupRestoreStage.Done, 100, 100, false);
                                 completionSource.SetResult(restoreResult);
                             };
@@ -540,6 +542,7 @@ namespace Tango.PPC.Common.BackupRestore
                         else
                         {
                             LogManager.Log("Firmware package file not found, skipping...");
+                            UpdateTangoVersionAfterRestore(backupFile);
                             OnProgress(BackupRestoreStage.Done, 100, 100, false);
                             completionSource.SetResult(restoreResult);
                         }
@@ -554,6 +557,33 @@ namespace Tango.PPC.Common.BackupRestore
             return completionSource.Task;
         }
 
+        private void UpdateTangoVersionAfterRestore(BackupFile backupFile)
+        {
+            LogManager.Log("Creating a new TangoUpdate entry after successful restore...");
+            try
+            {
+                using (ObservablesContext db = ObservablesContext.CreateDefault())
+                {
+                    db.TangoUpdates.Add(new TangoUpdate()
+                    {
+                        EndDate = DateTime.UtcNow,
+                        LastUpdated = DateTime.UtcNow,
+                        MachineGuid = _machineProvider.Machine.Guid,
+                        StartDate = DateTime.UtcNow,
+                        Status = (int)TangoUpdateStatuses.OfflineUpdateCompleted,
+                        ApplicationVersion = backupFile.ApplicationVersion,
+                        FirmwareVersion = backupFile.FirmwareVersion,
+                    });
+
+                    db.SaveChanges();
+                }
+            }
+            catch (Exception ex)
+            {
+                LogManager.Log(ex, "Error creating TangoUpdate entry...");
+            }
+        }
+
         private void OnRestoreException(Exception ex, TaskCompletionSource<RestoreResult> completionSource, bool shouldRollback, String dbRollbackFile, TemporaryFolder tempFolder)
         {
             if (shouldRollback)