waas: use getLocalTime instead of Date.now() (#614)

Author: patrislav

packages/waas/src/auth.ts

@@ -18,7 +18,8 @@ import {
   SignedIntent,
   SignMessageArgs,
   getTimeDrift,
-  updateTimeDrift
+  updateTimeDrift,
+  getLocalTime
 } from './intents'
 import {
   FeeOptionsResponse,
@@ -609,6 +610,7 @@ export class SequenceWaaS {
       session.clear()
       await this.waas.completeSignOut()
       await this.deviceName.set(undefined)
+      updateTimeDrift(undefined)
     }
   }
 
@@ -672,9 +674,9 @@ export class SequenceWaaS {
   }
 
   async waitForSessionValid(timeout: number = 600000, pollRate: number = 2000) {
-    const start = Date.now()
+    const start = getLocalTime()
 
-    while (Date.now() - start < timeout) {
+    while (getLocalTime() - start < timeout) {
       if (await this.isSessionValid()) {
         return true
       }

packages/waas/src/intents/base.ts

@@ -18,6 +18,10 @@ function isSessionStorageAvailable() {
   return typeof window === 'object' && typeof window.sessionStorage === 'object'
 }
 
+export function getLocalTime() {
+  return new Date().getTime()
+}
+
 export function getTimeDrift() {
   if (isSessionStorageAvailable()) {
     const drift = window.sessionStorage.getItem(timeDriftKey)
@@ -28,16 +32,24 @@ export function getTimeDrift() {
   return timeDrift
 }
 
-export function updateTimeDrift(serverTime: Date) {
-  timeDrift = (Date.now() - serverTime.getTime()) / 1000
+export function updateTimeDrift(serverTime?: Date) {
+  if (!serverTime) {
+    timeDrift = undefined
+    if (isSessionStorageAvailable()) {
+      window.sessionStorage.removeItem(timeDriftKey)
+    }
+    return
+  }
+
+  timeDrift = (getLocalTime() - serverTime.getTime()) / 1000
   if (isSessionStorageAvailable()) {
     window.sessionStorage.setItem(timeDriftKey, timeDrift.toString(10))
   }
 }
 
 export function makeIntent<T>(name: IntentName, lifespan: number, data: T): Intent<T> {
   const drift = Math.abs(Math.floor(getTimeDrift() || 0))
-  const issuedAt = Math.floor(Date.now() / 1000 - drift)
+  const issuedAt = Math.floor(getLocalTime() / 1000 - drift)
   const expiresAt = issuedAt + lifespan + 2 * drift
   return {
     version: VERSION,

packages/waas/src/intents/utils.ts

@@ -1,5 +1,7 @@
+import { getLocalTime } from "./base"
+
 export function useLifespan(lifespan: number) {
-  const issuedAt = Math.floor(Date.now() / 1000)
+  const issuedAt = Math.floor(getLocalTime() / 1000)
   return {
     issuedAt,
     expiresAt: issuedAt + lifespan