@roochnetwork/rooch-sdk-kit
Version:
Rooch SDK Kit
89 lines (82 loc) • 2.88 kB
JavaScript
// 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