|
1 | 1 | import { useLocalStorage } from "@migtools/lib-ui";
|
| 2 | +import { ColumnSetting } from "../types"; |
| 3 | +import { useEffect } from "react"; |
2 | 4 |
|
3 | 5 | export interface ColumnState<TColumnKey extends string> {
|
4 | 6 | id: TColumnKey;
|
5 | 7 | label: string;
|
6 | 8 | isVisible: boolean;
|
| 9 | + isIdentity?: boolean; |
7 | 10 | }
|
8 | 11 |
|
9 | 12 | export interface IColumnState<TColumnKey extends string> {
|
10 | 13 | columns: ColumnState<TColumnKey>[];
|
| 14 | + defaultColumns: ColumnState<TColumnKey>[]; |
11 | 15 | setColumns: (newColumns: ColumnState<TColumnKey>[]) => void;
|
12 | 16 | }
|
13 | 17 |
|
14 | 18 | interface IColumnStateArgs<TColumnKey extends string> {
|
15 |
| - initialColumns: ColumnState<TColumnKey>[]; |
| 19 | + initialColumns?: Partial<Record<TColumnKey, ColumnSetting>>; |
16 | 20 | columnsKey: string;
|
| 21 | + supportedColumns: Record<TColumnKey, string>; |
17 | 22 | }
|
18 | 23 |
|
19 |
| -export const useColumnState = <TColumnKey extends string>( |
20 |
| - args: IColumnStateArgs<TColumnKey> |
21 |
| -): IColumnState<TColumnKey> => { |
| 24 | +export const useColumnState = <TColumnKey extends string>({ |
| 25 | + initialColumns, |
| 26 | + supportedColumns, |
| 27 | + columnsKey, |
| 28 | +}: IColumnStateArgs<TColumnKey>): IColumnState<TColumnKey> => { |
| 29 | + const defaultColumns = ( |
| 30 | + Object.entries(supportedColumns) as [TColumnKey, string][] |
| 31 | + ).map(([id, label]) => ({ |
| 32 | + id, |
| 33 | + label, |
| 34 | + isVisible: initialColumns?.[id]?.isVisible ?? true, |
| 35 | + isIdentity: initialColumns?.[id]?.isIdentity, |
| 36 | + })); |
22 | 37 | const [columns, setColumns] = useLocalStorage<ColumnState<TColumnKey>[]>({
|
23 |
| - key: args.columnsKey, |
24 |
| - defaultValue: args.initialColumns, |
| 38 | + key: columnsKey, |
| 39 | + defaultValue: defaultColumns, |
25 | 40 | });
|
26 | 41 |
|
27 |
| - return { columns, setColumns }; |
| 42 | + useEffect(() => { |
| 43 | + const valid = columns.filter(({ id }) => |
| 44 | + defaultColumns.find((it) => id === it.id) |
| 45 | + ); |
| 46 | + if (valid.length !== defaultColumns.length) { |
| 47 | + setColumns( |
| 48 | + defaultColumns.map((it) => ({ |
| 49 | + ...it, |
| 50 | + isVisible: |
| 51 | + valid.find(({ id }) => id === it.id)?.isVisible ?? it.isVisible, |
| 52 | + })) |
| 53 | + ); |
| 54 | + } |
| 55 | + }, [defaultColumns, columns, setColumns]); |
| 56 | + |
| 57 | + return { columns, setColumns, defaultColumns }; |
28 | 58 | };
|
0 commit comments