Merge pull request #50 from blocknative/enhancement/subscription-init…

…ial-state Subscription Initial State
blocknative · Nov 5, 2019 · 8ec8417 · 8ec8417
2 parents 0e92b8c + 1e6097c
commit 8ec8417
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 10 deletions.
diff --git a/src/@types/libraries.d.ts b/src/@types/libraries.d.ts
@@ -1,7 +1,7 @@
 declare module "promise-cancelable" {
   export default class Cancelable extends Promise<any> {
     constructor(executor: any)
-    cancel(): any
+    cancel(func: () => void): any
     isFulfilled(): any
     isResolved(): any
     isRejected(): any

diff --git a/src/onboard.ts b/src/onboard.ts
@@ -56,7 +56,9 @@ function init(initialization: Initialization): API {
   if (subscriptions) {
     if (subscriptions.address) {
       address.subscribe((address: string | null) => {
-        address && subscriptions.address(address)
+        if (address !== null) {
+          subscriptions.address(address)
+        }
       })
     }
 
@@ -68,7 +70,9 @@ function init(initialization: Initialization): API {
 
     if (subscriptions.balance) {
       balance.subscribe((balance: string) => {
-        balance && subscriptions.balance(balance)
+        if (balance !== null) {
+          subscriptions.balance(balance)
+        }
       })
     }
 

diff --git a/src/stores.ts b/src/stores.ts
@@ -41,7 +41,7 @@ export const network: WalletStateSliceStore = createWalletStateSliceStore({
   parameter: "network",
   initialState: null
 })
-export const balance: BalanceStore = createBalanceStore("")
+export const balance: BalanceStore = createBalanceStore(null)
 export const wallet: WritableStore = writable({
   name: null,
   provider: null,
@@ -106,14 +106,14 @@ function createWalletInterfaceStore(
 
 function createWalletStateSliceStore(options: {
   parameter: string
-  initialState: string | number | null
+  initialState: string | number | null | undefined
 }): WalletStateSliceStore {
   const { parameter, initialState } = options
   const { subscribe, set } = writable(initialState)
 
   return {
     subscribe,
-    reset: () => set(initialState),
+    reset: () => set(undefined),
     setStateSyncer: (stateSyncer: StateSyncer) => {
       validateType({ name: "stateSyncer", value: stateSyncer, type: "object" })
 
@@ -155,7 +155,7 @@ function createWalletStateSliceStore(options: {
   }
 }
 
-function createBalanceStore(initialState: string): BalanceStore {
+function createBalanceStore(initialState: string | null): BalanceStore {
   let stateSyncer: StateSyncer
   let emitter
 
@@ -181,7 +181,7 @@ function createBalanceStore(initialState: string): BalanceStore {
         emitter.on("all", () => false)
       } else {
         // no address, so set balance back to null
-        set && set(initialState)
+        set && set(undefined)
       }
     }
   })
@@ -234,7 +234,7 @@ function syncStateWithTimeout(options: {
             "There was a problem getting the balance of this wallet"
         })
       }
-    )
+    ).catch(() => {})
   )
 
   balanceSyncStatus.syncing = prom
@@ -251,7 +251,7 @@ function syncStateWithTimeout(options: {
 
   timedOut.then(() => {
     if (!prom.isFulfilled()) {
-      prom.cancel()
+      prom.cancel(() => {})
     }
   })
 }