Add extra wallet check customization options

Author: lnbc1QWFyb24

src/interfaces.ts

@@ -249,11 +249,20 @@ export type AllWalletInitOptions = CommonWalletOptions &
   ImTokenOptions &
   TrustWalletOptions & { networkId: number }
 
-export interface WalletCheckInit {
-  checkName: string
+export interface WalletCheckCustomOptions {
   heading?: string
   description?: string
   minimumBalance?: string
+  icon?: string
+  button?: {
+    text: string
+    onclick: () => void
+  }
+  html?: string
+}
+
+export interface WalletCheckInit extends WalletCheckCustomOptions {
+  checkName: string
 }
 
 export interface WalletSelectFunction {

src/modules/check/balance.ts

@@ -1,24 +1,16 @@
-import { WalletCheckModal, StateAndHelpers } from '../../interfaces'
-import { validateType } from '../../validation'
+import {
+  WalletCheckModal,
+  StateAndHelpers,
+  WalletCheckCustomOptions
+} from '../../interfaces'
 import { balanceIcon } from './icons'
 
 function balance(
-  options: {
-    minimumBalance: string
-    heading?: string
-    description?: string
-    icon?: string
-  } = { minimumBalance: '0' }
+  options: WalletCheckCustomOptions & { minimumBalance: string } = {
+    minimumBalance: '0'
+  }
 ): (currentState: StateAndHelpers) => Promise<WalletCheckModal | undefined> {
-  validateType({ name: 'balance options', value: options, type: 'object' })
-
-  const { minimumBalance, heading, description, icon } = options
-
-  validateType({
-    name: 'minimumBalance',
-    value: minimumBalance,
-    type: 'string'
-  })
+  const { minimumBalance, heading, description, icon, html, button } = options
 
   return async (StateAndHelpers: StateAndHelpers) => {
     const { balance, BigNumber, stateSyncStatus, stateStore } = StateAndHelpers
@@ -44,7 +36,9 @@ function balance(
             .div(BigNumber('1000000000000000000'))
             .toString(10)} ETH.`,
         eventCode: 'nsfFail',
-        icon: icon || balanceIcon
+        icon: icon || balanceIcon,
+        html,
+        button
       }
     }
   }

src/modules/check/connect.ts

@@ -1,18 +1,13 @@
 import {
   WalletCheckModal,
   StateAndHelpers,
-  WalletCheckModule
+  WalletCheckModule,
+  WalletCheckCustomOptions
 } from '../../interfaces'
 import { connectIcon } from './icons'
 
-function connect(
-  options: {
-    heading?: string
-    description?: string
-    icon?: string
-  } = {}
-): WalletCheckModule {
-  const { heading, description, icon } = options
+function connect(options: WalletCheckCustomOptions = {}): WalletCheckModule {
+  const { heading, description, icon, html, button } = options
 
   return async (
     stateAndHelpers: StateAndHelpers
@@ -42,7 +37,9 @@ function connect(
           `This dapp requires access to your wallet, please login and authorize access to your ${wallet.name} accounts to continue.`,
         eventCode: 'loginFail',
         action: wallet.connect,
-        icon: icon || connectIcon
+        icon: icon || connectIcon,
+        html,
+        button
       }
     }
   }

src/modules/check/network.ts

@@ -1,15 +1,15 @@
 import { networkName } from '../../utilities'
-import { WalletCheckModal, StateAndHelpers } from '../../interfaces'
+import {
+  WalletCheckModal,
+  StateAndHelpers,
+  WalletCheckCustomOptions
+} from '../../interfaces'
 import { networkIcon } from './icons'
 
 function network(
-  options: {
-    heading?: string
-    description?: string
-    icon?: string
-  } = {}
+  options: WalletCheckCustomOptions = {}
 ): (currentState: StateAndHelpers) => Promise<WalletCheckModal | undefined> {
-  const { heading, description, icon } = options
+  const { heading, description, icon, html, button } = options
 
   return async (stateAndHelpers: StateAndHelpers) => {
     const {
@@ -48,13 +48,14 @@ function network(
             appNetworkId
           )} network</b> for this Dapp. <br><br> <i style="font-size: inherit; font-family: inherit;">*Some wallets may not support changing networks. If you can not change networks in your wallet you may consider switching to a different wallet.</i>`,
         eventCode: 'networkFail',
-        button: {
+        button: button || {
           onclick: () => {
             exit()
             walletSelect()
           },
           text: 'Switch Wallet'
         },
+        html,
         icon: icon || networkIcon
       }
     }

src/validation.ts

@@ -342,12 +342,23 @@ function validateWalletCheck(
         heading,
         description,
         minimumBalance,
+        html,
+        icon,
+        button,
         ...otherParams
       } = check
 
       invalidParams(
         otherParams,
-        ['checkName', 'heading', 'description', 'minimumBalance'],
+        [
+          'checkName',
+          'heading',
+          'description',
+          'html',
+          'icon',
+          'button',
+          'minimumBalance'
+        ],
         'walletCheck item'
       )
 
@@ -367,6 +378,27 @@ function validateWalletCheck(
         optional: true
       })
 
+      validateType({
+        name: 'html',
+        value: html,
+        type: 'string',
+        optional: true
+      })
+
+      validateType({
+        name: 'icon',
+        value: icon,
+        type: 'string',
+        optional: true
+      })
+
+      validateType({
+        name: 'button',
+        value: button,
+        type: 'object',
+        optional: true
+      })
+
       validateType({
         name: 'minimumBalance',
         value: minimumBalance,
@@ -543,33 +575,6 @@ export function validateWalletInterface(
   })
 }
 
-export function validateWalletCheckInit(
-  walletCheckInit: WalletCheckInit[]
-): void | never {
-  validateType({
-    name: 'walletCheckInit',
-    value: walletCheckInit,
-    type: 'array'
-  })
-
-  walletCheckInit.forEach(init => {
-    validateType({ name: 'walletCheckInit', value: init, type: 'object' })
-
-    validateType({
-      name: 'walletCheckInit.checkName',
-      value: init.checkName,
-      type: 'string'
-    })
-
-    validateType({
-      name: 'walletCheckInit.minimumBalance',
-      value: init.minimumBalance,
-      type: 'string',
-      optional: true
-    })
-  })
-}
-
 export function validateWalletInit(
   walletInit: WalletInitOptions
 ): void | never {