1 files changed, 14 insertions, 5 deletions
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UpdateService/MachineStudioUpdateService.svc.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UpdateService/MachineStudioUpdateService.svc.cs
index 9b03cda30..4438ad188 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UpdateService/MachineStudioUpdateService.svc.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UpdateService/MachineStudioUpdateService.svc.cs
@@ -30,6 +30,8 @@ namespace Tango.MachineStudio.UpdateService
 
             public String Comments { get; set; }
 
+            public bool ForcedUpdate { get; set; }
+
             public String FilePath { get; set; }
         }
 
@@ -48,7 +50,7 @@ namespace Tango.MachineStudio.UpdateService
             {
                 CheckForUpdatesResponse response = new CheckForUpdatesResponse();
 
-                using (ObservablesContext db = ObservablesContext.CreateDefaultForWeb())
+                using (ObservablesContext db = ObservablesContext.CreateDefault(GetServerAddress()))
                 {
                     db.Configuration.LazyLoadingEnabled = false;
 
@@ -98,7 +100,7 @@ namespace Tango.MachineStudio.UpdateService
             {
                 UploadVersionResponse response = new UploadVersionResponse();
 
-                using (ObservablesContext db = ObservablesContext.CreateDefaultForWeb())
+                using (ObservablesContext db = ObservablesContext.CreateDefault(GetServerAddress()))
                 {
                     db.Configuration.LazyLoadingEnabled = false;
 
@@ -134,6 +136,7 @@ namespace Tango.MachineStudio.UpdateService
                             {
                                 UserGuid = user.Guid,
                                 Comments = request.Comments,
+                                ForcedUpdate = request.ForcedUpdate,
                                 Token = response.Token,
                                 Version = request.Version,
                                 FilePath = response.FilePath,
@@ -169,7 +172,7 @@ namespace Tango.MachineStudio.UpdateService
                 {
                     _pendingUploads.RemoveAll(x => x.Token == upload.Token);
 
-                    using (ObservablesContext db = ObservablesContext.CreateDefaultForWeb())
+                    using (ObservablesContext db = ObservablesContext.CreateDefault(GetServerAddress()))
                     {
                         db.Configuration.LazyLoadingEnabled = false;
 
@@ -178,7 +181,8 @@ namespace Tango.MachineStudio.UpdateService
                             Comments = upload.Comments,
                             FtpFilePath = upload.FilePath,
                             UserGuid = upload.UserGuid,
-                            Version = upload.Version
+                            Version = upload.Version,
+                            ForceUpdate = upload.ForcedUpdate,
                         });
 
                         db.SaveChanges();
@@ -200,7 +204,7 @@ namespace Tango.MachineStudio.UpdateService
         {
             try
             {
-                using (ObservablesContext db = ObservablesContext.CreateDefaultForWeb())
+                using (ObservablesContext db = ObservablesContext.CreateDefault(GetServerAddress()))
                 {
                     return db.MachineStudioVersions.FirstOrDefault().Version;
                 }
@@ -210,5 +214,10 @@ namespace Tango.MachineStudio.UpdateService
                 throw new FaultException(ex.ToString());
             }
         }
+
+        private String GetServerAddress()
+        {
+            return ConfigurationManager.AppSettings["ServerAddress"].ToString();
+        }
     }
 }