Verso/services/web/frontend/js/features/file-tree/contexts/file-tree-main.tsx

import { createContext, FC, useContext, useState } from 'react'

type ContextMenuCoords = { top: number; left: number }

const FileTreeMainContext = createContext<
  | {
      refProviders: object
      setRefProviderEnabled: (provider: string, value: boolean) => void
      setStartedFreeTrial: (value: boolean) => void
      contextMenuCoords: ContextMenuCoords | null
      setContextMenuCoords: (value: ContextMenuCoords | null) => void
      contextMenuEntityId: string | null
      setContextMenuEntityId: (value: string | null) => void
    }
  | undefined
>(undefined)

export function useFileTreeMainContext() {
  const context = useContext(FileTreeMainContext)

  if (!context) {
    throw new Error(
      'useFileTreeMainContext is only available inside FileTreeMainProvider'
    )
  }

  return context
}

export const FileTreeMainProvider: FC<
  React.PropsWithChildren<{
    refProviders: object
    setRefProviderEnabled: (provider: string, value: boolean) => void
    setStartedFreeTrial: (value: boolean) => void
  }>
> = ({
  refProviders,
  setRefProviderEnabled,
  setStartedFreeTrial,
  children,
}) => {
  const [contextMenuCoords, setContextMenuCoords] =
    useState<ContextMenuCoords | null>(null)
  const [contextMenuEntityId, setContextMenuEntityId] = useState<string | null>(
    null
  )

  return (
    <FileTreeMainContext.Provider
      value={{
        refProviders,
        setRefProviderEnabled,
        setStartedFreeTrial,
        contextMenuCoords,
        setContextMenuCoords,
        contextMenuEntityId,
        setContextMenuEntityId,
      }}
    >
      {children}
    </FileTreeMainContext.Provider>
  )
}