110 lines
2.8 KiB
TypeScript
110 lines
2.8 KiB
TypeScript
import { useFirestoreQueryData } from '@react-query-firebase/firestore'
|
|
import { isEqual } from 'lodash'
|
|
import { useEffect, useRef, useState } from 'react'
|
|
import {
|
|
Contract,
|
|
listenForActiveContracts,
|
|
listenForContract,
|
|
listenForContracts,
|
|
listenForHotContracts,
|
|
listenForInactiveContracts,
|
|
listenForNewContracts,
|
|
getUserBetContractsQuery,
|
|
} from 'web/lib/firebase/contracts'
|
|
import { QueryClient } from 'react-query'
|
|
|
|
export const useContracts = () => {
|
|
const [contracts, setContracts] = useState<Contract[] | undefined>()
|
|
|
|
useEffect(() => {
|
|
return listenForContracts(setContracts)
|
|
}, [])
|
|
|
|
return contracts
|
|
}
|
|
|
|
export const useActiveContracts = () => {
|
|
const [activeContracts, setActiveContracts] = useState<
|
|
Contract[] | undefined
|
|
>()
|
|
const [newContracts, setNewContracts] = useState<Contract[] | undefined>()
|
|
|
|
useEffect(() => {
|
|
return listenForActiveContracts(setActiveContracts)
|
|
}, [])
|
|
|
|
useEffect(() => {
|
|
return listenForNewContracts(setNewContracts)
|
|
}, [])
|
|
|
|
if (!activeContracts || !newContracts) return undefined
|
|
|
|
return [...activeContracts, ...newContracts]
|
|
}
|
|
|
|
export const useInactiveContracts = () => {
|
|
const [contracts, setContracts] = useState<Contract[] | undefined>()
|
|
|
|
useEffect(() => {
|
|
return listenForInactiveContracts(setContracts)
|
|
}, [])
|
|
|
|
return contracts
|
|
}
|
|
|
|
export const useHotContracts = () => {
|
|
const [hotContracts, setHotContracts] = useState<Contract[] | undefined>()
|
|
|
|
useEffect(() => listenForHotContracts(setHotContracts), [])
|
|
|
|
return hotContracts
|
|
}
|
|
|
|
export const useUpdatedContracts = (contracts: Contract[] | undefined) => {
|
|
const [__, triggerUpdate] = useState(0)
|
|
const contractDict = useRef<{ [id: string]: Contract }>({})
|
|
|
|
useEffect(() => {
|
|
if (contracts === undefined) return
|
|
|
|
contractDict.current = Object.fromEntries(contracts.map((c) => [c.id, c]))
|
|
|
|
const disposes = contracts.map((contract) => {
|
|
const { id } = contract
|
|
|
|
return listenForContract(id, (contract) => {
|
|
const curr = contractDict.current[id]
|
|
if (!isEqual(curr, contract)) {
|
|
contractDict.current[id] = contract as Contract
|
|
triggerUpdate((n) => n + 1)
|
|
}
|
|
})
|
|
})
|
|
|
|
triggerUpdate((n) => n + 1)
|
|
|
|
return () => {
|
|
disposes.forEach((dispose) => dispose())
|
|
}
|
|
}, [!!contracts])
|
|
|
|
return contracts && Object.keys(contractDict.current).length > 0
|
|
? contracts.map((c) => contractDict.current[c.id])
|
|
: undefined
|
|
}
|
|
|
|
const queryClient = new QueryClient()
|
|
|
|
export const prefetchUserBetContracts = (userId: string) =>
|
|
queryClient.prefetchQuery(['contracts', 'bets', userId], () =>
|
|
getUserBetContractsQuery(userId)
|
|
)
|
|
|
|
export const useUserBetContracts = (userId: string) => {
|
|
const result = useFirestoreQueryData(
|
|
['contracts', 'bets', userId],
|
|
getUserBetContractsQuery(userId)
|
|
)
|
|
return result.data
|
|
}
|