about summary refs log tree commit diff
path: root/src/components/Portal.tsx
blob: 1813d9e05e9b7add00a0a5dbff22405a4ec3496e (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
import React from 'react'

type Component = React.ReactElement

type ContextType = {
  outlet: Component | null
  append(id: string, component: Component): void
  remove(id: string): void
}

type ComponentMap = {
  [id: string]: Component
}

export const Context = React.createContext<ContextType>({
  outlet: null,
  append: () => {},
  remove: () => {},
})

export function Provider(props: React.PropsWithChildren<{}>) {
  const map = React.useRef<ComponentMap>({})
  const [outlet, setOutlet] = React.useState<ContextType['outlet']>(null)

  const append = React.useCallback<ContextType['append']>((id, component) => {
    if (map.current[id]) return
    map.current[id] = <React.Fragment key={id}>{component}</React.Fragment>
    setOutlet(<>{Object.values(map.current)}</>)
  }, [])

  const remove = React.useCallback<ContextType['remove']>(id => {
    delete map.current[id]
    setOutlet(<>{Object.values(map.current)}</>)
  }, [])

  return (
    <Context.Provider value={{outlet, append, remove}}>
      {props.children}
    </Context.Provider>
  )
}

export function Outlet() {
  const ctx = React.useContext(Context)
  return ctx.outlet
}

export function Portal({children}: React.PropsWithChildren<{}>) {
  const {append, remove} = React.useContext(Context)
  const id = React.useId()
  React.useEffect(() => {
    append(id, children as Component)
    return () => remove(id)
  }, [id, children, append, remove])
  return null
}