newsnow/src/hooks/useRefetch.ts

import type { SourceID } from "@shared/types"

function initRefetchSources() {
  let time = 0
  // useOnReload
  // 没有放在 useOnReload 里面, 可以避免初始化后再修改 refetchSourceAtom，导致多次请求 API
  const _ = localStorage.getItem("quitTime")
  const now = Date.now()
  const quitTime = _ ? Number(_) : 0
  if (!Number.isNaN(quitTime) && now - quitTime < 1000) {
    time = now
  }
  return Object.fromEntries(Object.keys(sources).map(k => [k, time])) as Record<SourceID, number>
}

const refetchSourcesAtom = atom(initRefetchSources())
export function useRefetch() {
  const [refetchSource, setRefetchSource] = useAtom(refetchSourcesAtom)

  const refresh = useCallback((...sources: SourceID[]) => {
    const obj = Object.fromEntries(sources.map(id => [id, Date.now()]))
    setRefetchSource(prev => ({
      ...prev,
      ...obj,
    }))
  }, [setRefetchSource])

  const getRefreshId = useCallback((id: SourceID) => refetchSource[id], [refetchSource])

  return {
    refresh,
    getRefreshId,
  }
}