fix: clear extension storage when disconnecting the wallet (#174)

* Set storage values to null instead of undefined

* Fix types

src/background/services/storage.ts

@@ -12,10 +12,10 @@ const defaultStorage = {
   connected: false,
   enabled: true,
   exceptionList: {},
-  walletAddress: undefined,
-  amount: undefined,
-  token: undefined,
-  grant: undefined
+  walletAddress: null,
+  amount: null,
+  token: null,
+  grant: null,
 } satisfies Omit<Storage, 'publicKey' | 'privateKey' | 'keyId'>
 
 export class StorageService {

src/popup/lib/context.tsx

@@ -1,13 +1,13 @@
 import React from 'react'
 import { getContextData } from '@/popup/lib/messages'
-import { PopupStore } from '@/shared/types'
+import { DeepNonNullable, PopupStore } from '@/shared/types'
 
 export enum ReducerActionType {
   SET_DATA = 'SET_DATA',
   TOGGLE_WM = 'TOGGLE_WM'
 }
 
-export type PopupState = Required<NonNullable<PopupStore>>
+export type PopupState = Required<DeepNonNullable<PopupStore>>
 
 export interface PopupContext {
   state: Required<NonNullable<PopupState>>

src/shared/types.ts

@@ -34,13 +34,13 @@ export interface Storage {
   /** If a wallet is connected or not */
   connected: boolean
   /** User wallet address information */
-  walletAddress?: WalletAddress | undefined
+  walletAddress?: WalletAddress | undefined | null
   /** Overall amount */
-  amount?: WalletAmount | undefined
+  amount?: WalletAmount | undefined | null
   /** Access token for quoting & outgoing payments  */
-  token?: AccessToken | undefined
+  token?: AccessToken | undefined | null
   /** Grant details - continue access token & uri for canceling the grant */
-  grant?: GrantDetails | undefined
+  grant?: GrantDetails | undefined | null
   /** Exception list with websites and each specific amount */
   exceptionList: {
     [website: string]: Amount
@@ -58,3 +58,8 @@ export type PopupStore = Omit<
 > & {
   website: WebsiteData
 }
+
+export type DeepNonNullable<T> = {
+        [P in keyof T]?: NonNullable<T[P]>;
+}
+