types(n4s): simplify the types for

ealush · Nov 10, 2021 · 13830b9 · 13830b9
1 parent e09503d
commit 13830b9
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/packages/n4s/src/modifiers/partial.ts b/packages/n4s/src/modifiers/partial.ts
@@ -1,10 +1,14 @@
 import enforce from 'enforce';
-import type { TShapeObject, TLazy } from 'genEnforceLazy';
 
-export default function partial(shapeObject: TShapeObject): TShapeObject {
-  const output: TShapeObject = {};
+// Help needed improving the typings of this file.
+// Ideally, we'd be able to extend TShapeObject, but that's not possible.
+export default function partial<T extends Record<any, any>>(shapeObject: T): T {
+  const output = {} as T;
   for (const key in shapeObject) {
-    output[key] = enforce.optional(shapeObject[key]) as TLazy;
+    output[key] = enforce.optional(shapeObject[key]) as T[Extract<
+      keyof T,
+      string
+    >];
   }
 
   return output;