logos-messaging-frontend/src/hooks/useWallet.ts

import React from "react";
import { useStore } from "./useStore";
import { isEthereumEvenEmitterValid } from "@/utils/ethereum";
import { useRLN } from "./useRLN";

type UseWalletResult = {
  onWalletConnect: () => void;
};

export const useWallet = (): UseWalletResult => {
  const { rln } = useRLN();
  const { setEthAccount, setChainID, setWallet } = useStore();

  React.useEffect(() => {
    const ethereum = window.ethereum;
    if (!isEthereumEvenEmitterValid(ethereum)) {
      console.log("Cannot subscribe to ethereum events.");
      return;
    }

    const onAccountsChanged = (accounts: string[]) => {
      setEthAccount(accounts[0] || "");
    };
    ethereum.on("accountsChanged", onAccountsChanged);

    const onChainChanged = (chainID: string) => {
      const ID = parseInt(chainID, 16);
      setChainID(ID);
    };
    ethereum.on("chainChanged", onChainChanged);

    return () => {
      ethereum.removeListener("chainChanged", onChainChanged);
      ethereum.removeListener("accountsChanged", onAccountsChanged);
    };
  }, [setEthAccount, setChainID]);

  const onWalletConnect = async () => {
    const ethereum = window.ethereum;
    
    if (!ethereum) {
      console.log("No ethereum instance found.");
      return;
    }

    if (!rln?.rlnInstance) {
      console.log("RLN instance is not initialized.");
      return;
    }
    
    try {
      const accounts = await window.ethereum.request({ method: 'eth_requestAccounts' }) as unknown as string[];
      await rln.initRLNContract(rln.rlnInstance);
      setWallet(accounts?.[0] || "");
    } catch(error) {
      console.error("Failed to conenct to wallet.");
    }
  };

  return {
    onWalletConnect,
  };
};