import type { DepartmentListQuery, OrgVO } from './types';
|
|
/** 收集组织树中所有节点 ID，用于默认展开全部节点。 */
|
export const collectOrgIds = (nodes: OrgVO[]): number[] =>
|
nodes.flatMap((node) => [node.id, ...collectOrgIds(node.children ?? [])]);
|
|
/**
|
* 按名称 / 类型 / 状态过滤组织树。
|
* 命中节点保留完整子树；节点自身不命中但存在命中的后代时，保留其祖先链路。
|
*/
|
export const filterOrgTree = (nodes: OrgVO[], query: DepartmentListQuery): OrgVO[] => {
|
const keyword = query.orgName?.trim().toLowerCase();
|
|
const filterNodes = (list: OrgVO[]): OrgVO[] =>
|
list.flatMap((node) => {
|
const matchesSelf =
|
(!keyword || node.orgName.toLowerCase().includes(keyword)) &&
|
(!query.orgType || node.orgType === query.orgType) &&
|
(!query.status || String(node.status) === query.status);
|
|
if (matchesSelf) return [node];
|
|
const children = filterNodes(node.children ?? []);
|
return children.length ? [{ ...node, children }] : [];
|
});
|
|
return filterNodes(nodes);
|
};
|
|
export interface OrgTreeSelectOption {
|
value: number;
|
title: string;
|
children?: OrgTreeSelectOption[];
|
}
|
|
/** 组织树转为 TreeSelect 数据；编辑时排除自身及其子树，避免把上级设为自身。 */
|
export const toOrgTreeSelectData = (
|
nodes: OrgVO[],
|
excludeId?: number,
|
): OrgTreeSelectOption[] =>
|
nodes.flatMap((node) => {
|
if (node.id === excludeId) return [];
|
|
const children = toOrgTreeSelectData(node.children ?? [], excludeId);
|
return [
|
{
|
value: node.id,
|
title: node.orgName,
|
children: children.length ? children : undefined,
|
},
|
];
|
});
|
