@mysten/dapp-kit
Version:
A collection of React hooks and components for interacting with the Sui blockchain and wallets.
114 lines (106 loc) • 3.49 kB
JavaScript
// src/hooks/wallet/useSwitchAccount.ts
import { useMutation } 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 WalletAccountNotFoundError = class extends Error {
};
// src/hooks/wallet/useWalletStore.ts
import { useContext } from "react";
import { useStore } from "zustand";
// src/contexts/walletContext.ts
import { createContext } from "react";
var WalletContext = createContext(null);
// src/hooks/wallet/useWalletStore.ts
function useWalletStore(selector) {
const store = useContext(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/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/useSwitchAccount.ts
function useSwitchAccount({
mutationKey,
...mutationOptions
} = {}) {
const { currentWallet } = useCurrentWallet();
const setAccountSwitched = useWalletStore((state) => state.setAccountSwitched);
return useMutation({
mutationKey: walletMutationKeys.switchAccount(mutationKey),
mutationFn: async ({ account }) => {
if (!currentWallet) {
throw new WalletNotConnectedError("No wallet is connected.");
}
const accountToSelect = currentWallet.accounts.find(
(walletAccount) => walletAccount.address === account.address
);
if (!accountToSelect) {
throw new WalletAccountNotFoundError(
`No account with address ${account.address} is connected to ${currentWallet.name}.`
);
}
setAccountSwitched(accountToSelect);
},
...mutationOptions
});
}
export {
useSwitchAccount
};
//# sourceMappingURL=useSwitchAccount.js.map