UNPKG

@arco-design/web-react

Version:

Arco Design React UI Library.

39 lines (38 loc) 1.95 kB
import { useMemo } from 'react'; import cs from '../../_util/classNames'; import { isObject } from '../../_util/is'; // get sticky cell's className function useStickyClassNames(groupColumns, columns, prefixCls) { var colFixed = columns.map(function (c) { return c.fixed; }); function getClassNameFromColumn(column, index) { var _a; return cs((_a = {}, _a[prefixCls + "-col-fixed-left"] = column.fixed === 'left', _a[prefixCls + "-col-fixed-right"] = column.fixed === 'right', _a[prefixCls + "-col-fixed-left-last"] = column.fixed === 'left' && (isObject(columns[index + 1]) ? columns[index + 1].fixed !== 'left' : true), _a[prefixCls + "-col-fixed-right-first"] = column.fixed === 'right' && (isObject(columns[index - 1]) ? columns[index - 1].fixed !== 'right' : true), _a)); } var stickyClassNames = useMemo(function () { return columns.map(function (column, index) { return getClassNameFromColumn(column, index); }); }, [colFixed.join('-')]); var groupStickyClassNames = useMemo(function () { return groupColumns.map(function (gc) { return gc.map(function (column, i) { var index = i; var columnIndex = column.$$columnIndex; if (Array.isArray(columnIndex) && columnIndex.length === 2) { index = column.fixed === 'right' ? columnIndex[0] : columnIndex[1]; } else if (typeof columnIndex === 'number') { index = columnIndex; } return getClassNameFromColumn(column, index); }); }); }, [groupColumns.map(function (c) { return "|" + c.map(function (a) { return a.fixed || 'undefined'; }).join('-') + "|"; }).join('_')]); return [groupStickyClassNames, stickyClassNames]; } export default useStickyClassNames;