1 files changed, 27 insertions, 14 deletions

diff --git a/Software/Visual_Studio/Web/Tango.MachineService/Controllers/MachineStudioController.cs b/Software/Visual_Studio/Web/Tango.MachineService/Controllers/MachineStudioController.cs
index 054c8d781..29cc067d6 100644
--- a/Software/Visual_Studio/Web/Tango.MachineService/Controllers/MachineStudioController.cs
+++ b/Software/Visual_Studio/Web/Tango.MachineService/Controllers/MachineStudioController.cs
@@ -18,18 +18,28 @@ using Tango.MachineStudio.Common.Web;
 using Tango.Web.Controllers;
 using Tango.Web.Helpers;
 using Tango.Web.Storage;
-using Tango.Web.Authentication;
+using Tango.Web.Security;
 using Tango.Web.ActiveDirectory;
 using Tango.MachineService.Filters;
+using Tango.MachineService.Security;
 
 namespace Tango.MachineService.Controllers
 {
-    public class MachineStudioController : JsonController
+    public class MachineStudioController : TangoController<MachineStudioController.TokenObject>
     {
         private static List<MachineStudioPendingUpload> _pendingUploads;
         private ActiveDirectoryManager _ad_manager;
 
-        public static TokensManager<String> TokensManager { get; set; }
+        public class TokenObject
+        {
+            public String UserGuid { get; set; }
+        }
+
+        public class RefreshTokenObject
+        {
+            public String Email { get; set; }
+            public String Password { get; set; }
+        }
 
         #region Constructors
 
@@ -38,8 +48,6 @@ namespace Tango.MachineService.Controllers
         /// </summary>
         static MachineStudioController()
         {
-            TokensManager = new TokensManager<string>();
-            TokensManager.ExpirationTime = TimeSpan.FromDays(10);
             _pendingUploads = new List<MachineStudioPendingUpload>();
         }
 
@@ -61,11 +69,13 @@ namespace Tango.MachineService.Controllers
         /// <param name="request">The request.</param>
         /// <returns></returns>
         [HttpPost]
-        [MachineStudioLoginFilter]
+        [JwtTokenFilter]
         public CheckForUpdatesResponse CheckForUpdates(CheckForUpdatesRequest request)
         {
             LogManager.Log("Request received...");
 
+            var userID = RequestToken.Object.UserGuid;
+
             CheckForUpdatesResponse response = new CheckForUpdatesResponse();
 
             using (ObservablesContext db = ObservablesContextHelper.CreateContext())
@@ -82,7 +92,7 @@ namespace Tango.MachineService.Controllers
 
                 if (latestVersion != null && Version.Parse(latestVersion.Version) > currentVersion)
                 {
-                    var manager = new StorageManager();
+                    var manager = new BlobStorageManager();
                     var container = manager.GetContainer(MachineServiceConfig.MACHINE_STUDIO_VERSIONS_CONTAINER);
                     var blob = container.GetBlockBlobReference(latestVersion.BlobName);
 
@@ -103,7 +113,7 @@ namespace Tango.MachineService.Controllers
         /// <param name="request">The request.</param>
         /// <returns></returns>
         [HttpPost]
-        [MachineStudioLoginFilter]
+        [JwtTokenFilter]
         public DownloadLatestVersionResponse DownloadLatestVersion(DownloadLatestVersionRequest request)
         {
             LogManager.Log("Request received...");
@@ -118,7 +128,7 @@ namespace Tango.MachineService.Controllers
 
                 if (latestVersion != null)
                 {
-                    var manager = new StorageManager();
+                    var manager = new BlobStorageManager();
                     var container = manager.GetContainer(MachineServiceConfig.MACHINE_STUDIO_VERSIONS_CONTAINER);
                     var blob = container.GetBlockBlobReference(latestVersion.BlobName);
 
@@ -138,14 +148,14 @@ namespace Tango.MachineService.Controllers
         /// <exception cref="System.ArgumentException">New version must be greater than latest version.</exception>
         /// <exception cref="AuthenticationException">Invalid user credentials.</exception>
         [HttpPost]
-        [MachineStudioLoginFilter]
+        [JwtTokenFilter]
         public UploadVersionResponse UploadVersion(UploadVersionRequest request)
         {
             UploadVersionResponse response = new UploadVersionResponse();
 
             using (ObservablesContext db = ObservablesContextHelper.CreateContext())
             {
-                String userID = TokensManager.GetTokenObject(request.AccessToken);
+                String userID = RequestToken.Object.UserGuid;
 
                 var user = new UserBuilder(db).Set(userID).WithRolesAndPermissions().Build();
 
@@ -158,7 +168,7 @@ namespace Tango.MachineService.Controllers
                     {
                         String newVersionFileName = "Machine Studio Version" + " " + currentVersion.ToString() + ".zip";
 
-                        var manager = new StorageManager();
+                        var manager = new BlobStorageManager();
                         var container = manager.GetContainer(MachineServiceConfig.MACHINE_STUDIO_VERSIONS_CONTAINER);
                         var blob = container.CreateEmptyBlob(newVersionFileName);
 
@@ -195,7 +205,7 @@ namespace Tango.MachineService.Controllers
         /// <returns></returns>
         /// <exception cref="System.ArgumentException">Invalid Token.</exception>
         [HttpPost]
-        [MachineStudioLoginFilter]
+        [JwtTokenFilter]
         public UploadCompletedResponse NotifyUploadCompleted(UploadCompletedRequest request)
         {
             MachineStudioPendingUpload upload = _pendingUploads.FirstOrDefault(x => x.Token == request.Token);
@@ -347,7 +357,10 @@ namespace Tango.MachineService.Controllers
                     Password = request.Password,
                 },
 
-                WebToken = TokensManager.GetOrCreate(user.Guid),
+                AccessToken = WebToken<TokenObject>.CreateNew(MachineServiceConfig.JWT_TOKEN_SECRET, new TokenObject()
+                {
+                    UserGuid = user.Guid,
+                }, DateTime.UtcNow.AddDays(1)).AccessToken,
                 VersionChangeRequired = versionChangeRequired,
                 RequiredVersion = requiredVersion,
             };