updeep-remeda/lib/update.js
2018-11-27 16:56:13 -08:00

105 lines
2.3 KiB
JavaScript

import isPlainObject from 'lodash/isPlainObject'
import _omitBy from 'lodash/omitBy'
import wrap from './wrap'
import constant from './constant'
const innerOmitted = { __omitted: true }
export const omitted = constant(innerOmitted)
function isEmpty(object) {
return !Object.keys(object).length
}
function reduce(object, callback, initialValue) {
return Object.keys(object).reduce(
(acc, key) => callback(acc, object[key], key),
initialValue
)
}
function resolveUpdates(updates, object) {
return reduce(
updates,
(acc, value, key) => {
let updatedValue = value
if (
!Array.isArray(value) &&
value !== null &&
typeof value === 'object'
) {
updatedValue = update(value, object[key]) // eslint-disable-line no-use-before-define
} else if (typeof value === 'function') {
updatedValue = value(object[key])
}
if (object[key] !== updatedValue) {
acc[key] = updatedValue // eslint-disable-line no-param-reassign
}
return acc
},
{}
)
}
function updateArray(updates, object) {
const newArray = [...object]
Object.keys(updates).forEach(key => {
newArray[key] = updates[key]
})
return newArray
}
/**
* Recursively update an object or array.
*
* Can update with values:
* update({ foo: 3 }, { foo: 1, bar: 2 });
* // => { foo: 3, bar: 2 }
*
* Or with a function:
* update({ foo: x => (x + 1) }, { foo: 2 });
* // => { foo: 3 }
*
* @function
* @name update
* @param {Object|Function} updates
* @param {Object|Array} object to update
* @return {Object|Array} new object with modifications
*/
function update(updates, object, ...args) {
if (typeof updates === 'function') {
return updates(object, ...args)
}
if (!isPlainObject(updates)) {
return updates
}
const defaultedObject =
typeof object === 'undefined' || object === null ? {} : object
const resolvedUpdates = resolveUpdates(updates, defaultedObject)
if (isEmpty(resolvedUpdates)) {
return defaultedObject
}
if (Array.isArray(defaultedObject)) {
return updateArray(resolvedUpdates, defaultedObject).filter(
value => value !== innerOmitted
)
}
return _omitBy(
{ ...defaultedObject, ...resolvedUpdates },
value => value === innerOmitted
)
}
export default wrap(update, 2)