import { ChainType } from '@lifi/sdk'
import { SuiClientContext } from '@mysten/dapp-kit'
import { type FC, type PropsWithChildren, useContext } from 'react'
import { useInternalWalletProvider } from '../../hooks/useInternalWalletProvider.js'
import { isItemAllowed } from '../../utils/item.js'
import { useWidgetConfig } from '../WidgetProvider/WidgetProvider.js'
import { SuiBaseProvider } from './SuiBaseProvider.js'
import { SuiExternalContext } from './SuiExternalContext.js'

function useInSuiContext(): boolean {
  const { chains } = useWidgetConfig()
  const context = useContext(SuiClientContext)

  return Boolean(context) && isItemAllowed(ChainType.MVM, chains?.types)
}

export const SuiProvider: FC<PropsWithChildren> = ({ children }) => {
  const inSuiContext = useInSuiContext()
  const useInternalWallet = useInternalWalletProvider(inSuiContext)

  return useInternalWallet ? (
    <SuiBaseProvider>{children}</SuiBaseProvider>
  ) : (
    <SuiExternalContext.Provider value={inSuiContext}>
      {children}
    </SuiExternalContext.Provider>
  )
}