48 lines
1.4 KiB
TypeScript
48 lines
1.4 KiB
TypeScript
import { useEffect } from 'react'
|
|
import { useFirestoreDocumentData } from '@react-query-firebase/firestore'
|
|
import {
|
|
Contract,
|
|
contracts,
|
|
getContractFromId,
|
|
listenForContract,
|
|
} from 'web/lib/firebase/contracts'
|
|
import { useStateCheckEquality } from './use-state-check-equality'
|
|
import { doc, DocumentData } from 'firebase/firestore'
|
|
import { useQuery } from 'react-query'
|
|
|
|
export const useContract = (contractId: string) => {
|
|
const result = useFirestoreDocumentData<DocumentData, Contract>(
|
|
['contracts', contractId],
|
|
doc(contracts, contractId),
|
|
{ subscribe: true, includeMetadataChanges: true }
|
|
)
|
|
|
|
return result.isLoading ? undefined : result.data
|
|
}
|
|
|
|
export const useContractsFromIds = (contractIds: string[]) => {
|
|
const contractResult = useQuery(['contracts', contractIds], () =>
|
|
Promise.all(contractIds.map(getContractFromId))
|
|
)
|
|
const contracts = contractResult.data?.filter(
|
|
(contract): contract is Contract => !!contract
|
|
)
|
|
|
|
return contractResult.isLoading ? undefined : contracts
|
|
}
|
|
|
|
export const useContractWithPreload = (
|
|
initial: Contract | null | undefined
|
|
) => {
|
|
const [contract, setContract] = useStateCheckEquality<
|
|
Contract | null | undefined
|
|
>(initial)
|
|
const contractId = initial?.id
|
|
|
|
useEffect(() => {
|
|
if (contractId) return listenForContract(contractId, setContract)
|
|
}, [contractId, setContract])
|
|
|
|
return contract
|
|
}
|