@mezo-org/passport 0.4.0-dev.8 → 0.4.0-dev.80

package/src/hooks/useTokensBalances.ts

@@ -0,0 +1,187 @@
+import { useReadContracts } from "wagmi"
+import { useCallback, useMemo } from "react"
+import { Abi } from "viem"
+import { useQueryClient } from "@tanstack/react-query"
+import { usePassportContext } from "./usePassportContext"
+import {
+  mainnetTokenContracts,
+  testnetTokenContracts,
+  MezoChainToken,
+} from "../lib/contracts"
+import useWalletAccount from "./useWalletAccount"
+import { CHAIN_ID } from "../constants"
+
+// Wagmi handles typesafety with ABI const assertions. TypeScript doesn't
+// support importing JSON as const yet so types cannot be inferred from the
+// imported contract. As a workaround there is minimal ABI definition that can
+// be asserted types from.
+// Ref: https://wagmi.sh/core/typescript#const-assert-abis-typed-data
+
+const BALANCE_OF_ABI = [
+  {
+    inputs: [
+      {
+        internalType: "address",
+        name: "account",
+        type: "address",
+      },
+    ],
+    name: "balanceOf",
+    outputs: [
+      {
+        internalType: "uint256",
+        name: "",
+        type: "uint256",
+      },
+    ],
+    stateMutability: "view",
+    type: "function",
+  },
+] as const satisfies Abi
+
+const BALANCE_TOKENS: MezoChainToken[] = [
+  "mcbBTC",
+  "mDAI",
+  "mFBTC",
+  "mSolvBTC",
+  "mswBTC",
+  "mT",
+  "mUSDC",
+  "mUSDe",
+  "mUSDT",
+  "mxSolvBTC",
+  "MUSD",
+]
+
+type UseMezoChainTokensBalancesOptions<T> = {
+  tokens?: T
+  queryOptions?: object
+}
+
+/**
+ * Hook to get the balance of a list of tokens for the current account
+ * @param   options.tokens The list of tokens to get the balance for. It will
+ *          fallback to all tokens if not provided.
+ * @param   options.queryOptions The query options to pass to the
+ *          `useReadContracts`
+ * @returns Tanstack's `useQuery` returnings with balance of tokens for the
+ *          current account in form of typesafe object with token names as keys
+ *          and balances as values.
+ * @dev     In case of error data fallbacks to `0n`. To determine if an error
+ *          occurred, check the `isError` property.
+ * @example
+ * const mezoTokensBalance = useTokensBalances({
+ *   tokens: ["mT", "mxSolvBTC"],
+ * })
+ * console.log(mezoTokensBalance?.data.mT) // Eg. 0n
+ * console.log(Object.keys(mezoTokensBalance?.data)) // ["mT", "mxSolvBTC"]
+ */
+export function useTokensBalances<T extends MezoChainToken[]>(
+  options: UseMezoChainTokensBalancesOptions<T> = {},
+) {
+  const { environment = "mainnet" } = usePassportContext()
+
+  const { accountAddress } = useWalletAccount()
+
+  const contractsMap = useMemo(() => {
+    const contracts =
+      environment === "mainnet" ? mainnetTokenContracts : testnetTokenContracts
+
+    const tokenContracts = Object.fromEntries(
+      Object.entries(contracts).filter(([key]) =>
+        BALANCE_TOKENS.includes(key as MezoChainToken),
+      ),
+    ) as typeof contracts
+
+    return tokenContracts
+  }, [environment])
+
+  const { tokens = Object.keys(contractsMap) as T, queryOptions = {} } = options
+
+  const contracts = useMemo(() => {
+    if (!accountAddress) return []
+
+    return tokens.map((token) => {
+      const { address } = contractsMap[token]
+
+      return {
+        address,
+        abi: BALANCE_OF_ABI,
+        functionName: "balanceOf",
+        args: [accountAddress],
+        chainId: CHAIN_ID[environment],
+      }
+    })
+  }, [accountAddress, tokens, contractsMap, environment])
+
+  return useReadContracts({
+    contracts,
+    query: {
+      select: (data) =>
+        data.reduce(
+          (acc, item) => {
+            const token = tokens[data.indexOf(item)]
+
+            return {
+              ...acc,
+              [token]: item.result ?? 0n,
+            }
+          },
+          {} as Record<T[number], bigint>,
+        ),
+      ...queryOptions,
+    },
+  })
+}
+
+/**
+ * Hook for invalidating current user's token balances. Can be used to
+ * invalidate the balances manually, which forces the data to be re-fetched.
+ * @param tokens The list of tokens (as string arrays) for which we want to
+ * invalidate query for.
+ * @returns Function `invalidateTokenBalances` that invalidates token balances
+ * @example
+ * const { invalidateTokenBalances } = useInvalidateTokensBalances(["mT", "mUSD"])
+ * (...)
+ * await invalidateTokenBalances()
+ */
+export function useInvalidateTokensBalances(tokens?: MezoChainToken[]) {
+  const queryClient = useQueryClient()
+
+  const { queryKey } = useTokensBalances({ tokens })
+
+  const invalidateTokensBalancesHandler = useCallback(
+    () => queryClient.invalidateQueries({ queryKey }),
+    [queryClient, queryKey],
+  )
+
+  return {
+    invalidateTokensBalances: invalidateTokensBalancesHandler,
+  }
+}
+
+/**
+ * Hook for resetting current user's token balances. Can be used to reset the
+ * balances manually, which forces the data to be re-fetched.
+ * @param tokens The list of tokens (as string arrays) for which we want to
+ * reset query for.
+ * @returns Function `resetTokenBalances` that invalidates token balances
+ * @example
+ * const { resetTokenBalances } = useResetTokensBalances(["mT", "mUSD"])
+ * (...)
+ * await resetTokenBalances()
+ */
+export function useResetTokensBalances(tokens?: MezoChainToken[]) {
+  const queryClient = useQueryClient()
+
+  const { queryKey } = useTokensBalances({ tokens })
+
+  const resetTokenBalancesHandler = useCallback(
+    () => queryClient.resetQueries({ queryKey }),
+    [queryClient, queryKey],
+  )
+
+  return {
+    resetTokenBalances: resetTokenBalancesHandler,
+  }
+}

package/src/hooks/useValidateMezoId.ts

@@ -0,0 +1,31 @@
+import { UseQueryResult } from "@tanstack/react-query"
+import { validateMezoId, MezoId } from "../utils/mezoId"
+import { useGetAccountByMezoId } from "./useGetAccountByMezoId"
+
+export function useValidateMezoId(mezoId: MezoId) {
+  const validationErrors = validateMezoId(mezoId)
+
+  return useGetAccountByMezoId(mezoId, {
+    placeholderData: {
+      // @ts-expect-error TODO: Couldn't fix type for placeholderData which for
+      // some reason should return the same data as the data returned in
+      // queryFn.
+      isValid: validationErrors.length === 0,
+      isAvailable: validationErrors.length === 0,
+      errors: validationErrors,
+    },
+    enabled: validationErrors.length === 0,
+    // @ts-expect-error TODO: Couldn't fix type for select which for some reason
+    // should return the same data as the data returned in queryFn.
+    select: (data) => {
+      const isValid = validationErrors.length === 0
+      const isAvailable = isValid ? !data : false
+
+      return { isValid, isAvailable, errors: validationErrors }
+    },
+  }) as UseQueryResult<{
+    isValid: boolean
+    isAvailable: boolean
+    errors: string[]
+  }>
+}

package/src/hooks/useWalletAccount.ts

@@ -1,10 +1,12 @@
-import { useBitcoinAccount } from "@mezo-org/orangekit"
-import { useEffect, useMemo, useState } from "react"
+import { OrangeKitConnector } from "@mezo-org/orangekit"
+import { useQuery } from "@tanstack/react-query"
+import { useMemo } from "react"
 import { Address } from "viem"
 import { Connector, useAccount } from "wagmi"
 
 type UseWalletAccountReturn = {
-  address?: string | Address
+  accountAddress?: Address
+  walletAddress?: string
   isConnected: boolean
   networkFamily: "bitcoin" | "evm"
   connector?: Connector
@@ -13,27 +15,35 @@ type UseWalletAccountReturn = {
 
 export default function useWalletAccount(): UseWalletAccountReturn {
   const { address: evmAddress, connector, chainId } = useAccount()
-  const { btcAddress } = useBitcoinAccount()
-  const [isConnected, setIsConnected] = useState(false)
-
-  useEffect(() => {
-    if (!connector) {
-      setIsConnected(false)
-      return
-    }
-
-    const isBitcoinFlow = connector.type === "orangekit"
-    setIsConnected(isBitcoinFlow ? !!btcAddress : !!evmAddress)
-  }, [btcAddress, connector, evmAddress])
 
   const networkFamily = useMemo(
-    () => (btcAddress ? "bitcoin" : "evm"),
-    [btcAddress],
+    () => (connector?.type !== "orangekit" ? "evm" : "bitcoin"),
+    [connector],
   )
 
+  const { data: btcData } = useQuery({
+    queryKey: [connector?.id],
+    queryFn: async () => {
+      const bitcoinProvider = (
+        connector as unknown as OrangeKitConnector
+      ).getBitcoinProvider()
+
+      const address = await bitcoinProvider.getAddress()
+      const balance = await bitcoinProvider.getBalance()
+
+      return {
+        address,
+        balance,
+      }
+    },
+  })
+
+  const { address: btcAddress } = btcData || {}
+
   return {
-    address: btcAddress || evmAddress,
-    isConnected,
+    accountAddress: evmAddress,
+    walletAddress: btcAddress ?? evmAddress,
+    isConnected: networkFamily === "bitcoin" ? !!btcAddress : !!evmAddress,
     networkFamily,
     connector,
     chainId,

package/src/hooks/useWatchTransferEvents.ts

@@ -0,0 +1,74 @@
+import { useMemo } from "react"
+import { useAccount, useWatchContractEvent } from "wagmi"
+import {
+  mainnetTokenContracts,
+  MezoChainToken,
+  testnetTokenContracts,
+} from "../lib/contracts"
+import { usePassportContext } from "./usePassportContext"
+import { CHAIN_ID } from "../constants"
+import { useRefreshPassport } from "./useRefreshPassport"
+import { simpleWssMezoConfig } from "../config"
+
+function useWatchTransferEvents(token: MezoChainToken) {
+  const { address: userAddress } = useAccount()
+  const { environment = "mainnet" } = usePassportContext()
+  const { refreshTokensBalances, refreshBorrowData } = useRefreshPassport()
+
+  const contract = useMemo(() => {
+    const contracts =
+      environment === "mainnet" ? mainnetTokenContracts : testnetTokenContracts
+
+    return contracts[token]
+  }, [environment, token])
+
+  useWatchContractEvent({
+    address: contract.address,
+    abi: contract.abi,
+    eventName: "Transfer",
+    args: {
+      from: userAddress,
+    },
+    chainId: CHAIN_ID[environment],
+    enabled: !!userAddress,
+    config: simpleWssMezoConfig,
+    onLogs() {
+      console.log(`[${token}] Transfer from: ${userAddress}`)
+      refreshTokensBalances()
+      refreshBorrowData()
+    },
+  })
+
+  useWatchContractEvent({
+    address: contract.address,
+    abi: contract.abi,
+    eventName: "Transfer",
+    args: {
+      to: userAddress,
+    },
+    chainId: CHAIN_ID[environment],
+    enabled: !!userAddress,
+    config: simpleWssMezoConfig,
+    onLogs() {
+      console.log(`[${token}] Transfer to: ${userAddress}`)
+      refreshTokensBalances()
+      refreshBorrowData()
+    },
+  })
+}
+
+function useWatchTransferEventsForAllTokens() {
+  useWatchTransferEvents("mcbBTC")
+  useWatchTransferEvents("mDAI")
+  useWatchTransferEvents("mFBTC")
+  useWatchTransferEvents("mSolvBTC")
+  useWatchTransferEvents("mswBTC")
+  useWatchTransferEvents("mT")
+  useWatchTransferEvents("mUSDC")
+  useWatchTransferEvents("mUSDe")
+  useWatchTransferEvents("mUSDT")
+  useWatchTransferEvents("mxSolvBTC")
+  useWatchTransferEvents("MUSD")
+}
+
+export { useWatchTransferEvents, useWatchTransferEventsForAllTokens }

package/src/index.ts

@@ -1,6 +1,17 @@
 export * from "./components"
-export * from "./config"
+export {
+  defaultConfig,
+  unisatWalletMezoMainnet,
+  unisatWalletMezoTestnet,
+  okxWalletMezoMainnet,
+  okxWalletMezoTestnet,
+  xverseWalletMezoMainnet,
+  xverseWalletMezoTestnet,
+  getDefaultWallets,
+  getConfig,
+} from "./config"
 export * from "./constants"
 export * from "./hooks"
 export * from "./provider"
 export * from "./wallet"
+export { DISALLOWED_MEZO_ID_PATTERNS } from "./utils/mezoId"

package/src/lib/contracts/index.ts

@@ -0,0 +1,92 @@
+import cbbtcMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mcbBTC.json"
+import daiMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mDAI.json"
+import fbtcMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mFBTC.json"
+import solvbtcMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mSolvBTC.json"
+import swbtcMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mswBTC.json"
+import tMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mT.json"
+import usdcMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mUSDC.json"
+import usdeMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mUSDe.json"
+import usdtMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mUSDT.json"
+import xsolvbtcMainnet from "@mezo-org/mezod-contracts/deployments/mainnet/mxSolvBTC.json"
+import musdMainnet from "@mezo-org/musd-contracts/deployments/mainnet/MUSD.json"
+import troveManagerMainnet from "@mezo-org/musd-contracts/deployments/mainnet/TroveManager.json"
+import borrowerOperationsMainnet from "@mezo-org/musd-contracts/deployments/mainnet/BorrowerOperations.json"
+
+import cbbtcTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mcbBTC.json"
+import daiTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mDAI.json"
+import fbtcTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mFBTC.json"
+import solvbtcTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mSolvBTC.json"
+import swbtcTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mswBTC.json"
+import tTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mT.json"
+import usdcTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mUSDC.json"
+import usdeTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mUSDe.json"
+import usdtTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mUSDT.json"
+import xsolvbtcTestnet from "@mezo-org/mezod-contracts/deployments/testnet/mxSolvBTC.json"
+import musdTestnet from "@mezo-org/musd-contracts/deployments/matsnet/MUSD.json"
+import troveManagerTestnet from "@mezo-org/musd-contracts/deployments/matsnet/TroveManager.json"
+import borrowerOperationsTestnet from "@mezo-org/musd-contracts/deployments/matsnet/BorrowerOperations.json"
+
+import { Abi, Address } from "viem"
+
+export type MezoChainToken =
+  | "mcbBTC"
+  | "mDAI"
+  | "mFBTC"
+  | "mSolvBTC"
+  | "mswBTC"
+  | "mT"
+  | "mUSDC"
+  | "mUSDe"
+  | "mUSDT"
+  | "mxSolvBTC"
+  | "MUSD"
+
+export type MezoBorrowContract = "TroveManager" | "BorrowerOperations"
+
+type ContractsMap<K extends string> = Record<
+  K,
+  {
+    address: Address
+    abi: Abi
+  }
+>
+
+export const testnetTokenContracts = {
+  mcbBTC: cbbtcTestnet,
+  mDAI: daiTestnet,
+  mFBTC: fbtcTestnet,
+  mSolvBTC: solvbtcTestnet,
+  mswBTC: swbtcTestnet,
+  mT: tTestnet,
+  mUSDC: usdcTestnet,
+  mUSDe: usdeTestnet,
+  mUSDT: usdtTestnet,
+  mxSolvBTC: xsolvbtcTestnet,
+  MUSD: musdTestnet,
+} as unknown as ContractsMap<MezoChainToken>
+
+export const mainnetTokenContracts = {
+  mcbBTC: cbbtcMainnet,
+  mDAI: daiMainnet,
+  mFBTC: fbtcMainnet,
+  mSolvBTC: solvbtcMainnet,
+  mswBTC: swbtcMainnet,
+  mT: tMainnet,
+  mUSDC: usdcMainnet,
+  mUSDe: usdeMainnet,
+  mUSDT: usdtMainnet,
+  mxSolvBTC: xsolvbtcMainnet,
+  MUSD: musdMainnet,
+} as unknown as ContractsMap<MezoChainToken>
+
+export const mainnetBorrowContracts = {
+  TroveManager: troveManagerMainnet,
+  BorrowerOperations: borrowerOperationsMainnet,
+} as unknown as ContractsMap<MezoBorrowContract>
+
+export const testnetBorrowContracts = {
+  TroveManager: troveManagerTestnet,
+  BorrowerOperations: borrowerOperationsTestnet,
+} as unknown as ContractsMap<MezoBorrowContract>
+
+export { default as priceOracleContract } from "./priceOracle"

package/src/lib/contracts/priceOracle.ts

@@ -0,0 +1,53 @@
+import { Address } from "viem"
+
+export default {
+  address: "0x7b7c000000000000000000000000000000000015" as Address,
+  abi: [
+    {
+      inputs: [],
+      name: "decimals",
+      outputs: [
+        {
+          internalType: "uint8",
+          name: "",
+          type: "uint8",
+        },
+      ],
+      stateMutability: "view",
+      type: "function",
+    },
+    {
+      inputs: [],
+      name: "latestRoundData",
+      outputs: [
+        {
+          internalType: "uint80",
+          name: "roundId",
+          type: "uint80",
+        },
+        {
+          internalType: "int256",
+          name: "answer",
+          type: "int256",
+        },
+        {
+          internalType: "uint256",
+          name: "startedAt",
+          type: "uint256",
+        },
+        {
+          internalType: "uint256",
+          name: "updatedAt",
+          type: "uint256",
+        },
+        {
+          internalType: "uint80",
+          name: "answeredInRound",
+          type: "uint80",
+        },
+      ],
+      stateMutability: "view",
+      type: "function",
+    },
+  ] as const,
+}

package/src/provider.ts

@@ -29,7 +29,7 @@ export function PassportProvider({
   portalApiUrl,
   children,
 }: PassportProviderProps) {
-  if (environment !== "mainnet" && environment !== "testnet") {
+  if (environment && !["mainnet", "testnet"].includes(environment)) {
     throw new Error("Wrong environment passed to PassportProvider.")
   }
 

package/src/stores/dropdownStore.ts

@@ -0,0 +1,20 @@
+import { create } from "zustand"
+
+export enum DropdownView {
+  ROOT = "ROOT",
+  RECEIVE = "RECEIVE",
+}
+
+interface DropdownStore {
+  view: DropdownView
+  setView: (view: DropdownView) => void
+  isNestedView: () => boolean
+}
+
+const useDropdownStore = create<DropdownStore>((set, get) => ({
+  view: DropdownView.ROOT,
+  setView: (view: DropdownView) => set({ view }),
+  isNestedView: () => get().view !== DropdownView.ROOT,
+}))
+
+export default useDropdownStore

package/src/utils/address.test.ts

@@ -10,12 +10,6 @@ describe("trimAddress", () => {
     const expected = "0x123...bcdef"
     expect(trimAddress(input)).toBe(expected)
   })
-
-  it("returns lowercase result", () => {
-    const input = "0xABCDEF1234567890"
-    const expected = "0xabc...67890"
-    expect(trimAddress(input)).toBe(expected)
-  })
 })
 
 describe("getAddressExplorerUrl", () => {
@@ -41,4 +35,14 @@ describe("getAddressExplorerUrl", () => {
     const url = getAddressExplorerUrl(evmAddress, "evm", true)
     expect(url).toBe(`https://sepolia.etherscan.io/address/${evmAddress}`)
   })
+
+  it("returns correct mainnet Mezo explorer URL", () => {
+    const url = getAddressExplorerUrl(evmAddress, "mezo", false)
+    expect(url).toBe(`https://explorer.mezo.org/address/${evmAddress}`)
+  })
+
+  it("returns correct testnet Mezo explorer URL", () => {
+    const url = getAddressExplorerUrl(evmAddress, "mezo", true)
+    expect(url).toBe(`https://explorer.test.mezo.org/address/${evmAddress}`)
+  })
 })

package/src/utils/address.ts

@@ -11,14 +11,16 @@ export function trimAddress(address: string): string {
   return trimmedAddress
 }
 
-const MAINNET_EXPLORER_URLS: Record<"bitcoin" | "evm", string> = {
+const MAINNET_EXPLORER_URLS: Record<"bitcoin" | "evm" | "mezo", string> = {
   bitcoin: "https://mempool.space",
   evm: "https://etherscan.io",
+  mezo: "https://explorer.mezo.org",
 }
 
-const TESTNET_EXPLORER_URLS: Record<"bitcoin" | "evm", string> = {
+const TESTNET_EXPLORER_URLS: Record<"bitcoin" | "evm" | "mezo", string> = {
   bitcoin: "https://mempool.space/testnet",
   evm: "https://sepolia.etherscan.io",
+  mezo: "https://explorer.test.mezo.org",
 }
 
 /**
@@ -30,7 +32,7 @@ const TESTNET_EXPLORER_URLS: Record<"bitcoin" | "evm", string> = {
  */
 export function getAddressExplorerUrl(
   address: string,
-  chainType: "bitcoin" | "evm",
+  chainType: "bitcoin" | "evm" | "mezo",
   isTestnet: boolean,
 ) {
   const EXPLORER_URLS = isTestnet