nocobase/packages/core/utils/src/getValuesByPath.ts
2023-10-09 08:46:46 +08:00

44 lines
977 B
TypeScript

export const getValuesByPath = (obj: object, path: string, defaultValue?: any) => {
if (!obj) {
return defaultValue;
}
const keys = path.split('.');
let result: any[] = [];
let currentValue = obj;
let shouldReturnArray = false;
for (let i = 0; i < keys.length; i++) {
const key = keys[i];
if (Array.isArray(currentValue)) {
shouldReturnArray = true;
for (const element of currentValue) {
const value = getValuesByPath(element, keys.slice(i).join('.'), defaultValue);
result = result.concat(value);
}
break;
}
if (currentValue?.[key] === undefined) {
break;
}
currentValue = currentValue[key];
if (i === keys.length - 1) {
result.push(currentValue);
}
}
result = result.filter((item) => item != null);
if (result.length === 0) {
return defaultValue;
}
if (shouldReturnArray) {
return result;
}
return result.length === 1 ? result[0] : result;
};