Session class - set StartTime and TimeOfDeath in SessionRefresh method,

fixes #253
thycotic-ps · Nov 17, 2021 · 2b0d11f · 2b0d11f
1 parent 4196aa4
commit 2b0d11f
Showing 1 changed file with 38 additions and 36 deletions.
diff --git a/src/Thycotic.SecretServer/classes/authentication/Session.cs b/src/Thycotic.SecretServer/classes/authentication/Session.cs
@@ -10,42 +10,6 @@
 
 namespace Thycotic.PowerShell.Authentication
 {
-    public class Request
-    {
-        public static IRestResponse AccessToken(string SecretServerHost, string Username, string Password, string ProxyServer, int Timeout = 0)
-        {
-            var client = new RestClient(SecretServerHost + "/oauth2/token");
-            client.Timeout = Timeout;
-            if (string.IsNullOrEmpty(ProxyServer))
-            {
-                client.Proxy = new WebProxy(ProxyServer);
-            }
-            var request = new RestRequest(Method.POST);
-            request.AddHeader("Content-Type", "application/x-www-form-urlencoded");
-            request.AddParameter("username", Username);
-            request.AddParameter("password", Password);
-            request.AddParameter("grant_type", "password");
-            IRestResponse response = client.Execute(request);
-            return response;
-        }
-
-        public static IRestResponse RefreshToken(string SecretServerHost, string TokenValue, string ProxyServer, int Timeout = 0)
-        {
-            var client = new RestClient(SecretServerHost + "/oauth2/token");
-            client.Timeout = Timeout;
-            if (string.IsNullOrEmpty(ProxyServer))
-            {
-                client.Proxy = new WebProxy(ProxyServer);
-            }
-            var request = new RestRequest(Method.POST);
-            request.AddHeader("Content-Type", "application/x-www-form-urlencoded");
-            request.AddParameter("refresh_token", TokenValue);
-            request.AddParameter("grant_type", "refresh_token");
-            IRestResponse response = client.Execute(request);
-            return response;
-        }
-    }
-
     public class Session
     {
         public string SecretServer { get; set; }
@@ -151,6 +115,8 @@ public bool SessionRefresh()
                 this.RefreshToken = jsonObj.refresh_token;
                 this.ExpiresIn = jsonObj.expires_in;
                 this.TokenType = jsonObj.token_type;
+                this.StartTime = DateTime.Now;
+                this.TimeOfDeath = DateTime.Now.Add(TimeSpan.FromSeconds(jsonObj.expires_in));
                 return true;
             }
             catch
@@ -167,4 +133,40 @@ private class ApiTokenResponse
             public int expires_in { get; set; }
         }
     }
+    public class Request
+    {
+        public static IRestResponse AccessToken(string SecretServerHost, string Username, string Password, string ProxyServer, int Timeout = 0)
+        {
+            var client = new RestClient(SecretServerHost + "/oauth2/token");
+            client.Timeout = Timeout;
+            if (string.IsNullOrEmpty(ProxyServer))
+            {
+                client.Proxy = new WebProxy(ProxyServer);
+            }
+            var request = new RestRequest(Method.POST);
+            request.AddHeader("Content-Type", "application/x-www-form-urlencoded");
+            request.AddParameter("username", Username);
+            request.AddParameter("password", Password);
+            request.AddParameter("grant_type", "password");
+            IRestResponse response = client.Execute(request);
+            return response;
+        }
+
+        public static IRestResponse RefreshToken(string SecretServerHost, string TokenValue, string ProxyServer, int Timeout = 0)
+        {
+            var client = new RestClient(SecretServerHost + "/oauth2/token");
+            client.Timeout = Timeout;
+            if (string.IsNullOrEmpty(ProxyServer))
+            {
+                client.Proxy = new WebProxy(ProxyServer);
+            }
+            var request = new RestRequest(Method.POST);
+            request.AddHeader("Content-Type", "application/x-www-form-urlencoded");
+            request.AddParameter("refresh_token", TokenValue);
+            request.AddParameter("grant_type", "refresh_token");
+            IRestResponse response = client.Execute(request);
+            return response;
+        }
+    }
+
 }