create relayer-provider and improve transfer-provider

packages/apps/src/hooks/use-relayer.ts

@@ -0,0 +1,4 @@
+import { RelayerContext } from "@/providers/relayer-provider";
+import { useContext } from "react";
+
+export const useRelayer = () => useContext(RelayerContext);

packages/apps/src/providers/relayer-provider.tsx

@@ -0,0 +1,250 @@
+"use client";
+
+import { LnBridgeDefault } from "@/bridges/lnbridge-default";
+import { LnBridgeOpposite } from "@/bridges/lnbridge-opposite";
+import { BridgeCategory } from "@/types/bridge";
+import { ChainConfig } from "@/types/chain";
+import { Token } from "@/types/token";
+import {
+  Dispatch,
+  PropsWithChildren,
+  SetStateAction,
+  createContext,
+  useCallback,
+  useEffect,
+  useMemo,
+  useState,
+} from "react";
+import { useAccount, usePublicClient, useWalletClient } from "wagmi";
+import { Subscription, forkJoin } from "rxjs";
+
+interface RelayerCtx {
+  margin: { formatted: bigint; value: string };
+  baseFee: { formatted: bigint; value: string };
+  feeRate: { formatted: number; value: string };
+  sourceAllowance: { value: bigint; token: Token } | undefined;
+  targetAllowance: { value: bigint; token: Token } | undefined;
+  sourceBalance: { value: bigint; token: Token } | undefined;
+  targetBalance: { value: bigint; token: Token } | undefined;
+  bridgeCategory: BridgeCategory | undefined;
+  sourceChain: ChainConfig | undefined;
+  targetChain: ChainConfig | undefined;
+  transferToken: Token | undefined;
+  defaultBridge: LnBridgeDefault | undefined;
+  oppositeBridge: LnBridgeOpposite | undefined;
+
+  setMargin: Dispatch<SetStateAction<{ formatted: bigint; value: string }>>;
+  setBaseFee: Dispatch<SetStateAction<{ formatted: bigint; value: string }>>;
+  setFeeRate: Dispatch<SetStateAction<{ formatted: number; value: string }>>;
+  setBridgeCategory: Dispatch<SetStateAction<BridgeCategory | undefined>>;
+  setSourceChain: Dispatch<SetStateAction<ChainConfig | undefined>>;
+  setTargetChain: Dispatch<SetStateAction<ChainConfig | undefined>>;
+  setTransferToken: Dispatch<SetStateAction<Token | undefined>>;
+  sourceApprove: (amount: bigint) => Promise<void>;
+  targetApprove: (amount: bigint) => Promise<void>;
+  depositMargin: (margin: bigint) => Promise<void>;
+  updateFeeAndMargin: (margin: bigint, baseFee: bigint, feeRate: number) => Promise<void>;
+}
+
+const defaultValue: RelayerCtx = {
+  margin: { formatted: 0n, value: "" },
+  baseFee: { formatted: 0n, value: "" },
+  feeRate: { formatted: 0, value: "" },
+  sourceAllowance: undefined,
+  targetAllowance: undefined,
+  sourceBalance: undefined,
+  targetBalance: undefined,
+  bridgeCategory: undefined,
+  sourceChain: undefined,
+  targetChain: undefined,
+  transferToken: undefined,
+  defaultBridge: undefined,
+  oppositeBridge: undefined,
+  setMargin: () => undefined,
+  setBaseFee: () => undefined,
+  setFeeRate: () => undefined,
+  setBridgeCategory: () => undefined,
+  setSourceChain: () => undefined,
+  setTargetChain: () => undefined,
+  setTransferToken: () => undefined,
+  sourceApprove: async () => undefined,
+  targetApprove: async () => undefined,
+  depositMargin: async () => undefined,
+  updateFeeAndMargin: async () => undefined,
+};
+
+export const RelayerContext = createContext(defaultValue);
+
+export default function RelayerProvider({ children }: PropsWithChildren<unknown>) {
+  const { data: walletClient } = useWalletClient();
+  const publicClient = usePublicClient();
+  const { address } = useAccount();
+
+  const [margin, setMargin] = useState(defaultValue.margin);
+  const [baseFee, setBaseFee] = useState(defaultValue.baseFee);
+  const [feeRate, setFeeRate] = useState(defaultValue.feeRate);
+
+  const [sourceAllowance, setSourceAllowance] = useState(defaultValue.sourceAllowance);
+  const [targetAllowance, setTargetAllowance] = useState(defaultValue.targetAllowance);
+  const [sourceBalance, setSourceBalance] = useState(defaultValue.sourceBalance);
+  const [targetBalance, setTargetBalance] = useState(defaultValue.targetBalance);
+
+  const [bridgeCategory, setBridgeCategory] = useState(defaultValue.bridgeCategory);
+  const [sourceChain, setSourceChain] = useState(defaultValue.sourceChain);
+  const [targetChain, setTargetChain] = useState(defaultValue.targetChain);
+  const [transferToken, setTransferToken] = useState(defaultValue.transferToken);
+
+  const { defaultBridge, oppositeBridge, bridgeClient } = useMemo(() => {
+    let defaultBridge: LnBridgeDefault | undefined;
+    let oppositeBridge: LnBridgeOpposite | undefined;
+
+    if (bridgeCategory === "lnbridgev20-default") {
+      defaultBridge = new LnBridgeDefault({
+        category: bridgeCategory,
+        sourceChain,
+        targetChain,
+        walletClient,
+        publicClient,
+      });
+    } else if (bridgeCategory === "lnbridgev20-opposite") {
+      oppositeBridge = new LnBridgeOpposite({
+        category: bridgeCategory,
+        sourceChain,
+        targetChain,
+        walletClient,
+        publicClient,
+      });
+    }
+
+    return { defaultBridge, oppositeBridge, bridgeClient: defaultBridge ?? oppositeBridge };
+  }, [sourceChain, targetChain, bridgeCategory, walletClient, publicClient]);
+
+  const sourceApprove = useCallback(
+    async (amount: bigint) => {
+      if (address && bridgeClient) {
+        try {
+          await bridgeClient.sourceApprove(amount, address);
+          setSourceAllowance(await bridgeClient.getSourceAllowance(address));
+        } catch (err) {
+          console.error(err);
+        }
+      }
+    },
+    [address, bridgeClient],
+  );
+
+  const targetApprove = useCallback(
+    async (amount: bigint) => {
+      if (address && bridgeClient) {
+        try {
+          await bridgeClient.targetApprove(amount, address);
+          setTargetAllowance(await bridgeClient.getTargetAllowance(address));
+        } catch (err) {
+          console.error(err);
+        }
+      }
+    },
+    [address, bridgeClient],
+  );
+
+  const depositMargin = useCallback(
+    async (margin: bigint) => {
+      if (address && defaultBridge) {
+        try {
+          await defaultBridge.depositMargin(margin);
+          const a = await defaultBridge.getTargetAllowance(address);
+          const b = await defaultBridge.getTargetBalance(address);
+          setTargetAllowance(a);
+          setTargetBalance(b);
+        } catch (err) {
+          console.error(err);
+        }
+      }
+    },
+    [address, defaultBridge],
+  );
+
+  const updateFeeAndMargin = useCallback(
+    async (margin: bigint, baseFee: bigint, feeRate: number) => {
+      if (address && oppositeBridge) {
+        try {
+          await oppositeBridge.updateFeeAndMargin(margin, baseFee, feeRate);
+          const a = await oppositeBridge.getSourceAllowance(address);
+          const b = await oppositeBridge.getSourceBalance(address);
+          setSourceAllowance(a);
+          setSourceBalance(b);
+        } catch (err) {
+          console.error(err);
+        }
+      }
+    },
+    [address, oppositeBridge],
+  );
+
+  useEffect(() => {
+    let sub$$: Subscription | undefined;
+
+    if (address && bridgeClient) {
+      sub$$ = forkJoin([
+        bridgeClient.getSourceAllowance(address),
+        bridgeClient.getTargetAllowance(address),
+        bridgeClient.getSourceBalance(address),
+        bridgeClient.getTargetBalance(address),
+      ]).subscribe({
+        next: ([as, at, bs, bt]) => {
+          setSourceAllowance(as);
+          setTargetAllowance(at);
+          setSourceBalance(bs);
+          setTargetBalance(bt);
+        },
+        error: (err) => {
+          console.error(err);
+          setSourceAllowance(defaultValue.sourceAllowance);
+          setTargetAllowance(defaultValue.targetAllowance);
+          setSourceBalance(defaultValue.sourceBalance);
+          setTargetBalance(defaultValue.targetBalance);
+        },
+      });
+    } else {
+      setSourceAllowance(defaultValue.sourceAllowance);
+      setTargetAllowance(defaultValue.targetAllowance);
+      setSourceBalance(defaultValue.sourceBalance);
+      setTargetBalance(defaultValue.targetBalance);
+    }
+
+    return () => sub$$?.unsubscribe();
+  }, [address, bridgeClient]);
+
+  return (
+    <RelayerContext.Provider
+      value={{
+        margin,
+        baseFee,
+        feeRate,
+        sourceBalance,
+        targetBalance,
+        sourceAllowance,
+        targetAllowance,
+        bridgeCategory,
+        sourceChain,
+        targetChain,
+        transferToken,
+        defaultBridge,
+        oppositeBridge,
+        setMargin,
+        setBaseFee,
+        setFeeRate,
+        setBridgeCategory,
+        setSourceChain,
+        setTargetChain,
+        setTransferToken,
+        sourceApprove,
+        targetApprove,
+        depositMargin,
+        updateFeeAndMargin,
+      }}
+    >
+      {children}
+    </RelayerContext.Provider>
+  );
+}