UNPKG

@roochnetwork/rooch-sdk-kit

Version:
609 lines (567 loc) 18.3 kB
// src/hooks/client/useCurrentNetwork.ts function useCurrentNetwork() { return useRoochContext().network; } // src/hooks/client/useRoochContext.ts import { useContext as useContext4 } from "react"; // src/provider/clientProvider.tsx import { createContext as createContext4, useMemo, useState } from "react"; import { getRoochNodeUrl, RoochClient } from "@roochnetwork/rooch-sdk"; // src/hooks/useSessionsStore.ts import { useContext as useContext3 } from "react"; import { useStore as useStore2 } from "zustand"; // src/provider/roochProvider.tsx import { createContext as createContext2, useRef } from "react"; // src/provider/globalProvider.tsx import { createContext, useContext } from "react"; import { jsx } from "react/jsx-runtime"; var GlobalContext = createContext(null); var useTriggerError = () => { const ctx = useContext(GlobalContext); if (!ctx) { throw new Error("useTriggerError must be used within a GlobalProvider"); } return ctx.triggerError; }; // src/provider/roochProvider.tsx import { jsx as jsx2, jsxs } from "react/jsx-runtime"; var RoochContext = createContext2(null); // src/provider/walletProvider.tsx import { useCallback, createContext as createContext3, useEffect, useRef as useRef2 } from "react"; import { BitcoinAddress } from "@roochnetwork/rooch-sdk"; // src/wellet/wallets.ts var wallets = void 0; var registered = /* @__PURE__ */ new Set(); var listeners = {}; function getWallets() { if (wallets) return wallets; wallets = Object.freeze({ register, get, on }); return wallets; } function register(...wallets2) { wallets2.forEach((wallet) => registered.add(wallet)); listeners["register"]?.forEach((listener) => guard(() => listener(...wallets2))); return function unregister() { wallets2.forEach((wallet) => registered.delete(wallet)); listeners["unregister"]?.forEach((listener) => guard(() => listener(...wallets2))); }; } function get() { return [...registered]; } function on(event, listener) { listeners[event]?.push(listener) || (listeners[event] = [listener]); return function off() { listeners[event] = listeners[event]?.filter((existingListener) => listener !== existingListener); }; } function guard(callback) { try { callback(); } catch (error) { console.error(error); } } // src/wellet/util.ts function getRegisteredWallets(preferredWallets, walletFilter) { const walletsApi = getWallets(); const wallets2 = walletsApi.get(); const Wallets = wallets2.filter((wallet) => !walletFilter || walletFilter(wallet)); return [ // Preferred wallets, in order: ...preferredWallets.map((name) => Wallets.find((wallet) => wallet.getName() === name)), // Wallets in default order: ...Wallets.filter((wallet) => !preferredWallets.includes(wallet.getName())) ].filter((wallet) => wallet !== void 0); } // src/hooks/wallet/useWalletStore.ts import { useContext as useContext2 } from "react"; import { useStore } from "zustand"; 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/provider/walletProvider.tsx import { jsx as jsx3 } from "react/jsx-runtime"; var WalletContext = createContext3(null); // src/hooks/useSessionsStore.ts function useSessionStore(selector) { const store = useContext3(RoochContext); if (!store) { throw new Error( "Could not find RoochSessionContext. Ensure that you have set up the RoochClientProvider." ); } return useStore2(store, selector); } // src/provider/clientProvider.tsx import { jsx as jsx4 } from "react/jsx-runtime"; var ClientContext = createContext4(null); var DEFAULT_NETWORKS = { localnet: { url: getRoochNodeUrl("localnet") } }; // src/hooks/client/useRoochContext.ts function useRoochContext() { const context = useContext4(ClientContext); if (!context) { throw new Error( "Could not find RoochClientContext. Ensure that you have set up the RoochClientProvider." ); } return context; } // src/hooks/client/useRoochClient.ts function useRoochClient() { return useRoochContext().client; } // src/hooks/client/useRoochClientQuery.ts import { useQuery } from "@tanstack/react-query"; function useRoochClientQuery(...args) { const [method, params, { queryKey = [], ...options } = {}] = args; const network = useCurrentNetwork(); const roochClient = useRoochClient(); return useQuery({ ...options, queryKey: [network, method, params, ...queryKey], queryFn: async () => { return await roochClient[method](params); } }); } // src/hooks/client/useSwitchNetwork.ts import { useMutation } from "@tanstack/react-query"; // src/constants/roochMutationKeys.ts function formMutationKeyFn(baseEntity) { return function mutationKeyFn(additionalKeys = []) { return [{ ...roochMutationKeys.all, baseEntity }, ...additionalKeys]; }; } var roochMutationKeys = { all: { baseScope: "rooch" }, addNetwork: formMutationKeyFn("add-network"), switchNetwork: formMutationKeyFn("switch-network"), removeNetwork: formMutationKeyFn("remove-network"), removeSession: formMutationKeyFn("remove-session"), transferObject: formMutationKeyFn("transfer-object"), transferCoin: formMutationKeyFn("transfer-coin"), signAndExecuteTransaction: formMutationKeyFn("sign-and-execute-transaction") }; // src/constants/walletMutationKeys.ts function formMutationKeyFn2(baseEntity) { return function mutationKeyFn(additionalKeys = []) { return [{ ...walletMutationKeys.all, baseEntity }, ...additionalKeys]; }; } var walletMutationKeys = { all: { baseScope: "wallet" }, connectWallet: formMutationKeyFn2("connect-wallet"), autoConnectWallet: formMutationKeyFn2("auto-connect-wallet"), switchAccount: formMutationKeyFn2("switch-account"), createSessionKey: formMutationKeyFn2("create-session-key") }; // src/hooks/client/useSwitchNetwork.ts function useSwitchNetwork({ mutationKey, ...mutationOptions } = {}) { const switchNetwork = useRoochContext().selectNetwork; return useMutation({ mutationKey: roochMutationKeys.switchNetwork(mutationKey), mutationFn: async (args) => { switchNetwork(args); }, ...mutationOptions }); } // src/hooks/client/useTransferObject.ts import { useMutation as useMutation2 } from "@tanstack/react-query"; function useTransferObject({ mutationKey, ...mutationOptions } = {}) { const client = useRoochClient(); return useMutation2({ mutationKey: roochMutationKeys.transferObject(mutationKey), mutationFn: async (args) => { const result = await client.transferObject(args); if (result.execution_info.status.type !== "executed") { Error("transfer failed" + result.execution_info.status.type); } }, ...mutationOptions }); } // src/hooks/client/useTransferCoin.ts import { useMutation as useMutation3 } from "@tanstack/react-query"; // src/hooks/useCurrentSession.ts function useCurrentSession() { return useSessionStore((state) => state.currentSession); } // src/hooks/client/useTransferCoin.ts function useTransferCoin({ mutationKey, ...mutationOptions } = {}) { const client = useRoochClient(); const curSession = useCurrentSession(); return useMutation3({ mutationKey: roochMutationKeys.transferCoin(mutationKey), mutationFn: async (args) => { const signer = args.signer || curSession; if (signer === null) { throw Error(""); } const result = await client.transfer({ ...args, signer: args.signer || curSession }); if (result.execution_info.status.type !== "executed") { Error("transfer failed" + result.execution_info.status.type); } }, ...mutationOptions }); } // src/hooks/client/networks-config.ts function createNetworkConfig(networkConfig) { function useNetworkConfig() { const { config } = useRoochContext(); if (!config) { throw new Error("No network config found"); } return config; } function useNetworkVariables() { const { variables } = useNetworkConfig(); return variables ?? {}; } function useNetworkVariable(name) { const variables = useNetworkVariables(); return variables[name]; } return { networkConfig, useNetworkConfig, useNetworkVariables, useNetworkVariable }; } // src/hooks/client/useSignAndExecuteTransaction.ts import { useMutation as useMutation4 } from "@tanstack/react-query"; function useSignAndExecuteTransaction({ mutationKey, ...mutationOptions } = {}) { const client = useRoochClient(); const session = useCurrentSession(); return useMutation4({ mutationKey: roochMutationKeys.signAndExecuteTransaction(mutationKey), mutationFn: async (args) => { if (!session) { throw Error("Create a session first"); } const result = await client.signAndExecuteTransaction({ transaction: args.transaction, signer: args.signer || session }); if (result.execution_info.status.type !== "executed" && result.execution_info.status) { Error("transfer failed" + result.execution_info.status.type); } return result; }, ...mutationOptions }); } // src/hooks/wallet/useAddresses.ts function useAddresses() { return useWalletStore((state) => state.addresses); } // src/hooks/wallet/useAutoConnectWallet.ts import { useQuery as useQuery2 } from "@tanstack/react-query"; import { useLayoutEffect, useState as useState2 } from "react"; function useAutoConnectWallet() { const { mutateAsync: connectWallet } = useConnectWallet(); const autoConnectEnabled = useWalletStore((state) => state.autoConnectEnabled); const lastConnectedWalletName = useWalletStore((state) => state.lastConnectedWalletName); const lastConnectedAddress = useWalletStore((state) => state.lastConnectedAddress); const { isConnected } = useCurrentWallet(); const wallets2 = useWallets(); const [clientOnly, setClientOnly] = useState2(false); useLayoutEffect(() => { setClientOnly(true); }, []); const { data, isError } = useQuery2({ queryKey: [ "@rooch/sdk-kit", "autoconnect", { isConnected, autoConnectEnabled, lastConnectedWalletName, lastConnectedAddress } ], queryFn: async () => { if (!autoConnectEnabled) { return "disabled"; } if (!lastConnectedWalletName || !lastConnectedAddress || isConnected) { return "attempted"; } let wallet = wallets2.find((wallet2) => wallet2.getName() === lastConnectedWalletName); if (wallet) { await connectWallet({ wallet }); } return "attempted"; }, enabled: autoConnectEnabled, persister: void 0, gcTime: 0, staleTime: 0, networkMode: "always", retry: (_) => { return false; }, retryOnMount: false, refetchInterval: 1e3, refetchIntervalInBackground: false, refetchOnMount: false, refetchOnReconnect: false, refetchOnWindowFocus: false }); if (!autoConnectEnabled) { return "disabled"; } if (!clientOnly) { return "idle"; } if (!lastConnectedWalletName) { return "attempted"; } return isError ? "attempted" : data ?? "idle"; } // src/hooks/wallet/useConnectWallet.ts import { useMutation as useMutation5 } from "@tanstack/react-query"; // src/hooks/useSessions.ts function useSessions() { return useSessionStore( (state) => state.sessions.sort((a, b) => b.getCreateTime() - a.getCreateTime()) ); } // src/hooks/wallet/useConnectWallet.ts function useConnectWallet({ mutationKey, ...mutationOptions } = {}) { const sessions = useSessions(); const setCurrentSession = useSessionStore((state) => state.setCurrentSession); const setWalletConnected = useWalletStore((state) => state.setWalletConnected); const setWalletDisconnected = useWalletStore((state) => state.setWalletDisconnected); const setConnectionStatus = useWalletStore((state) => state.setConnectionStatus); const triggerError = useTriggerError(); return useMutation5({ mutationKey: walletMutationKeys.connectWallet(mutationKey), mutationFn: async ({ wallet }) => { try { setConnectionStatus("connecting"); const connectAddress = await wallet.connect(); const selectedAddress = connectAddress[0]; setWalletConnected(wallet, connectAddress, selectedAddress); const cur = sessions.find( (item) => item.getRoochAddress().toStr() === selectedAddress?.genRoochAddress().toStr() ); setCurrentSession(cur); return connectAddress; } catch (error) { setWalletDisconnected(); if ("code" in error && "message" in error) { triggerError(error); } throw error; } }, ...mutationOptions }); } // src/hooks/wallet/useCurrentAddress.ts function useCurrentAddress() { return useWalletStore((state) => state.currentAddress); } // src/hooks/wallet/useCurrentWallet.ts function useCurrentWallet() { const currentWallet = useWalletStore((state) => state.currentWallet); const connectionStatus = useWalletStore((state) => state.connectionStatus); switch (connectionStatus) { case "connecting": return { status: connectionStatus, wallet: currentWallet, isDisconnected: false, isConnecting: true, isConnected: false }; case "disconnected": return { status: connectionStatus, wallet: currentWallet, isDisconnected: true, isConnecting: false, isConnected: false }; case "connected": { return { status: connectionStatus, wallet: currentWallet, isDisconnected: false, isConnecting: false, isConnected: true }; } } } // src/hooks/wallet/useWallets.ts function useWallets() { return useWalletStore((state) => state.wallets); } // src/hooks/wallet/useWalletChanged.ts import { useEffect as useEffect2 } from "react"; function useWalletChanged(preferredWallets, walletFilter) { const updateWallets = useWalletStore((state) => state.updateWallets); useEffect2(() => { const api = getWallets(); updateWallets(getRegisteredWallets(preferredWallets, walletFilter)); const unsubscribeFromRegister = api.on("register", () => { updateWallets(getRegisteredWallets(preferredWallets, walletFilter)); }); return () => { unsubscribeFromRegister(); }; }, [preferredWallets, updateWallets, walletFilter]); } // src/hooks/wallet/useConnectionStatus.ts function useConnectionStatus() { return useWalletStore((state) => state.connectionStatus); } // src/hooks/useSession.ts function useSession(scope) { return useSessionStore( (state) => state.sessions.find( (item) => scope.includes("::") ? item.getScopes().find((_scope) => _scope === scope) : item.getScopes().find((_scope) => _scope.startsWith(scope)) !== void 0 ) ); } // src/hooks/useRemoveSession.ts import { useMutation as useMutation6 } from "@tanstack/react-query"; function useRemoveSession({ mutationKey, ...mutationOptions } = {}) { const sessionsKeys = useSessions(); const removeSession = useSessionStore((state) => state.removeSession); const setCurrentSession = useSessionStore((state) => state.setCurrentSession); const currentSession = useCurrentSession(); const client = useRoochClient(); return useMutation6({ mutationKey: roochMutationKeys.removeSession(mutationKey), mutationFn: async (args) => { try { if (!currentSession) { return; } const result = await client.removeSession({ authKey: args.authKey, signer: currentSession }); if (result) { let cacheSession = sessionsKeys.find( (item) => item.getAuthKey() === args.authKey ); if (cacheSession) { removeSession(cacheSession); if (cacheSession.getAuthKey() === currentSession?.getAuthKey()) { const substitute = sessionsKeys.filter((item) => item.getAuthKey() !== cacheSession.getAuthKey()).sort((a, b) => b.getCreateTime() - a.getCreateTime()).filter((item) => !item.isSessionExpired()); if (substitute.length > 0) { setCurrentSession(substitute[0]); } else { setCurrentSession(void 0); } } } } } catch (e) { throw e; } }, ...mutationOptions }); } // src/hooks/useCreateSession.ts import { useMutation as useMutation7 } from "@tanstack/react-query"; // src/error/walletErrors.ts var WalletNotConnectedError = class extends Error { }; // src/hooks/useCreateSession.ts function useCreateSessionKey({ mutationKey, ...mutationOptions } = {}) { const client = useRoochClient(); const { wallet } = useCurrentWallet(); const setCurrentSession = useSessionStore((state) => state.setCurrentSession); const triggerError = useTriggerError(); return useMutation7({ mutationKey: walletMutationKeys.createSessionKey(mutationKey), mutationFn: async (args) => { const signer = args.signer || wallet; if (!signer) { throw new WalletNotConnectedError("No wallet is connected."); } try { const sessionAccount = await client.createSession({ signer, sessionArgs: args }); setCurrentSession(sessionAccount); return sessionAccount; } catch (error) { if ("code" in error && "message" in error) { triggerError(error); } throw error; } }, ...mutationOptions }); } export { createNetworkConfig, useAddresses, useAutoConnectWallet, useConnectWallet, useConnectionStatus, useCreateSessionKey, useCurrentAddress, useCurrentNetwork, useCurrentSession, useCurrentWallet, useRemoveSession, useRoochClient, useRoochClientQuery, useRoochContext, useSession, useSessions, useSignAndExecuteTransaction, useSwitchNetwork, useTransferCoin, useTransferObject, useWalletChanged, useWallets }; //# sourceMappingURL=index.js.map