Merge pull request #2 from Yeti-or/yeti-or.create-array

Add createObjFromObj
Yeti-or · Oct 25, 2016 · 3569a82 · 3569a82
2 parents c4b9bad + 1a52deb
commit 3569a82
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 1 deletion.
diff --git a/.jshintrc b/.jshintrc
@@ -13,6 +13,7 @@
     "unused": true,
 
     "node": true,
+    "esversion": 6,
 
     "expr": true,
     "sub": true

diff --git a/index.js b/index.js
@@ -7,6 +7,7 @@ module.exports.removeElementFromArray = array.removeElementFromArray;
 
 // Object helpers
 module.exports.createObject = object.createObject;
+module.exports.createObjectFromObject = object.createObjectFromObject;
 module.exports.removePropertyFromObject = object.removePropertyFromObject;
 module.exports.removePropertyFromObjectByKeyName = object.removePropertyFromObjectByKeyName;
 module.exports.getKeysFromObject = object.getKeysFromObject;

diff --git a/lib/object.js b/lib/object.js
@@ -74,6 +74,13 @@ function createObject(map, opts) {
     return new types.ObjectExpression(arr);
 }
 
+function createObjectFromObject(obj, opts) {
+    var map = obj.properties.reduce((map, cur) => {
+        return map.set(cur.key.value || cur.key.name, cur.value.cloneElement());
+    }, new Map());
+    return createObject(map, opts || {});
+}
+
 function keyVal(propKey) {
     return propKey.name || propKey.value;
 }
@@ -218,6 +225,7 @@ function removePropertyFromObjectByKeyName(obj, keyName) {
 module.exports.removePropertyFromObject = removePropertyFromObject;
 module.exports.removePropertyFromObjectByKeyName = removePropertyFromObjectByKeyName;
 module.exports.createObject = createObject;
+module.exports.createObjectFromObject = createObjectFromObject;
 module.exports.getKeysFromObject = getKeysFromObject;
 module.exports.getValueFromObject = getValueFromObject;
 module.exports.getValuesFromObject = getValuesFromObject;

diff --git a/tests/object.js b/tests/object.js
@@ -28,7 +28,7 @@ describe('Object:', () => {
             expect(() => helpers.createObject(one)).to.throw(Error);
         });
 
-        it('should create simple object object', () => {
+        it('should create simple object', () => {
             var one = new types.NumericLiteral([Token.create('Numeric', 1)]);
             var map = new Map();
             map.set(1, one);
@@ -88,6 +88,28 @@ describe('Object:', () => {
                 .to.eql('{1: 1, 2: x, three: \'@@@\', 4: false}');
         });
 
+        describe('from another', () => {
+            it('should create object from another object', () => {
+                var one = new types.NumericLiteral([Token.create('Numeric', 1)]);
+                var map = new Map();
+                map.set(1, one);
+                var obj = helpers.createObject(map);
+                var obj2 = helpers.createObjectFromObject(obj);
+
+                expect(obj2.getSourceCode()).to.eql('{1: 1}');
+            });
+
+            it('should create object from another object with opts', () => {
+                var one = new types.NumericLiteral([Token.create('Numeric', 1)]);
+                var map = new Map();
+                map.set(1, one);
+                var obj = helpers.createObject(map);
+                var obj2 = helpers.createObjectFromObject(obj, {paddingNewLinesInObjects: true});
+
+                expect(obj2.getSourceCode()).to.eql('{\n1: 1\n}');
+            });
+        });
+
         describe('opts', () => {
             beforeEach(() => {
                 this.createObj = (opts) => {