Update login script to only login when the JWT expires

NASA-AMMOS · Nov 22, 2023 · bddc103 · bddc103
1 parent 7625619
commit bddc103
Showing 1 changed file with 38 additions and 37 deletions.
diff --git a/src/packages/api-playground/api-playground.ts b/src/packages/api-playground/api-playground.ts
@@ -7,48 +7,49 @@ export default (app: Express) => {
   const initialQuery = '{ plan { id name } }';
   const initialPreRequestScript =
   `
-  // Fetch a new token from the Gateway
-  const res = await altair.helpers.request(
-    'POST',
-    '/auth/login', // AUTH ENDPOINT OF THE DEPLOYMENT
-  {
-    body: { "username": "<YOUR_AERIE_USERNAME>", "password": "<YOUR_AERIE_PASSWORD>"}, // CREDENTIALS TO LOG IN AS
-    headers: {"Content-Type": "application/json"}
-  });
-  if(res.success) {
-    const token = res.token;
-    await altair.helpers.setEnvironment("user", token);
-  } else {
-    altair.log(res);
-  }`;
+  const nowInSeconds = () => Date.now() / 1000;
+  const tokenExpiry = await altair.storage.get("token_exp") || 0;
+
+  if (nowInSeconds() >= Number(tokenExpiry)) {
+    // Fetch a new token from the Gateway
+    const res = await altair.helpers.request(
+      'POST',
+      '/auth/login', // AUTH ENDPOINT OF THE DEPLOYMENT
+      {
+        body: { "username": "<YOUR_AERIE_USERNAME>", "password": "<YOUR_AERIE_PASSWORD>"}, // CREDENTIALS TO LOG IN AS
+        headers: {"Content-Type": "application/json"}
+      });
+    if(res.success) {
+      const token = res.token;
+      await altair.storage.set("token", token);
+      // Set JWT expiry
+      const atob = await altair.importModule('atob');
+      const body = JSON.parse(atob(token.split('.')[1]));
+      await altair.storage.set("token_exp", body.exp);
+    } else { altair.log(res); }
+  }
+  // Set the token in the environment
+  const token = await altair.storage.get("token");
+  altair.helpers.setEnvironment('user', token);`;
+
   const initialHeaders = {"Authorization": "Bearer {{user}}", "x-hasura-role": "viewer"};
   const initialSettings = {
-      addQueryDepthLimit: 5,
-      enableExperimental: true,
-      "plugin.list": [
-          "altair-graphql-plugin-graphql-explorer"
-      ],
-      "request.withCredentials": true,
-      "schema.reloadOnStart": true,
-      "script.allowedCookies": [
-          "user"
-      ],
-      tabSize: 2,
-      theme: "dracula",
+    addQueryDepthLimit: 5,
+    enableExperimental: true,
+    "plugin.list": [
+      "altair-graphql-plugin-graphql-explorer"
+    ],
+    "request.withCredentials": true,
+    "schema.reloadOnStart": true,
+    "script.allowedCookies": [
+      "user"
+    ],
+    tabSize: 2,
+    theme: "dracula",
   }
 
-  /*
-          'initialEnvironments',
-          'instanceStorageNamespace',
-          'initialSettings',
-          'initialSubscriptionsProvider',
-          'initialSubscriptionsPayload',
-          'preserveState',
-          'initialHttpMethod',
-          'initialWindows',
-          'disableAccount',
-   */
   app.use('/api-playground', altairExpress({
+      disableAccount:true,
       endpointURL,
       initialHeaders,
       initialPreRequestScript,