UNPKG

@mysten/dapp-kit

Version:

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

207 lines (194 loc) 6.71 kB
// src/hooks/wallet/useSignTransaction.ts 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/useSignTransaction.ts function useSignTransaction({ mutationKey, ...mutationOptions } = {}) { const { currentWallet } = useCurrentWallet(); const currentAccount = useCurrentAccount(); const { client, network } = useSuiClientContext(); const { mutate: reportTransactionEffects } = useReportTransactionEffects(); return useMutation2({ mutationKey: walletMutationKeys.signTransaction(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 { bytes, signature } = await signTransaction(currentWallet, { ...signTransactionArgs, transaction: { toJSON: async () => { return typeof transaction === "string" ? transaction : await transaction.toJSON({ supportedIntents: [], client }); } }, account: signerAccount, chain }); return { bytes, signature, reportTransactionEffects: (effects) => { reportTransactionEffects({ effects, account: signerAccount, chain }); } }; }, ...mutationOptions }); } export { useSignTransaction }; //# sourceMappingURL=useSignTransaction.js.map