feat: add object compact

afeiship · Jan 17, 2019 · 4325f1d · 4325f1d
1 parent b2f99ab
commit 4325f1d
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 14 deletions.
diff --git a/dist/next-compact.js b/dist/next-compact.js
@@ -2,12 +2,14 @@
   var global = global || this || window || Function('return this')();
   var nx = global.nx || require('next-js-core2');
 
-  nx.compact = function(inArray) {
-    var result = [];
-    for (var index = 0; index < inArray.length; index++) {
-      var value = inArray[index];
-      value && result.push(value);
-    }
+  nx.compact = function(inTarget) {
+    var isary = inTarget instanceof Array;
+    var result = isary ? [] : {};
+    nx.each(inTarget, function(key, value) {
+      if (value) {
+        isary ? result.push(value) : (result[key] = value);
+      }
+    });
     return result;
   };
 

diff --git a/dist/next-compact.min.js b/dist/next-compact.min.js
diff --git a/src/next-compact.js b/src/next-compact.js
@@ -2,12 +2,14 @@
   var global = global || this || window || Function('return this')();
   var nx = global.nx || require('next-js-core2');
 
-  nx.compact = function(inArray) {
-    var result = [];
-    for (var index = 0; index < inArray.length; index++) {
-      var value = inArray[index];
-      value && result.push(value);
-    }
+  nx.compact = function(inTarget) {
+    var isary = inTarget instanceof Array;
+    var result = isary ? [] : {};
+    nx.each(inTarget, function(key, value) {
+      if (value) {
+        isary ? result.push(value) : (result[key] = value);
+      }
+    });
     return result;
   };
 

diff --git a/test/test.js b/test/test.js
@@ -3,9 +3,21 @@ require('../src/next-compact');
 
 describe('nx.remove', function() {
   var arr1 = [0, 1, false, 2, '', 3];
+  var obj = {
+    name: 'fei',
+    age: 0,
+    test: 12,
+    ni: null,
+    tes2: undefined
+  };
 
-  test('return truthy vaulle', function() {
+  test('array: return truthy vaulle', function() {
     var result = nx.compact(arr1);
     expect(result).toEqual([1, 2, 3]);
   });
+
+  test('object: return truthy vaulle', function() {
+    var result = nx.compact(obj);
+    expect(result).toEqual({ name: 'fei', test: 12 });
+  });
 });