nocobase/packages/core/utils/src/getValuesByPath.ts

/**
 * This file is part of the NocoBase (R) project.
 * Copyright (c) 2020-2024 NocoBase Co., Ltd.
 * Authors: NocoBase Team.
 *
 * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
 * For more information, please refer to: https://www.nocobase.com/agreement.
 */

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;
};