improve typing of array-omit

angus-c · Sep 9, 2022 · 2261ca8 · 2261ca8
1 parent af091d7
commit 2261ca8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/packages/object-omit/index.d.ts b/packages/object-omit/index.d.ts
@@ -1,3 +1,12 @@
-declare function omit<Obj extends object, Key extends string>(obj: Obj, remove: Key[]): Omit<Obj, Key>;
-declare function omit<Obj extends object, Key extends string>(obj: Obj, remove1: Key, ...removeN: Key[]): Omit<Obj, Key>;
+declare function omit<Obj extends object, Key extends keyof Obj>(
+  obj: Obj,
+  remove: Key[]
+): Omit<Obj, Key>;
+
+declare function omit<Obj extends object, Key extends keyof Obj>(
+  obj: Obj,
+  remove1: Key,
+  ...removeN: Key[]
+): Omit<Obj, Key>;
+
 export default omit
diff --git a/packages/object-omit/index.tests.ts b/packages/object-omit/index.tests.ts
@@ -9,9 +9,6 @@ const obj = {
 {
 	const test: { c: number } = omit(obj, ['a', 'b'])
 }
-{
-	const test: { a: number } = omit(obj, ['b', 'c', 'd'])
-}
 {
 	const test: { a: number, c: number } = omit(obj, ['b', 'b', 'b'])
 }
@@ -47,4 +44,4 @@ omit({}, {}, 'a');
 // @ts-expect-error
 omit({ a: 1 }, { a: 1 });
 // @ts-expect-error
-omit([1, 2, 3], 0);
+omit({ a: 1}, ['a', 'b'])