feat: store saveToken

Jeff-Tian · Nov 12, 2019 · d48dc45 · d48dc45
1 parent fb2c795
commit d48dc45
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -113,21 +113,18 @@ export default class CitiOAuth {
         }
       : getToken
 
-    if (
-      !saveToken &&
-      (process.env.NODE_ENV === 'production' || process.env.NODE_ENV === 'prod')
-    ) {
+    if (!saveToken) {
       this.logger.warn(
         `Please don't save oauth token into memory under production!`
       )
-    }
 
-    if (!saveToken) {
       this.saveToken = (openid: string, token: object) => {
         this.store[openid] = token
 
         return Promise.resolve()
       }
+    } else {
+      this.saveToken = saveToken
     }
 
     this.endpoint = 'https://sandbox.apihub.citi.com/gcb/api'