object-omit -> TypeScript

angus-c · Mar 8, 2019 · a093c6d · jednano · Mar 8, 2019 · jednano
1 parent 0ca08ba
commit a093c6d
Show file tree

Hide file tree

Showing 5 changed files with 37 additions and 5 deletions.
diff --git a/packages/object-omit/index.d.ts b/packages/object-omit/index.d.ts
@@ -0,0 +1,2 @@
+declare function omit<T extends object, K extends keyof T>(obj: T | null | undefined, remove: keyof T | Array<K | ReadonlyArray<K>>, ...rest: Array<K | ReadonlyArray<K>>): any;
+export = omit;
diff --git a/packages/object-omit/index.js b/packages/object-omit/index.js
diff --git a/packages/object-omit/index.js.map b/packages/object-omit/index.js.map
diff --git a/packages/object-omit/index.ts b/packages/object-omit/index.ts
@@ -0,0 +1,28 @@
+/*
+  var obj = {a: 3, b: 5, c: 9};
+  omit(obj, ['a', 'c']); // {b: 5}
+  omit(obj, a, c); // {b: 5}
+  omit(obj, ['a', 'b', 'd']); // {c: 9}
+  omit(obj, ['a', 'a']); // {b: 5, c: 9}
+*/
+
+function omit<T extends object, K extends keyof T>(
+  obj: T | null | undefined,
+  remove: keyof T | Array<K | ReadonlyArray<K>>,
+  ...rest: Array<K | ReadonlyArray<K>>
+) {
+  var result = {} as any;
+  if (typeof remove === 'string') {
+    remove = [remove as any].concat(rest)
+  }
+  for (var prop in obj) {
+    if (!obj.hasOwnProperty || obj.hasOwnProperty(prop)) {
+      if ((remove as Array<K | ReadonlyArray<K>>).indexOf(prop as unknown as K) === -1) {
+        result[prop as string] = obj[prop];
+      }
+    }
+  }
+  return result;
+}
+
+export = omit;
diff --git a/test/object-omit/index.js b/test/object-omit/index.js
@@ -1,3 +1,4 @@
+// @ts-check
 var test = require('../util/test')(__filename);
 var omit = require('../../packages/object-omit');
 var compare = require('../../packages/collection-compare');
@@ -35,6 +36,7 @@ test('omit using arguments', function(t) {
 test('omit using a non-existent key', function(t) {
   t.plan(1);
   var obj = {a: 3, b: 5, c: 9};
+  // @ts-ignore
   t.ok(compare(omit(obj, ['a', 'b', 'd']), {c: 9}));
   t.end();
 });