src/components/ContextMenu/context.tsx


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

import React from 'react'

import {
  type ContextType,
  type ItemContextType,
  type MenuContextType,
} from '#/components/ContextMenu/types'

export const Context = React.createContext<ContextType | null>(null)
Context.displayName = 'ContextMenuContext'

export const MenuContext = React.createContext<MenuContextType | null>(null)
MenuContext.displayName = 'ContextMenuMenuContext'

export const ItemContext = React.createContext<ItemContextType | null>(null)
ItemContext.displayName = 'ContextMenuItemContext'

export function useContextMenuContext() {
  const context = React.useContext(Context)

  if (!context) {
    throw new Error(
      'useContextMenuContext must be used within a Context.Provider',
    )
  }

  return context
}

export function useContextMenuMenuContext() {
  const context = React.useContext(MenuContext)

  if (!context) {
    throw new Error(
      'useContextMenuMenuContext must be used within a Context.Provider',
    )
  }

  return context
}

export function useContextMenuItemContext() {
  const context = React.useContext(ItemContext)

  if (!context) {
    throw new Error(
      'useContextMenuItemContext must be used within a Context.Provider',
    )
  }

  return context
}