about summary refs log tree commit diff
path: root/src/state/unstable-post-source.tsx
blob: 1fb4af287293a6dcf8b53d4cf28b2c3f2db010ce (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import {createContext, useCallback, useContext, useState} from 'react'
import {type AppBskyFeedDefs} from '@atproto/api'

import {type FeedDescriptor} from './queries/post-feed'

/**
 * For passing the source of the post (i.e. the original post, from the feed) to the threadview,
 * without using query params. Deliberately unstable to avoid using query params, use for FeedFeedback
 * and other ephemeral non-critical systems.
 */

type Source = {
  post: AppBskyFeedDefs.FeedViewPost
  feed?: FeedDescriptor
}

const SetUnstablePostSourceContext = createContext<
  (key: string, source: Source) => void
>(() => {})
const ConsumeUnstablePostSourceContext = createContext<
  (uri: string) => Source | undefined
>(() => undefined)

export function Provider({children}: {children: React.ReactNode}) {
  const [sources, setSources] = useState<Map<string, Source>>(() => new Map())

  const setUnstablePostSource = useCallback((key: string, source: Source) => {
    setSources(prev => {
      const newMap = new Map(prev)
      newMap.set(key, source)
      return newMap
    })
  }, [])

  const consumeUnstablePostSource = useCallback(
    (uri: string) => {
      const source = sources.get(uri)
      if (source) {
        setSources(prev => {
          const newMap = new Map(prev)
          newMap.delete(uri)
          return newMap
        })
      }
      return source
    },
    [sources],
  )

  return (
    <SetUnstablePostSourceContext.Provider value={setUnstablePostSource}>
      <ConsumeUnstablePostSourceContext.Provider
        value={consumeUnstablePostSource}>
        {children}
      </ConsumeUnstablePostSourceContext.Provider>
    </SetUnstablePostSourceContext.Provider>
  )
}

export function useSetUnstablePostSource() {
  return useContext(SetUnstablePostSourceContext)
}

/**
 * DANGER - This hook is unstable and should only be used for FeedFeedback
 * and other ephemeral non-critical systems. Does not change when the URI changes.
 */
export function useUnstablePostSource(uri: string) {
  const consume = useContext(ConsumeUnstablePostSourceContext)

  const [source] = useState(() => consume(uri))
  return source
}