UNPKG

@roochnetwork/rooch-sdk-kit

Version:
89 lines (82 loc) 2.88 kB
// src/hooks/wallet/useWalletStore.ts import { useContext } from "react"; import { useStore } from "zustand"; // src/provider/walletProvider.tsx import { useCallback, createContext, useEffect, useRef } 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/provider/walletProvider.tsx import { jsx } from "react/jsx-runtime"; 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/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]); } export { useWalletChanged }; //# sourceMappingURL=useWalletChanged.js.map