@addev-be/ui 0.2.5 → 0.2.7

package/src/components/data/SqlRequestDataGrid/index.tsx

@@ -1,252 +1,347 @@
-import {
-  ConditionDTO,
-  OrderByDTO,
-  useSqlRequestHandler,
-} from '../../../services/sqlRequests';
-import { DataGridFilters, DataGridSort } from '../DataGrid/types';
-import _, { debounce } from 'lodash';
-import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
-
-import { DataGrid } from '../DataGrid';
-import { SqlRequestDataGridProps } from './types';
-import { convertSqlFiltersToConditions } from './helpers';
-import { isColumnVisible } from '../DataGrid/helpers';
-
-export const SqlRequestDataGrid = <R,>({
-  onSelectionChange: onSelectionChangeFromProps,
-  ...props
-}: SqlRequestDataGridProps<R>) => {
-  const currentRows = useRef<R[]>([]);
-  const [rows, setRows] = useState<R[]>([]);
-  const [start, setStart] = useState(0);
-  const [length, setLength] = useState(50);
-  const [total, setTotal] = useState(-1);
-  const [sqlRequest, sqlIdRequest] = useSqlRequestHandler<R>(props.type);
-
-  const [conditions, setConditions] = useState<Record<string, ConditionDTO>>(
-    {}
-  );
-  const [orderBy, setOrderBy] = useState<OrderByDTO[]>(
-    Object.entries(props.initialSorts ?? {}).map(
-      ([columnKey, direction]): OrderByDTO => ({
-        field: props.columns[columnKey].field?.fieldAlias ?? columnKey,
-        direction: direction.toUpperCase() as 'ASC' | 'DESC',
-      })
-    )
-  );
-
-  const [columnsKeys, visibleColumnsKeys] = useMemo(
-    () => [
-      [...Object.keys(props.columns), ...(props.hiddenColumns ?? [])],
-      [
-        ...Object.keys(props.columns).filter((key) =>
-          isColumnVisible(props.columns[key])
-        ),
-        ...(props.hiddenColumns ?? []),
-      ],
-    ],
-    [props.columns, props.hiddenColumns]
-  );
-
-  const refresh = useCallback(() => {
-    setRows([]);
-    setStart(0);
-    setLength(50);
-    setTotal(-1);
-  }, []);
-
-  const onFiltersChanged = useCallback((filters: DataGridFilters) => {
-    const newConditions = convertSqlFiltersToConditions(filters);
-    setTotal(-1);
-    setConditions(newConditions);
-  }, []);
-
-  const onSortsChanged = useCallback(
-    (sorts: Record<string, DataGridSort>) => {
-      refresh();
-      const newOrderBy = Object.entries(sorts).map(
-        ([columnKey, direction]) =>
-          ({
-            field: columnKey,
-            direction: direction.toUpperCase(),
-          } as OrderByDTO)
-      );
-      setOrderBy(newOrderBy);
-    },
-    [refresh]
-  );
-
-  const loadRows = useRef(
-    debounce(
-      (
-        columns: string[],
-        returnColumns: string[],
-        conditions: ConditionDTO[] = [],
-        orderBy: OrderByDTO[] = [],
-        start = 0,
-        length = 100,
-        getTotal = false
-      ) => {
-        sqlRequest({
-          columns: columns.includes('Id') ? columns : [...columns, 'Id'],
-          returnColumns: returnColumns.includes('Id')
-            ? returnColumns
-            : [...returnColumns, 'Id'],
-          conditions,
-          orderBy,
-          start,
-          length,
-          getTotal,
-        }).then((response) => {
-          if (getTotal) {
-            currentRows.current = Array(response.total).fill(null);
-            if (getTotal) setTotal(response.total ?? 0);
-          }
-          const parsedRows = props.parser
-            ? response.data.map(props.parser)
-            : (response.data as R[]);
-          currentRows.current.splice(start, length, ...parsedRows);
-          setRows([...currentRows.current]);
-        });
-      },
-      100,
-      {
-        leading: true,
-        trailing: true,
-      }
-    )
-  );
-
-  const loadFilterValues = useCallback(
-    (columnKey: string) => {
-      return sqlRequest({
-        columns: columnsKeys,
-        returnColumns: [columnKey],
-        conditions: [
-          ...(props.conditions ?? []),
-          ...Object.values(
-            _.pickBy(conditions, (condition, key) => key !== columnKey)
-          ),
-        ].filter((condition) => condition.field !== columnKey),
-        orderBy: [
-          {
-            field:
-              props.columns[columnKey].filterField ??
-              props.columns[columnKey].field?.fieldAlias ??
-              props.columns[columnKey].field?.fieldName ??
-              columnKey,
-            direction: 'ASC',
-          },
-        ],
-        getTotal: false,
-        unique: true,
-      }).then((response) =>
-        response.data.map(
-          (row) => props.columns[columnKey].filter?.getter?.(row) ?? null
-        )
-      );
-    },
-    [columnsKeys, conditions, props.columns, props.conditions, sqlRequest]
-  );
-
-  useEffect(
-    () =>
-      loadRows.current(
-        columnsKeys,
-        visibleColumnsKeys,
-        [...(props.conditions ?? []), ...(Object.values(conditions) ?? [])],
-        orderBy,
-        start,
-        length,
-        total < 0
-      ),
-    [
-      props.columns,
-      conditions,
-      orderBy,
-      start,
-      length,
-      total,
-      props.conditions,
-      columnsKeys,
-      visibleColumnsKeys,
-    ]
-  );
-
-  const loadCopyRows = useCallback(
-    () =>
-      sqlRequest({
-        columns: columnsKeys,
-        returnColumns: visibleColumnsKeys,
-        conditions: [
-          ...(props.conditions ?? []),
-          ...(Object.values(conditions) ?? []),
-        ],
-        orderBy,
-        start: 0,
-        length: total,
-      }).then((response) =>
-        props.parser ? response.data.map(props.parser) : (response.data as R[])
-      ),
-    [
-      sqlRequest,
-      columnsKeys,
-      visibleColumnsKeys,
-      props.conditions,
-      props.parser,
-      conditions,
-      orderBy,
-      total,
-    ]
-  );
-
-  const loadAllIds = useCallback(
-    () =>
-      sqlIdRequest({
-        columns: columnsKeys,
-        returnColumns: ['Id'],
-        conditions: [
-          ...(props.conditions ?? []),
-          ...(Object.values(conditions) ?? []),
-        ],
-        orderBy,
-        start: 0,
-        length: total,
-      }).then((response) => response.data.map((row) => row['Id'])),
-    [columnsKeys, conditions, orderBy, props.conditions, sqlIdRequest, total]
-  );
-
-  const onVisibleRowsChanged = useCallback(
-    (newStart: number, newLength: number) => {
-      if (newStart !== start || newLength !== length) {
-        setStart(newStart);
-        setLength(newLength);
-      }
-    },
-    [length, start]
-  );
-
-  const onSelectionChange = useCallback(
-    (selectedKeys: string[]) => {
-      onSelectionChangeFromProps?.(selectedKeys);
-    },
-    [onSelectionChangeFromProps]
-  );
-
-  return (
-    <DataGrid
-      onVisibleRowsChange={onVisibleRowsChanged}
-      filter={false}
-      sort={false}
-      onFiltersChanged={onFiltersChanged}
-      onSortsChanged={onSortsChanged}
-      filterValuesLoader={loadFilterValues}
-      rows={rows}
-      loadCopyRows={loadCopyRows}
-      refresh={refresh}
-      onSelectionChange={onSelectionChange}
-      getAllIds={loadAllIds}
-      {...props}
-    />
-  );
-};
+import {
+  ConditionDTO,
+  OrderByDTO,
+  SqlRequestFooterFunction,
+  SqlRequestRow,
+  useSqlRequestHandler,
+} from '../../../services/sqlRequests';
+import {
+  DataGridFilters,
+  DataGridFooterFunction,
+  DataGridSort,
+} from '../DataGrid/types';
+import _, { debounce } from 'lodash';
+import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
+
+import { DataGrid } from '../DataGrid';
+import { SqlRequestDataGridProps } from './types';
+import { convertSqlFiltersToConditions } from './helpers';
+import { isColumnVisible } from '../DataGrid/helpers';
+
+export const SqlRequestDataGrid = <R,>({
+  onSelectionChange: onSelectionChangeFromProps,
+  ...props
+}: SqlRequestDataGridProps<R>) => {
+  const currentRows = useRef<R[]>([]);
+  const [rows, setRows] = useState<R[]>([]);
+  const [start, setStart] = useState(0);
+  const [length, setLength] = useState(50);
+  const [count, setCount] = useState(-1);
+  const [sqlRequest, sqlIdRequest, sqlPartialRequest] = useSqlRequestHandler<R>(
+    props.type
+  );
+
+  const [conditions, setConditions] = useState<Record<string, ConditionDTO>>(
+    {}
+  );
+  const [orderBy, setOrderBy] = useState<OrderByDTO[]>(
+    Object.entries(props.initialSorts ?? {}).map(
+      ([columnKey, direction]): OrderByDTO => ({
+        field: props.columns[columnKey].field?.fieldAlias ?? columnKey,
+        type: props.columns[columnKey].type ?? 'text',
+        direction: direction.toUpperCase() as 'ASC' | 'DESC',
+      })
+    )
+  );
+
+  const [columnsKeys, visibleColumnsKeys] = useMemo(
+    () => [
+      [...Object.keys(props.columns), ...(props.hiddenColumns ?? [])],
+      [
+        ...Object.keys(props.columns).filter((key) =>
+          isColumnVisible(props.columns[key])
+        ),
+        ...(props.hiddenColumns ?? []),
+      ],
+    ],
+    [props.columns, props.hiddenColumns]
+  );
+  const columnTypes = useMemo(
+    () =>
+      visibleColumnsKeys.map((key) =>
+        String(props.columns[key].type ?? 'text')
+      ),
+    [visibleColumnsKeys, props.columns]
+  );
+
+  const refresh = useCallback(() => {
+    setRows([]);
+    setStart(0);
+    setLength(50);
+    setCount(-1);
+  }, []);
+
+  const onFiltersChanged = useCallback((filters: DataGridFilters) => {
+    const newConditions = convertSqlFiltersToConditions(filters);
+    setCount(-1);
+    setConditions(newConditions);
+  }, []);
+
+  const onSortsChanged = useCallback(
+    (sorts: Record<string, DataGridSort>) => {
+      refresh();
+      const newOrderBy = Object.entries(sorts).map(
+        ([columnKey, direction]) =>
+          ({
+            field: columnKey,
+            direction: direction.toUpperCase(),
+          } as OrderByDTO)
+      );
+      setOrderBy(newOrderBy);
+    },
+    [refresh]
+  );
+
+  const loadRows = useRef(
+    debounce(
+      (
+        columns: string[],
+        returnColumns: string[],
+        conditions: ConditionDTO[] = [],
+        orderBy: OrderByDTO[] = [],
+        start = 0,
+        length = 100,
+        getCount = false
+      ) => {
+        sqlRequest({
+          columns: columns.includes('Id') ? columns : [...columns, 'Id'],
+          returnColumns: returnColumns.includes('Id')
+            ? returnColumns
+            : [...returnColumns, 'Id'],
+          columnTypes: columnTypes.includes('Id')
+            ? columnTypes
+            : [...columnTypes, 'Id'],
+          conditions,
+          orderBy,
+          start,
+          length,
+          getCount,
+        }).then((response) => {
+          if (getCount) {
+            currentRows.current = Array(response.count).fill(null);
+            if (getCount) setCount(response.count ?? 0);
+          }
+          const parsedRows = props.parser
+            ? response.data.map(props.parser)
+            : (response.data as R[]);
+          currentRows.current.splice(start, length, ...parsedRows);
+          setRows([...currentRows.current]);
+        });
+      },
+      100,
+      {
+        leading: true,
+        trailing: true,
+      }
+    )
+  );
+
+  const loadFilterValues = useCallback(
+    (columnKey: string) => {
+      return sqlRequest({
+        columns: columnsKeys,
+        returnColumns: [columnKey],
+        columnTypes: [props.columns[columnKey].type ?? 'text'],
+        conditions: [
+          ...(props.conditions ?? []),
+          ...Object.values(_.pickBy(conditions, (_, key) => key !== columnKey)),
+        ].filter((condition) => condition.field !== columnKey),
+        orderBy: [
+          {
+            field:
+              props.columns[columnKey].filterField ??
+              props.columns[columnKey].field?.fieldAlias ??
+              props.columns[columnKey].field?.fieldName ??
+              columnKey,
+            type: props.columns[columnKey].type ?? 'text',
+            direction: 'ASC',
+          },
+        ],
+        getCount: false,
+        unique: true,
+      }).then((response) =>
+        response.data.map(
+          (row) => props.columns[columnKey].filter?.getter?.(row) ?? null
+        )
+      );
+    },
+    [columnsKeys, conditions, props.columns, props.conditions, sqlRequest]
+  );
+
+  useEffect(
+    () =>
+      loadRows.current(
+        columnsKeys,
+        visibleColumnsKeys,
+        [...(props.conditions ?? []), ...(Object.values(conditions) ?? [])],
+        orderBy,
+        start,
+        length,
+        count < 0
+      ),
+    [
+      props.columns,
+      conditions,
+      orderBy,
+      start,
+      length,
+      count,
+      props.conditions,
+      columnsKeys,
+      visibleColumnsKeys,
+    ]
+  );
+
+  const loadCopyRows = useCallback(
+    () =>
+      sqlRequest({
+        columns: columnsKeys,
+        returnColumns: visibleColumnsKeys,
+        columnTypes,
+        conditions: [
+          ...(props.conditions ?? []),
+          ...(Object.values(conditions) ?? []),
+        ],
+        orderBy,
+        start: 0,
+        length: count,
+      }).then((response) =>
+        props.parser ? response.data.map(props.parser) : (response.data as R[])
+      ),
+    [
+      sqlRequest,
+      columnsKeys,
+      visibleColumnsKeys,
+      columnTypes,
+      props.conditions,
+      props.parser,
+      conditions,
+      orderBy,
+      count,
+    ]
+  );
+
+  const loadAllIds = useCallback(
+    () =>
+      sqlIdRequest({
+        columns: columnsKeys.includes('Id')
+          ? columnsKeys
+          : [...columnsKeys, 'Id'],
+        returnColumns: ['Id'],
+        columnTypes: ['text'],
+        conditions: [
+          ...(props.conditions ?? []),
+          ...(Object.values(conditions) ?? []),
+        ],
+        orderBy,
+        start: 0,
+        length: count,
+      }).then((response) => response.data.map((row) => row['Id'])),
+    [columnsKeys, conditions, orderBy, props.conditions, sqlIdRequest, count]
+  );
+
+  const onVisibleRowsChanged = useCallback(
+    (newStart: number, newLength: number) => {
+      if (newStart !== start || newLength !== length) {
+        setStart(newStart);
+        setLength(newLength);
+      }
+    },
+    [length, start]
+  );
+
+  const onSelectionChange = useCallback(
+    (selectedKeys: string[]) => {
+      onSelectionChangeFromProps?.(selectedKeys);
+    },
+    [onSelectionChangeFromProps]
+  );
+
+  /** FOOTERS */
+  const [footers, setFooters] = useState<Record<string, string>>(
+    props.initialFooters ?? {}
+  );
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  const [footerData, setFooterData] = useState<SqlRequestRow<any>>({});
+
+  const loadFooters = useCallback(() => {
+    if (Object.keys(footers).length === 0) {
+      setFooterData({});
+    } else {
+      sqlPartialRequest({
+        columns: columnsKeys.includes('Id')
+          ? columnsKeys
+          : [...columnsKeys, 'Id'],
+        returnColumns: [],
+        columnTypes: [],
+        totalColumns: footers as Record<string, SqlRequestFooterFunction>,
+        conditions: [
+          ...(props.conditions ?? []),
+          ...(Object.values(conditions) ?? []),
+        ],
+        orderBy,
+      }).then((response) => setFooterData(response.totals ?? {}));
+    }
+  }, [
+    columnsKeys,
+    conditions,
+    footers,
+    orderBy,
+    props.conditions,
+    sqlPartialRequest,
+  ]);
+
+  useEffect(() => {
+    loadFooters();
+  }, [loadFooters]);
+
+  const footerFunctions = useMemo(
+    () =>
+      !footerData
+        ? {}
+        : Object.entries(footers).reduce((acc, [columnKey, footerKey]) => {
+            const column = props.columns[columnKey];
+            const footerFunc =
+              typeof column?.footer === 'function'
+                ? column.footer
+                : column?.footer?.[footerKey];
+            const render = footerFunc
+              ? // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                (data: any) => footerFunc([data], [], [])
+              : column?.render ?? _.identity;
+            if (!column) {
+              return acc;
+            }
+            acc[columnKey] = () => render(footerData, column);
+            return acc;
+          }, {} as Record<string, DataGridFooterFunction<R>>),
+    [footerData, footers, props.columns]
+  );
+
+  const contextOverride = useMemo(
+    () => ({
+      footers,
+      setFooters,
+      footerFunctions,
+    }),
+    [footers, setFooters, footerFunctions]
+  );
+
+  return (
+    <DataGrid
+      onVisibleRowsChange={onVisibleRowsChanged}
+      filter={false}
+      sort={false}
+      onFiltersChanged={onFiltersChanged}
+      onSortsChanged={onSortsChanged}
+      filterValuesLoader={loadFilterValues}
+      rows={rows}
+      loadCopyRows={loadCopyRows}
+      refresh={refresh}
+      onSelectionChange={onSelectionChange}
+      getAllIds={loadAllIds}
+      contextOverride={contextOverride}
+      {...props}
+    />
+  );
+};