[SDK] Expose useInvalidateBalances() (#8373)

Author: joaquim-verges

.changeset/clever-horses-turn.md

@@ -0,0 +1,5 @@
+---
+"thirdweb": patch
+---
+
+Respect from address when simulating with Engine.serverWallet

.changeset/curvy-pillows-return.md

@@ -0,0 +1,5 @@
+---
+"thirdweb": patch
+---
+
+Expose useInvalidateBalances() react hook

packages/thirdweb/src/engine/server-wallet.ts

@@ -332,6 +332,7 @@ export function serverWallet(options: ServerWalletOptions): ServerWallet {
       if (args.simulate) {
         serializedTransaction = await toSerializableTransaction({
           transaction: args.transaction,
+          from: address,
         });
       } else {
         const [to, data, value] = await Promise.all([

packages/thirdweb/src/exports/react.native.ts

@@ -22,6 +22,7 @@ export { useContractEvents } from "../react/core/hooks/contract/useContractEvent
 // contract
 export { useReadContract } from "../react/core/hooks/contract/useReadContract.js";
 export { useWaitForReceipt } from "../react/core/hooks/contract/useWaitForReceipt.js";
+export { useInvalidateBalances } from "../react/core/hooks/others/useInvalidateBalances.js";
 export { useInvalidateContractQuery } from "../react/core/hooks/others/useInvalidateQueries.js";
 export { useWalletBalance } from "../react/core/hooks/others/useWalletBalance.js";
 export {

packages/thirdweb/src/exports/react.ts

@@ -29,6 +29,7 @@ export { useReadContract } from "../react/core/hooks/contract/useReadContract.js
 export { useWaitForReceipt } from "../react/core/hooks/contract/useWaitForReceipt.js";
 // chain hooks
 export { useChainMetadata } from "../react/core/hooks/others/useChainQuery.js";
+export { useInvalidateBalances } from "../react/core/hooks/others/useInvalidateBalances.js";
 export { useInvalidateContractQuery } from "../react/core/hooks/others/useInvalidateQueries.js";
 export { useWalletBalance } from "../react/core/hooks/others/useWalletBalance.js";
 export {

packages/thirdweb/src/react/core/hooks/others/useInvalidateBalances.ts

@@ -0,0 +1,18 @@
+import { useQueryClient } from "@tanstack/react-query";
+import { invalidateWalletBalance } from "../../providers/invalidateWalletBalance.js";
+
+/**
+ * Invalidate the balances for a given chainId. If no chainId is provided, invalidate all balances.
+ * @example
+ * ```ts
+ * const invalidateBalances = useInvalidateBalances();
+ * invalidateBalances();
+ * ```
+ */
+export function useInvalidateBalances() {
+  const queryClient = useQueryClient();
+
+  return ({ chainId }: { chainId?: number }) => {
+    invalidateWalletBalance(queryClient, chainId);
+  };
+}