49 lines
1.4 KiB
TypeScript
49 lines
1.4 KiB
TypeScript
import { useQueryClient } from 'react-query'
|
|
import { useFirestoreQueryData } from '@react-query-firebase/firestore'
|
|
import { DAY_MS, HOUR_MS, MINUTE_MS, sleep } from 'common/util/time'
|
|
import {
|
|
getPortfolioHistory,
|
|
getPortfolioHistoryQuery,
|
|
Period,
|
|
} from 'web/lib/firebase/users'
|
|
|
|
const getCutoff = (period: Period) => {
|
|
const nowRounded = Math.round(Date.now() / HOUR_MS) * HOUR_MS
|
|
return periodToCutoff(nowRounded, period).valueOf()
|
|
}
|
|
|
|
export const usePrefetchPortfolioHistory = (userId: string, period: Period) => {
|
|
const queryClient = useQueryClient()
|
|
const cutoff = getCutoff(period)
|
|
return queryClient.prefetchQuery(
|
|
['portfolio-history', userId, cutoff],
|
|
() => sleep(1000).then(() => getPortfolioHistory(userId, cutoff)),
|
|
{ staleTime: 15 * MINUTE_MS }
|
|
)
|
|
}
|
|
|
|
export const usePortfolioHistory = (userId: string, period: Period) => {
|
|
const cutoff = getCutoff(period)
|
|
const result = useFirestoreQueryData(
|
|
['portfolio-history', userId, cutoff],
|
|
getPortfolioHistoryQuery(userId, cutoff),
|
|
{},
|
|
{ staleTime: 15 * MINUTE_MS }
|
|
)
|
|
return result.data
|
|
}
|
|
|
|
const periodToCutoff = (now: number, period: Period) => {
|
|
switch (period) {
|
|
case 'daily':
|
|
return now - 1 * DAY_MS
|
|
case 'weekly':
|
|
return now - 7 * DAY_MS
|
|
case 'monthly':
|
|
return now - 30 * DAY_MS
|
|
case 'allTime':
|
|
default:
|
|
return new Date(0)
|
|
}
|
|
}
|