fix: optimize.

dotbitHQ · Apr 12, 2024 · 79baaea · 79baaea
1 parent 5e80f00
commit 79baaea
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/src/lib/wallets/WalletConnectorHandler/UnisatConnector.ts b/src/lib/wallets/WalletConnectorHandler/UnisatConnector.ts
@@ -42,12 +42,8 @@ export class UnisatConnector extends WalletConnector {
   }
 
   async switchNetwork(chainId: string): Promise<void> {
-    try {
-      const { provider } = this.context
-      await provider.switchNetwork(chainId)
-    } catch (err) {
-      console.error(err)
-    }
+    const { provider } = this.context
+    await provider.switchNetwork(chainId)
   }
 
   async signData(data: SignDataType): Promise<string | undefined> {

diff --git a/src/lib/wallets/WalletEventListenerHandler/TronLinkEventListener.ts b/src/lib/wallets/WalletEventListenerHandler/TronLinkEventListener.ts
@@ -26,6 +26,7 @@ export class TronLinkEventListener extends WalletEventListener {
       const isAuth = event.data?.message?.data?.data?.isAuth
 
       if (!_chainId || !isAuth || eventTabReplyDebounce) {
+        // shim disconnect
         if (_chainId && isAuth === false) {
           this.removeEvents()
           this.context.address = undefined
@@ -37,6 +38,7 @@ export class TronLinkEventListener extends WalletEventListener {
         return
       }
 
+      // switch network check
       eventTabReplyDebounce = true
       globalThis.setTimeout(() => {
         eventTabReplyDebounce = false
@@ -129,6 +131,7 @@ export class TronLinkEventListener extends WalletEventListener {
     }
 
     if (event.data?.message?.action === 'disconnect') {
+      // shim disconnect
       this.removeEvents()
       this.context.address = undefined
       this.context.chainId = undefined

diff --git a/src/lib/wallets/index.ts b/src/lib/wallets/index.ts
@@ -180,7 +180,7 @@ class WalletSDK {
 
   async sendTransaction(data: ISendTrxParams): Promise<string | undefined> {
     const isInit = await this.initWallet()
-    if (!isInit) {
+    if (!isInit && !this.walletTransaction) {
       // eslint-disable-next-line lingui/no-unlocalized-strings
       throw new CustomError(errno.failedToInitializeWallet, 'sendTransaction: Please initialize wallet first')
     }