safe-dig/index.js

A safe dig function for navigating nested objects and arrays in JavaScript, with an optional prototype method for all objects.

// The core safeDig function
function safeDig(obj, ...args) {
  try {
    return args.reduce((acc, key) => {
      // If the accumulator is null or not an object/array, return null
      if (acc && (typeof acc === 'object' || Array.isArray(acc))) {
        return acc[key] !== undefined ? acc[key] : null; // Explicitly return null when key is not found
      }
      return null; // Return null if acc is not an object or array
    }, obj);
  } catch (e) {
    return null; // Return null in case of an error
  }
}

// Optional function to apply safeDig to Object prototype
function addSafeDigPrototype() {
  if (!Object.prototype.safeDig) {
    Object.prototype.safeDig = function (...args) {
      return safeDig(this, ...args);
    };
  }
}

module.exports = { safeDig, addSafeDigPrototype };