53 lines
1.5 KiB
TypeScript
53 lines
1.5 KiB
TypeScript
import _, { Dictionary } from 'lodash'
|
|
import { useState, useEffect } from 'react'
|
|
import type { feed } from 'common/feed'
|
|
import { useTimeSinceFirstRender } from './use-time-since-first-render'
|
|
import { trackLatency } from 'web/lib/firebase/tracking'
|
|
import { User } from 'common/user'
|
|
import { getCategoryFeeds, getUserFeed } from 'web/lib/firebase/users'
|
|
import {
|
|
getRecentBetsAndComments,
|
|
getTopWeeklyContracts,
|
|
} from 'web/lib/firebase/contracts'
|
|
|
|
export const useAlgoFeed = (
|
|
user: User | null | undefined,
|
|
category: string
|
|
) => {
|
|
const [allFeed, setAllFeed] = useState<feed>()
|
|
const [categoryFeeds, setCategoryFeeds] = useState<Dictionary<feed>>()
|
|
|
|
const getTime = useTimeSinceFirstRender()
|
|
|
|
useEffect(() => {
|
|
if (user) {
|
|
getUserFeed(user.id).then((feed) => {
|
|
if (feed.length === 0) {
|
|
getDefaultFeed().then((feed) => setAllFeed(feed))
|
|
} else setAllFeed(feed)
|
|
|
|
trackLatency('feed', getTime())
|
|
console.log('"all" feed load time', getTime())
|
|
})
|
|
|
|
getCategoryFeeds(user.id).then((feeds) => {
|
|
setCategoryFeeds(feeds)
|
|
console.log('category feeds load time', getTime())
|
|
})
|
|
}
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, [user?.id])
|
|
|
|
const feed = category === 'all' ? allFeed : categoryFeeds?.[category]
|
|
|
|
return feed
|
|
}
|
|
|
|
const getDefaultFeed = async () => {
|
|
const contracts = await getTopWeeklyContracts()
|
|
const feed = await Promise.all(
|
|
contracts.map((c) => getRecentBetsAndComments(c))
|
|
)
|
|
return feed
|
|
}
|