Merge pull request #286 from blocknative/enhancement/trust-balance

Enhancement: Trust Balance.

Closes #285

Author: lnbc1QWFyb24

src/modules/select/wallets/trust.ts

@@ -3,24 +3,62 @@ import { WalletModule, Helpers, CommonWalletOptions } from '../../../interfaces'
 
 import trustIcon from '../wallet-icons/icon-trust'
 
-function trust (options: CommonWalletOptions): WalletModule {
-  const { preferred, label, iconSrc, svg } = options
+function trust(
+  options: CommonWalletOptions & { rpcUrl: string }
+): WalletModule {
+  const { preferred, label, iconSrc, svg, rpcUrl } = options
 
   return {
     name: label || 'Trust',
     svg: svg || trustIcon,
     iconSrc,
     wallet: async (helpers: Helpers) => {
-      const { getProviderName, createLegacyProviderInterface } = helpers
-      const provider =
+      const { getProviderName, getAddress, getNetwork, getBalance } = helpers
+      const trustProvider =
         (window as any).web3 && (window as any).web3.currentProvider
 
+      const isTrust = getProviderName(trustProvider) === 'Trust'
+      let createProvider
+
+      if (isTrust && rpcUrl) {
+        createProvider = (await import('./providerEngine')).default
+      }
+
+      const provider = createProvider ? createProvider({ rpcUrl }) : null
+
+      let warned = false
+
       return {
-        provider,
-        interface:
-          provider && getProviderName(provider) === 'Trust'
-            ? createLegacyProviderInterface(provider)
-            : null
+        provider: trustProvider,
+        interface: isTrust
+          ? {
+              address: {
+                get: () => getAddress(trustProvider)
+              },
+              network: {
+                get: () => getNetwork(trustProvider)
+              },
+              balance: {
+                get: async () => {
+                  if (!provider) {
+                    if (!warned) {
+                      console.warn(
+                        'The Trust provider does not allow rpc calls preventing Onboard.js from getting the balance. You can pass in a "rpcUrl" to the Trust wallet initialization object to get the balance.'
+                      )
+                      warned = true
+                    }
+
+                    return null
+                  }
+
+                  const address = await getAddress(trustProvider)
+
+                  return getBalance(provider, address)
+                }
+              },
+              name: getProviderName(trustProvider)
+            }
+          : null
       }
     },
     type: 'injected',