Overleaf/services/web/frontend/js/features/source-editor/context/review-panel/review-panel-context.tsx

import { createContext, useContext } from 'react'
import useAngularReviewPanelState from './hooks/use-angular-review-panel-state'
import {
  ReviewPanelReactIdeUpdaterFnsContext,
  ReviewPanelReactIdeValueContext,
} from '@/features/ide-react/context/review-panel/review-panel-context'
import { useIdeContext } from '@/shared/context/ide-context'
import { ReviewPanelState } from './types/review-panel-state'

const ReviewPanelValueContext = createContext<
  ReviewPanelState['values'] | undefined
>(undefined)

const ReviewPanelUpdaterFnsContext = createContext<
  ReviewPanelState['updaterFns'] | undefined
>(undefined)

type ReviewPanelProviderProps = {
  children?: React.ReactNode
}

export function ReviewPanelProvider({ children }: ReviewPanelProviderProps) {
  const { values, updaterFns } = useAngularReviewPanelState()

  return (
    <ReviewPanelValueContext.Provider value={values}>
      <ReviewPanelUpdaterFnsContext.Provider value={updaterFns}>
        {children}
      </ReviewPanelUpdaterFnsContext.Provider>
    </ReviewPanelValueContext.Provider>
  )
}

export function useReviewPanelValueContext() {
  const contextAngularIde = useContext(ReviewPanelValueContext)
  const contextReactIde = useContext(ReviewPanelReactIdeValueContext)
  const { isReactIde } = useIdeContext()
  const context = isReactIde ? contextReactIde : contextAngularIde

  if (!context) {
    throw new Error(
      'ReviewPanelValueContext is only available inside ReviewPanelProvider'
    )
  }
  return context
}

export function useReviewPanelUpdaterFnsContext() {
  const contextAngularIde = useContext(ReviewPanelUpdaterFnsContext)
  const contextReactIde = useContext(ReviewPanelReactIdeUpdaterFnsContext)
  const { isReactIde } = useIdeContext()
  const context = isReactIde ? contextReactIde : contextAngularIde

  if (!context) {
    throw new Error(
      'ReviewPanelUpdaterFnsContext is only available inside ReviewPanelProvider'
    )
  }
  return context
}