Query Inside JS object like lodash _.get but handles arrays as well

query.js

function queryObject(object, path) {
  const parts = path.split('.');
  let objects = wrapArray(object);
  while (parts.length > 0 && objects.length > 0) {
    const prop = parts.shift();
    objects = objects
      .flatMap((o) => wrapArray(o[prop]))
      .filter((o) => o != null);
  }
  return objects;
}

function containsArray(object, path) {
  if (object == null) {
    return null;
  }
  if (Array.isArray(object)) {
    return true;
  }
  const parts = path.split('.');
  while (parts.length > 0) {
    const prop = parts.shift();
    object = object[prop];
    if (Array.isArray(object)) {
      return true;
    }
  }
  return false;
}

function wrapArray(obj) {
  if (Array.isArray(obj)) {
    return obj;
  } else {
    return [obj];
  }
}

module.exports = { queryObject, containsArray };