portainer/app/edge/devices/components/EdgeDevicesDatatable/columns/RowContext.tsx

import { createContext, useContext, useMemo, PropsWithChildren } from 'react';

interface RowContextState {
  disableTrustOnFirstConnect: boolean;
  isOpenAmtEnabled: boolean;
  groupName?: string;
}

const RowContext = createContext<RowContextState | null>(null);

export interface RowProviderProps {
  disableTrustOnFirstConnect: boolean;
  groupName?: string;
  isOpenAmtEnabled: boolean;
}

export function RowProvider({
  disableTrustOnFirstConnect,
  groupName,
  isOpenAmtEnabled,
  children,
}: PropsWithChildren<RowProviderProps>) {
  const state = useMemo(
    () => ({ disableTrustOnFirstConnect, groupName, isOpenAmtEnabled }),
    [disableTrustOnFirstConnect, groupName, isOpenAmtEnabled]
  );

  return <RowContext.Provider value={state}>{children}</RowContext.Provider>;
}

export function useRowContext() {
  const context = useContext(RowContext);
  if (!context) {
    throw new Error('should be nested under RowProvider');
  }

  return context;
}