🐛 Handle rejection in WebGL wallet adapter (#167)

* 🐛 Handle rejection in WebGL wallet adapter

* ⚡ Remove websocket dependency

Author: GabrielePicco

Runtime/Plugins/SolanaWalletAdapterWebGL/SolanaWalletAdapterWebGL.jslib

@@ -24,6 +24,7 @@ mergeInto(LibraryManager.library, {
             Module.dynCall_vi(callback, walletsPtr);
         } catch (err) {
             console.error(err.message);
+            Module.dynCall_vi(callback, null);
         }
     },
     ExternConnectWallet: async function (walletNamePtr, callback) {
@@ -41,6 +42,7 @@ mergeInto(LibraryManager.library, {
                 Module.dynCall_vi(callback, pubKeyPtr);
          } catch (err) {
             console.error(err.message);
+            Module.dynCall_vi(callback, null);
          }
     },
     ExternSignTransactionWallet: async function (walletNamePtr, transactionPtr, callback) {
@@ -61,6 +63,7 @@ mergeInto(LibraryManager.library, {
                 Module.dynCall_vi(callback, txPtr);          
          } catch (err) {
             console.error(err.message);
+            Module.dynCall_vi(callback, null);
          }
     },
     ExternSignMessageWallet: async function (walletNamePtr, messagePtr, callback) {
@@ -82,6 +85,7 @@ mergeInto(LibraryManager.library, {
                 Module.dynCall_vi(callback, signaturePtr);          
          } catch (err) {
             console.error(err.message);
+            Module.dynCall_vi(callback, null);
          }
     },
      ExternSignAllTransactionsWallet: async function (walletNamePtr, transactionsPtr, callback) {
@@ -113,6 +117,7 @@ mergeInto(LibraryManager.library, {
                 Module.dynCall_vi(callback, txsPtr);
         } catch (err) {
             console.error(err.message);
+            Module.dynCall_vi(callback, null);
         }
     },
 } );

Runtime/codebase/SolanaWalletAdapterWebGL/SolanaWalletAdapterWebGL.cs

@@ -191,9 +191,15 @@ protected override Task<Transaction[]> _SignAllTransactions(Transaction[] transa
         [MonoPInvokeCallback(typeof(Action<string>))]
         private static void OnWalletConnected(string walletPubKey)
         {
+            if (walletPubKey == null)
+            {
+                _loginTaskCompletionSource.TrySetException(new Exception("Login cancelled"));
+                _loginTaskCompletionSource.TrySetResult(null);
+                return;
+            }
             Debug.Log($"Wallet {walletPubKey} connected!");
             _account = new Account("", walletPubKey);
-            _loginTaskCompletionSource.SetResult(_account);
+            _loginTaskCompletionSource.TrySetResult(_account);
         }
 
         /// <summary>
@@ -203,6 +209,12 @@ private static void OnWalletConnected(string walletPubKey)
         [MonoPInvokeCallback(typeof(Action<string>))]
         public static void OnTransactionSigned(string transaction)
         {
+            if (transaction == null)
+            {
+                _signedTransactionTaskCompletionSource.TrySetException(new Exception("Transaction signing cancelled"));
+                _signedTransactionTaskCompletionSource.TrySetResult(null);
+                return;
+            }
             var tx = Transaction.Deserialize(Convert.FromBase64String(transaction));
             _signedTransactionTaskCompletionSource.SetResult(tx);
         }
@@ -214,6 +226,12 @@ public static void OnTransactionSigned(string transaction)
         [MonoPInvokeCallback(typeof(Action<string>))]
         public static void OnAllTransactionsSigned(string signatures)
         {
+            if (signatures == null)
+            {
+                _signedAllTransactionsTaskCompletionSource.TrySetException(new Exception("Transactions signing cancelled"));
+                _signedAllTransactionsTaskCompletionSource.TrySetResult(null);
+                return;
+            }
             string[] signaturesList = signatures.Split(',');
             for (int i = 0; i < signaturesList.Length; i++)
             {
@@ -234,6 +252,12 @@ public static void OnAllTransactionsSigned(string signatures)
         [MonoPInvokeCallback(typeof(Action<string>))]
         public static void OnMessageSigned(string signature)
         {
+            if (signature == null)
+            {
+                _signedMessageTaskCompletionSource.TrySetException(new Exception("Message signing cancelled"));
+                _signedMessageTaskCompletionSource.TrySetResult(null);
+                return;
+            }
             _signedMessageTaskCompletionSource.SetResult(Convert.FromBase64String(signature));
         }