all files / scour/utilities/ set.js

100% Statements 16/16
100% Branches 8/8
100% Functions 1/1
100% Lines 16/16
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39                    17× 17×   17× 36× 17×   19×   19×         17× 53× 17×   36× 36×       17×    
'use strict'
 
const clone = require('./clone')
 
/**
 * Sets a `keypath` into an `object` immutably.
 *
 *     data = { users: { bob: { name: 'john' } } }
 *     result = set(data, ['users', 'bob', 'name'], 'robert')
 */
 
module.exports = function set (object, keypath, value) {
  let results = {}
  let parents = {}
 
  for (let i = 0, len = keypath.length; i < len; i++) {
    if (i === 0) {
      parents[i] = object
    } else {
      parents[i] = parents[i - 1][keypath[i - 1]] || {}
      // handle cases when it isn't an object
      if (typeof parents[i] !== 'object') {
        parents[i] = {}
      }
    }
  }
 
  for (let i = keypath.length; i >= 0; i--) {
    if (!parents[i]) {
      results[i] = value
    } else {
      results[i] = clone(parents[i])
      results[i][keypath[i]] = results[i + 1]
    }
  }
 
  return results[0]
}