mango-ui-v3/hooks/useConnection.tsx

import { useEffect, useMemo } from 'react'
import { Account, Connection } from '@solana/web3.js'
// import { IDS } from '@blockworks-foundation/mango-client'
import useMangoStore, {
  programId,
  serumProgramId,
} from '../stores/useMangoStore'

const useConnection = () => {
  const setMangoStore = useMangoStore((s) => s.set)
  const {
    cluster,
    current: connection,
    endpoint,
  } = useMangoStore((s) => s.connection)

  const sendConnection = useMemo(
    () => new Connection(endpoint, 'recent'),
    [endpoint]
  )

  useEffect(() => {
    if (connection && endpoint === connection['_rpcEndpoint']) return

    const newConnection = new Connection(endpoint, 'recent')
    setMangoStore((state) => {
      state.connection.current = newConnection
    })
  }, [endpoint])

  useEffect(() => {
    if (connection && endpoint === connection['_rpcEndpoint']) return
    const id = connection.onAccountChange(new Account().publicKey, () => {})
    return () => {
      connection.removeAccountChangeListener(id)
    }
  }, [endpoint])

  useEffect(() => {
    if (connection && endpoint === connection['_rpcEndpoint']) return
    const id = connection.onSlotChange(() => null)
    return () => {
      connection.removeSlotChangeListener(id)
    }
  }, [endpoint])

  // const programId = useMemo(() => IDS[cluster].mango_program_id, [cluster])
  // const dexProgramId = useMemo(() => IDS[cluster]?.dex_program_id, [cluster])
  const dexProgramId = serumProgramId

  return { connection, dexProgramId, cluster, programId, sendConnection }
}

export default useConnection