From d706c31c7f0a2af4585210a9145a3a9faca4482b Mon Sep 17 00:00:00 2001
From: Chris <chrisbreuer93@gmail.com>
Date: Sun, 24 Nov 2024 23:18:33 +0100
Subject: [PATCH] chore: improve `only` typing

---
 src/collect.ts | 8 +++++---
 src/types.ts   | 6 +++++-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/collect.ts b/src/collect.ts
index 72562c8..175b0c5 100644
--- a/src/collect.ts
+++ b/src/collect.ts
@@ -349,11 +349,13 @@ function createCollectionOperations<T>(collection: Collection<T>): CollectionOpe
       return collect(merged) as CollectionOperations<RecordMerge<T, U>>
     },
 
-    only<K extends keyof T>(...keys: K[]) {
+    only<K extends string>(...keys: K[]) {
       return this.map((item) => {
-        const result = {} as Pick<T, K>
+        const result = {} as { [P in K & keyof T]?: T[P] }
         keys.forEach((key) => {
-          result[key] = item[key]
+          if (key in item) {
+            result[key as keyof T] = item[key as keyof T]
+          }
         })
         return result
       })
diff --git a/src/types.ts b/src/types.ts
index 09768ff..0f14c0c 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -236,7 +236,11 @@ export interface CollectionOperations<T> extends Collection<T> {
   ) => Map<K, V>
   merge: <U extends T>(other: U[] | CollectionOperations<U>) => CollectionOperations<T | U>
   mergeRecursive: <U>(other: U[] | CollectionOperations<U>) => CollectionOperations<RecordMerge<T, U>>
-  only: <K extends keyof T>(...keys: K[]) => CollectionOperations<Pick<T, K>>
+  only: <K extends string>(
+    ...keys: K[]
+  ) => CollectionOperations<{
+    [P in K & keyof T]?: T[P]
+  }>
   pad: (size: number, value: T) => CollectionOperations<T>
   pop: () => T | undefined
   prepend: (value: T) => CollectionOperations<T>