unleash.unleash/frontend/src/hooks/useConditionallyHiddenColumns.ts

import { useEffect } from 'react';

interface IConditionallyHiddenColumns {
    condition: boolean;
    columns: string[];
}

export const useConditionallyHiddenColumns = (
    conditionallyHiddenColumns: IConditionallyHiddenColumns[],
    setHiddenColumns: (
        columns: string[] | ((columns: string[]) => string[])
    ) => void,
    columnsDefinition: unknown[]
) => {
    useEffect(() => {
        const columnsToHide = conditionallyHiddenColumns
            .filter(({ condition }) => condition)
            .flatMap(({ columns }) => columns);

        const columnsToShow = conditionallyHiddenColumns
            .flatMap(({ columns }) => columns)
            .filter(column => !columnsToHide.includes(column));

        setHiddenColumns(columns => [
            ...new Set(
                [...columns, ...columnsToHide].filter(
                    column => !columnsToShow.includes(column)
                )
            ),
        ]);
    }, [
        ...conditionallyHiddenColumns.map(({ condition }) => condition),
        setHiddenColumns,
        columnsDefinition,
    ]);
};