Use getFees (#158)

magicblock-labs · Jul 20, 2023 · b3b779f · b3b779f
1 parent a34b308
commit b3b779f
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/Runtime/codebase/SessionWallet.cs b/Runtime/codebase/SessionWallet.cs
@@ -18,7 +18,7 @@ public class SessionWallet : InGameWallet
     {
         public PublicKey TargetProgram { get; protected set; }
         public PublicKey SessionTokenPDA { get; protected set; }
-        
+
         public static SessionWallet Instance;
 
         private SessionWallet(RpcCluster rpcCluster = RpcCluster.DevNet,
@@ -89,9 +89,9 @@ public static async Task<SessionWallet> GetSessionWallet(PublicKey targetProgram
                     sessionWallet.Logout();
                     return await GetSessionWallet(targetProgram, password, rpcCluster, customRpcUri, customStreamingRpcUri, autoConnectOnStartup);
                 }
-                
+
                 sessionWallet.SessionTokenPDA = FindSessionToken(targetProgram, sessionWallet.Account, Web3.Account);
-                
+
                 Debug.Log(sessionWallet.SessionTokenPDA);
 
                 if (!await sessionWallet.IsSessionTokenInitialized())
@@ -116,7 +116,7 @@ public static async Task<SessionWallet> GetSessionWallet(PublicKey targetProgram
             sessionWallet.SessionTokenPDA = FindSessionToken(targetProgram, sessionWallet.Account, Web3.Account);
             return sessionWallet;
         }
-        
+
 
         /// <summary>
         /// Creates a transaction instruction to create a new session token account and initialize it with the provided session signer and target program.
@@ -197,8 +197,8 @@ private void DeleteSessionWallet()
             PlayerPrefs.DeleteKey(EncryptedKeystoreKey);
             PlayerPrefs.Save();
         }
-        
-        
+
+
         /// <summary>
         /// Prepares the session wallet for logout by revoking the session, issuing a refund, and purging the keystore.
         /// NOTE: You must call PrepareLogout before calling Logout to ensure that the session token account is revoked and the refund is issued.
@@ -217,7 +217,7 @@ public async Task PrepareLogout()
 
             // Get balance and calculate refund
             var balance = (await GetBalance(Account.PublicKey)) * SolLamports;
-            var estimatedFees = await ActiveRpcClient.GetFeeCalculatorForBlockhashAsync(tx.RecentBlockHash);
+            var estimatedFees = await ActiveRpcClient.GetFeesAsync();
             var refund = balance - (estimatedFees.Result.Value.FeeCalculator.LamportsPerSignature * 1);
             Debug.Log($"LAMPORTS Balance: {balance}, Refund: {refund}");