ant-design-vue/components/vc-cascader/utils/treeUtil.ts

55 lines
1.5 KiB
TypeScript

import type { Key } from '../../_util/type';
import type { SingleValueType, DefaultOptionType, InternalFieldNames } from '../Cascader';
import type { OptionsInfo } from '../hooks/useEntities';
export function formatStrategyValues(
pathKeys: Key[],
keyPathEntities: OptionsInfo['pathKeyEntities'],
) {
const valueSet = new Set(pathKeys);
return pathKeys.filter(key => {
const entity = keyPathEntities[key];
const parent = entity ? entity.parent : null;
if (parent && !parent.node.disabled && valueSet.has(parent.key)) {
return false;
}
return true;
});
}
export function toPathOptions(
valueCells: SingleValueType,
options: DefaultOptionType[],
fieldNames: InternalFieldNames,
// Used for loadingKeys which saved loaded keys as string
stringMode = false,
) {
let currentList = options;
const valueOptions: {
value: SingleValueType[number];
index: number;
option: DefaultOptionType;
}[] = [];
for (let i = 0; i < valueCells.length; i += 1) {
const valueCell = valueCells[i];
const foundIndex = currentList?.findIndex(option => {
const val = option[fieldNames.value];
return stringMode ? String(val) === String(valueCell) : val === valueCell;
});
const foundOption = foundIndex !== -1 ? currentList?.[foundIndex] : null;
valueOptions.push({
value: foundOption?.[fieldNames.value] ?? valueCell,
index: foundIndex,
option: foundOption,
});
currentList = foundOption?.[fieldNames.children];
}
return valueOptions;
}