feat(set): Support for arguments as parameters instead of object.

jusx · Mar 3, 2019 · 02bc072 · 02bc072
1 parent a456614
commit 02bc072
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/__tests__/object-dot.js b/__tests__/object-dot.js
@@ -30,6 +30,12 @@ describe('set', () => {
     odjectd.set({ object, path: 'a.b', value: 'foo' })
     expect(object).toEqual({ a: { exist: true, b: 'foo' } })
   })
+
+  test('with arguments instead of destructing object', () => {
+    let object = { a: { exist: true } }
+    odjectd.set(object, 'a.b', 'foo')
+    expect(object.a.b).toBe('foo')
+  })
 })
 
 describe('get', () => {

diff --git a/lib/object-dot.js b/lib/object-dot.js
@@ -1,5 +1,8 @@
 
 function set ({ object, path, value, separator = '.' }) {
+  // support arguments as parameter instead of object.
+  if (arguments.length > 1) [ object, path, value, separator = '.' ] = arguments
+
   let properties = (Array.isArray(path)) ? path : path.split(separator)
   if (properties.length === 1) {
     object[path] = value
@@ -13,11 +16,9 @@ function set ({ object, path, value, separator = '.' }) {
 
 function get ({ object, path, value }) {
   let properties = path.split('.')
-
   while (properties.length && object) {
     object = object[properties.shift()]
   }
-
   return (value === undefined) ? object : value
 }