UNPKG

@mysten/dapp-kit

Version:

A collection of React hooks and components for interacting with the Sui blockchain and wallets.

225 lines (212 loc) 7.49 kB
// src/hooks/wallet/useSignAndExecuteTransaction.ts import { toBase64 as toBase642 } from "@mysten/sui/utils"; import { signTransaction } from "@mysten/wallet-standard"; import { useMutation as useMutation2 } from "@tanstack/react-query"; // src/constants/walletMutationKeys.ts var walletMutationKeys = { all: { baseScope: "wallet" }, connectWallet: formMutationKeyFn("connect-wallet"), autoconnectWallet: formMutationKeyFn("autoconnect-wallet"), disconnectWallet: formMutationKeyFn("disconnect-wallet"), signPersonalMessage: formMutationKeyFn("sign-personal-message"), signTransaction: formMutationKeyFn("sign-transaction"), signAndExecuteTransaction: formMutationKeyFn("sign-and-execute-transaction"), switchAccount: formMutationKeyFn("switch-account"), reportTransactionEffects: formMutationKeyFn("report-transaction-effects") }; function formMutationKeyFn(baseEntity) { return function mutationKeyFn(additionalKeys = []) { return [{ ...walletMutationKeys.all, baseEntity }, ...additionalKeys]; }; } // src/errors/walletErrors.ts var WalletNotConnectedError = class extends Error { }; var WalletNoAccountSelectedError = class extends Error { }; var WalletFeatureNotSupportedError = class extends Error { }; // src/hooks/useSuiClient.ts import { useContext } from "react"; // src/components/SuiClientProvider.tsx import { getFullnodeUrl, isSuiClient, SuiClient } from "@mysten/sui/client"; import { createContext, useMemo, useState } from "react"; import { jsx } from "react/jsx-runtime"; var SuiClientContext = createContext(null); var DEFAULT_NETWORKS = { localnet: { url: getFullnodeUrl("localnet") } }; // src/hooks/useSuiClient.ts function useSuiClientContext() { const suiClient = useContext(SuiClientContext); if (!suiClient) { throw new Error( "Could not find SuiClientContext. Ensure that you have set up the SuiClientProvider" ); } return suiClient; } // src/hooks/wallet/useWalletStore.ts import { useContext as useContext2 } from "react"; import { useStore } from "zustand"; // src/contexts/walletContext.ts import { createContext as createContext2 } from "react"; var WalletContext = createContext2(null); // src/hooks/wallet/useWalletStore.ts function useWalletStore(selector) { const store = useContext2(WalletContext); if (!store) { throw new Error( "Could not find WalletContext. Ensure that you have set up the WalletProvider." ); } return useStore(store, selector); } // src/hooks/wallet/useCurrentAccount.ts function useCurrentAccount() { return useWalletStore((state) => state.currentAccount); } // src/hooks/wallet/useCurrentWallet.ts function useCurrentWallet() { const currentWallet = useWalletStore((state) => state.currentWallet); const connectionStatus = useWalletStore((state) => state.connectionStatus); const supportedIntents = useWalletStore((state) => state.supportedIntents); switch (connectionStatus) { case "connecting": return { connectionStatus, currentWallet: null, isDisconnected: false, isConnecting: true, isConnected: false, supportedIntents: [] }; case "disconnected": return { connectionStatus, currentWallet: null, isDisconnected: true, isConnecting: false, isConnected: false, supportedIntents: [] }; case "connected": { return { connectionStatus, currentWallet, isDisconnected: false, isConnecting: false, isConnected: true, supportedIntents }; } } } // src/hooks/wallet/useReportTransactionEffects.ts import { toBase64 } from "@mysten/sui/utils"; import { useMutation } from "@tanstack/react-query"; function useReportTransactionEffects({ mutationKey, ...mutationOptions } = {}) { const { currentWallet } = useCurrentWallet(); const currentAccount = useCurrentAccount(); return useMutation({ mutationKey: walletMutationKeys.reportTransactionEffects(mutationKey), mutationFn: async ({ effects, chain = currentWallet?.chains[0], account = currentAccount }) => { if (!currentWallet) { throw new WalletNotConnectedError("No wallet is connected."); } if (!account) { throw new WalletNoAccountSelectedError( "No wallet account is selected to report transaction effects for" ); } const reportTransactionEffectsFeature = currentWallet.features["sui:reportTransactionEffects"]; if (reportTransactionEffectsFeature) { return await reportTransactionEffectsFeature.reportTransactionEffects({ effects: Array.isArray(effects) ? toBase64(new Uint8Array(effects)) : effects, account, chain: chain ?? currentWallet?.chains[0] }); } }, ...mutationOptions }); } // src/hooks/wallet/useSignAndExecuteTransaction.ts function useSignAndExecuteTransaction({ mutationKey, execute, ...mutationOptions } = {}) { const { currentWallet, supportedIntents } = useCurrentWallet(); const currentAccount = useCurrentAccount(); const { client, network } = useSuiClientContext(); const { mutate: reportTransactionEffects } = useReportTransactionEffects(); const executeTransaction = execute ?? (async ({ bytes, signature }) => { const { digest, rawEffects } = await client.executeTransactionBlock({ transactionBlock: bytes, signature, options: { showRawEffects: true } }); return { digest, rawEffects, effects: toBase642(new Uint8Array(rawEffects)), bytes, signature }; }); return useMutation2({ mutationKey: walletMutationKeys.signAndExecuteTransaction(mutationKey), mutationFn: async ({ transaction, ...signTransactionArgs }) => { if (!currentWallet) { throw new WalletNotConnectedError("No wallet is connected."); } const signerAccount = signTransactionArgs.account ?? currentAccount; if (!signerAccount) { throw new WalletNoAccountSelectedError( "No wallet account is selected to sign the transaction with." ); } if (!currentWallet.features["sui:signTransaction"] && !currentWallet.features["sui:signTransactionBlock"]) { throw new WalletFeatureNotSupportedError( "This wallet doesn't support the `signTransaction` feature." ); } const chain = signTransactionArgs.chain ?? `sui:${network}`; const { signature, bytes } = await signTransaction(currentWallet, { ...signTransactionArgs, transaction: { async toJSON() { return typeof transaction === "string" ? transaction : await transaction.toJSON({ supportedIntents, client }); } }, account: signerAccount, chain }); const result = await executeTransaction({ bytes, signature }); let effects; if ("effects" in result && result.effects?.bcs) { effects = result.effects.bcs; } else if ("rawEffects" in result) { effects = toBase642(new Uint8Array(result.rawEffects)); } else { throw new Error("Could not parse effects from transaction result."); } reportTransactionEffects({ effects, account: signerAccount, chain }); return result; }, ...mutationOptions }); } export { useSignAndExecuteTransaction }; //# sourceMappingURL=useSignAndExecuteTransaction.js.map