@antv/s2
Version:
effective spreadsheet render core lib
77 lines • 4.03 kB
JavaScript
import { __rest } from "tslib";
import { isEmpty, isNumber } from 'lodash';
import { EXTRA_FIELD } from '../../common/constant';
import { generateId } from '../../utils/layout/generate-id';
import { Node } from '../layout/node';
import { layoutHierarchy } from './layout-hooks';
/**
* 自定义🌲结构设计原则：
* 1、渲染的节点由传入的数据结构决定, 不管是平铺,还是树状, 本质上都是树状结构
* 2、没有总计小计的概念
* 3、是否展开和收起完全由 customTreeNode.collapsed 来控制（默认都展开）
* 4、支持行头/列头自定义
* 5、支持隐藏列头
* @param params
*/
export const buildCustomTreeHierarchy = (params) => {
var _a;
const { tree = [], level, parentNode, hierarchy, spreadsheet, isRowHeader, } = params;
const { collapseFields, collapseAll, expandDepth } = (_a = spreadsheet.options.style) === null || _a === void 0 ? void 0 : _a.rowCell;
const hiddenColumnsDetail = spreadsheet.store.get('hiddenColumnsDetail') || [];
tree.forEach((treeNode) => {
var _a, _b;
const { field, title, collapsed, children } = treeNode, rest = __rest(treeNode, ["field", "title", "collapsed", "children"]);
const isHiddenNode = hiddenColumnsDetail.some(({ hideColumnNodes }) => hideColumnNodes.find((hideNode) => hideNode.field === field));
// query 只与值本身有关，不会涉及到 parent 节点
const valueQuery = { [EXTRA_FIELD]: field };
// 使用 field 作为 id, 保证其唯一性, 复制时再做二次转换
const nodeId = generateId(parentNode.id, field);
const defaultCollapsed = collapsed !== null && collapsed !== void 0 ? collapsed : false;
const isDefaultCollapsed = (_a = collapseFields === null || collapseFields === void 0 ? void 0 : collapseFields[nodeId]) !== null && _a !== void 0 ? _a : collapseFields === null || collapseFields === void 0 ? void 0 : collapseFields[field];
// 如果 level 大于 rowExpandDepth 或者没有配置层级展开配置时，返回 null，保证能正确降级到 collapseAll
const isLevelCollapsed = isNumber(expandDepth) ? level > expandDepth : null;
const isCollapsed = (_b = isDefaultCollapsed !== null && isDefaultCollapsed !== void 0 ? isDefaultCollapsed : isLevelCollapsed) !== null && _b !== void 0 ? _b : (collapseAll || defaultCollapsed);
// TODO: 平铺模式支持 折叠/展开: https://github.com/antvis/S2/issues/2019
const isCollapsedNode = spreadsheet.isHierarchyTreeType() && isRowHeader && isCollapsed;
const isLeaf = isEmpty(children);
const node = new Node({
id: nodeId,
field,
value: title,
level,
parent: parentNode,
// 自定义行头不会存在总计概念
isTotals: false,
isCollapsed: isCollapsedNode,
hierarchy,
query: valueQuery,
spreadsheet,
isLeaf,
extra: Object.assign(Object.assign({}, rest), { isCustomNode: true }),
});
if (isLeaf || isHiddenNode || isCollapsedNode) {
hierarchy.pushIndexNode(node);
}
if (isHiddenNode) {
return;
}
if (level > hierarchy.maxLevel && !node.isSeriesNumberNode()) {
hierarchy.maxLevel = level;
hierarchy.sampleNodesForAllLevels.push(node);
hierarchy.sampleNodeForLastLevel = node;
hierarchy.maxLevel = level;
}
const expandCurrentNode = layoutHierarchy(spreadsheet, parentNode, node, hierarchy);
if (!isEmpty(children) && !isCollapsedNode && expandCurrentNode) {
buildCustomTreeHierarchy({
spreadsheet,
parentNode: node,
level: level + 1,
hierarchy,
tree: (children || []),
isRowHeader,
});
}
});
};
//# sourceMappingURL=build-row-custom-tree-hierarchy.js.map