supabase/apps/studio/components/interfaces/Database/Schemas/SchemaGraphContext.tsx

import { Edge } from '@xyflow/react'
import { createContext, useContext, type ReactNode } from 'react'

export type SchemaGraphContextType = {
  isDownloading: boolean
  selectedEdge: Edge | undefined
  onEditColumn: (tableId: number, columnId: string) => void
  onEditTable: (tableId: number) => void
}

export const SchemaGraphContext = createContext<SchemaGraphContextType | null>(null)

export const SchemaGraphContextProvider = ({
  children,
  value,
}: {
  children: ReactNode
  value: SchemaGraphContextType
}) => <SchemaGraphContext.Provider value={value}>{children}</SchemaGraphContext.Provider>

export const useSchemaGraphContext = () => {
  const context = useContext(SchemaGraphContext)
  if (!context)
    throw new Error('useSchemaGraphContext must be used inside a <SchemaGraphContextProvider>')
  return context
}